mirror of
https://we.phorge.it/source/phorge.git
synced 2024-11-10 00:42:41 +01:00
226 lines
6 KiB
PHP
Executable file
226 lines
6 KiB
PHP
Executable file
#!/usr/bin/env php
|
|
<?php
|
|
|
|
/*
|
|
* Copyright 2011 Facebook, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
$root = dirname(dirname(dirname(__FILE__)));
|
|
require_once $root.'/scripts/__init_script__.php';
|
|
require_once $root.'/scripts/__init_env__.php';
|
|
|
|
phutil_require_module('phabricator', 'infrastructure/daemon/control');
|
|
$control = new PhabricatorDaemonControl();
|
|
|
|
switch (isset($argv[1]) ? $argv[1] : 'help') {
|
|
case 'list':
|
|
$err = $control->executeListCommand();
|
|
exit($err);
|
|
|
|
case 'status':
|
|
$err = $control->executeStatusCommand();
|
|
exit($err);
|
|
|
|
case 'stop':
|
|
$err = $control->executeStopCommand();
|
|
exit($err);
|
|
|
|
case 'repository-launch-readonly':
|
|
$need_launch = phd_load_tracked_repositories_of_type('git');
|
|
if (!$need_launch) {
|
|
echo "There are no repositories with tracking enabled.\n";
|
|
} else {
|
|
will_launch($control);
|
|
|
|
foreach ($need_launch as $repository) {
|
|
$name = $repository->getName();
|
|
$callsign = $repository->getCallsign();
|
|
$desc = "'{$name}' ({$callsign})";
|
|
$phid = $repository->getPHID();
|
|
|
|
echo "Launching 'git fetch' daemon on the {$desc} repository...\n";
|
|
$control->launchDaemon(
|
|
'PhabricatorRepositoryGitFetchDaemon',
|
|
array(
|
|
$phid,
|
|
));
|
|
}
|
|
}
|
|
break;
|
|
|
|
case 'repository-launch-master':
|
|
$need_launch = phd_load_tracked_repositories();
|
|
if (!$need_launch) {
|
|
echo "There are no repositories with tracking enabled.\n";
|
|
} else {
|
|
will_launch($control);
|
|
|
|
foreach ($need_launch as $repository) {
|
|
$name = $repository->getName();
|
|
$callsign = $repository->getCallsign();
|
|
$desc = "'{$name}' ({$callsign})";
|
|
$phid = $repository->getPHID();
|
|
|
|
switch ($repository->getVersionControlSystem()) {
|
|
case 'git':
|
|
echo "Launching 'git fetch' daemon on the {$desc} repository...\n";
|
|
$control->launchDaemon(
|
|
'PhabricatorRepositoryGitFetchDaemon',
|
|
array(
|
|
$phid,
|
|
));
|
|
echo "Launching discovery daemon on the {$desc} repository...\n";
|
|
$control->launchDaemon(
|
|
'PhabricatorRepositoryGitCommitDiscoveryDaemon',
|
|
array(
|
|
$phid,
|
|
));
|
|
break;
|
|
case 'svn':
|
|
echo "Launching discovery daemon on the {$desc} repository...\n";
|
|
$control->launchDaemon(
|
|
'PhabricatorRepositorySvnCommitDiscoveryDaemon',
|
|
array(
|
|
$phid,
|
|
));
|
|
break;
|
|
}
|
|
}
|
|
|
|
echo "Launching CommitTask daemon...\n";
|
|
$control->launchDaemon(
|
|
'PhabricatorRepositoryCommitTaskDaemon',
|
|
array());
|
|
|
|
echo "Done.\n";
|
|
}
|
|
break;
|
|
|
|
case 'launch':
|
|
case 'debug':
|
|
$is_debug = ($argv[1] == 'debug');
|
|
|
|
$daemon = idx($argv, 2);
|
|
if (!$daemon) {
|
|
throw new Exception("Daemon name required!");
|
|
}
|
|
|
|
$pass_argv = array_slice($argv, 3);
|
|
|
|
$n = 1;
|
|
if (!$is_debug) {
|
|
if (is_numeric($daemon)) {
|
|
$n = $daemon;
|
|
if ($n < 1) {
|
|
throw new Exception("Count must be at least 1!");
|
|
}
|
|
$daemon = idx($argv, 3);
|
|
if (!$daemon) {
|
|
throw new Exception("Daemon name required!");
|
|
}
|
|
$pass_argv = array_slice($argv, 4);
|
|
}
|
|
}
|
|
|
|
$loader = new PhutilSymbolLoader();
|
|
$symbols = $loader
|
|
->setAncestorClass('PhutilDaemon')
|
|
->selectSymbolsWithoutLoading();
|
|
|
|
$symbols = ipull($symbols, 'name');
|
|
$match = array();
|
|
foreach ($symbols as $symbol) {
|
|
if (stripos($symbol, $daemon) !== false) {
|
|
if (strtolower($symbol) == strtolower($daemon)) {
|
|
$match = array($symbol);
|
|
break;
|
|
} else {
|
|
$match[] = $symbol;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (count($match) == 0) {
|
|
throw new Exception(
|
|
"No daemons match! Use 'phd list' for a list of daemons.");
|
|
} else if (count($match) > 1) {
|
|
throw new Exception(
|
|
"Which of these daemons did you mean?\n".
|
|
" ".implode("\n ", $match));
|
|
} else {
|
|
$daemon = reset($match);
|
|
}
|
|
|
|
will_launch($control);
|
|
|
|
if ($is_debug) {
|
|
echo "Launching {$daemon} in debug mode (nondaemonized)...\n";
|
|
} else {
|
|
echo "Launching {$n} x {$daemon}";
|
|
}
|
|
|
|
for ($ii = 0; $ii < $n; $ii++) {
|
|
$control->launchDaemon($daemon, $pass_argv, $is_debug);
|
|
if (!$is_debug) {
|
|
echo ".";
|
|
}
|
|
}
|
|
|
|
echo "\n";
|
|
echo "Done.\n";
|
|
|
|
break;
|
|
|
|
case '--help':
|
|
case 'help':
|
|
default:
|
|
$err = $control->executeHelpCommand();
|
|
exit($err);
|
|
}
|
|
|
|
function phd_load_tracked_repositories_of_type($type) {
|
|
$repositories = phd_load_tracked_repositories();
|
|
|
|
foreach ($repositories as $key => $repository) {
|
|
if ($repository->getVersionControlSystem() != $type) {
|
|
unset($repositories[$key]);
|
|
}
|
|
}
|
|
|
|
return $repositories;
|
|
}
|
|
|
|
function phd_load_tracked_repositories() {
|
|
phutil_require_module(
|
|
'phabricator',
|
|
'applications/repository/storage/repository');
|
|
|
|
$repositories = id(new PhabricatorRepository())->loadAll();
|
|
foreach ($repositories as $key => $repository) {
|
|
if (!$repository->getDetail('tracking-enabled')) {
|
|
unset($repositories[$key]);
|
|
}
|
|
}
|
|
|
|
return $repositories;
|
|
}
|
|
|
|
function will_launch($control) {
|
|
echo "Staging launch...\n";
|
|
$control->pingConduit();
|
|
$log_dir = $control->getControlDirectory('log').'/daemons.log';
|
|
echo "NOTE: Logs will appear in '{$log_dir}'.\n\n";
|
|
}
|
|
|