From 8d010b9721f7d5614b8cf8b0afef5e3eb13de3f5 Mon Sep 17 00:00:00 2001
From: Florian <florian.schmidt.stargatewissen@gmail.com>
Date: Mon, 18 Apr 2016 20:03:53 +0200
Subject: [PATCH] Add maintenance script to add sites to sites table

Also: Add language code validation to Site::setLanguageCode().

Bug: T132937
Change-Id: I763ec65cb06d5250a3886a66eefdde8701b2299c
---
 autoload.php            |  1 +
 includes/site/Site.php  |  3 ++
 maintenance/addSite.php | 91 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 95 insertions(+)
 create mode 100755 maintenance/addSite.php

diff --git a/autoload.php b/autoload.php
index 18635770de..e56b681453 100644
--- a/autoload.php
+++ b/autoload.php
@@ -11,6 +11,7 @@ $wgAutoloadLocalClasses = [
 	'ActiveUsersPager' => __DIR__ . '/includes/specials/pagers/ActiveUsersPager.php',
 	'ActivityUpdateJob' => __DIR__ . '/includes/jobqueue/jobs/ActivityUpdateJob.php',
 	'AddRFCAndPMIDInterwiki' => __DIR__ . '/maintenance/addRFCandPMIDInterwiki.php',
+	'AddSite' => __DIR__ . '/maintenance/addSite.php',
 	'AjaxDispatcher' => __DIR__ . '/includes/AjaxDispatcher.php',
 	'AjaxResponse' => __DIR__ . '/includes/AjaxResponse.php',
 	'AllMessagesTablePager' => __DIR__ . '/includes/specials/pagers/AllMessagesTablePager.php',
diff --git a/includes/site/Site.php b/includes/site/Site.php
index 6a97a50209..28f19f9abf 100644
--- a/includes/site/Site.php
+++ b/includes/site/Site.php
@@ -463,6 +463,9 @@ class Site implements Serializable {
 	 * @param string $languageCode
 	 */
 	public function setLanguageCode( $languageCode ) {
+		if ( !Language::isValidCode( $languageCode ) ) {
+			throw new InvalidArgumentException( "$languageCode is not a valid language code." );
+		}
 		$this->languageCode = $languageCode;
 	}
 
diff --git a/maintenance/addSite.php b/maintenance/addSite.php
new file mode 100755
index 0000000000..1ebc4f6643
--- /dev/null
+++ b/maintenance/addSite.php
@@ -0,0 +1,91 @@
+<?php
+
+use MediaWiki\MediaWikiServices;
+
+$basePath = getenv( 'MW_INSTALL_PATH' ) !== false ? getenv( 'MW_INSTALL_PATH' ) : __DIR__ . '/..';
+
+require_once $basePath . '/maintenance/Maintenance.php';
+
+/**
+ * Maintenance script for adding a site definition into the sites table.
+ *
+ * @since 1.29
+ *
+ * @license GNU GPL v2+
+ * @author Florian Schmidt
+ */
+class AddSite extends Maintenance {
+
+	public function __construct() {
+		$this->addDescription( 'Add a site definition into the sites table.' );
+
+		$this->addArg( 'globalid', 'The global id of the site to add, e.g. "wikipedia".', true );
+		$this->addArg( 'group', 'In which group this site should be sorted in.', true );
+		$this->addOption( 'language', 'The language code of the site, e.g. "de".' );
+		$this->addOption( 'interwiki-id', 'The interwiki ID of the site.' );
+		$this->addOption( 'navigation-id', 'The navigation ID of the site.' );
+		$this->addOption( 'pagepath', 'The URL to pages of this site, e.g.' .
+			' https://example.com/wiki/\$1.' );
+		$this->addOption( 'filepath', 'The URL to files of this site, e.g. https://example
+		.com/w/\$1.' );
+
+		parent::__construct();
+	}
+
+	/**
+	 * Imports the site described by the parameters (see self::__construct()) passed to this
+	 * maintenance sccript into the sites table of MediaWiki.
+	 */
+	public function execute() {
+		$siteStore = MediaWikiServices::getInstance()->getSiteStore();
+		$siteStore->reset();
+
+		$globalId = $this->getArg( 0 );
+		$group = $this->getArg( 1 );
+		$language = $this->getOption( 'language' );
+		$interwikiId = $this->getOption( 'interwiki-id' );
+		$navigationId = $this->getOption( 'navigation-id' );
+		$pagepath = $this->getOption( 'pagepath' );
+		$filepath = $this->getOption( 'filepath' );
+
+		if ( !is_string( $globalId ) || !is_string( $group ) ) {
+			echo "Arguments globalid and group need to be strings.\n";
+			return false;
+		}
+
+		if ( $siteStore->getSite( $globalId ) !== null ) {
+			echo "Site with global id $globalId already exists.\n";
+			return false;
+		}
+
+		$site = new MediaWikiSite();
+		$site->setGlobalId( $globalId );
+		$site->setGroup( $group );
+		if ( $language !== null ) {
+			$site->setLanguageCode( $language );
+		}
+		if ( $interwikiId !== null ) {
+			$site->addInterwikiId( $interwikiId );
+		}
+		if ( $navigationId !== null ) {
+			$site->addNavigationId( $navigationId );
+		}
+		if ( $pagepath !== null ) {
+			$site->setPagePath( $pagepath );
+		}
+		if ( $filepath !== null ) {
+			$site->setFilePath( $filepath );
+		}
+
+		$siteStore->saveSites( [ $site ] );
+
+		if ( method_exists( $siteStore, 'reset' ) ) {
+			$siteStore->reset();
+		}
+
+		echo "Done.\n";
+	}
+}
+
+$maintClass = 'AddSite';
+require_once RUN_MAINTENANCE_IF_MAIN;
-- 
2.20.1