Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F96123707
PhabricatorLipsumGenerateWorkflow.php
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sun, Dec 22, 21:51
Size
5 KB
Mime Type
text/x-php
Expires
Tue, Dec 24, 21:51 (1 d, 23 h)
Engine
blob
Format
Raw Data
Handle
23127030
Attached To
rPH Phabricator
PhabricatorLipsumGenerateWorkflow.php
View Options
<?php
final
class
PhabricatorLipsumGenerateWorkflow
extends
PhabricatorLipsumManagementWorkflow
{
protected
function
didConstruct
()
{
$this
->
setName
(
'generate'
)
->
setExamples
(
'**generate**'
)
->
setSynopsis
(
pht
(
'Generate synthetic test objects.'
))
->
setArguments
(
array
(
array
(
'name'
=>
'force'
,
'short'
=>
'f'
,
'help'
=>
pht
(
'Generate objects without prompting for confirmation.'
),
),
array
(
'name'
=>
'quickly'
,
'help'
=>
pht
(
'Generate objects as quickly as possible.'
),
),
array
(
'name'
=>
'args'
,
'wildcard'
=>
true
,
),
));
}
public
function
execute
(
PhutilArgumentParser
$args
)
{
$config_key
=
'phabricator.developer-mode'
;
if
(!
PhabricatorEnv
::
getEnvConfig
(
$config_key
))
{
throw
new
PhutilArgumentUsageException
(
pht
(
'lipsum is a development and testing tool and may only be run '
.
'on installs in developer mode. Enable "%s" in your configuration '
.
'to enable lipsum.'
,
$config_key
));
}
$all_generators
=
id
(
new
PhutilClassMapQuery
())
->
setAncestorClass
(
'PhabricatorTestDataGenerator'
)
->
setUniqueMethod
(
'getGeneratorKey'
)
->
execute
();
$argv
=
$args
->
getArg
(
'args'
);
$is_force
=
$args
->
getArg
(
'force'
);
$is_quickly
=
$args
->
getArg
(
'quickly'
);
$all
=
'all'
;
if
(
isset
(
$all_generators
[
$all
]))
{
throw
new
Exception
(
pht
(
'A lipsum generator is registered with key "%s". This key is '
.
'reserved.'
,
$all
));
}
if
(!
$argv
)
{
ksort
(
$all_generators
);
$names
=
array
();
foreach
(
$all_generators
as
$generator
)
{
$names
[]
=
tsprintf
(
'%s (%s)'
,
$generator
->
getGeneratorKey
(),
$generator
->
getGeneratorName
());
}
$list
=
id
(
new
PhutilConsoleList
())
->
setWrap
(
false
)
->
addItems
(
$names
);
id
(
new
PhutilConsoleBlock
())
->
addParagraph
(
pht
(
'Choose which type or types of test data you want to generate, '
.
'or select "%s".'
,
$all
))
->
addList
(
$list
)
->
draw
();
return
0
;
}
$generators
=
array
();
foreach
(
$argv
as
$arg_original
)
{
$arg
=
phutil_utf8_strtolower
(
$arg_original
);
if
(
$arg
==
'all'
)
{
$matches
=
$all_generators
;
}
else
{
$matches
=
array
();
foreach
(
$all_generators
as
$generator
)
{
$name
=
phutil_utf8_strtolower
(
$generator
->
getGeneratorKey
());
// If there's an exact match, select just that generator.
if
(
$arg
==
$name
)
{
$matches
=
array
(
$generator
);
break
;
}
// If there's a partial match, match that generator but continue.
if
(
strpos
(
$name
,
$arg
)
!==
false
)
{
$matches
[]
=
$generator
;
}
}
if
(!
$matches
)
{
throw
new
PhutilArgumentUsageException
(
pht
(
'Argument "%s" does not match the name of any generators.'
,
$arg_original
));
}
if
(
count
(
$matches
)
>
1
)
{
throw
new
PhutilArgumentUsageException
(
pht
(
'Argument "%s" is ambiguous, and matches multiple '
.
'generators: %s.'
,
$arg_original
,
implode
(
', '
,
mpull
(
$matches
,
'getGeneratorName'
))));
}
}
foreach
(
$matches
as
$match
)
{
$generators
[]
=
$match
;
}
}
$generators
=
mpull
(
$generators
,
null
,
'getGeneratorKey'
);
echo
tsprintf
(
"**<bg:blue> %s </bg>** %s
\n
"
,
pht
(
'GENERATORS'
),
pht
(
'Selected generators: %s.'
,
implode
(
', '
,
mpull
(
$generators
,
'getGeneratorName'
))));
if
(!
$is_force
)
{
echo
tsprintf
(
"**<bg:yellow> %s </bg>** %s
\n
"
,
pht
(
'WARNING'
),
pht
(
'This command generates synthetic test data, including user '
.
'accounts. It is intended for use in development environments so '
.
'you can test features more easily. There is no easy way to delete '
.
'this data or undo the effects of this command. If you run it in a '
.
'production environment, it will pollute your data with large '
.
'amounts of meaningless garbage that you can not get rid of.'
));
$prompt
=
pht
(
'Are you sure you want to generate piles of garbage?'
);
if
(!
phutil_console_confirm
(
$prompt
,
true
))
{
return
;
}
}
echo
tsprintf
(
"**<bg:green> %s </bg>** %s
\n
"
,
pht
(
'LIPSUM'
),
pht
(
'Generating synthetic test objects forever. '
.
'Use ^C to stop when satisfied.'
));
$this
->
generate
(
$generators
,
$is_quickly
);
}
protected
function
generate
(
array
$generators
,
$is_quickly
)
{
$viewer
=
$this
->
getViewer
();
foreach
(
$generators
as
$generator
)
{
$generator
->
setViewer
(
$this
->
getViewer
());
}
while
(
true
)
{
$generator
=
$generators
[
array_rand
(
$generators
)];
try
{
$object
=
$generator
->
generateObject
();
}
catch
(
Exception
$ex
)
{
echo
tsprintf
(
"**<bg:yellow> %s </bg>** %s
\n
"
,
pht
(
'OOPS'
),
pht
(
'Generator ("%s") was unable to generate an object.'
,
$generator
->
getGeneratorName
()));
echo
tsprintf
(
"%B
\n
"
,
$ex
->
getMessage
());
continue
;
}
if
(
is_string
(
$object
))
{
$object_phid
=
$object
;
}
else
{
$object_phid
=
$object
->
getPHID
();
}
$handles
=
$viewer
->
loadHandles
(
array
(
$object_phid
));
echo
tsprintf
(
"%s
\n
"
,
pht
(
'Generated "%s": %s'
,
$handles
[
$object_phid
]->
getTypeName
(),
$handles
[
$object_phid
]->
getFullName
()));
if
(!
$is_quickly
)
{
sleep
(
1
);
}
}
}
}
Event Timeline
Log In to Comment