From b7b2ab744a59d0f0e2407d4a0cd2fe2485525446 Mon Sep 17 00:00:00 2001
From: Geoffrey Frogeye <geoffrey@frogeye.fr>
Date: Sat, 10 Dec 2016 21:44:21 +0100
Subject: [PATCH] Gogs SSH keys support

---
 index.php | 63 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 62 insertions(+), 1 deletion(-)

diff --git a/index.php b/index.php
index 870258b..b472244 100644
--- a/index.php
+++ b/index.php
@@ -311,6 +311,63 @@ function argAssert($arg, $data, $args) {
     return false;
 }
 
+// Hooks
+//
+function gogsRequest($route, $meth = 'GET', $data = null) {
+    global $GOGS_API;
+    global $GOGS_TOKEN;
+    $ch = curl_init();
+    curl_setopt($ch, CURLOPT_URL, $GOGS_API.'/'.$route);
+    curl_setopt($ch, CURLOPT_HTTPHEADER, array('Authorization: token '.$GOGS_TOKEN));
+    curl_setopt($ch, CURLOPT_CUSTOMREQUEST, $meth);
+    if ($data) {
+        $textdata = '';
+            foreach ($data as $key => $value) {
+            $textdata .= '&'.$key.'='.urlencode($value);
+        }
+        curl_setopt($ch, CURLOPT_POST, count($data));
+        curl_setopt($ch, CURLOPT_POSTFIELDS, ltrim($textdata, '&'));
+    }
+    curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
+    $raw = curl_exec($ch);
+    curl_close($ch);
+    return json_decode($raw);
+}
+
+function updateGogsKeys($keys) {
+    global $GOGS_API;
+    global $GOGS_TOKEN;
+    global $SSH_KEY_REGEX;
+    if (isset($GOGS_API) && isset($GOGS_TOKEN)) {
+        $existing = gogsRequest('user/keys');
+        $toDelete = [];
+        foreach ($existing as $ekey) {
+            $toDelete[$ekey->id] = $ekey->key;
+        }
+
+        foreach (explode("\n", $keys) as $key) {
+            $found = false;
+            foreach ($toDelete as $id => $ekey) {
+                if ($key == $ekey) {
+                    unset($toDelete[$id]);
+                    $found = true;
+                    break;
+                }
+            }
+            if (!$found) {
+                gogsRequest('user/keys', 'POST', array(
+                    "title" => ltrim(preg_replace($SSH_KEY_REGEX, '', $key)),
+                    "key" => $key
+                ));
+            }
+        }
+
+        foreach ($toDelete as $id => $ekey) {
+            gogsRequest('user/keys/'.$id, 'DELETE');
+        }
+    }
+}
+
 switch ($route[0]) {
 case 'machine':
 case 'network':
@@ -513,8 +570,12 @@ case 'akey':
             file_put_contents('akey/'.$networkName.'.authorized_keys', getKeys($networkName ? $network : null));
             file_put_contents('akey/'.$networkName.'.authorized_keys.sha256', $sign);
 
+            if ($networkName == 'gogs') {
+                updateGogsKeys(getKeys($network));
+            }
+
             http_response_code(201);
-            logActivity('Updated key '.$networkName);
+            logActivity('Updated akeys '.$networkName);
         } else {
             http_response_code(404);
             die("Unknown network\n");