@recaptime-dev's working patches + fork for Phorge, a community fork of Phabricator. (Upstream dev and stable branches are at upstream/main and upstream/stable respectively.)
hq.recaptime.dev/wiki/Phorge
phorge
phabricator
1<?php
2
3final class HeraldWebhookCallManagementWorkflow
4 extends HeraldWebhookManagementWorkflow {
5
6 protected function didConstruct() {
7 $this
8 ->setName('call')
9 ->setExamples('**call** --id __id__ [--object __object__]')
10 ->setSynopsis(pht('Call a webhook.'))
11 ->setArguments(
12 array(
13 array(
14 'name' => 'id',
15 'param' => 'id',
16 'help' => pht('Webhook ID to call'),
17 ),
18 array(
19 'name' => 'object',
20 'param' => 'object',
21 'help' => pht('Submit transactions for a particular object.'),
22 ),
23 array(
24 'name' => 'silent',
25 'help' => pht('Set the "silent" flag on the request.'),
26 ),
27 array(
28 'name' => 'secure',
29 'help' => pht('Set the "secure" flag on the request.'),
30 ),
31 array(
32 'name' => 'count',
33 'param' => 'N',
34 'help' => pht('Make a total of __N__ copies of the call.'),
35 ),
36 array(
37 'name' => 'background',
38 'help' => pht(
39 'Instead of making calls in the foreground, add the tasks '.
40 'to the daemon queue.'),
41 ),
42 ));
43 }
44
45 public function execute(PhutilArgumentParser $args) {
46 $viewer = $this->getViewer();
47
48 $id = $args->getArg('id');
49 if (!$id) {
50 throw new PhutilArgumentUsageException(
51 pht(
52 'Specify a webhook to call with "--id".'));
53 }
54
55 $count = $args->getArg('count');
56 if ($count === null) {
57 $count = 1;
58 }
59
60 if ($count <= 0) {
61 throw new PhutilArgumentUsageException(
62 pht(
63 'Specified "--count" must be larger than 0.'));
64 }
65
66 $hook = id(new HeraldWebhookQuery())
67 ->setViewer($viewer)
68 ->withIDs(array($id))
69 ->executeOne();
70 if (!$hook) {
71 throw new PhutilArgumentUsageException(
72 pht(
73 'Unable to load specified webhook ("%s").',
74 $id));
75 }
76
77 $object_name = $args->getArg('object');
78 if ($object_name === null) {
79 $object = $hook;
80 } else {
81 $objects = id(new PhabricatorObjectQuery())
82 ->setViewer($viewer)
83 ->withNames(array($object_name))
84 ->execute();
85 if (!$objects) {
86 throw new PhutilArgumentUsageException(
87 pht(
88 'Unable to load specified object ("%s").',
89 $object_name));
90 }
91 $object = head($objects);
92 }
93
94 $is_background = $args->getArg('background');
95
96 $xaction_query =
97 PhabricatorApplicationTransactionQuery::newQueryForObject($object);
98
99 $xactions = $xaction_query
100 ->withObjectPHIDs(array($object->getPHID()))
101 ->setViewer($viewer)
102 ->setLimit(10)
103 ->execute();
104
105 $application_phid = id(new PhabricatorHeraldApplication())->getPHID();
106
107 if ($is_background) {
108 echo tsprintf(
109 "%s\n",
110 pht(
111 'Queueing webhook calls...'));
112 $progress_bar = id(new PhutilConsoleProgressBar())
113 ->setTotal($count);
114 } else {
115 echo tsprintf(
116 "%s\n",
117 pht(
118 'Calling webhook...'));
119 PhabricatorWorker::setRunAllTasksInProcess(true);
120 }
121
122 for ($ii = 0; $ii < $count; $ii++) {
123 $request = HeraldWebhookRequest::initializeNewWebhookRequest($hook)
124 ->setObjectPHID($object->getPHID())
125 ->setIsTestAction(true)
126 ->setIsSilentAction((bool)$args->getArg('silent'))
127 ->setIsSecureAction((bool)$args->getArg('secure'))
128 ->setTriggerPHIDs(array($application_phid))
129 ->setTransactionPHIDs(mpull($xactions, 'getPHID'))
130 ->save();
131
132 $request->queueCall();
133
134 if ($is_background) {
135 $progress_bar->update(1);
136 } else {
137 $request->reload();
138
139 echo tsprintf(
140 "%s\n",
141 pht(
142 'Success, got HTTP %s from webhook.',
143 $request->getErrorCode()));
144 }
145 }
146
147 if ($is_background) {
148 $progress_bar->done();
149 }
150
151 return 0;
152 }
153
154}