DennyLoko · February 5, 2016 17:38 · Feb 5, 2016
diff --git a/akamai.php b/akamai.php
@@ -0,0 +1,179 @@
+#!/usr/bin/php
+<?php
+date_default_timezone_set('America/Sao_Paulo');
+
+/**
+ * Script responsible to handle communication with Akamai
+ *
+ * @author  Danniel Magno
+ * @version 1.0
+ */
+class AkamaiTool {
+    private $user = '';
+    private $pass = '';
+    private $baseUrl = 'https://api.ccu.akamai.com/ccu/v2/';
+
+    private $verbose = false;
+
+    private $curl;
+
+    public function __construct()
+    {
+        // Instantiate the request
+        $this->createRequest();
+
+        // Get the params
+        $args = getopt('hqs:w:p:v', ['help', 'viewqueue', 'jobstatus', 'watch', 'purge']);
+
+        if (isset($args['v'])) {
+            $this->verbose = true;
+            unset($args['v']);
+        }
+
+        // Check the values
+        if (count($args) === 0) {
+            echo 'Invalid option!' . PHP_EOL;
+            $this->showHelp();
+        } elseif (isset($args['h']) || isset($args['help'])) {
+            $this->showHelp();
+        } elseif (isset($args['q']) || isset($args['viewqueue'])) {
+            $this->viewQueue();
+        } elseif (isset($args['s']) || isset($args['jobstatus'])) {
+            $jobId = (isset($args['s']) ? $args['s'] : $args['jobstatus']);
+            $this->viewJobStatus($jobId);
+        } elseif (isset($args['w']) || isset($args['watch'])) {
+            $jobId = (isset($args['w']) ? $args['w'] : $args['watch']);
+            $this->watchjob($jobId);
+        } elseif (isset($args['p']) || isset($args['purge'])) {
+            $object = (isset($args['p']) ? $args['p'] : $args['purge']);
+            $this->purgeObject($object);
+        }
+    }
+
+    public function __destruct()
+    {
+        curl_close($this->curl);
+    }
+
+    private function createRequest()
+    {
+        // Init cURL
+        $this->curl = curl_init();
+
+        // Set the defaults
+        curl_setopt_array($this->curl, [
+            CURLOPT_RETURNTRANSFER      => true,
+            CURLOPT_HTTPHEADER          => array("Content-Type: application/json"),
+            CURLOPT_USERPWD             => $this->user . ':' . $this->pass,
+            CURLOPT_CONNECTTIMEOUT_MS   => 5000,
+        ]);
+    }
+
+    private function makeRequest($url, $postFields = [])
+    {
+        // Set the url
+        curl_setopt($this->curl, CURLOPT_URL, $this->baseUrl . $url);
+
+        // We should send an POST?
+        if (!empty($postFields)) {
+            curl_setopt($this->curl, CURLOPT_POST, 1);
+            curl_setopt($this->curl, CURLOPT_POSTFIELDS, json_encode($postFields));
+        }
+
+        // Do the request
+        $result = curl_exec($this->curl);
+
+        // Returns the result
+        return $result;
+    }
+
+    private function showHelp()
+    {
+        echo 'Usage: akamai.php -q' . PHP_EOL;
+        echo '       akamai.php -s <jobId>' . PHP_EOL;
+        echo '       akamai.php -w <jobId>' . PHP_EOL;
+        echo '       akamai.php -p <object>' . PHP_EOL;
+        echo PHP_EOL;
+
+        echo '  -q, --viewqueue   Show the queue length' . PHP_EOL;
+        echo '  -s, --jobstatus   Show the status of a job' . PHP_EOL;
+        echo '  -w, --watch       Watch a job, polling Akamai every 30s' . PHP_EOL;
+        echo '  -p, --purge       Purge a single object' . PHP_EOL;
+    }
+
+    private function viewQueue()
+    {
+        $queue = json_decode($this->makeRequest('queues/default'));
+        echo 'Length: ' . $queue->queueLength . PHP_EOL;
+    }
+
+    private function getJobStatus($jobId)
+    {
+        return $this->makeRequest('purges/' . $jobId);
+    }
+
+    private function viewJobStatus($jobId)
+    {
+        echo $this->getJobStatus($jobId);
+    }
+
+    private function purgeObject($object)
+    {
+        // Check if the object has http/https prefix
+        if (preg_match('/^http(s)?\:\/\//', $object) == 0) {
+            echo 'The object must start with http:// or https://' . PHP_EOL;
+        } else {
+            $job = json_decode($this->makeRequest('queues/default', [
+                'type'    => 'arl',
+                'objects' => [$object],
+            ]));
+
+            if ($job->httpStatus == 201) {
+                echo 'Purge queued!' . PHP_EOL;
+                echo 'Use "akamai.php -w ' . $job->purgeId . '" to watch the job.' . PHP_EOL;
+            } else {
+                echo 'Error while queuing the job!' . PHP_EOL;
+                echo $job->title . '(' . $job->detail . ')' . PHP_EOL;
+            }
+        }
+    }
+
+    private function watchJob($jobId)
+    {
+        $repeat = false;
+
+        do {
+            // Get the job data
+            $job = $this->getJobStatus($jobId);
+            $jobObj = json_decode($job);
+            $jobStatus = strtolower($jobObj->purgeStatus);
+
+            if ($jobStatus === 'in-progress') {
+                echo 'Purge in progress...' . PHP_EOL;
+
+                $repeat = true;
+
+                sleep($jobObj->pingAfterSeconds / 2);
+            } elseif ($jobStatus === 'done') {
+                $submission = new DateTime($jobObj->submissionTime);
+                $completion = new DateTime($jobObj->completionTime);
+                $diff = $completion->diff($submission);
+
+                echo 'Job done!' . PHP_EOL;
+                echo 'Submission: ' . $submission->format('H:i:s d/m/Y') . ' | ';
+                echo 'Completion: ' . $completion->format('H:i:s d/m/Y') . ' | ';
+                echo 'Duration: ' . (($diff->format('%d') * 1440) + ($diff->format('%h') * 60) + $diff->format('%i')) . 'min and ' . $diff->format('%s') . 'secs' . PHP_EOL;
+
+                $repeat = false;
+            } elseif ($jobStatus === 'unknown') {
+                echo 'Job is in unknown status.' . PHP_EOL;
+                $repeat = false;
+            } else {
+                echo 'Unknown job status' . PHP_EOL;
+                echo $job;
+            }
+        } while ($repeat === true);
+    }
+}
+
+$akamai = new AkamaiTool($argv);