Start working on management API implementation

maubot/management/api/plugin.py

@@ -0,0 +1,83 @@
+# maubot - A plugin-based Matrix bot system.
+# Copyright (C) 2018 Tulir Asokan
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+from aiohttp import web
+from io import BytesIO
+import os.path
+
+from ...loader import PluginLoader, ZippedPluginLoader, MaubotZipImportError
+from .responses import ErrPluginNotFound, ErrPluginInUse, RespDeleted
+from . import routes, config
+
+
+def _plugin_to_dict(plugin: PluginLoader) -> dict:
+    return {
+        **plugin.to_dict(),
+        "instances": [instance.to_dict() for instance in plugin.references]
+    }
+
+
+@routes.get("/plugins")
+async def get_plugins(_) -> web.Response:
+    return web.json_response([_plugin_to_dict(plugin) for plugin in PluginLoader.id_cache.values()])
+
+
+@routes.get("/plugin/{id}")
+async def get_plugin(request: web.Request) -> web.Response:
+    plugin_id = request.match_info.get("id", None)
+    plugin = PluginLoader.id_cache.get(plugin_id, None)
+    if not plugin:
+        return ErrPluginNotFound
+    return web.json_response(_plugin_to_dict(plugin))
+
+
+@routes.delete("/plugin/{id}")
+async def delete_plugin(request: web.Request) -> web.Response:
+    plugin_id = request.match_info.get("id", None)
+    plugin = PluginLoader.id_cache.get(plugin_id, None)
+    if not plugin:
+        return ErrPluginNotFound
+    elif len(plugin.references) > 0:
+        return ErrPluginInUse
+    plugin.delete()
+    return RespDeleted
+
+
+@routes.post("/plugins/upload")
+async def upload_plugin(request: web.Request) -> web.Response:
+    content = await request.read()
+    file = BytesIO(content)
+    try:
+        pid, version = ZippedPluginLoader.verify_meta(file)
+    except MaubotZipImportError as e:
+        return web.json_response({
+            "error": str(e),
+            "errcode": "invalid_plugin",
+        }, status=web.HTTPBadRequest)
+    plugin = PluginLoader.id_cache.get(pid, None)
+    if not plugin:
+        path = os.path.join(config["plugin_directories.upload"], f"{pid}-{version}.mbp")
+        with open(path, "wb") as p:
+            p.write(content)
+        try:
+            ZippedPluginLoader.get(path)
+        except MaubotZipImportError as e:
+            trash(path)
+            return web.json_response({
+                "error": str(e),
+                "errcode": "invalid_plugin",
+            }, status=web.HTTPBadRequest)
+    else:
+        pass