148 lines
3.0 KiB
Perl
148 lines
3.0 KiB
Perl
#!/usr/bin/perl
|
|
|
|
# runit sv utility ignores evironment variable SVDIR
|
|
# and unable to list services, fully relying to user this knowlege
|
|
|
|
# TODO handle root user case as system mode!
|
|
# TODO handle system mode correctly
|
|
|
|
use strict;
|
|
use warnings "all";
|
|
use JSON::PP;
|
|
|
|
sub help;
|
|
sub ctrl ($);
|
|
sub status;
|
|
|
|
my $sep = $/;
|
|
$/ = '';
|
|
|
|
open (CONF, "$ENV{HOME}/.local/etc/svc.json") or die "No config at $ENV{HOME}/.local/etc/svc.json\n";
|
|
|
|
my $c = <CONF>;
|
|
close CONF;
|
|
$/ = $sep;
|
|
|
|
$c = decode_json($c);
|
|
|
|
if ($c->{mode} eq 'user') {
|
|
die "\$HOME is no set\n" unless (defined($ENV{HOME}));
|
|
die "$ENV{HOME} does not exist" unless (-d $ENV{HOME});
|
|
chdir $ENV{HOME};
|
|
die "No such directory $ENV{HOME}$c->{dir}\n" unless(-d "$ENV{HOME}$c->{dir}");
|
|
chdir "$ENV{HOME}$c->{dir}";
|
|
} else {
|
|
die "No such directory $c->{dir}" unless(-d $c->{dir});
|
|
chdir $c->{dir};
|
|
}
|
|
|
|
unless(defined($ARGV[0])) {
|
|
help;
|
|
} else {
|
|
if ($ARGV[0] ne 'list') {
|
|
help unless (defined($ARGV[1]));
|
|
}
|
|
}
|
|
|
|
if ($ARGV[0] eq 'list') {
|
|
my $d;
|
|
|
|
if (defined($ARGV[1]) && $ARGV[1] eq '--all') {
|
|
$d = "$ENV{HOME}$c->{servicedir}";
|
|
} else {
|
|
$d = "$ENV{HOME}$c->{dir}";
|
|
}
|
|
|
|
opendir(D, $d) or die "Unable to list services\n";
|
|
print join("\n", sort(grep {!/^\.{1,2}$/} readdir(D))) . "\n";
|
|
closedir D;
|
|
} elsif ($ARGV[1] eq 'disable') {
|
|
my $d = "$ENV{HOME}$c->{dir}/$ARGV[0]";
|
|
|
|
if (-l $d) {
|
|
unlink("$d") or die "Unable to disable $ARGV[0]\n";
|
|
} else {
|
|
die "No such service $ARGV[0]\n";
|
|
}
|
|
} elsif ($ARGV[1] eq 'enable') {
|
|
my $d = "$ENV{HOME}$c->{dir}/$ARGV[0]";
|
|
|
|
if (-l $d) {
|
|
die "Service $ARGV[0] already enabled\n";
|
|
} else {
|
|
my $svcdir = "$ENV{HOME}$c->{servicedir}/$ARGV[0]";
|
|
|
|
if (-d $svcdir) {
|
|
symlink($svcdir, $d) or die "Unable to enable service $ARGV[0]\n";
|
|
} else {
|
|
die "No such service $ARGV[0]\n";
|
|
}
|
|
}
|
|
} elsif ((-l "./$ARGV[0]") || (-d "./$ARGV[0]")) {
|
|
chdir "./$ARGV[0]";
|
|
die "Is service running?\n" unless (-d './supervise');
|
|
chdir './supervise';
|
|
|
|
if ($ARGV[1] eq 'start') {
|
|
die "Service $ARGV[0] already running.\n" if (status() eq 'run');
|
|
ctrl('u');
|
|
} elsif ($ARGV[1] eq 'restart') {
|
|
die "Service $ARGV[0] already stopped.\n" if (status() eq 'down');
|
|
ctrl('d');
|
|
|
|
for (my $cnt = 0; $cnt < $c->{timeout}; $cnt++) {
|
|
if (status() eq 'down') {
|
|
last;
|
|
}
|
|
|
|
sleep 1;
|
|
}
|
|
|
|
ctrl('k') if (status() ne 'down');
|
|
sleep 1;
|
|
ctrl('u');
|
|
} elsif ($ARGV[1] eq 'stop') {
|
|
die "Service $ARGV[0] already stopped.\n" if (status() eq 'down');
|
|
ctrl('d');
|
|
|
|
for (my $cnt = 0; $cnt < $c->{timeout}; $cnt++) {
|
|
if (status() eq 'down') {
|
|
last;
|
|
}
|
|
|
|
sleep 1;
|
|
}
|
|
|
|
ctrl('k') if (status() ne 'down');
|
|
sleep 1;
|
|
} elsif ($ARGV[1] eq 'status') {
|
|
print status() . "\n";
|
|
}
|
|
} else {
|
|
die "No such service.\n";
|
|
}
|
|
|
|
exit 0;
|
|
|
|
sub help {
|
|
print "Usage $0 SERVICE [start|stop|restart|status|enable|disable]\n";
|
|
print " $0 list [--all]\n";
|
|
exit 0;
|
|
}
|
|
|
|
sub ctrl ($) {
|
|
my $command = shift;
|
|
open (C, ">", "control") or die;
|
|
print C $command;
|
|
close C;
|
|
}
|
|
|
|
sub status {
|
|
open (S, "stat") or die;
|
|
my $status = <S>;
|
|
chomp($status);
|
|
close S;
|
|
return $status;
|
|
}
|
|
# vim: set ft=perl noet ai :
|