Fix passing correct key, re-add getopt trait as this is needed, my bad
[catagits/Catalyst-Runtime.git] / lib / Catalyst / Script / Server.pm
index 2795439..87e7b24 100644 (file)
@@ -2,102 +2,186 @@ package Catalyst::Script::Server;
 
 BEGIN {
     $ENV{CATALYST_ENGINE} ||= 'HTTP';
-    $ENV{CATALYST_SCRIPT_GEN} = 31;
     require Catalyst::Engine::HTTP;
 }
 
-use FindBin qw/$Bin/;
-use lib "$Bin/../lib";
-use Pod::Usage;
 use Moose;
-use Catalyst::Engine::HTTP;
-use namespace::clean -except => [ qw(meta) ];
+use Catalyst::Restarter;
+use MooseX::Types::Moose qw/ArrayRef Str Bool Int/;
+use namespace::autoclean;
 
-with 'MooseX::Getopt';
+with 'Catalyst::ScriptRole';
 
-has help => ( 
+has debug => (
     traits => [qw(Getopt)],
-    cmd_aliases => 'h',
-    isa => 'Bool',   
-    is => 'ro', 
-    required => 0, 
-    default => 0,  
+    cmd_aliases => 'd',
+    isa => Bool,
+    is => 'ro',
+    documentation => q{Force debug mode},
 );
 
-has host => ( 
-    isa => 'Str',    
-    is => 'ro', 
-    required => 0, 
-    default =>  "localhost" 
+has host => (
+    traits => [qw(Getopt)],
+    isa => Str,
+    is => 'ro',
+    default => 'localhost',
+    documentation => 'Specify a host for the server to run on',
 );
 
-has fork => ( 
+has fork => (
     traits => [qw(Getopt)],
     cmd_aliases => 'f',
-    isa => 'Bool',
-    is => 'ro', 
-    required => 0 
+    isa => Bool,
+    is => 'ro',
+    documentation => 'Fork the server',
 );
 
-has listen => ( 
+has listen => (
     traits => [qw(Getopt)],
     cmd_aliases => 'l',
-    isa => 'Int',
-    is => 'ro', 
-    required => 0, 
-    default => "3000" 
+    isa => Int,
+    is => 'ro',
+    default => 3000,
+    documentation => 'Specify a different listening port',
 );
 
-has pidfile => ( 
+has pidfile => (
     traits => [qw(Getopt)],
-    cmd_aliases => 'pf',
-    isa => 'Str',    
-    is => 'ro', 
-    required => 0 
+    cmd_aliases => 'pid',
+    isa => Str,
+    is => 'ro',
+    documentation => 'Specify a pidfile',
 );
 
-has keepalive => ( 
+has keepalive => (
     traits => [qw(Getopt)],
     cmd_aliases => 'k',
-    isa => 'Bool',   
-    is => 'ro', 
-    required => 0, 
-    default => 0 
+    isa => Bool,
+    is => 'ro',
+    documentation => 'Server keepalive',
+
+);
+
+has background => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'bg',
+    isa => Bool,
+    is => 'ro',
+    documentation => 'Run in the background',
+);
+
+has restart => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'r',
+    isa => Bool,
+    is => 'ro',
+    documentation => 'use Catalyst::Restarter to detect code changes',
+);
+
+has restart_directory => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'rdir',
+    isa => ArrayRef[Str],
+    is  => 'ro',
+    predicate => '_has_restart_directory',
+    documentation => 'Restarter directory to watch',
+);
+
+has restart_delay => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'rdel',
+    isa => Int,
+    is => 'ro',
+    predicate => '_has_restart_delay',
+    documentation => 'Set a restart delay',
+);
+
+has restart_regex => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'rxp',
+    isa => Str,
+    is => 'ro',
+    predicate => '_has_restart_regex',
+    documentation => 'Restart regex',
 );
 
-has background      => ( isa => 'Bool',   is => 'ro', required => 0 );
-has app             => ( isa => 'Str',    is => 'ro', required => 1 ); # THIS IS FUCKING RETARDED HALP PLZ
-has restart         => ( isa => 'Bool',   is => 'ro', required => 0 );
-has restart_delay   => ( isa => 'Int',    is => 'ro', required => 0 );
-has restart_regex   => ( isa => 'Str',    is => 'ro', required => 0 );
-has follow_symlinks => ( isa => 'Bool',   is => 'ro', required => 0 );
+has follow_symlinks => (
+    traits => [qw(Getopt)],
+    cmd_aliases => 'sym',
+    isa => Bool,
+    is => 'ro',
+    predicate => '_has_follow_symlinks',
+    documentation => 'Follow symbolic links',
 
-my @argv = @ARGV;
+);
 
 sub run {
-    my $self = shift;
-    
-    pod2usage() if $self->help;
-    my $app = $self->app;
-    Class::MOP::load_class($app);
-    $app->run(
-        $self->listen, $self->host,
-        {  
-           'fork'     => $self->fork,
-           keepalive  => $self->keepalive,
-           background => $self->background,
-           pidfile    => $self->pidfile,
-           keepalive         => $self->keepalive,
-           restart           => $self->restart,
-           restart_delay     => $self->restart_delay,
-           restart_regex     => qr/$self->restart_regex/,
-# FIXME    restart_directory => $self->restart_directory,
-           follow_symlinks   => $self->follow_symlinks,
-        }  
-    );
+    my ($self) = shift;
+
+    if ( $self->debug ) {
+        $ENV{CATALYST_DEBUG} = 1;
+    }
+
+    # If we load this here, then in the case of a restarter, it does not
+    # need to be reloaded for each restart.
+    require Catalyst;
+
+    # If this isn't done, then the Catalyst::Devel tests for the restarter
+    # fail.
+    $| = 1 if $ENV{HARNESS_ACTIVE};
+
+    if ( $self->restart ) {
+        die "Cannot run in the background and also watch for changed files.\n"
+            if $self->background;
+
+        require Catalyst::Restarter;
+
+        my $subclass = Catalyst::Restarter->pick_subclass;
+
+        my %args;
+        $args{follow_symlinks} = $self->follow_symlinks
+            if $self->_has_follow_symlinks;
+        $args{directories}     = $self->restart_directory
+            if $self->_has_restart_directory;
+        $args{sleep_interval}  = $self->restart_delay
+            if $self->_has_restart_delay;
+        $args{filter} = qr/$self->restart_regex/
+            if $self->_has_restart_regex;
+
+        my $restarter = $subclass->new(
+            %args,
+            start_sub => sub { $self->_run },
+            argv      => \$self->ARGV,
+        );
+
+        $restarter->run_and_watch;
+    }
+    else {
+        $self->_run_application;
+    }
+
 
 }
 
+sub _application_args {
+    my ($self) = shift;
+    return (
+        $self->listen,
+        $self->host,
+        {
+           map { $_ => $self->$_ } qw/
+                fork
+                keepalive
+                background
+                pidfile
+                keepalive
+                follow_symlinks
+            /,
+        },
+    );
+}
+
+__PACKAGE__->meta->make_immutable;
 
 1;
 
@@ -110,27 +194,27 @@ sub run {
 [% appprefix %]_server.pl [options]
 
  Options:
-   -d -debug          force debug mode
-   -f -fork           handle each request in a new process
-                      (defaults to false)
-   -? -help           display this help and exits
-      -host           host (defaults to all)
-   -p -port           port (defaults to 3000)
-   -k -keepalive      enable keep-alive connections
-   -r -restart        restart when files get modified
+   -d     --debug          force debug mode
+   -f     --fork           handle each request in a new process
                       (defaults to false)
-   -rd -restartdelay  delay between file checks
+   -h     --help           display this help and exits
+          --host           host (defaults to all)
+   -p     --port           port (defaults to 3000)
+   -k     --keepalive      enable keep-alive connections
+   -r     --restart        restart when files get modified
+                       (defaults to false)
+   --rd   --restartdelay  delay between file checks
                       (ignored if you have Linux::Inotify2 installed)
-   -rr -restartregex  regex match files that trigger
+   --rr   --restartregex  regex match files that trigger
                       a restart when modified
                       (defaults to '\.yml$|\.yaml$|\.conf|\.pm$')
-   -restartdirectory  the directory to search for
+   --rdir --restartdirectory  the directory to search for
                       modified files, can be set mulitple times
                       (defaults to '[SCRIPT_DIR]/..')
-   -follow_symlinks   follow symlinks in search directories
+   --sym  --follow_symlinks   follow symlinks in search directories
                       (defaults to false. this is a no-op on Win32)
-   -background        run the process in the background
-   -pidfile           specify filename for pid file
+   --bg   --background        run the process in the background
+   --pid  --pidfile           specify filename for pid file
 
  See also:
    perldoc Catalyst::Manual