| [93] | 1 | #!/usr/bin/perl -w |
|---|
| 2 | use strict; |
|---|
| [54] | 3 | |
|---|
| [93] | 4 | use FindBin; |
|---|
| [103] | 5 | use lib "$FindBin::RealBin/../lib"; |
|---|
| [65] | 6 | |
|---|
| [94] | 7 | use Getopt::Long; |
|---|
| 8 | use YAML; |
|---|
| [93] | 9 | use Plack::Runner; |
|---|
| [103] | 10 | use File::Pid; |
|---|
| [115] | 11 | use File::Spec::Functions qw{catfile rel2abs}; |
|---|
| 12 | use Cwd; |
|---|
| [93] | 13 | |
|---|
| [94] | 14 | GetOptions( |
|---|
| [115] | 15 | 'file=s' => \my $CONFIG_FILE, |
|---|
| 16 | 'verbose|v' => \my $VERBOSE, |
|---|
| 17 | 'D=s' => \my %DEFINES, |
|---|
| [94] | 18 | ); |
|---|
| [93] | 19 | |
|---|
| [103] | 20 | my %run = ( |
|---|
| [110] | 21 | run => \&run_server, |
|---|
| 22 | start => \&start_server, |
|---|
| [115] | 23 | stop => sub { signal_server('QUIT') }, |
|---|
| 24 | restart => sub { signal_server('HUP') }, |
|---|
| [104] | 25 | ); |
|---|
| [94] | 26 | |
|---|
| [115] | 27 | my %default_config = ( |
|---|
| 28 | server_root => cwd, |
|---|
| 29 | port => 5000, |
|---|
| 30 | access_log => 'access', |
|---|
| 31 | error_log => 'error', |
|---|
| 32 | pid_file => 'pid', |
|---|
| 33 | ); |
|---|
| 34 | |
|---|
| [110] | 35 | my $command = shift or die "Usage: $0 <" . join('|', keys %run) . ">\n"; |
|---|
| 36 | |
|---|
| [104] | 37 | exists $run{$command} or die "Unrecognized command $command\n"; |
|---|
| [115] | 38 | $run{$command}->(); |
|---|
| [103] | 39 | |
|---|
| [115] | 40 | sub get_config { |
|---|
| 41 | $CONFIG_FILE ||= 'server.yml'; |
|---|
| 42 | my $config_from_file = -e $CONFIG_FILE ? YAML::LoadFile($CONFIG_FILE) : {}; |
|---|
| 43 | |
|---|
| 44 | my $config = { |
|---|
| 45 | %default_config, |
|---|
| 46 | %{ $config_from_file }, |
|---|
| 47 | %DEFINES, |
|---|
| 48 | }; |
|---|
| 49 | |
|---|
| 50 | # make config paths absolute before handing off to the app |
|---|
| 51 | for my $file_key (qw{dbname htdigest access_log error_log pid_file}) { |
|---|
| 52 | if ($config->{$file_key}) { |
|---|
| 53 | $config->{$file_key} = rel2abs($config->{$file_key}, $config->{server_root}); |
|---|
| 54 | } |
|---|
| 55 | } |
|---|
| 56 | |
|---|
| 57 | warn Dump($config) if $VERBOSE; |
|---|
| 58 | |
|---|
| 59 | return $config; |
|---|
| 60 | } |
|---|
| 61 | |
|---|
| [110] | 62 | sub run_server { |
|---|
| [115] | 63 | my $config = get_config(); |
|---|
| 64 | |
|---|
| [104] | 65 | require BookmarksApp; |
|---|
| 66 | my $app = BookmarksApp->new({ config => $config })->to_app; |
|---|
| [103] | 67 | |
|---|
| [110] | 68 | my $runner = Plack::Runner->new(server => 'Starman'); |
|---|
| 69 | $runner->parse_options( |
|---|
| [115] | 70 | '--listen', ':' . $config->{port}, |
|---|
| [110] | 71 | ); |
|---|
| 72 | $runner->run($app); |
|---|
| 73 | } |
|---|
| 74 | |
|---|
| 75 | sub start_server { |
|---|
| [115] | 76 | my $config = get_config(); |
|---|
| 77 | |
|---|
| [110] | 78 | require BookmarksApp; |
|---|
| 79 | my $app = BookmarksApp->new({ config => $config })->to_app; |
|---|
| 80 | |
|---|
| [104] | 81 | my $runner = Plack::Runner->new(server => 'Starman'); |
|---|
| 82 | $runner->parse_options( |
|---|
| 83 | '--daemonize', |
|---|
| [115] | 84 | '--listen', ':' . $config->{port}, |
|---|
| 85 | '--pid', $config->{pid_file}, |
|---|
| 86 | '--error-log', $config->{error_log}, |
|---|
| 87 | '--access-log', $config->{access_log}, |
|---|
| [104] | 88 | ); |
|---|
| 89 | $runner->run($app); |
|---|
| 90 | } |
|---|
| [103] | 91 | |
|---|
| [115] | 92 | sub signal_server { |
|---|
| 93 | my $config = get_config(); |
|---|
| 94 | my $signal = shift; |
|---|
| [103] | 95 | |
|---|
| [115] | 96 | my $pid_path = $config->{pid_file}; |
|---|
| [104] | 97 | unless (-e $pid_path) { |
|---|
| 98 | warn "$pid_path does not exist\n"; |
|---|
| 99 | return; |
|---|
| 100 | } |
|---|
| 101 | |
|---|
| 102 | my $pid_file = File::Pid->new({ |
|---|
| 103 | file => $pid_path, |
|---|
| 104 | }); |
|---|
| 105 | |
|---|
| 106 | if (my $pid = $pid_file->running) { |
|---|
| [115] | 107 | kill $signal, $pid; |
|---|
| [104] | 108 | } |
|---|
| 109 | } |
|---|