From 43c8ad0ff66e2d24ed8f21ae758fe8a807e9db26 Mon Sep 17 00:00:00 2001
From: jesopo <github@lolnerd.net>
Date: Thu, 4 Jul 2019 13:15:51 +0100
Subject: [PATCH] Add !editserver command

---
 modules/admin.py | 38 ++++++++++++++++++++++++++++++++++++++
 src/Database.py  |  6 ++++++
 2 files changed, 44 insertions(+)

diff --git a/modules/admin.py b/modules/admin.py
index 3128afb8..b9ba25e1 100644
--- a/modules/admin.py
+++ b/modules/admin.py
@@ -135,3 +135,41 @@ class Module(ModuleManager.BaseModule):
                 exc_info=True)
             return
         event["stdout"].write("Added server '%s'" % alias)
+
+    @utils.hook("received.command.editserver")
+    @utils.kwarg("min_args", 3)
+    @utils.kwarg("help", "Edit server details")
+    @utils.kwarg("usage", "<alias> <option> <value>")
+    @utils.kwarg("permission", "editserver")
+    def edit_server(self, event):
+        alias = event["args_split"][0]
+        server = self.bot.get_server_by_alias(alias)
+        if server == None:
+            raise utils.EventError("Unknown server '%s'" % alias)
+
+        option = event["args_split"][1].lower()
+        value = " ".join(event["args_split"][2:])
+        value_parsed = None
+
+        if option == "hostname":
+            value_parsed = value
+        elif option == "port":
+            if not value.isdigit():
+                raise utils.EventError("Invalid port")
+            value_parsed = int(value.lstrip("0"))
+        elif option == "tls":
+            value_lower = value.lower()
+            if not value_lower in ["yes", "no"]:
+                raise utils.EventError("TLS should be either 'yes' or 'no'")
+            value_parsed = value_lower == "yes"
+        elif option == "password":
+            value_parsed = value
+        elif option == "bindhost":
+            value_parsed = value
+        elif option in ["nickname", "username"]:
+            value_parsed = value
+        else:
+            raise utils.EventError("Unknown option '%s'" % option)
+
+        self.bot.database.servers.edit(server.id, option, value_parsed)
+        event["stdout"].write("Set %s for %s" % (option, alias))
diff --git a/src/Database.py b/src/Database.py
index 6a5a4d58..995fc643 100644
--- a/src/Database.py
+++ b/src/Database.py
@@ -39,6 +39,12 @@ class Servers(Table):
         if value:
             return value[0]
         return value
+    def edit(self, id: int, column: str, value: typing.Any):
+        if not column in ["alias", "hostname", "port", "password", "tls",
+                "bindhost", "nickname", "username", "realname"]:
+            raise ValueError("Unknown column on servers table '%s'" % column)
+        self.database.execute(
+            "UPDATE servers SET %s=? WHERE server_id=?" % column, [value, id])
 
 class Channels(Table):
     def add(self, server_id: int, name: str):