nextcloud-server/apps/federation/lib/AppInfo/Application.php

<?php
/**
 * @copyright Copyright (c) 2016, ownCloud, Inc.
 *
 * @author Björn Schießle <bjoern@schiessle.org>
 * @author Robin Appelman <robin@icewind.nl>
 * @author Thomas Müller <thomas.mueller@tmit.eu>
 *
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */

namespace OCA\Federation\AppInfo;

use OCA\Federation\API\OCSAuthAPI;
use OCA\Federation\Controller\SettingsController;
use OCA\Federation\DAV\FedAuth;
use OCA\Federation\DbHandler;
use OCA\Federation\Hooks;
use OCA\Federation\Middleware\AddServerMiddleware;
use OCA\Federation\SyncFederationAddressBooks;
use OCA\Federation\SyncJob;
use OCA\Federation\TrustedServers;
use OCP\API;
use OCP\App;
use OCP\AppFramework\IAppContainer;
use OCP\SabrePluginEvent;
use OCP\Util;
use Sabre\DAV\Auth\Plugin;

class Application extends \OCP\AppFramework\App {

	/**
	 * @param array $urlParams
	 */
	public function __construct($urlParams = array()) {
		parent::__construct('federation', $urlParams);
		$this->registerService();
		$this->registerMiddleware();
	}

	/**
	 * register setting scripts
	 */
	public function registerSettings() {
		App::registerAdmin('federation', 'settings/settings-admin');
	}

	private function registerService() {
		$container = $this->getContainer();

		$container->registerService('addServerMiddleware', function(IAppContainer $c) {
			return new AddServerMiddleware(
				$c->getAppName(),
				\OC::$server->getL10N($c->getAppName()),
				\OC::$server->getLogger()
			);
		});

		$container->registerService('DbHandler', function(IAppContainer $c) {
			return new DbHandler(
				\OC::$server->getDatabaseConnection(),
				\OC::$server->getL10N($c->getAppName())
			);
		});

		$container->registerService('TrustedServers', function(IAppContainer $c) {
			$server = $c->getServer();
			return new TrustedServers(
				$c->query('DbHandler'),
				$server->getHTTPClientService(),
				$server->getLogger(),
				$server->getJobList(),
				$server->getSecureRandom(),
				$server->getConfig(),
				$server->getEventDispatcher()
			);
		});

		$container->registerService('SettingsController', function (IAppContainer $c) {
			$server = $c->getServer();
			return new SettingsController(
				$c->getAppName(),
				$server->getRequest(),
				$server->getL10N($c->getAppName()),
				$c->query('TrustedServers')
			);
		});

	}

	private function registerMiddleware() {
		$container = $this->getContainer();
		$container->registerMiddleware('addServerMiddleware');
	}

	/**
	 * register OCS API Calls
	 */
	public function registerOCSApi() {

		$container = $this->getContainer();
		$server = $container->getServer();

		$auth = new OCSAuthAPI(
			$server->getRequest(),
			$server->getSecureRandom(),
			$server->getJobList(),
			$container->query('TrustedServers'),
			$container->query('DbHandler'),
			$server->getLogger()

		);

		API::register('get',
			'/apps/federation/api/v1/shared-secret',
			array($auth, 'getSharedSecret'),
			'federation',
			API::GUEST_AUTH
		);

		API::register('post',
			'/apps/federation/api/v1/request-shared-secret',
			array($auth, 'requestSharedSecret'),
			'federation',
			API::GUEST_AUTH
		);

	}

	/**
	 * listen to federated_share_added hooks to auto-add new servers to the
	 * list of trusted servers.
	 */
	public function registerHooks() {

		$container = $this->getContainer();
		$hooksManager = new Hooks($container->query('TrustedServers'));

		Util::connectHook(
				'OCP\Share',
				'federated_share_added',
				$hooksManager,
				'addServerHook'
		);

		$dispatcher = $this->getContainer()->getServer()->getEventDispatcher();
		$dispatcher->addListener('OCA\DAV\Connector\Sabre::authInit', function($event) use($container) {
			if ($event instanceof SabrePluginEvent) {
				$authPlugin = $event->getServer()->getPlugin('auth');
				if ($authPlugin instanceof Plugin) {
					$h = new DbHandler($container->getServer()->getDatabaseConnection(),
							$container->getServer()->getL10N('federation')
					);
					$authPlugin->addBackend(new FedAuth($h));
				}
			}
		});
	}

	/**
	 * @return SyncFederationAddressBooks
	 */
	public function getSyncService() {
		$syncService = \OC::$server->query('CardDAVSyncService');
		$dbHandler = $this->getContainer()->query('DbHandler');
		return new SyncFederationAddressBooks($dbHandler, $syncService);
	}

}