View Options

	diff --git a/src/infrastructure/daemon/bot/PhabricatorBot.php b/src/infrastructure/daemon/bot/PhabricatorBot.php
	index 656c5861fd..d4bac41ff5 100644
	--- a/src/infrastructure/daemon/bot/PhabricatorBot.php
	+++ b/src/infrastructure/daemon/bot/PhabricatorBot.php
	@@ -1,139 +1,143 @@
	<?php

	/**
	* Simple IRC bot which runs as a Phabricator daemon. Although this bot is
	* somewhat useful, it is also intended to serve as a demo of how to write
	* "system agents" which communicate with Phabricator over Conduit, so you can
	* script system interactions and integrate with other systems.
	*
	* NOTE: This is super janky and experimental right now.
	*
	* @group irc
	*/
	final class PhabricatorBot extends PhabricatorDaemon {

	private $handlers;

	private $conduit;
	private $config;
	private $pollFrequency;

	public function run() {
	$argv = $this->getArgv();
	if (count($argv) !== 1) {
	throw new Exception("usage: PhabricatorBot <json_config_file>");
	}

	$json_raw = Filesystem::readFile($argv[0]);
	$config = json_decode($json_raw, true);
	if (!is_array($config)) {
	throw new Exception("File '{$argv[0]}' is not valid JSON!");
	}

	$nick = idx($config, 'nick', 'phabot');
	$handlers = idx($config, 'handlers', array());
	$protocol_adapter_class = idx(
	$config,
	'protocol-adapter',
	'PhabricatorIRCProtocolAdapter');
	$this->pollFrequency = idx($config, 'poll-frequency', 1);

	$this->config = $config;

	foreach ($handlers as $handler) {
	$obj = newv($handler, array($this));
	$this->handlers[] = $obj;
	}

	$conduit_uri = idx($config, 'conduit.uri');
	if ($conduit_uri) {
	$conduit_user = idx($config, 'conduit.user');
	$conduit_cert = idx($config, 'conduit.cert');

	// Normalize the path component of the URI so users can enter the
	// domain without the "/api/" part.
	$conduit_uri = new PhutilURI($conduit_uri);
	$conduit_uri->setPath('/api/');
	$conduit_uri = (string)$conduit_uri;

	$conduit = new ConduitClient($conduit_uri);
	$response = $conduit->callMethodSynchronous(
	'conduit.connect',
	array(
	'client' => 'PhabricatorBot',
	'clientVersion' => '1.0',
	'clientDescription' => php_uname('n').':'.$nick,
	'user' => $conduit_user,
	'certificate' => $conduit_cert,
	));

	$this->conduit = $conduit;
	}

	// Instantiate Protocol Adapter, for now follow same technique as
	// handler instantiation
	$this->protocolAdapter = newv($protocol_adapter_class, array());
	$this->protocolAdapter
	->setConfig($this->config)
	->connect();

	$this->runLoop();
	}

	public function getConfig($key, $default = null) {
	return idx($this->config, $key, $default);
	}

	private function runLoop() {
	do {
	$this->stillWorking();

	$messages = $this->protocolAdapter->getNextMessages($this->pollFrequency);
	if (count($messages) > 0) {
	foreach ($messages as $message) {
	$this->routeMessage($message);
	}
	}

	foreach ($this->handlers as $handler) {
	$handler->runBackgroundTasks();
	}
	} while (true);
	}

	public function writeMessage(PhabricatorBotMessage $message) {
	return $this->protocolAdapter->writeMessage($message);
	}

	private function routeMessage(PhabricatorBotMessage $message) {
	$ignore = $this->getConfig('ignore');
	if ($ignore) {
	$sender = $message->getSender();
	if ($sender && in_array($sender->getName(), $ignore)) {
	return;
	}
	}

	+ if ($message->getCommand() == 'LOG') {
	+ $this->log("[LOG] ".$message->getBody());
	+ }
	+
	foreach ($this->handlers as $handler) {
	try {
	$handler->receiveMessage($message);
	} catch (Exception $ex) {
	phlog($ex);
	}
	}
	}

	public function getAdapter() {
	return $this->protocolAdapter;
	}

	public function getConduit() {
	if (empty($this->conduit)) {
	throw new Exception(
	"This bot is not configured with a Conduit uplink. Set 'conduit.uri', ".
	"'conduit.user' and 'conduit.cert' in the configuration to connect.");
	}
	return $this->conduit;
	}

	}

File Metadata

Mime Type: text/x-diff
Expires: Jan 19 2025, 21:16 (6 w, 1 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 1128843
Default Alt Text: (4 KB)

No OneTemporary
Actions

View Options

File Metadata

Event Timeline

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions