--- /dev/null
+# xdcc autogetter, to automate searching and downloading xdcc packs based on chosen search strings
+# requires that you enable dcc_autoget and dcc_autoresume. Also requires File::Dir which can be installed with the command >>> sudo perl -MCPAN -e 'install "File::HomeDir"'
+# made because torrents are watched by the feds, but xdcc lacks RSS feeds.
+# if you encounter any problems, fix it yourself you lazy bastard (or get me to), then contact me so I can add your fix and bump that version #
+# BeerWare License. Use any part you want, but buy me a beer if you ever meet me and liked this hacked together broken PoS
+# Somewhat based off of DCC multiget by Kaveh Moini.
+# USE: for help : ag_help
+# to run : ag_run
+# to halt at next : ag_stop
+# to set the server: ag_server
+# to add a bot : ag_botadd BOT1 BOT2 *** BOTN
+# to remove a bot : ag_botrem BOT1 BOT2 *** BOTN
+# to add string : ag_add "[TEXT STRING OF TV SHOW/CHINESE CARTOON/ETC]","[ETC]",***,"[ETC]"
+# to remove strings: ag_rem "[TEXT STRING OF TV SHOW/CHINESE CARTOON/ETC]","[ETC]",***,"[ETC]"
+# ag_next_delay : delay between full transfers
+# ag_dcc_closed_retry_delay : delay after premature transfer
+# ag_bot_delay: delay between request and when you "SHOULD" get it
+# ag_interrun_delay : delay (in minutes, the rest seconds) between finishing a round and starting another
+# ag_autorun : whether to run on startup
+# ag_xdcc_send_prefix : the xdcc message before the pack #
+# ag_xdcc_find_prefix : the xdcc message before the search term
+# ag_bot_file : where your bot list is stored
+# ag_search_file : where your search list is stored
+
+use strict;
+
+use Irssi;
+use Text::ParseWords;
+use autodie; # die if problem reading or writing a file
+use File::HomeDir;
+use File::Copy;
+use Irssi 20090331;
+use vars qw($VERSION %IRSSI);
+
+$VERSION = 1.0;
+%IRSSI = (
+ name => "autoget",
+ description => "XDCC Autoget, for automated searching and downloading of xdcc packs",
+ license => "BeerWare Version 42",
+ changed => "$VERSION",
+ authors => "MarshalMeatball",
+ contact => "mobilegundamseed\@hotmail.com",
+);
+
+my @totags = (); #timeout tags (need to be purged between send requests maybe)
+
+my $nexdelay = 5; #delay for next pack
+my $dcrdelay = 10; #delay if transfer closed prematurely
+my $botdelay = 30; #max time to wait for the bot to respond
+my $exedelay = 15; #delay (in minutes) between finishing one run and starting another
+
+my $initflag = 1; #flag controls whether AG starts on IRSSI boot (if in autorun), or on LOAD
+my $msgflag = 1; #flag controls whether bot has responded to search request
+my $pact = 0; #3 state flag to avoid recursive ag_reqpack calls
+
+my $sendprefix = "xdcc send"; #virtually universal xdcc send, cancel, and find prefixes
+my $cancelprefix = "xdcc cancel";
+my $findprefix = "!find";
+my $botsfilename = File::HomeDir->my_home . "/.irssi/scripts/bots.txt"; #werks on my machine (tm).
+my $searchesfilename = File::HomeDir->my_home . "/.irssi/scripts/searches.txt";
+
+my $dccflag = 0; #flag so that dccs aren't mistakenly thought of belonging to AG
+
+my @terms; #lists of search terms, bots, and pack numbers (for current bot only)
+my @bots;
+my @packs;
+
+my $termcounter = 0; #counters for array position
+my $botcounter = 0;
+my $packcounter = 0;
+
+my $server; #current server
+
+sub ag_init #init system
+{
+ Irssi::print "AG | Autoget initiated";
+ Irssi::print "AG | /ag_help for help";
+ &ag_initserver;
+}
+
+sub ag_initserver #init server
+{
+ $server = Irssi::active_server(); #keep trying to get server until it works, then continue after 5 seconds
+ if ($server !~ m/^Irssi::Irc::Server=HASH/) {Irssi::timeout_add_once(1000, sub {&ag_initserver;} , []);}
+ else {Irssi::timeout_add_once(5000, sub {&ag_run;} , []);}
+}
+
+sub ag_help
+{
+ Irssi::print "for this help : ag_help";
+ Irssi::print "to run : ag_run";
+ Irssi::print "to halt at next : ag_stop";
+ Irssi::print "to set the server: ag_server";
+ Irssi::print "to add a bot : ag_botadd BOT1 BOT2 *** BOTN";
+ Irssi::print "to remove a bot : ag_botrem BOT1 BOT2 *** BOTN";
+ Irssi::print "to add string : ag_add \"[TEXT STRING OF SEARCH]\",\"[ETC]\",***,\"[ETC]\"";
+ Irssi::print "to remove strings: ag_rem \"[TEXT STRING OF SEARCH]\",\"[ETC]\",***,\"[ETC]\"";
+ Irssi::print "ag_next_delay : delay between full transfers";
+ Irssi::print "ag_dcc_closed_retry_delay: delay after premature transfer";
+ Irssi::print "ag_bot_delay : max time to wait for the bot to respond";
+ Irssi::print "ag_interrun_delay : delay (in minutes, the rest seconds) between finishing a round and starting another";
+ Irssi::print "ag_autorun : whether to run on startup";
+ Irssi::print "ag_xdcc_send_prefix : the xdcc message before the pack #";
+ Irssi::print "ag_xdcc_cancel_prefix : the xdcc message to cancel a transfer";
+ Irssi::print "ag_xdcc_find_prefix : the xdcc message before the search term";
+ Irssi::print "ag_bot_file : where your bot list is stored";
+ Irssi::print "ag_search_file : where your search list is stored}";
+}
+
+sub ag_server #should only be run when you have a server, or it'll break (probably unfixable without a bunch of ugly dumb checks everywhere)
+{
+ $server = Irssi::active_server();
+}
+
+sub ag_getbots #reads in bot list
+{
+ open(bots, "<", $botsfilename);
+ @bots = <bots>;
+ chomp(@bots);
+ close(bots);
+}
+
+sub ag_getterms #reads in search term list
+{
+ open(searches, "<", $searchesfilename);
+ @terms = <searches>;
+ chomp(@terms);
+ close(searches);
+}
+
+sub ag_search #searches current bot for current term
+{
+ $msgflag = 0;
+ Irssi::signal_add("message irc notice", "ag_getmsg");
+ $server->command("msg $bots[$botcounter] $findprefix $terms[$termcounter]");
+ Irssi::timeout_add_once($botdelay * 1000, sub { &ag_skip } , []); #skip search if no results given
+}
+
+sub ag_skip
+{
+ if ($msgflag == 0)
+ {
+ if ($#terms != $termcounter)
+ {
+ Irssi::print "AG | No packs found or Bot " . $bots[$botcounter] . " unresponsive or nonexistent. Skipping to next search";
+ Irssi::signal_remove("message irc notice", "ag_getmsg");
+ $termcounter++;
+ &ag_search;
+ }
+ elsif ($#bots != $botcounter)
+ {
+ Irssi::print "AG | No packs found or Bot " . $bots[$botcounter] . " unresponsive or nonexistent. Skipping to next bot";
+ Irssi::signal_remove("message irc notice", "ag_getmsg");
+ $termcounter = 0;
+ $botcounter++;
+ &ag_search;
+ }
+ else
+ {
+ Irssi::print "AG | No packs found or Bot " . $bots[$botcounter] . " unresponsive or nonexistent. End of list";
+ Irssi::signal_remove("message irc notice", "ag_getmsg");
+ $botcounter = 0;
+ $termcounter = 0;
+ Irssi::print "AG | Waiting " . $exedelay . " minutes until next search";
+ Irssi::timeout_add_once($exedelay * 1000 * 60, sub { &ag_run; } , []);
+ }
+ }
+}
+
+sub ag_getmsg #runs when bot sends privmsg. Avoid talking to bots to keep this from sending useless shit that breaks things
+{
+ my $message = @_[1];
+ my $botname = @_[2];
+ $botname =~ tr/[A-Z]/[a-z]/;
+ $bots[$botcounter] =~ tr/[A-Z]/[a-z]/;
+ if ($botname == $bots[$botcounter])
+ {
+ $msgflag = 1;
+ &parseresponse($message);
+ }
+}
+
+sub parseresponse #takes a single message and finds all instances of "#[XDCC NUMBER]:" since most bots reply to !find requests like that. If a bot uses another method and this doesn't work, fix it yourself and send me the code =D
+{
+ my($message) = @_;
+ my @temp = split(' ', $message);
+ foreach my $n (@temp){ if ($n =~ m{#(\d+):}) {push(@packs, $1);} }
+ @packs = ag_uniq(@packs);
+ if ($pact == 0 and $#packs >= 0 and $packs[$packcounter] ne "") #initiallizes the actual xdcc get system only once per search term/bot (pact should be >0 until the whole process is finished)
+ {
+ $pact = 1;
+ &ag_reqpack();
+ }
+}
+
+sub ag_uniq #only returns unique entries
+{
+ my %seen;
+ grep !$seen{$_}++, @_;
+}
+
+sub ag_reqpack #sends the xdcc send request, and retries on failure
+{
+ if ($dccflag == 0){Irssi::signal_add("dcc get receive", "ag_opendcc");} #init DCC recieve init flag
+ $dccflag = 1;
+ $server->command("msg $bots[$botcounter] $sendprefix $packs[$packcounter]");
+ push(@totags, Irssi::timeout_add_once($botdelay * 1000, sub { if ($pact < 2) { Irssi::print "AG | Connection failed"; Irssi::print "AG | retrying: " . $bots[$botcounter] . " " .$packs[$packcounter]; &ag_reqpack(); } } , []));
+}
+
+sub ag_opendcc #runs on DCC recieve init
+{
+ Irssi::signal_remove("dcc get receive", "ag_opendcc"); #stops any suplicate sends (there should only ever be one)
+ $dccflag = 0;
+
+ my ($gdcc) = @_; #current pack
+ my $botname = $gdcc->{'nick'};
+ $botname =~ tr/[A-Z]/[a-z]/;
+ $bots[$botcounter] =~ tr/[A-Z]/[a-z]/;
+ if ($botname eq $bots[$botcounter]) #if it's our bot, let user know, and stop any further AG pack requests until finished
+ {
+ Irssi::print "AG | received connection for bot: " . $botname . ", #" . $packs[$packcounter];
+ $pact = 2;
+ }
+ else #if not, allow this to rerun on next get
+ {
+ Irssi::signal_add("dcc get receive", "ag_opendcc");
+ $dccflag = 1;
+ }
+}
+
+sub ag_closedcc #deals with DCC closes
+{
+ my ($dcc) = @_; #current pack
+ my $botname = $dcc->{'nick'}; #get the bots name, and checks if it's the one we want
+ $botname =~ tr/[A-Z]/[a-z]/;
+ $bots[$botcounter] =~ tr/[A-Z]/[a-z]/;
+ if ($botname eq $bots[$botcounter])
+ {
+ if ($dccflag == 0) {Irssi::signal_add("dcc get receive", "ag_opendcc");} #if so, reinits DCC get signal for the next pack
+ $dccflag = 1;
+ Irssi::print "AG | pack " . $packs[$packcounter] . " size was: " . $dcc->{'size'} . " transferred: " . $dcc->{'transfd'} . " skipped: " . $dcc->{'skipped'}; #nerdy info on pack recieved
+ foreach my $to (@totags) #clears timeouts to defuckify everything
+ {
+ Irssi::timeout_remove($to);
+ }
+ @totags = ();
+ if ($dcc->{'transfd'} == $dcc->{'size'}) #checks if the transfer actually ran to completion
+ {
+ Irssi::print "AG | transfer successful"; #if so, does next pack/search/bot (in that order)
+ if ($dcc->{'skipped'} == $dcc->{'size'})
+ {
+ $server->command("msg $bots[$botcounter] $cancelprefix"); #workaround because IRSSI doesn't actually 'get' packs if they're already downloaded, causing long stalls if left unattended.
+ }
+ if ($packcounter < $#packs)
+ {
+ $pact = 1; #allow pack requests now that transfer is finished
+ $packcounter += 1;
+ push(@totags, Irssi::timeout_add_once($nexdelay * 1000, sub { Irssi::print "AG | Getting next pack in list"; &ag_reqpack(); }, []));
+ Irssi::print "AG | waiting " . $nexdelay . " seconds";
+ }
+ elsif ($termcounter < $#terms)
+ {
+ $pact = 0; #allow the system to parse new responses
+ @packs = (); #delete last terms packlist
+ $termcounter += 1;
+ $packcounter = 0;
+ push(@totags, Irssi::timeout_add_once($nexdelay * 1000, sub { Irssi::print "AG | Packlist finished. Searching next term"; &ag_search(); }, []));
+ Irssi::print "AG | waiting " . $nexdelay . " seconds";
+ }
+ elsif ($botcounter < $#bots)
+ {
+ $pact = 0; #allow the system to parse new responses
+ @packs = (); #delete last bots packlist
+ $botcounter += 1;
+ $termcounter = 0;
+ $packcounter = 0;
+ push(@totags, Irssi::timeout_add_once($nexdelay * 1000, sub { Irssi::print "AG | All searches complete. Searching next bot"; &ag_search(); }, []));
+ Irssi::print "AG | waiting " . $nexdelay . " seconds";
+ }
+ else #if last pack on last search on last bot finished, then resets counters and starts over
+ {
+ $pact = 0;
+ @packs = (); #delete last bots packlist
+ $botcounter = 0;
+ $termcounter = 0;
+ $packcounter = 0;
+ Irssi::signal_remove("message irc notice", "ag_getmsg");
+ Irssi::print "AG | Waiting " . $exedelay . " minutes until next search";
+ Irssi::timeout_add_once($exedelay * 1000 * 60, sub { &ag_run; } , []);
+ }
+ }
+ else #if not, retry transfer
+ {
+ Irssi::print "AG | transfer failed";
+ Irssi::print "AG | " . $dcrdelay . " seconds until retry";
+ push(@totags, Irssi::timeout_add_once($dcrdelay * 1000, sub { Irssi::print "AG | retrying: " .$bots[$botcounter] . " " . $packs[$packcounter]; &ag_reqpack(); }, []));
+ }
+ }
+}
+
+sub ag_parseadd #parses add arguments for storage
+{
+ my ($file, @args) = @_;
+ open(file, ">>", $file);
+ foreach my $arg (@args)
+ {
+ print file $arg . "\n"; #print to file
+ }
+ close(file);
+ copy($file, "/tmp/temp") or die "COPY FAILED"; #copy to temp file so that duplicate lines [searches/bots] can be removed
+ unlink "$file";
+ open(temp, "<", "/tmp/temp");
+ open(file, ">", $file);
+ my %hTmp;
+ while (my $sLine = <temp>) #remove duplicate lines
+ {
+ next if $sLine =~ m/^\s*$/; #remove empty lines. Without this, still destroys empty lines except for the first one.
+ $sLine=~s/^\s+//; #strip leading/trailing whitespace
+ $sLine=~s/\s+$//;
+ print file qq{$sLine\n} unless ($hTmp{$sLine}++);
+ }
+ unlink "/tmp/temp";
+ close(file);
+}
+
+sub ag_parserem #parses remove arguments for deletion from file
+{
+ my ($file, @args) = @_;
+ open(temp, ">>", "/tmp/temp");
+ foreach my $arg (@args)
+ {
+ Irssi::print "AG | removing term: " . $arg;
+ print temp $arg . "\n";
+ }
+ close(temp);
+ open(temp2, ">", "/tmp/temp2");
+ open(file, "<", $file);
+ my %hTmp;
+ while( my $fileLine = file->getline() ) #get each entry already stored
+ {
+ open(temp, "<", "/tmp/temp");
+ while( my $tempLine = temp->getline() )
+ {
+ if ($fileLine eq $tempLine) #if entry in file and arguments
+ {
+ $hTmp{$fileLine}++; #set flag to not copy
+ }
+ print temp2 qq{$fileLine} unless $hTmp{$fileLine}; #copy other lines to other temp file
+ }
+ close(temp);
+ }
+ close(temp2);
+ copy("/tmp/temp2", $file) or die "COPY FAILED"; #rewrite old file
+ copy($file, "/tmp/temp") or die "COPY FAILED";
+ unlink "$file";
+ open(temp, "<", "/tmp/temp");
+ open(searches, ">", $file);
+ my %hTmp;
+ while (my $sLine = <temp>) #remove duplicate lines
+ {
+ next if $sLine =~ m/^\s*$/; #remove empty lines. Without this, still destroys empty lines except for the first one.
+ $sLine=~s/^\s+//; #strip leading/trailing whitespace
+ $sLine=~s/\s+$//;
+ print file qq{$sLine\n} unless ($hTmp{$sLine}++);
+ }
+ unlink "/tmp/temp";
+ unlink "/tmp/temp2";
+ close(file);
+}
+
+sub ag_add #add search terms
+{
+ &ag_server;
+ my @args = quotewords('\s+', 0, @_[0]); #split arguments (words in brackets not seperated)
+ if ($#args < 0)
+ {
+ Irssi::print "AG | too few arguments";
+ Irssi::print "AG | usage: ag_add <search terms>";
+ return;
+ }
+ &ag_parseadd($searchesfilename, @args);
+}
+
+sub ag_rem #remove ssearch terms
+{
+ &ag_server;
+ my @args = quotewords('\s+', 0, @_[0]);
+ if ($#args < 0)
+ {
+ Irssi::print "AG | too few arguments";
+ Irssi::print "AG | usage: ag_rem <search terms>";
+ return;
+ }
+ &ag_parserem($searchesfilename, @args);
+}
+
+sub ag_botadd #add bots
+{
+ &ag_server;
+ my @args = quotewords('\s+', 0, @_[0]);
+ if ($#args < 0)
+ {
+ Irssi::print "AG | too few arguments";
+ Irssi::print "AG | usage: ag_botsadd <bots>";
+ return;
+ }
+ &ag_parseadd($botsfilename, @args);
+}
+
+sub ag_botrem #remove bots
+{
+ &ag_server;
+ my @args = quotewords('\s+', 0, @_[0]);
+ if ($#args < 0)
+ {
+ Irssi::print "AG | too few arguments";
+ Irssi::print "AG | usage: ag_rem <search terms>";
+ return;
+ }
+ &ag_parserem($botsfilename, @args);
+}
+
+sub ag_run #main loop
+{
+ Irssi::print "AG | Search and get cycle Initiated";
+ &ag_getbots;
+ foreach my $n (@bots)
+ {
+ Irssi::print "AG | Bots: " . $n;
+ }
+ &ag_getterms;
+ foreach my $n (@terms)
+ {
+ Irssi::print "AG | Terms: " . $n;
+ }
+ &ag_search;
+}
+
+sub ag_stop
+{
+ Irssi::print "AG | killed";
+ Irssi::signal_remove("dcc get receive", "ag_opendcc");
+ $botcounter = 0;
+ $termcounter = 0;
+ $packcounter = 0;
+ @bots = ();
+ @terms = ();
+ @packs = ();
+ foreach my $to (@totags)
+ {
+ Irssi::timeout_remove($to);
+ }
+ @totags = ();
+}
+
+sub ag_settings
+{
+ ($nexdelay, $dcrdelay, $botdelay, $exedelay, $initflag, $sendprefix, $cancelprefix, $findprefix, $botsfilename, $searchesfilename) = (Irssi::settings_get_int("ag_next_delay"), Irssi::settings_get_int("ag_dcc_closed_retry_delay"), Irssi::settings_get_int("ag_bot_delay"), Irssi::settings_get_int("ag_interrun_delay"), Irssi::settings_get_bool("ag_autorun"), Irssi::settings_get_str("ag_xdcc_send_prefix"), Irssi::settings_get_str("ag_xdcc_cancel_prefix"), Irssi::settings_get_str("ag_xdcc_find_prefix"), Irssi::settings_get_str("ag_bot_file"), Irssi::settings_get_str("ag_search_file"));
+}
+
+sub ag_reset
+{
+ my $nexdelay = 5;
+ my $dcrdelay = 10;
+ my $botdelay = 30;
+ my $exedelay = 15;
+ my $initflag = 1;
+ my $sendprefix = "xdcc send";
+ my $findprefix = "!find";
+ my $botsfilename = "$FindBin::Bin/.irssi/scripts/bots.txt";
+ my $searchesfilename = "$FindBin::Bin/.irssi/scripts/searches.txt";
+ Irssi::settings_get_int("ag_next_delay");
+ Irssi::settings_get_int("ag_dcc_closed_retry_delay");
+ Irssi::settings_get_int("ag_bot_delay");
+ Irssi::settings_get_int("ag_interrun_delay");
+ Irssi::settings_get_bool("ag_autorun");
+ Irssi::settings_get_str("ag_xdcc_send_prefix");
+ Irssi::settings_get_str("ag_xdcc_find_prefix");
+ Irssi::settings_get_str("ag_bot_file");
+ Irssi::settings_get_str("ag_search_file");
+ Irssi::print "AG | all settings reset to default values";
+}
+
+open(bots, ">>", $botsfilename); #makes bots and searches file if they don't exist
+close(bots);
+open(searches, ">>", $searchesfilename);
+close(searches);
+if ($initflag) {&ag_init();}
+
+Irssi::signal_add("dcc closed", "ag_closedcc");
+
+Irssi::settings_add_int("ag", "ag_next_delay", $nexdelay);
+Irssi::settings_add_int("ag", "ag_dcc_closed_retry_delay", $dcrdelay);
+Irssi::settings_add_int("ag", "ag_bot_delay", $botdelay);
+Irssi::settings_add_int("ag", "ag_interrun_delay", $exedelay);
+Irssi::settings_add_bool("ag", "ag_autorun", $initflag);
+Irssi::settings_add_str("ag", "ag_xdcc_send_prefix", $sendprefix);
+Irssi::settings_add_str("ag", "ag_xdcc_cancel_prefix", $cancelprefix);
+Irssi::settings_add_str("ag", "ag_xdcc_find_prefix", $findprefix);
+Irssi::settings_add_str("ag", "ag_bot_file", $botsfilename);
+Irssi::settings_add_str("ag", "ag_search_file", $searchesfilename);
+
+Irssi::command_bind("ag_help", "ag_help");
+Irssi::command_bind("ag_run", "ag_run");
+Irssi::command_bind("ag_stop", "ag_run");
+Irssi::command_bind("ag_server", "ag_server");
+Irssi::command_bind("ag_add", "ag_add");
+Irssi::command_bind("ag_rem", "ag_rem");
+Irssi::command_bind("ag_botadd", "ag_botadd");
+Irssi::command_bind("ag_botrem", "ag_botrem");
+
git.99rst.org / irssi-scripts.irssi.org.git / commitdiff