File Coverage

blib/lib/Rex/JobControl/Mojolicious/Command/jobcontrol.pm

Criterion	Covered	Total	%
statement	18	130	13.8
branch	0	40	0.0
condition	0	3	0.0
subroutine	6	21	28.5
pod			n/a
total	24	194	12.3

package Mojolicious::Command::jobcontrol;
$Mojolicious::Command::jobcontrol::VERSION = '0.6.0';
use Mojo::Base 'Mojolicious::Command';

use Data::Dumper;
use Digest::Bcrypt;
use Getopt::Long qw(GetOptionsFromArray :config no_auto_abbrev no_ignore_case);
use File::Path;
use File::Basename;

has description => 'JobControl Commands';
has usage => sub { shift->extract_usage };

sub run {
  my ( $self, $command, @args ) = @_;

  if ( !$command || $command eq "help" ) {
    print "Usage:\n";
    print "$0 jobcontrol  []\n";
    print "\n";
    print "Commands:\n";
    print "\n";
    print "setup         create all required folders\n";
    print "systemd -c    create systemd unit files for JobControl and Minion.\n";
    print "upstart -c    create upstart files for JobControl and Minion.\n";
    print "\n";
    print "User Management:\n";
    print "\n";
    print "adduser -u username -p password        will add a new user\n";
    print "deluser -u username                    will delete a user\n";
    print "listuser                               list all users\n";
    exit 0;
  }

  if($command eq "systemd") {
    my ($create_unit, $start, $stop, $restart) = @_;

    GetOptionsFromArray \@args,
      'c|create-unit'     => sub { $create_unit = 1 },
      's|start' => sub { $start = 1 },
      'r|restart' => sub { $restart = 1 },
      'k|stop' => sub { $stop = 1 };

    if($create_unit) {
      $self->create_systemd_unit();
    }
  }

  if($command eq "upstart") {
    my ($create_unit, $start, $stop, $restart) = @_;

    GetOptionsFromArray \@args,
      'c|create-unit'     => sub { $create_unit = 1 },
      's|start' => sub { $start = 1 },
      'r|restart' => sub { $restart = 1 },
      'k|stop' => sub { $stop = 1 };

    if($create_unit) {
      $self->create_upstart_unit();
    }
  }

  if ( $command eq "setup" ) {
    my $changed = 0;

    if(! -d $self->app->config->{project_path} ) {
      $self->app->log->info("Creating project path: " . $self->app->config->{project_path});
      File::Path::make_path($self->app->config->{project_path});
      $changed = 1;
    }

    if(! -d dirname($self->app->config->{minion_db_file}) ) {
      $self->app->log->info("Creating minion db path: " . dirname($self->app->config->{minion_db_file}));
      File::Path::make_path(dirname($self->app->config->{minion_db_file}));
      $changed = 1;
    }

    if(! -d dirname($self->app->config->{auth}->{passwd}) ) {
      $self->app->log->info("Creating passwd path: " . dirname($self->app->config->{auth}->{passwd}));
      File::Path::make_path(dirname($self->app->config->{auth}->{passwd}));
      $changed = 1;
    }

    if(! -f $self->app->config->{auth}->{passwd}) {
      $self->app->log->info("No passwd file found. Creating one with user 'admin' and password 'admin'.");
      $self->add_user("admin", "admin");
      $changed = 1;
    }

    if($changed == 0) {
      $self->app->log->info("Everything seems ok.");
    }

  }

  if ( $command eq "adduser" ) {
    my ( $user, $password );

    GetOptionsFromArray \@args,
      'u|user=s'     => sub { $user     = $_[1] },
      'p|password=s' => sub { $password = $_[1] };

    $self->add_user($user, $password);
  }

  if ( $command eq "deluser" ) {

    my $user;

    GetOptionsFromArray \@args, 'u|user=s' => sub { $user = $_[1] };

    my @lines =
      grep { !m/^$user:/ }
      eval { local (@ARGV) = ( $self->app->config->{auth}->{passwd} ); <>; };

    open( my $fh, ">", $self->app->config->{auth}->{passwd} ) or die($!);
    print $fh join( "\n", @lines );
    close($fh);
  }

  if ( $command eq "listuser" ) {
    my @lines =
      eval { local (@ARGV) = ( $self->app->config->{auth}->{passwd} ); <>; };
    chomp @lines;

    for my $l (@lines) {
      my ( $user, $pass ) = split( /:/, $l );
      print "> $user\n";
    }
  }

}

sub add_user {
  my ($self, $user, $password) = @_;

  $self->app->log->debug("Creating new user $user with password $password");

  my $salt = $self->app->config->{auth}->{salt};
  my $cost = $self->app->config->{auth}->{cost};

  my $bcrypt = Digest::Bcrypt->new;
  $bcrypt->salt($salt);
  $bcrypt->cost($cost);
  $bcrypt->add($password);

  my $pw = $bcrypt->hexdigest;

  open( my $fh, ">>", $self->app->config->{auth}->{passwd} ) or die($!);
  print $fh "$user:$pw\n";
  close($fh);

}

sub create_upstart_unit {
  my ($self) = @_;

  my $pid_file = $self->app->config->{hypnotoad}->{pid_file};

  my $whereis_hypnotoad = qx{which hypnotoad};
  chomp $whereis_hypnotoad;

  open(my $fh, ">", "/etc/init/rex-jobcontrol.conf") or die ($!);
  print $fh qq~# Rex::JobControl

description     "Rex::JobControl upstart job"

start on runlevel [2345]
stop on runlevel [!2345]

pre-start exec $whereis_hypnotoad $0
post-stop exec $whereis_hypnotoad $0 -s
~;
  close($fh);

  open(my $fh, ">", "/etc/init/rex-jobcontrol-minion.conf") or die ($!);
  print $fh qq~# Rex::JobControl Minion

description     "Rex::JobControl Minion upstart job"

start on runlevel [2345]
stop on runlevel [!2345]

exec /usr/local/bin/rex_job_control minion worker
~;
  close($fh);

}

sub create_systemd_unit {
  my ($self) = @_;

  my $pid_file = $self->app->config->{hypnotoad}->{pid_file};

  my $whereis_hypnotoad = qx{which hypnotoad};
  chomp $whereis_hypnotoad;

  open(my $fh, ">", "/lib/systemd/system/rex-jobcontrol.service") or die ($!);
  print $fh qq~[Unit]
Description=Rex JobControl Server
After=network.target

[Service]
Type=simple
SyslogIdentifier=rex-jobcontrol
PIDFile=$pid_file
ExecStart=$whereis_hypnotoad -f $0
ExecStop=$whereis_hypnotoad -s $0
ExecReload=$whereis_hypnotoad $0
~;
  close($fh);

  open(my $fh_m, ">", "/lib/systemd/system/rex-jobcontrol-minion.service") or die ($!);
  print $fh_m qq~[Unit]
Description=Rex JobControl Minion
After=network.target

[Service]
Type=simple
SyslogIdentifier=rex-jobcontrol-minion
ExecStart=$0 minion worker
~;
  close($fh_m);

}

1;