batch.php

<?php
// Only run this from CLI
if (!isset($argv)) exit;

require "env.php";
require "db.php";
require "telegram.php";
require "f2pool.php";

$sleep_interval = 10;
$db = new f2poolbot_db();
$tg = new Telegram();
$pool = new F2Pool();
$ERROR = [
	"CANNOT_RETRIEVE_POOL_INFO"	=>	"Error: Couldn't retreive Pool Information",
	"F2_USERNAME_UNDEFINED"		=>	"Error: F2 Username is not defined",
];
$default_stat_counter = 60;
$stat_counter_logfile = $default_stat_counter;
$stat_counter_tg = $default_stat_counter;

if (isset($argv[1])) {
  if ($argv[1] == "on") {
    $db->turnBatchSwitchOn();
  } elseif ($argv[1] == "off") {
    $db->turnBatchSwitchOff();
  } elseif ($argv[1] == "status") {
	  $m = "\n";
	  $m .= "Batch Switch : ";
	  if ($db->isBatchSwitchOn()) {
		  $m .= "on";
	  } else {
		  $m .= "off";
	  }
	  $m .= "\n";
	  $m .= "Debug Mode : ";
	  if ($db->isDebugModeOn()) {
		  $m .= "on";
	  } else {
		  $m .= "off";
	  }
	  $m .= "\n";
	  $m .= "Status Mode: ";
	  if ($db->isShowStatMode()) {
		  $m .= "yes";
	  } else {
		  $m .= "no";
	  }
	  $m .= "\n";
	  $m .= "Batch Running? ";
	  if ($db->isBatchRunning()) {
		  $m .= "yes";
	  } else {
		  $m .= "no";
	  }
	  $m .= "\n";
	  $m .= "Status Mode: ";
          if ($db->isShowStatMode()) {
                  $m .= "yes";
          } else {
                  $m .= "no";
	  }
	  $m .= "\n\n";
	  $m .= makeStatusMessage();
          $m .= "\n\n";
	  echo $m;
  } elseif ($argv[1] == "debug-on") {
	  if($ENV["ENV"] == "DEV") {
	  	$db->setDebugModeOn();
	  } else {
		$db->setDebugModeOff();
	  }
  } elseif ($argv[1] == "debug-off") {
	  $db->setDebugModeOff();
  } elseif ($argv[1] == "stat-on") {
	  $db->setShowStatModeOn();
  } elseif ($argv[1] == "stat-off") {
	  $db->setShowStatModeOff();
  } elseif ($argv[1] == "prod"
	    || $argv[1] == "dev"
  ) {
	  while($db->isBatchSwitchOn()) {
	  // MAIN LOOP
	  	$db->setBatchRunningStatusOn();
	  	$userIds = $db->getOfflineAlertOnUserIds();
	
		foreach ($userIds as $uid) {
	  		$f2_username = $db->getF2Username($uid);
			if ($f2_username != "") {
				$pool = new F2Pool();
				$pool->setUsername($f2_username);
				if ($pool->fetchPoolInfo()) {
					if($pool->numOfOfflineWorkers()
					|| $db->isDebugModeOn()
					) {
						$tg = new Telegram();
						$tg->setChatId($db->getChatId($uid));
						$tg->returnTgMessage($pool->getOfflineAlertMessage());
						$tg = null;
						$db->setOfflineAlertOff($uid);
				  	}
				} else {
					$m = date('Y/m/d H:i:s') . ": " . $ERROR["CANNOT_RETRIEVE_POOL_INFO"];
					echo $m;
				}
				$pool = null;
		  	}
	  	}

	  	// get all f2_username where auto_monitor == on for that user
	  	$userIds = $db->getAutoMonitorModeOnUserIds();
  	  	// for each f2_username get next_batch_run_time
	  	foreach	($userIds as $uid) {
			$f2_username = $db->getF2Username($uid);
			if ($f2_username != "") {
				$next_run_time = $db->getNextBatchRunTime($uid);
				// if current_time > next_batch_run_time then run the command
				if ($next_run_time <= time()) {
					// run command
					$pool = new F2Pool();
					$pool->setUsername($f2_username);
					if ($pool->fetchPoolInfo()) {
						$m = "[Auto Monitor]\n";
						$m .= $pool->getStatusSummaryMessage();
						$tg = new Telegram();
						$tg->setChatId($db->getChatId($uid));
						$tg->returnTgMessage($m);
						$tg = null;
					} else {
						$m = date('Y/m/d H:i:s') . ": " . $ERROR["CANNOT_RETRIEVE_POOL_INFO"];
						echo $m;
						//$tg->returnTgMessage($m);
					}
					$pool = null;
					// set next_batch_run_time for that user
					$db->setNextBatchRunTime($uid);
				} else {
				}
			}
	  	}
		// SHOW STATS
		if($db->isShowStatMode()) {
			$tg = new Telegram();
			$tg->setChatId($db->getChatId($ENV["ADMIN_UID"]));
			$tg->returnTgMessage(makeStatusMessage());
			$tg = null;
			$db->setShowStatModeOff();
		}
		if($stat_counter_logfile <= 0) {
			$m = "\n\n";
			$m .=  makeStatusMessage();
			$m .= "\n";
			echo $m;
			$stat_counter_logfile = $default_stat_counter;
		} else {
			$stat_counter_logfile--;
		}
		sleep($sleep_interval);
	  }
	  $db->setBatchRunningStatusOff();
  }
} else {
}

function makeStatusMessage() {
	global $db;
	$m = date('Y/m/d H:i:s') . "\n";
	$m .= "Number of TG users: " . $db->getNumOfTelegramUsernames() . "\n";
	$m .= "Memory Usage:\n";
	$m .= "memory_get_usage(true): " . memory_get_usage(true) . "\n";
	$m .= "memory_get_usage(false): " . memory_get_usage(false) . "\n";
	$m .= "memory_get_peak_usage(true): " . memory_get_peak_usage(true) . "\n";
	$m .= "memory_get_peak_usage(false): " . memory_get_peak_usage(false) . "\n";
	return $m; 
}
?>