mediawiki-extensions-Echo/includes/ForeignNotifications.php

<?php

class EchoForeignNotifications {
	/**
	 * @var bool|EchoUnreadWikis
	 */
	protected $unreadWikis = false;

	/**
	 * @var array [(str) section => (int) count, ...]
	 */
	protected $counts = array( EchoAttributeManager::ALERT => 0, EchoAttributeManager::MESSAGE => 0 );

	/**
	 * @var array [(str) section => (string[]) wikis, ...]
	 */
	protected $wikis = array( EchoAttributeManager::ALERT => array(), EchoAttributeManager::MESSAGE => array() );

	/**
	 * @var array [(str) section => (MWTimestamp) timestamp, ...]
	 */
	protected $timestamps = array( EchoAttributeManager::ALERT => false, EchoAttributeManager::MESSAGE => false );

	/**
	 * @var bool
	 */
	protected $populated = false;

	/**
	 * @param User $user
	 */
	public function __construct( User $user ) {
		if ( $user->getOption( 'echo-cross-wiki-notifications' ) ) {
			$this->unreadWikis = EchoUnreadWikis::newFromUser( $user );
		}
	}

	/**
	 * @param string|null $section Name of section or null for all
	 * @return int
	 */
	public function getCount( $section = null ) {
		$this->populate();

		if ( $section === null ) {
			return array_sum( $this->counts );
		}

		return isset( $this->counts[$section] ) ? $this->counts[$section] : 0;
	}

	/**
	 * @param string|null $section Name of section or null for all
	 * @return MWTimestamp|false
	 */
	public function getTimestamp( $section = null ) {
		$this->populate();

		if ( $section === null ) {
			$max = false;
			/** @var MWTimestamp $timestamp */
			foreach ( $this->timestamps as $timestamp ) {
				// $timestamp < $max = invert 0
				// $timestamp > $max = invert 1
				if ( $max === false || $timestamp->diff( $max )->invert === 1 ) {
					$max = $timestamp;
				}
			}

			return $timestamp;
		}

		return isset( $this->timestamps[$section] ) ? $this->timestamps[$section] : false;
	}

	/**
	 * @param string|null $section Name of section or null for all
	 * @return string[]
	 */
	public function getWikis( $section = null ) {
		$this->populate();

		if ( $section === null ) {
			$all = array();
			foreach ( $this->wikis as $wikis ) {
				$all = array_merge( $all, $wikis );
			}

			return array_unique( $all );
		}

		return isset( $this->wikis[$section] ) ? $this->wikis[$section] : array();
	}

	protected function populate() {
		if ( $this->populated ) {
			return;
		}

		if ( $this->unreadWikis === false ) {
			return;
		}

		$unreadCounts = $this->unreadWikis->getUnreadCounts();
		if ( !$unreadCounts ) {
			return;
		}

		foreach ( $unreadCounts as $wiki => $sections ) {
			// exclude current wiki
			if ( $wiki === wfWikiID() ) {
				continue;
			}

			foreach ( $sections as $section => $data ) {
				if ( $data['count'] > 0 ) {
					$this->counts[$section] += $data['count'];
					$this->wikis[$section][] = $wiki;
					$this->timestamps[$section] = new MWTimestamp( $data['ts'] );
				}
			}
		}

		$this->populated = true;
	}

	/**
	 * @param string[] $wikis
	 * @return array[] [(string) wiki => (array) data]
	 */
	public function getApiEndpoints( array $wikis ) {
		global $wgConf;

		$data = array();
		foreach ( $wikis as $wiki ) {
			$data[$wiki] = array(
				'title' => $wiki,
				'url' => $wgConf->get( 'wgServer', $wiki ) .
					$wgConf->get( 'wgScriptPath', $wiki ) .
					'/api.php'
			);
		}

		return $data;
	}
}