User:PockBot/SourceCode/106
Appearance
#!/usr/bin/perl -- use strict; #use warnings; use CGI; use CGI::Carp "fatalsToBrowser"; use LWP::Simple; use LWP::UserAgent; use HTTP::Request; use HTTP::Request::Common qw(GET); use HTTP::Response; #Disable buffering to allow progress bar to work. $|=1; #______________________________________________________________________________# # PockBot.pl TRIAL VERSION # # Author Dan Adams , (User:PocklingtonDan) # #______________________________________________________________________________# #______________________________________________________________________________# # RIGHTS MANAGEMENT ETC # # # # The source code for PockBot is supplied solely for the purposes of allowing # # other editors to comment on and improve the code, and/or to run the code as # # a clone. It may be distributed and modified as required for these purposes. # #______________________________________________________________________________# #______________________________________________________________________________# # CHANGES STILL TO MAKE # # # # none.code complete. # # # # RECENT CHANGES # # # # 05.12.06 - Version 0.01 - source code released # # 05.12.06 - Version 0.02 - does not run now for non-existent categories # # 06.12.06 - Version 0.03 - Now writes to wikipedia # # 06.12.06 - Version 0.04 - Now adds signature to posts # # 06.12.06 - Version 0.05 - Now prints in DIV scrollbox to take up less room # # 06.12.06 - Version 0.06 - Now monitors server load and advises user # # 06.12.06 - Version 0.07 - Now gets correct category for all articles # # 06.12.06 - Version 1.00 - Released for trial. # # 07.12.06 - Version 1.01 - Colour-filling article classes as per templates # # 07.12.06 - Version 1.02 - Sortable DHTML columns added # # 07.12.06 - Version 1.03 - Added edit attribution to user running bot # # 07.12.06 - Version 1.04 - Logs IP Address of end user # # 08.12.06 - Version 1.05 - implemented 100-subcat limit to set finite limit # # 08.12.06 - Version 1.06 - Added progress bar to stop timeouts. # #______________________________________________________________________________# #______________________________________________________________________________# # WHAT THE SCRIPT DOES # # # # This script is a wikipedia bot. It acts as a web spider. Given a wikipedia # # category page to start from, it finds all articles listed in that category # # as well as all subcategories of that category. For every subcategory it # # pulls a list of articles. For all articles retrieved (a list of all articles # # in that category and its subcategories) it then retrieves the CLASS flag for # # each page from wikipedia. It then presents these resulsts in tabulated form. # # # # INTENDED USE # # # # It is intended that this script would be useful to those trying to monitor # # all pages within a category for purposes of administration or for a project # # in order to monitor which articles need bringing up from stub or start class # # to full article status. # # # # CODE FORMATTING # # # # Code is formatted for ease of editing with Textad (www.textpad.com) or # # similar editor with colour-coding meta-markup. It may be difficult to scan # # using a no-frills text editor. # #______________________________________________________________________________# #______________________________________________________________________________# # MAIN ROUTINE # #______________________________________________________________________________# use CGI qw(:standard Vars); my $action = param('action') || 'startBot'; if ($action eq 'intro') {&startBot;} elsif ($action eq 'disableBot') {&disableBot;} elsif ($action eq 'enableBot') {&enableBot;} elsif ($action eq 'getMainCategory') {&getMainCategory;} else {&error("Unrecognised action request");} exit; #______________________________________________________________________________# # SUBROUTINES # #______________________________________________________________________________# sub startBot { my @gettheip = split(/\./,$ENV{'REMOTE_ADDR'}); my $remotehost = "$gettheip[0].$gettheip[1].$gettheip[2].$gettheip[3]"; &checkIfBotOnline; &logAction("Bot requested"); &printOnlineHeader; print "<FORM action=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi\" method=\"post\"><fieldset style=\"width: 425px;\">"; print "<legend style=\"font-family: arial, sans-serif; font-size: 10\">Please enter the wikipedia Category you wish to process</legend>"; print "<p><font face=\"arial\" size=\"2\" color=\"red\"><b>*</b></font> <font face=\"arial\" size=\"2\">Category:"; print "<INPUT type=\"text\" style=\"font-family: arial, serif; font-size: 12px;\" size=\"50\" name=\"category_specified\" value=\"Enter category name here!\"><br>"; print "<b><em>mandatory</em></b></font></p>"; print "<p><font face=\"arial\" size=\"2\">Your wikipedia username:"; print "<INPUT type=\"text\" style=\"font-family: arial, serif; font-size: 12px;\" size=\"35\" name=\"wikipedia_user\" value=\"\"><br>"; print "<b><em>optional but useful to attribute PockBot edits</em></b></font></p>"; print "<INPUT type=\"hidden\" name=\"action\" value=\"getMainCategory\">"; print "<INPUT type=\"hidden\" name=\"userIPAddress\" value=\"$remotehost\">"; print "<INPUT type=\"submit\" value=\"Start Pockbot\">"; print " </fieldset></FORM>"; print "<p><font face=\"arial\" size=\"2\"><b>Notes:</b><br><em>Do not run for a top-level category.</em><br><em>Bot may take over an hour to run for categories with many nested subcategories.</em></font></p>"; &printFooter; } #______________________________________________________________________________# sub getArticlesinCategory { my $content_articles = $_[0]; &logAction("Searching for articles in this category "); # if its not a wikipedia category page, return empty array unless ($content_articles =~ m/<div id="mw-pages">/){ $content_articles = ""; my @found_articles = split(/\|/,$content_articles); &logAction("Found 0 articles in this category "); return (@found_articles); } # empty array if no articles, else populate with article names if ($content_articles =~ m/There are 0 pages in this section of this category/){ $content_articles = ""; &logAction("Found 0 articles in this category"); } else { $content_articles =~ s/[\s\S]*<div id="mw-pages">//; $content_articles =~ s/<\/div>[\s\S]*/<\/div>/; $content_articles =~ s/[\s\S]*?<ul>/<ul>/; $content_articles =~ s/<h3>[\s\S]*?<\/h3>//g; $content_articles =~ s/<ul>//g; $content_articles =~ s/<\/ul>//g; $content_articles =~ s/<td>//g; $content_articles =~ s/<\/td>//g; $content_articles =~ s/<\/div>//g; $content_articles =~ s/<\/tr>//g; $content_articles =~ s/<\/table>//g; $content_articles =~ s/<\/li>/|/g; $content_articles =~ s/<li>/|/g; $content_articles =~ s/\n//g; $content_articles =~ s/\|\|/\|/g; $content_articles =~ s/<a[\s\S]*?>//g; $content_articles =~ s/<\/a>//g; $content_articles =~ s/\|$//; $content_articles =~ s/^\|//; $content_articles =~ s/_/ /g; $content_articles =~ s/\s\|/\|/g; &logAction("Found 1 or more articles in this category"); } my @found_articles = split(/\|/,$content_articles); return (@found_articles); } #______________________________________________________________________________# sub getSubCatsinCategory { my $content_subcats = $_[0]; &logAction("Searching for subcats in this category"); # if its not a wikipedia category page, empty array unless ($content_subcats =~ m/<div id="mw-subcategories">/){ $content_subcats = ""; my @found_subcats = split(/\|/,$content_subcats); &logAction("Found 0 subcats in this category"); return (@found_subcats); } # empty array if no subcats, else populate with subcat names if ($content_subcats =~ m/There are 0 subcategories to this category/){ $content_subcats = ""; &logAction("Found 0 subcats in this category"); } else { $content_subcats =~ s/[\s\S]*<div id="mw-subcategories">//; $content_subcats =~ s/<div id="mw-pages">[\s\S]*//; $content_subcats =~ s/<h3>[\s\S]*?<\/h3>//g; $content_subcats =~ s/<div[\s\S]*?>//g; $content_subcats =~ s/<\/div>//g; $content_subcats =~ s/<span[\s\S]*?<\/span>//g; $content_subcats =~ s/[\s\S]*?<ul>/<ul>/; $content_subcats =~ s/<ul>//g; $content_subcats =~ s/<\/ul>//g; $content_subcats =~ s/<\/li>/|/g; $content_subcats =~ s/<li>/|/g; $content_subcats =~ s/<a[\s\S]*?>//g; $content_subcats =~ s/<\/a>//g; $content_subcats =~ s/\n//g; $content_subcats =~ s/\|\|/\|/g; $content_subcats =~ s/<td>//g; $content_subcats =~ s/<\/td>//g; $content_subcats =~ s/<\/tr>//g; $content_subcats =~ s/<\/table>//g; $content_subcats =~ s/[\s]*?\|/\|/g; $content_subcats =~ s/\|$//; $content_subcats =~ s/^\|//; $content_subcats =~ s/\|\|/\|/g; &logAction("Found 1 or more subcats in category $content_subcats"); } my @found_subcats = split(/\|/,$content_subcats); return (@found_subcats); } #______________________________________________________________________________# sub processContents { my $category = $_[0]; my $contents = $_[1]; my $userRunningBot = $_[2]; my $userIPAddress = $_[3]; $category =~ s/_/ /g; &logAction("Starting to process category $category"); #Check to make sure category is valid my ($testcategory, $testcontents) = fetchContents($category); if ($testcontents =~ m/noarticletext/) { &error("You specified an invalid category. Please check your spelling and capitalization and try again."); } else { #Seperate the page generation from spider work use threads; use threads::shared; use Config; if ($Config{useithreads}) { # We have threads # Let user know spider is on the job. &logAction("Notifying user bot starting"); &printOnlineHeader; print "<form><fieldset style=\"width: 425px;\"><p><font face=\"arial\" size=\"2\">Thank you for using PockBot. You have requested a list of article classes for "; print " wikipedia category <a href=\"http://en.wikipedia.org/wiki/Category:$category\">$category</a>.</font></p>"; print "<p><font face=\"arial\" size=\"2\">The content will take some time to generate, espcially for large categories. When complete, the results will be posted to wikipedia for you at the <a href=\"http://en.wikipedia.org/wiki/Category_talk:$category\">category's talk page</a>.<br><br>"; print "<b>If your browser times out you may get a blank page, The data will still be written as requested and not affected by this.</font></p></fieldset></form>"; print "<p><font face=\"arial\" size=\"2\"><b>Progress:</b><br><img src=\"http://www.thepaty.plus.com/working.gif\" align=\"middle\" width=\"20\" height=\"20\">Working "; &printFooter; #Another thread to print progress bar to keep brower from timing out? my $keepRunningProgressBar : shared = 1; my $progressBar = threads->create(sub { while ($keepRunningProgressBar == 1) {sleep(5); print "<img src=\"http://www.thepaty.plus.com/dot.gif\" align=\"middle\">";} }); $progressBar->detach; # Set spider to work on requested category, in separate thread my $threadForSpidering = threads->new(\&workthread, $category, $contents, $userRunningBot,$userIPAddress); $threadForSpidering->join; $keepRunningProgressBar = 0; sleep(6); } else { &error("PockBot requires threads. This perl installation is not built with threads activated. PockBot cannot run."); } } } #______________________________________________________________________________# sub removeDuplicates { my @articles = @_; my @articles_no_duplicates = (); &logAction("Removing duplicates from found articles list."); foreach my $suggested_article (@articles) { my $already_exists = 0; foreach my $existing_article (@articles_no_duplicates) { if ($suggested_article eq $existing_article) { $already_exists = 1; } } if ($already_exists == 0) { push(@articles_no_duplicates, $suggested_article); } } return (@articles_no_duplicates); } #______________________________________________________________________________# sub getAllArticlesIn { my @subcats = @_; my @new_articles = (); foreach my $individual_subcat (@subcats) { &logAction("Searching for new articles in subcat $individual_subcat"); my ($subcategory, $subcategorycontents) = fetchContents($individual_subcat); my @found_articles = getArticlesinCategory($subcategorycontents); foreach my $found_article (@found_articles) { push(@new_articles, $found_article); } } return (@new_articles); } #______________________________________________________________________________# sub getArticleClasses { my @articles_no_duplicates = @_; my %classes = (); foreach my $article_title (@articles_no_duplicates) { my ($article, $contents) = fetchTalkContents($article_title); my $class = "unclassified"; $article =~ s/_/ /g; &logAction("Getting article class for article $article_title"); if ($contents =~ m/as Start-Class/i) { $class = "Start"; } elsif ($contents =~ m/as Stub-Class/i) { $class = "Stub"; } elsif ($contents =~ m/as A-Class/i) { $class = "A"; } elsif ($contents =~ m/as B-Class/i) { $class = "B"; } elsif ($contents =~ m/as FA-Class/i) { $class = "Featured Article"; } elsif ($contents =~ m/as GA-Class/i) { $class = "Good Article"; } elsif ($contents =~ m/This page is not an article and does not require/i) { $class = "Non-Article"; } else { $class = "unclassified"; } # add details of article class to hash $classes{$article} = $class; } return (%classes) } #______________________________________________________________________________# sub writeResultsToFile { my $replacement_text = $_[0]; my $replacement_page = $_[1]; my $tagWhoRequestedEdit = $_[2]; my $userIPAddress = $_[3]; my $timeStamp = getTimeStamp(); my $replacement_summary = "PockBot (run by IP:$userIPAddress) - Category articles summary as of $timeStamp"; &logAction("Writing bot results to file."); use LWP::UserAgent; my $agent=LWP::UserAgent->new; $agent->agent('Perlwikipedia/0.90'); $agent->cookie_jar({file=> '.perlwikipedia-cookies'}); my $editor = "PockBot"; my $password = "**********"; #INTENTIONALLY BLANKED! my $login = HTTP::Request->new(POST => "http://en.wikipedia.org/w/index.php?title=Special:Userlogin&action=submitlogin&type=login"); $login->content_type('application/x-www-form-urlencoded'); $login->content("wpName=$editor&wpPassword=$password&wpRemember=1&wpLoginattempt=Log+in"); my $logger_inner = $agent->request($login); my $do_redirect=HTTP::Request->new(GET =>'http://en.wikipedia.org/w/index.php?title=Special:Userlogin&wpCookieCheck=login'); my $redirecter= $agent->request($do_redirect); my $is_success=$redirecter->content; if ($is_success=~m/\QYou have successfully signed in to Wikipedia as "$editor".\E/) { use HTML::Form; my $ua = LWP::UserAgent->new; $ua->agent("Perlwikipedia/0.90"); $ua->cookie_jar($agent->cookie_jar()); my $response = $ua->get("http://en.wikipedia.org/w/index.php?title=Category_talk:$replacement_page&action=edit§ion=new"); my $form = HTML::Form->parse($response); my $text = $form->find_input('wpTextbox1')->value; my $summary = $form->find_input('wpSummary')->value; my $save = $form->find_input('wpSave')->value; my $edittoken = $form->find_input('wpEditToken')->value; my $starttime = $form->find_input('wpStarttime')->value; my $edittime = $form->find_input('wpEdittime')->value; $form->value('wpTextbox1', $replacement_text); $form->value('wpSummary', $replacement_summary ); $response = $ua->request($form->click); return "success"; } else { &error("Login to wikipedia failed."); } } #______________________________________________________________________________# sub getTimeStamp { my @months = qw(Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec); my @weekDays = qw(Sun Mon Tue Wed Thu Fri Sat Sun); my ($second, $minute, $hour, $dayOfMonth, $month, $yearOffset, $dayOfWeek, $dayOfYear, $daylightSavings) = localtime(); my $year = 1900 + $yearOffset; my $timeNow = "$hour:$minute:$second, $weekDays[$dayOfWeek] $months[$month] $dayOfMonth, $year"; return ($timeNow); } #______________________________________________________________________________# sub logAction { #commenting out whole subroutine - was only used for debugging and removing it will reduce server load and decrease run time. #my $actionToLog = $_[0]; #my $log_file = '/files/home2/thepaty/cgi-bin/log.htm'; #my $timeStamp = getTimeStamp(); #open(LOGFILE,">>$log_file") || &error("Cannot open log file."); # flock(LOGFILE, 2) || &error("Cannot lock log file."); # print LOGFILE "$timeStamp: $actionToLog<br>"; # flock(LOGFILE, 8); #close (LOGFILE); #print "<img src=\"http://www.thepaty.plus.com/dot.gif\">"; } #______________________________________________________________________________# sub workthread { my $category = $_[0]; my $contents = $_[1]; my $userRunningBot = $_[2]; my $userIPAddress = $_[3]; my $tagWhoRequestedEdit = ""; if ($userRunningBot eq "") { $tagWhoRequestedEdit = "on behalf of an anonymous user"; } else { $tagWhoRequestedEdit = "on behalf of [[User:$userRunningBot|$userRunningBot]]"; } &logAction("Starting work thread for category $category"); my @subcats = getSubCatsinCategory($contents); my @articles = getArticlesinCategory($contents); my $new_subcats_found_this_round = 1; my @subcats_searched_aleady = (); my $subCatLimit = 100; my $hitSubcatLimit = "false"; # Keep searching until no new subcats are found.in any categories searched while (($new_subcats_found_this_round > 0) && ($hitSubcatLimit eq "false")) { $new_subcats_found_this_round = 0; my @proposed_extra_subcats = (); # Perform a search of every category we currently know of foreach my $existing_subcat (@subcats) { my $already_searched = 0; # If already searched this category in an earlier pass, skip it. foreach my $searched_subcat (@subcats_searched_aleady) { if ($existing_subcat eq $searched_subcat) { $already_searched = 1; } } # If not already searched, get all subcats of that category if ($already_searched == 0) { &logAction("Have not searched subcat $existing_subcat already"); my ($subcategory, $subcategorycontents) = fetchContents($existing_subcat); my @additional_subcats = getSubCatsinCategory($subcategorycontents); foreach my $proposed_additional_subcat (@additional_subcats) { push(@proposed_extra_subcats, $proposed_additional_subcat); &logAction("Found possible new subcat $proposed_additional_subcat"); } push(@subcats_searched_aleady, $existing_subcat); } else { &logAction("Have searched subcat $existing_subcat already"); } } # If this new found subcat isn't a duplicate of one we already know about... foreach my $proposed_new_subcat (@proposed_extra_subcats) { my $already_exists = 0; foreach my $existing_subcat (@subcats) { if ($proposed_new_subcat eq $existing_subcat) { $already_exists = 1; } } # then add it to our master list if ($already_exists == 0) { &logAction("subcat $proposed_new_subcat is a genuinely new subcategory, adding to master list"); push(@subcats, $proposed_new_subcat); $new_subcats_found_this_round++; if ($#subcats > $subCatLimit) { $hitSubcatLimit = "true"; } } else { &logAction("subcat $proposed_new_subcat already existed in master list, ignoring"); } } &logAction("$new_subcats_found_this_round new subcats found this round. If greater than zero, should run through again"); } # And now get a list of every article in every subcat my @new_articles = getAllArticlesIn(@subcats); my @articles = (@articles, @new_articles); # Remove duplicates from article list. my @articles_no_duplicates = removeDuplicates(@articles); # Search talk pages for each article to find "class=X" classification my %classes = getArticleClasses(@articles_no_duplicates); my $explainReducedResultsSet = ""; if ($hitSubcatLimit eq "true") { $explainReducedResultsSet = "'''Note: this category had more than $subCatLimit sub-categories. Only data from the first $subCatLimit sub-categories has been returned.'''<br><br>"; } else{ $explainReducedResultsSet = ""; } # Prepare text to print to results file my $text_to_print = ""; $text_to_print = "{{PockBotHeader}}\n"; foreach my $article_title (@articles_no_duplicates) { my $fetchedArticleClass = ""; if ($classes{$article_title}) { $fetchedArticleClass = $classes{$article_title}; } else { $fetchedArticleClass = "Error finding article class for $article_title"; } $fetchedArticleClass =~ s/Non-Article/NA/; $fetchedArticleClass =~ s/unclassified/''not yet classified''/; $fetchedArticleClass =~ s/Featured Article/FA/; $fetchedArticleClass =~ s/Good Article/GA/; my $cellColour = "white"; if ($fetchedArticleClass =~ m/Start/) { $cellColour = "#ffaa66"; } if ($fetchedArticleClass =~ m/Stub/) { $cellColour = "#ff6666"; } if ($fetchedArticleClass =~ m/^A$/) { $cellColour = "#66ffff"; } if ($fetchedArticleClass =~ m/B/) { $cellColour = "#ffff66"; } if ($fetchedArticleClass =~ m/NA/) { $cellColour = "whitesmoke"; } if ($fetchedArticleClass =~ m/''not yet classified''/) { $cellColour = "white"; } if ($fetchedArticleClass =~ m/FA/) { $cellColour = "#6699ff"; } if ($fetchedArticleClass =~ m/GA/) { $cellColour = "#66ff66"; } $text_to_print .= "{{PockBotData|$article_title|$fetchedArticleClass|$cellColour}}\n"; } $text_to_print .= "{{PockBotFooter|Edit by ~~~ ($tagWhoRequestedEdit)}}\n"; # write results to results.htm my $successfuledit = writeResultsToFile($text_to_print,$category,$tagWhoRequestedEdit,$userIPAddress); return "success"; } #______________________________________________________________________________# sub fetchContents { my $category = $_[0]; $category =~ s/\s/_/g; my $category_url = "http://en.wikipedia.org/wiki/Category:" . $category; &logAction("Fetching page contents for category $category"); my $browser = LWP::UserAgent->new(); $browser->timeout(60); my $request = HTTP::Request->new(GET => $category_url); my $response = $browser->request($request); #if ($response->is_error()) {printf "%s\n", $response->status_line;} my $contents = $response->content(); sleep(1); # don't hammer the server! One read request every 1 second. return($category,$contents); } #______________________________________________________________________________# sub fetchTalkContents { my $article = $_[0]; $article =~ s/\s/_/g; my $article_url = "http://en.wikipedia.org/wiki/Talk:$article"; &logAction("Fetching talk page contents for article $article"); my $browser = LWP::UserAgent->new(); $browser->timeout(60); my $request = HTTP::Request->new(GET => $article_url); my $response = $browser->request($request); if ($response->is_error()) {printf "%s\n", $response->status_line;} my $contents = $response->content(); sleep(1); # don't hammer the server! One read request every 1 second. return($article,$contents); } #______________________________________________________________________________# sub finishedRunning { my $category = $_[0]; my $category_url = "http://en.wikipedia.org/wiki/Category_talk:" . $category; &logAction("Finished processing category $category"); print "<br><img src=\"http://www.thepaty.plus.com/tick.gif\" align=\"middle\" width=\"20\" height=\"20\"> <b>Finished</b>.</font></p>"; &printFooter; } #______________________________________________________________________________# sub resetLogAndResultsFiles { my $log_file = '/files/home2/thepaty/cgi-bin/log.htm'; &logAction("Resetting log and results files to empty"); open(LOGFILE,">$log_file") || &error("Cannot open log file."); flock(LOGFILE, 2) || &error("Cannot lock log file."); print LOGFILE ""; flock(LOGFILE, 8); close (LOGFILE); } #______________________________________________________________________________# sub getMainCategory{ my $category = "BLANK"; $category = param('category_specified'); my $userRunningBot = param('wikipedia_user'); my $userIPAddress = param('userIPAddress'); &resetLogAndResultsFiles(); &logAction("Bot started for category $category"); if ($category eq "BLANK") { &error("Error receiving category name"); } else { my ($category, $contents) = fetchContents($category); &processContents($category,$contents,$userRunningBot,$userIPAddress); &finishedRunning($category); } } #______________________________________________________________________________# sub enableBot { my $status_file = "/files/home2/thepaty/cgi-bin/status.txt"; &logAction("Bot enable request made"); open(STATUSFILE,"$status_file") || &error("Cannot open bot status file."); flock(STATUSFILE, 2) || &error("Cannot lock bot status file."); my $current_status = <STATUSFILE>; flock(STATUSFILE, 8); close (STATUSFILE); chomp($current_status); my $bot_enabled = $current_status; if ($bot_enabled == 1) { &logAction("Bot already enabled, no action necesary"); &printOnlineHeader; print "<p><font face=\"arial\">PockBot is already enabled. <a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=disableBot\">Disable PockBot</a></font></p>"; &printFooter; exit; } elsif ($bot_enabled == 0) { &logAction("Bot currently disabled. Enabling bot."); open(STATUSFILE,">$status_file") || &error("Cannot open bot status file."); flock(STATUSFILE, 2) || &error("Cannot lock bot status file."); print STATUSFILE "1"; flock(STATUSFILE, 8); close (STATUSFILE); &printOnlineHeader; print "<p><font face=\"arial\">PockBot is now enabled. <a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=disableBot\">Disable Pockbot</a></font></p>"; &printFooter; exit; } else { &error("Unrecognised bot status. Something has gone wrong."); } } #______________________________________________________________________________# sub disableBot { my $status_file = "/files/home2/thepaty/cgi-bin/status.txt"; &logAction("Bot disable request made"); open(STATUSFILE,"$status_file") || &error("Cannot open bot status file."); flock(STATUSFILE, 2) || &error("Cannot lock bot status file."); my $current_status = <STATUSFILE>; flock(STATUSFILE, 8); close (STATUSFILE); chomp($current_status); my $bot_enabled = $current_status; if ($bot_enabled == 0) { &logAction("Bot is already disabled. No action necessary"); &printOfflineHeader; print "<p><font face=\"arial\">PockBot is already disabled. <a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=enableBot\">Enable PockBot</a></font></p>"; &printFooter; exit; } elsif ($bot_enabled == 1) { &logAction("Bot is currently enabled. Disabling bot."); open(STATUSFILE,">$status_file") || &error("Cannot open bot status file."); flock(STATUSFILE, 2) || &error("Cannot lock bot status file."); print STATUSFILE "0"; flock(STATUSFILE, 8); close (STATUSFILE); &printOfflineHeader; print "<p><font face=\"arial\">PockBot is now disabled. <a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=enableBot\">Enable Pockbot</a></font></p>"; &printFooter; exit; } else { &error("Unrecognised bot status. Something has gone wrong."); } } #______________________________________________________________________________# sub checkIfBotOnline { my $status_file = '/files/home2/thepaty/cgi-bin/status.txt'; &logAction("Checking if bot is online"); open(STATUSFILE,"$status_file") || &error("Cannot open bot status file."); flock(STATUSFILE, 2) || &error("Cannot lock bot status file."); my $current_status = <STATUSFILE>; flock(STATUSFILE, 8); close (STATUSFILE); chomp($current_status); my $bot_enabled = $current_status; if ($bot_enabled == 0) { &logAction("Bot is disabled, cannot perform action"); &printOfflineHeader; print "<p><font face=\"arial\">PockBot is currently disabled. If you are certain it has nt been disabled for a reason, you can <a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=enableBot\">Enable PockBot</a></font></p>"; &printFooter; exit; } elsif ($bot_enabled == 1) { &logAction("Bot is enabled, we are good to go."); #no action necessary } else { &error("Unrecognised bot status. Something has gone wrong."); } } #______________________________________________________________________________# sub getWikipediaLoad { my @months = qw(Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec); my @weekDays = qw(Sun Mon Tue Wed Thu Fri Sat Sun); my ($second, $minute, $hour, $dayOfMonth, $month, $yearOffset, $dayOfWeek, $dayOfYear, $daylightSavings) = localtime(); $hour = $hour - 6; # adjust to get US time from GMT of PockBot's server. my $currentServerLoad = "$hour"; $currentServerLoad =~ s/10/low/g; $currentServerLoad =~ s/11/low/g; $currentServerLoad =~ s/12/fairlylow/g; $currentServerLoad =~ s/13/fairlyhigh/g; $currentServerLoad =~ s/14/high/g; $currentServerLoad =~ s/15/veryhigh/g; $currentServerLoad =~ s/16/veryhigh/g; $currentServerLoad =~ s/17/high/g; $currentServerLoad =~ s/18/high/g; $currentServerLoad =~ s/19/veryhigh/g; $currentServerLoad =~ s/20/veryhigh/g; $currentServerLoad =~ s/21/veryhigh/g; $currentServerLoad =~ s/22/veryhigh/g; $currentServerLoad =~ s/0/fairlyhigh/g; $currentServerLoad =~ s/1/fairlylow/g; $currentServerLoad =~ s/2/low/g; $currentServerLoad =~ s/3/verylow/g; $currentServerLoad =~ s/4/verylow/g; $currentServerLoad =~ s/5/verylow/g; $currentServerLoad =~ s/6/verylow/g; $currentServerLoad =~ s/7/verylow/g; $currentServerLoad =~ s/8/low/g; $currentServerLoad =~ s/9/low/g; return ($currentServerLoad); } #______________________________________________________________________________# sub printOnlineHeader { print "Content-type: text/html\n\n"; print "<html><head><title>PockBot</title><script src=\"sorttable.js\"></script></head><body>"; print "<font face=\"arial\" size=\"1\"><a href=\"http://en.wikipedia.org/wiki/Main_Page\">Wikipedia</a> > <a href=\"http://en.wikipedia.org/wiki/User:PockBot\">Pockbot's User Page</a></font><br>"; print "<font face=\"arial\" size=\"1\"><b>Pockbot is currently ONLINE / ENABLED</b> (<a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=disableBot\">Disable PockBot</a>)</font><br>"; print "<img src=\"http://www.thepaty.plus.com/pockbot.gif\"><br>"; my$currentServerLoad = getWikipediaLoad(); print "<img src=\"http://www.thepaty.plus.com/load_$currentServerLoad.gif\">"; } sub printOfflineHeader { print "Content-type: text/html\n\n"; print "<html><head><title>PockBot</title></head><body>"; print "<p><font face=\"arial\" size=\"1\"><a href=\"http://en.wikipedia.org/wiki/Main_Page\">Wikipedia</a> > <a href=\"http://en.wikipedia.org/wiki/User:PockBot\">Pockbot's User Page</a></font></p>"; print "<p><font face=\"arial\" size=\"1\"><b>Pockbot is currently OFFLINE / DISABLED</b> (<a href=\"http://ccgi.thepaty.plus.com/cgi-bin/PockBot.cgi?action=enableBot\">Enable PockBot</a>)</font><p>"; print "<img src=\"http://www.thepaty.plus.com/pockbot.gif\">"; } #______________________________________________________________________________# sub printFooter { print "</body></html>"; } #______________________________________________________________________________# sub error { &checkIfBotOnline; &logAction("ERROR: $_[0]"); &printOnlineHeader; print "<p><font face=\"arial\">ERROR: $_[0]</font></p>"; &printFooter; exit; }