better recognize python and ruby scripts in code2color

code2color

@@ -1,11 +1,5 @@
 #!/usr/bin/env perl
 
-#use strict;
-use Getopt::Long;
-use File::Basename;
-use POSIX;
-use Term::ANSIColor;
-
 my $VERSION = "0.6";
 my $vernr = "0.9.2"; # this is the underlying version of Peter Palfraders script
 ########################################################################
@@ -33,6 +27,10 @@ my $vernr = "0.9.2"; # this is the underlying version of Peter Palfraders script
 
 my $LANG_TEST_LENGTH = 1024;
 
+use strict;
+use POSIX;
+use Term::ANSIColor;
+
 use Getopt::Std;
 our($opt_h, $opt_l, $opt_v);
 getopts('l:hv') || exit 2;
@@ -147,7 +145,7 @@ sub get_input_file {
 
 	if (not $langmode) {
 		my $test_code = substr($code, 0, $LANG_TEST_LENGTH);
-		# warn("language mode not given. guessing...\n");
+		#warn("language mode not given. guessing...\n");
 		$langmode = '';
 
 		for (sort keys %LANGUAGE) { # make it repeatable
@@ -1945,7 +1943,7 @@ $LANGUAGE{'povray'} = $LANGUAGE{'pov'};
 # by Tom Good
 $LANGUAGE{'python'} = {
 	'filename' => '(?i)\\.py$',
-	'regex'    => '^\\s*#\\s*![^\\s]*python',
+	'regex'    => '^\\s*#\\s*!([^\\s]*\\b|.*env\\s+)*python',
 	'patterns' => [
 		{
 			'name'       => 'python comment',
@@ -2049,7 +2047,7 @@ $LANGUAGE{'python'} = {
 # by Joshua Swink <jswink AT pacbell.net>
 $LANGUAGE{'ruby'} = {
 	'filename' => '\\.rb$',
-	'regex'    => '^\\s*#\\s*![^\\s]*\\bruby\\b',
+	'regex'    => '^\\s*#\\s*!([^\\s]*\\b|.*env\\s+)ruby\\b',
 	'patterns' => [
 		{
 			'name'       => 'comment',