Initial version
This commit is contained in:
eleksir
190
bin/meow-irc-bot
Executable file
190
bin/meow-irc-bot
Executable file
@@ -0,0 +1,190 @@
|
||||
#!/usr/bin/perl
|
||||
|
||||
use 5.030; ## no critic (ProhibitImplicitImport)
|
||||
use strict;
|
||||
use warnings;
|
||||
use utf8;
|
||||
use open qw (:std :utf8);
|
||||
use English qw ( -no_match_vars );
|
||||
use version; our $VERSION = qw (1.0);
|
||||
|
||||
my $workdir;
|
||||
|
||||
# before we run, change working dir
|
||||
BEGIN {
|
||||
use Cwd qw (chdir abs_path);
|
||||
my @CWD = split /\//, abs_path ($PROGRAM_NAME);
|
||||
|
||||
if ($#CWD > 1) {
|
||||
$#CWD = $#CWD - 2;
|
||||
}
|
||||
|
||||
$workdir = join '/', @CWD;
|
||||
chdir $workdir;
|
||||
}
|
||||
|
||||
use lib ("$workdir/lib", "$workdir/vendor_perl", "$workdir/vendor_perl/lib/perl5");
|
||||
no Cwd;
|
||||
use Cwd::utf8 qw (chdir abs_path);
|
||||
use POSIX qw (setgid setuid setsid);
|
||||
use File::Basename qw (basename dirname);
|
||||
use File::Path qw (make_path);
|
||||
use File::Spec ();
|
||||
use Log::Any qw ($log);
|
||||
use Log::Any::Adapter ();
|
||||
use Log::Any::For::Std ();
|
||||
use BotLib qw (SigIntHandler SigTermHandler SigQuitHandler);
|
||||
use BotLib::Conf qw (LoadConf);
|
||||
use IRCBot qw (RunIRCBot);
|
||||
|
||||
sub daemonize ();
|
||||
sub __fork ();
|
||||
|
||||
local $OUTPUT_AUTOFLUSH = 1;
|
||||
|
||||
my $piddir = '/var/run/' . basename (abs_path $PROGRAM_NAME);
|
||||
my $C = LoadConf ();
|
||||
my $loglevel = $C->{'loglevel'};
|
||||
$C->{loglevel} = 'notice' unless defined $C->{loglevel};
|
||||
my $logfile;
|
||||
# Global IRC Client state var/object $MAIN::IRC
|
||||
my $IRC; ## no critic (Variables::ProhibitUnusedVariables)
|
||||
|
||||
if (defined $C->{'log'}) {
|
||||
$logfile = $C->{'log'};
|
||||
} elsif (defined $C->{'debug_log'}) {
|
||||
$logfile = $C->{'debug_log'};
|
||||
}
|
||||
|
||||
# if we run under root, create pid dir and set correct owner for it, same for debug log if it is defined
|
||||
if ($UID == 0) {
|
||||
my $user = $C->{user};
|
||||
die "Unable to run as root, please define unprivileged user in config.json\n" unless (defined $user);
|
||||
my ($uid, $gid) = (getpwnam $user)[2,3];
|
||||
|
||||
unless (-d $piddir) {
|
||||
make_path (
|
||||
$piddir, {
|
||||
uid => $uid,
|
||||
group => $gid,
|
||||
},
|
||||
) or die "Unable to create $piddir: $OS_ERROR\n";
|
||||
}
|
||||
|
||||
if (defined $logfile) {
|
||||
my $logdir = dirname $logfile;
|
||||
|
||||
unless (-d $logdir) {
|
||||
make_path (
|
||||
$logdir, {
|
||||
uid => $uid,
|
||||
group => $gid,
|
||||
},
|
||||
) or die "Unable to create $logdir: $OS_ERROR\n";
|
||||
}
|
||||
}
|
||||
|
||||
# drop privs
|
||||
setgid ($gid) or die "Unable to switch to group of $user: $OS_ERROR\n";
|
||||
setuid ($uid) or die "Unable to switch to account $user: $OS_ERROR\n";
|
||||
} else {
|
||||
$piddir = $workdir;
|
||||
}
|
||||
|
||||
if (defined $logfile) {
|
||||
# loglevel warning suppresses logging from CHI.
|
||||
Log::Any::Adapter->set ('File', $logfile, log_level => $loglevel);
|
||||
|
||||
# Some annoying CHI logging events come with category set to CHI::Driver and log_level set to debug.
|
||||
# Until we use start using Log::Any native features and set whole loglevel accordingly, lets lower
|
||||
# verbosity paticulary for CHI.
|
||||
Log::Any::Adapter->set (
|
||||
{
|
||||
category => 'CHI::Driver',
|
||||
},
|
||||
'File',
|
||||
$logfile,
|
||||
log_level => 'warn',
|
||||
);
|
||||
} else {
|
||||
# no log defined so set loglevel to the least verbose
|
||||
Log::Any::Adapter->set ('File', File::Spec->devnull, 'log_level'=>'emergency');
|
||||
}
|
||||
|
||||
# double fork attach to /dev/null all stdio and go background
|
||||
if (defined $C->{daemonize} && $C->{daemonize}) {
|
||||
daemonize ();
|
||||
}
|
||||
|
||||
# write pidfile
|
||||
my $apath = abs_path $PROGRAM_NAME;
|
||||
|
||||
unless (defined $apath) {
|
||||
$log->fatal ("[FATAL] Unable to get abs_path: $OS_ERROR");
|
||||
exit 1;
|
||||
}
|
||||
|
||||
my $pidfile = sprintf '%s/%s.pid', $piddir, basename $apath;
|
||||
$log->info ("[INFO] Creating pidfile at $pidfile");
|
||||
|
||||
open (my $pid, '>', $pidfile) or do {
|
||||
$log->fatal ("[FATAL] Error opening $pidfile: $OS_ERROR");
|
||||
exit 1;
|
||||
};
|
||||
|
||||
print $pid $PID or do {
|
||||
$log->fatal ("[FATAL] Error writing to $pidfile: $OS_ERROR");
|
||||
exit 1;
|
||||
};
|
||||
|
||||
close $pid;
|
||||
|
||||
local $SIG{INT} = \&SigIntHandler;
|
||||
local $SIG{TERM} = \&SigTermHandler;
|
||||
local $SIG{QUIT} = \&SigQuitHandler;
|
||||
|
||||
# set correct proctitle
|
||||
$PROGRAM_NAME = abs_path $PROGRAM_NAME; ## no critic (Variables::RequireLocalizedPunctuationVars)
|
||||
|
||||
RunIRCBot ();
|
||||
|
||||
if (-e $pidfile) {
|
||||
unlink $pidfile;
|
||||
}
|
||||
|
||||
exit 0;
|
||||
|
||||
# Демонизирует программу (double fork() && detach from terminal, etc)
|
||||
# undef daemonize()
|
||||
sub daemonize () {
|
||||
__fork ();
|
||||
setsid ();
|
||||
open STDIN, '<', File::Spec->devnull; ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
|
||||
# we handle STDOUT && STDERR via Log::Any::For::Std, so do not close them explicitly
|
||||
#open (STDOUT, '>', '/dev/null'); ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
|
||||
#open (STDERR, '>', '/dev/null'); ## no critic (InputOutput::RequireCheckedSyscalls, InputOutput::RequireCheckedOpen)
|
||||
__fork ();
|
||||
umask 0;
|
||||
return;
|
||||
}
|
||||
|
||||
# Форкает программу и завершает родителя
|
||||
# undef __fork()
|
||||
sub __fork () {
|
||||
my $pid = fork;
|
||||
|
||||
unless (defined $pid) {
|
||||
$log->fatal ("[FATAL] Can't fork: $OS_ERROR");
|
||||
exit 1;
|
||||
}
|
||||
|
||||
if ($pid != 0) {
|
||||
exit 0;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
__END__
|
||||
|
||||
# vim: set ft=perl noet ai ts=4 sw=4 sts=4:
|
||||
Reference in New Issue
Block a user