allow overriding of public cgi-bin paths
[catagits/Catalyst-Controller-WrapCGI.git] / lib / Catalyst / Controller / CGIBin.pm
index b2eaab0..5859ad8 100644 (file)
@@ -3,13 +3,18 @@ package Catalyst::Controller::CGIBin;
 use strict;
 use warnings;
 
-use Class::C3;
-use URI::Escape;
+use MRO::Compat;
+use mro 'c3';
 use File::Slurp 'slurp';
 use File::Find::Rule ();
-use Cwd;
 use Catalyst::Exception ();
-use File::Spec::Functions 'splitdir';
+use File::Spec::Functions qw/splitdir abs2rel/;
+use IPC::Open3;
+use Symbol 'gensym';
+use List::MoreUtils 'any';
+use IO::File ();
+use Carp;
+use namespace::clean -except => 'meta';
 
 use parent 'Catalyst::Controller::WrapCGI';
 
@@ -19,11 +24,11 @@ Catalyst::Controller::CGIBin - Serve CGIs from root/cgi-bin
 
 =head1 VERSION
 
-Version 0.001
+Version 0.006
 
 =cut
 
-our $VERSION = '0.001';
+our $VERSION = '0.006';
 
 =head1 SYNOPSIS
 
@@ -52,7 +57,10 @@ In your .conf:
 
 =head1 DESCRIPTION
 
-Dispatches to executable CGI files in root/cgi-bin for /cgi-bin/ paths.
+Dispatches to CGI files in root/cgi-bin for /cgi-bin/ paths.
+
+Unlike L<ModPerl::Registry> this module does _NOT_ stat and recompile the CGI
+for every invocation. If this is something you need, let me know.
 
 CGI paths are converted into action names using cgi_action (below.)
 
@@ -70,41 +78,37 @@ module for configuration information.
 sub register_actions {
     my ($self, $app) = @_;
 
-    my $cwd = getcwd;
-
     my $cgi_bin = $app->path_to('root', 'cgi-bin');
 
-    chdir $cgi_bin ||
-        Catalyst::Exception->throw(
-            message => 'You have no root/cgi-bin directory'
-        );
-
     my $namespace = $self->action_namespace($app);
 
     my $class = ref $self || $self;
 
-    for my $file (File::Find::Rule->executable->file->in(".")) {
+    for my $file (File::Find::Rule->file->in($cgi_bin)) {
+        my $cgi_path = abs2rel($file, $cgi_bin);
+
+        next if any { $_ eq '.svn' } splitdir $cgi_path;
+
+        my $path        = join '/' => splitdir($cgi_path);
+        my $action_name = $self->cgi_action($path);
+        my $public_path = $self->cgi_path($path);
+        my $reverse     = $namespace ? "$namespace/$action_name" : $action_name;
+        my $attrs       = { Path => [ $public_path ], Args => [ 0 ] };
+
         my ($cgi, $type);
-        my $code = do { no warnings; eval 'sub { '.slurp($file).' }' };
 
-        if (!$@) {
-            $cgi = $code;
+        if ($self->is_perl_cgi($file)) { # syntax check passed
             $type = 'Perl';
+            $cgi  = $self->wrap_perl_cgi($file, $action_name);
         } else {
-            $cgi = sub { system "$cgi_bin/$file" };
             $type = 'Non-Perl';
-            undef $@;
+            $cgi  = $self->wrap_nonperl_cgi($file, $action_name);
         }
 
-        $app->log->info("Registering root/cgi-bin/$file as a $type CGI.")
+        $app->log->info("Registering root/cgi-bin/$cgi_path as a $type CGI.")
             if $app->debug;
 
-        my $action_name = $self->cgi_action($file);
-        my $path        = join '/' => splitdir($file);
-        my $reverse     = $namespace ? "$namespace/$action_name" : $action_name;
-        my $attrs       = { Path => [ "cgi-bin/$path" ], Args => [ 0 ] };
-
-        $code = sub {
+        my $code = sub {
             my ($controller, $context) = @_;
             $controller->cgi_to_response($context, $cgi)
         };
@@ -121,29 +125,136 @@ sub register_actions {
         $app->dispatcher->register($app, $action);
     }
 
-    chdir $cwd;
-
     $self->next::method($app, @_);
+
+# Tell Static::Simple to ignore the cgi-bin dir.
+    if (!any{ $_ eq 'cgi-bin' } @{ $app->config->{static}{ignore_dirs}||[] }) {
+        push @{ $app->config->{static}{ignore_dirs} }, 'cgi-bin';
+    }
 }
 
 =head1 METHODS
 
-=head2 $self->cgi_action($cgi_path)
+=head2 $self->cgi_action($cgi)
 
 Takes a path to a CGI from C<root/cgi-bin> such as C<foo/bar.cgi> and returns
-the action name it is registered as.
+the action name it is registered as. See L</DESCRIPTION> for a discussion on how
+CGI actions are named.
 
 =cut
 
 sub cgi_action {
     my ($self, $cgi) = @_;
 
-    my $action_name = 'CGI_' . join '_' => splitdir($cgi);
+    my $action_name = 'CGI_' . join '_' => split '/' => $cgi;
     $action_name    =~ s/\W/_/g;
 
     $action_name
 }
 
+=head2 $self->cgi_path($cgi)
+
+Takes a path to a CGI from C<root/cgi-bin> such as C<foo/bar.cgi> and returns
+the public path it should be registered under.
+
+The default is C<cgi-bin/$cgi>.
+
+=cut
+
+sub cgi_path {
+    my ($self, $cgi) = @_;
+    return "cgi-bin/$cgi";
+}
+
+=head2 $self->is_perl_cgi($path)
+
+Tries to figure out whether the CGI is Perl or not.
+
+If it's Perl, it will be inlined into a sub instead of being forked off, see
+wrap_perl_cgi (below.)
+
+If it's not doing what you expect, you might want to override it, and let me
+know as well!
+
+=cut
+
+sub is_perl_cgi {
+    my ($self, $cgi) = @_;
+
+    my $shebang = IO::File->new($cgi)->getline;
+
+    return 0 if $shebang !~ /perl/ && $cgi !~ /\.pl\z/;
+
+    my $taint_check = $shebang =~ /-T/ ?  '-T' : '';
+
+    open NULL, '>', File::Spec->devnull;
+    my $pid = open3(gensym, '&>NULL', '&>NULL', "$^X $taint_check -c $cgi");
+    close NULL;
+    waitpid $pid, 0;
+
+    $? >> 8 == 0
+}
+
+=head2 $self->wrap_perl_cgi($path, $action_name)
+
+Takes the path to a Perl CGI and returns a coderef suitable for passing to
+cgi_to_response (from L<Catalyst::Controller::WrapCGI>.)
+
+C<$action_name> is the generated name for the action representing the CGI file.
+
+This is similar to how L<ModPerl::Registry> works, but will only work for
+well-written CGIs. Otherwise, you may have to override this method to do
+something more involved (see L<ModPerl::PerlRun>.)
+
+=cut
+
+sub wrap_perl_cgi {
+    my ($self, $cgi, $action_name) = @_;
+
+    my $code = slurp $cgi;
+
+    $code =~ s/^__DATA__\r?\n(.*)//ms;
+    my $data = $1;
+
+    my $coderef = do {
+        no warnings;
+        eval ' 
+            package Catalyst::Controller::CGIBin::_CGIs_::'.$action_name.';
+            sub {'
+                . 'local *DATA;'
+                . q{open DATA, '<', \$data;}
+                . $code
+         . '}';
+    };
+
+    croak __PACKAGE__ . ": Could not compile $cgi to coderef: $@" if $@;
+
+    $coderef
+}
+
+=head2 $self->wrap_nonperl_cgi($path, $action_name)
+
+Takes the path to a non-Perl CGI and returns a coderef for executing it.
+
+C<$action_name> is the generated name for the action representing the CGI file.
+
+By default returns:
+
+    sub { system $path }
+
+=cut
+
+sub wrap_nonperl_cgi {
+    my ($self, $cgi, $action_name) = @_;
+
+    sub { system $cgi }
+}
+
+=head1 SEE ALSO
+
+L<Catalyst::Controller::WrapCGI>, L<CatalystX::GlobalContext>,
+L<Catalyst::Controller>, L<CGI>, L<Catalyst>
+
 =head1 AUTHOR
 
 Rafael Kitover, C<< <rkitover at cpan.org> >>