Implement purpose-based templates

bin/mkexec

@@ -4,7 +4,8 @@
 use strict;
 use warnings;
 
-sub usage; sub mkexec; sub guesstype; sub getcmd; sub launch_editor;
+sub usage; sub mkexec; sub guesstype; sub guesspurpose;
+sub getcmd; sub launch_editor;
 
 my $DEFAULT_TYPE = 'sh';
 my $DEFAULT_MODE = 0755;
@@ -17,7 +18,7 @@ my $DEFAULT_EDIT = 0;
 ## '''' ##   ' ' '  '   '    '     '      '       '        '         '
 
 my $force = $DEFAULT_FORCE;
-my $name; my $type; my $edit;
+my $name; my $type; my $purpose; my $edit;
 
 foreach (@ARGV) {
     if (m/-f|--force/) {
@@ -26,6 +27,8 @@ foreach (@ARGV) {
         $edit++;
     } elsif (m/-E|--no-edit/) {
         $edit = 0;
+    } elsif (defined $type) {
+        $purpose = $_;
     } elsif (defined $name) {
         $type = $_;
     } else {
@@ -35,6 +38,7 @@ foreach (@ARGV) {
 
 usage unless defined $name;
 $type = guesstype $name unless defined $type;
+$purpose = guesspurpose $name unless defined $purpose;
 
 my $bangs = {
     pl   => `which perl`,
@@ -46,8 +50,18 @@ my $bangs = {
 };
 
 my $templates = {
-    pl   => "use strict;\nuse warnings;\n",
-    py   => "if __name__ == '__main__':\n"
+    py => {
+        ''      => "if __name__ == '__main__':\n",
+        test    => "import unittest\n\n\n"
+                   . "class TestCase(unittest.TestCase):\n"
+                   . "    pass\n\n\n"
+                   . "if __name__=='__main__':\n"
+                   ."    unittest.main()\n",
+    },
+    pl => {
+        ''      => "use strict;\nuse warnings;\n",
+        test    => "use strict;\nuse warnings;\nuse Test::More;\n\n",
+    }
 };
 
 my $editors = [ qw/ vim editor / ];
@@ -57,6 +71,11 @@ $cmds->{vim}->{run}     = "vim +\"normal G\$\" '%s'";
 $cmds->{editor}->{test} = "editor --version 2>/dev/null";
 $cmds->{editor}->{run}  = "editor '%s'";
 
+unless (exists $templates->{$type}->{$purpose}) {
+    warn "undefined purpose '$purpose' for type '$type'\n";
+    $purpose = "";
+}
+
 
 ## ... ##   # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
 ## RUN ## ----------------------------------------------------------------- #
@@ -66,7 +85,7 @@ if (exists $bangs->{$type}) {
     $force and -e $name
         and (defined `cp "$name" "$name~"` or die $!);
     (not -e $name or $force)
-        and mkexec $name, mkbody($type);
+        and mkexec $name, mkbody($type, $purpose);
     chmod $DEFAULT_MODE, $name;
     launch_editor $name if $edit;
 } else {
@@ -79,7 +98,7 @@ if (exists $bangs->{$type}) {
 ## '''' ##   ..............................................................-'
 
 sub usage {
-    print STDERR "usage: $0 [-f|-E] filename [type]\n";
+    print STDERR "usage: $0 [-f|-E] filename [type] [purpose]\n";
     exit 0;
 }
 
@@ -89,10 +108,23 @@ sub guesstype {
     return ( $ext ? $ext : $DEFAULT_TYPE);
 }
 
+sub guesspurpose {
+    my $name = shift;
+    my $testword = '[tT][eE][sS][tT]';
+    return 'test' if $name =~ m|^test_|i;
+    return 'test' if $name =~ m|_test.py$|i;
+    return 'test' if $name =~ m|_test.pl$|i;
+    return 'test' if $name =~ m|\btest\w*/|i;
+    return 'test' if $name =~ m|.*Test/|i;
+    return 'test' if $name =~ m|\bt/|;
+    return ''
+}
+
 sub mkbody {
     my $type = shift;
+    my $purpose = shift;
     my $tmpl = "";
-    $tmpl .= $templates->{$type} if exists $templates->{$type};
+    $tmpl .= $templates->{$type}->{$purpose} if exists $templates->{$type};
     return sprintf "#!%s\n%s\n", $bangs->{$type}, $tmpl;
 }