mirror of
https://we.phorge.it/source/phorge.git
synced 2025-03-30 22:18:13 +02:00
da0dfc057d
Summary: Fixes T13326. In D20571, I slightly generalized construction of an iterator over a set of files, but missed some code in other "bin/files ..." commands which was also affected. Today, basically all of these workflows define their own "--all" and "names" flags. Pull these definitions up and implement them more consistently. Test Plan: Ran multiple different `bin/files` commands with different combinations of arguments, saw consistent handling of iterator construction. Reviewers: amckinley Reviewed By: amckinley Maniphest Tasks: T13326 Differential Revision: https://secure.phabricator.com/D20614
140 lines
3.3 KiB
PHP
140 lines
3.3 KiB
PHP
<?php
|
|
|
|
final class PhabricatorFilesManagementEncodeWorkflow
|
|
extends PhabricatorFilesManagementWorkflow {
|
|
|
|
protected function didConstruct() {
|
|
$arguments = $this->newIteratorArguments();
|
|
|
|
$arguments[] = array(
|
|
'name' => 'as',
|
|
'param' => 'format',
|
|
'help' => pht('Select the storage format to use.'),
|
|
);
|
|
|
|
$arguments[] = array(
|
|
'name' => 'key',
|
|
'param' => 'keyname',
|
|
'help' => pht('Select a specific storage key.'),
|
|
);
|
|
|
|
$arguments[] = array(
|
|
'name' => 'force',
|
|
'help' => pht(
|
|
'Re-encode files which are already stored in the target '.
|
|
'encoding.'),
|
|
);
|
|
|
|
$this
|
|
->setName('encode')
|
|
->setSynopsis(
|
|
pht('Change the storage encoding of files.'))
|
|
->setArguments($arguments);
|
|
}
|
|
|
|
public function execute(PhutilArgumentParser $args) {
|
|
$iterator = $this->buildIterator($args);
|
|
|
|
$force = (bool)$args->getArg('force');
|
|
|
|
$format_list = PhabricatorFileStorageFormat::getAllFormats();
|
|
$format_list = array_keys($format_list);
|
|
$format_list = implode(', ', $format_list);
|
|
|
|
$format_key = $args->getArg('as');
|
|
if (!strlen($format_key)) {
|
|
throw new PhutilArgumentUsageException(
|
|
pht(
|
|
'Use --as <format> to select a target encoding format. Available '.
|
|
'formats are: %s.',
|
|
$format_list));
|
|
}
|
|
|
|
$format = PhabricatorFileStorageFormat::getFormat($format_key);
|
|
if (!$format) {
|
|
throw new PhutilArgumentUsageException(
|
|
pht(
|
|
'Storage format "%s" is not valid. Available formats are: %s.',
|
|
$format_key,
|
|
$format_list));
|
|
}
|
|
|
|
$key_name = $args->getArg('key');
|
|
if (strlen($key_name)) {
|
|
$format->selectMasterKey($key_name);
|
|
}
|
|
|
|
$engines = PhabricatorFileStorageEngine::loadAllEngines();
|
|
|
|
$failed = array();
|
|
foreach ($iterator as $file) {
|
|
$monogram = $file->getMonogram();
|
|
|
|
$engine_key = $file->getStorageEngine();
|
|
$engine = idx($engines, $engine_key);
|
|
|
|
if (!$engine) {
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht(
|
|
'%s: Uses unknown storage engine "%s".',
|
|
$monogram,
|
|
$engine_key));
|
|
$failed[] = $file;
|
|
continue;
|
|
}
|
|
|
|
if ($engine->isChunkEngine()) {
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht(
|
|
'%s: Stored as chunks, no data to encode directly.',
|
|
$monogram));
|
|
continue;
|
|
}
|
|
|
|
if (($file->getStorageFormat() == $format_key) && !$force) {
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht(
|
|
'%s: Already encoded in target format.',
|
|
$monogram));
|
|
continue;
|
|
}
|
|
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht(
|
|
'%s: Changing encoding from "%s" to "%s".',
|
|
$monogram,
|
|
$file->getStorageFormat(),
|
|
$format_key));
|
|
|
|
try {
|
|
$file->migrateToStorageFormat($format);
|
|
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht('Done.'));
|
|
} catch (Exception $ex) {
|
|
echo tsprintf(
|
|
"%B\n",
|
|
pht('Failed! %s', (string)$ex));
|
|
$failed[] = $file;
|
|
}
|
|
}
|
|
|
|
if ($failed) {
|
|
$monograms = mpull($failed, 'getMonogram');
|
|
|
|
echo tsprintf(
|
|
"%s\n",
|
|
pht('Failures: %s.', implode(', ', $monograms)));
|
|
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
}
|