コード例 #1
0
 public function execute()
 {
     $stripProtocols = (bool) $this->getOption('strip-protocols', false);
     $forceProtocol = $this->getOption('force-protocol', null);
     $url = $this->getOption('load-from', 'https://meta.wikimedia.org/w/api.php');
     $scriptPath = $this->getOption('script-path', '/w/$1');
     $articlePath = $this->getOption('article-path', '/wiki/$1');
     $expandGroup = !$this->getOption('no-expand-group', false);
     $siteGroup = $this->getOption('site-group');
     $wikiId = $this->getOption('wiki');
     if ($stripProtocols && is_string($forceProtocol)) {
         $this->error("You can't use both strip-protocols and force-protocol", 1);
     }
     $protocol = true;
     if ($stripProtocols) {
         $protocol = false;
     } elseif (is_string($forceProtocol)) {
         $protocol = $forceProtocol;
     }
     // @todo make it configurable, such as from a config file.
     $validGroups = array('wikipedia', 'wikivoyage', 'wikiquote', 'wiktionary', 'wikibooks', 'wikisource', 'wikiversity', 'wikinews');
     try {
         $json = $this->getSiteMatrixData($url);
         $siteMatrixParser = new SiteMatrixParser($scriptPath, $articlePath, $protocol, $expandGroup);
         $sites = $siteMatrixParser->sitesFromJson($json);
         $store = SiteSQLStore::newInstance();
         $sitesBuilder = new SitesBuilder($store, $validGroups);
         $sitesBuilder->buildStore($sites, $siteGroup, $wikiId);
     } catch (MWException $e) {
         $this->output($e->getMessage());
     }
     $this->output("done.\n");
 }
コード例 #2
0
 /**
  * @dataProvider sitesFromJsonProvider
  */
 public function testSitesFromJson($scriptPath, $articlePath, $protocol, $expected)
 {
     $json = $this->getSiteMatrixJson();
     $siteMatrixParser = new SiteMatrixParser($scriptPath, $articlePath, $protocol);
     $sites = $siteMatrixParser->sitesFromJson($json);
     ksort($expected);
     ksort($sites);
     $this->assertEquals($expected, $sites);
 }