From 0abe44cb05d1e6ef10f178ae93e1d5ad8fd7d91e Mon Sep 17 00:00:00 2001
From: Dustin Pianalto <dustin@djpianalto.com>
Date: Mon, 30 Dec 2019 22:40:31 -0900
Subject: [PATCH] Add GeeksbotAPI class to manage queries to the API

---
 geeksbot/imports/geeksbot.py     |  3 +-
 geeksbot/imports/geeksbot_api.py | 78 ++++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+), 1 deletion(-)
 create mode 100644 geeksbot/imports/geeksbot_api.py

diff --git a/geeksbot/imports/geeksbot.py b/geeksbot/imports/geeksbot.py
index cb26e26..7d487ee 100644
--- a/geeksbot/imports/geeksbot.py
+++ b/geeksbot/imports/geeksbot.py
@@ -12,6 +12,7 @@ import discord
 from discord.ext import commands
 from discord.ext.commands.context import Context
 from geeksbot.imports.strings import MyStringView
+from geeksbot.imports.geeksbot_api import GeeksbotAPI
 
 geeksbot_logger = logging.getLogger("Geeksbot")
 
@@ -45,8 +46,8 @@ class Geeksbot(commands.Bot):
         self.token = self.settings_cache.get("DISCORD_TOKEN")
         self.api_token = self.settings_cache.get("API_TOKEN")
         self.aio_session = aiohttp.ClientSession(loop=self.loop)
-        self.auth_header = {"Authorization": f"Token {self.api_token}"}
         self.api_base = "https://geeksbot.app/api"
+        self.api = GeeksbotAPI(self.api_token, self.api_base, self.loop)
         with open(f"{self.config_dir}/{self.config_file}") as f:
             self.bot_config = json.load(f)
         self.embed_color = discord.Colour.from_rgb(49, 107, 111)
diff --git a/geeksbot/imports/geeksbot_api.py b/geeksbot/imports/geeksbot_api.py
new file mode 100644
index 0000000..3eb7ca3
--- /dev/null
+++ b/geeksbot/imports/geeksbot_api.py
@@ -0,0 +1,78 @@
+import aiohttp
+import asyncio
+import logging
+
+logger = logging.getLogger("Geeksbot API")
+
+
+class APIError500(Exception):
+    pass
+
+
+class APIError(Exception):
+    pass
+
+
+class GeeksbotAPI:
+    def __init__(
+        self,
+        token: str,
+        base_url: str = "https://geeksbot.app/api",
+        loop: asyncio.AbstractEventLoop = None,
+    ):
+        self.loop = loop or asyncio.get_event_loop()
+        self.session = aiohttp.ClientSession(loop=self.loop)
+        self.base_url = base_url
+        self.token = token
+        self.auth_header = {"Authorization": f"Token {self.token}"}
+
+    def clean_endpoint(endpoint: str) -> str:
+        endpoint = endpoint[1:] if endpoint.startswith("/") else endpoint
+        endpoint += "/" if not endpoint.endswith("/") else ""
+        return endpoint
+
+    async def query(
+        self, method: str, endpoint: str, data: dict = None, query: dict = None
+    ):
+        endpoint = self.clean_endpoint(endpoint)
+        if method.lower() == "get":
+            resp = await self.session.get(
+                f"{self.base_url}/{endpoint}{query_str}",
+                headers=self.auth_header,
+                params=query,
+                json=data,
+            )
+        elif method.lower() == "post":
+            resp = await self.session.post(
+                f"{self.base_url}/{endpoint}{query_str}",
+                headers=self.auth_header,
+                params=query,
+                json=data,
+            )
+        elif method.lower() == "put":
+            resp = await self.session.put(
+                f"{self.base_url}/{endpoint}{query_str}",
+                headers=self.auth_header,
+                params=query,
+                json=data,
+            )
+        elif method.lower() == "delete":
+            resp = await self.session.delete(
+                f"{self.base_url}/{endpoint}{query_str}",
+                headers=self.auth_header,
+                params=query,
+                json=data,
+            )
+        else:
+            raise APIError(f"That is not a valid method. {method}")
+
+        if resp.status == 200:
+            return await resp.json()
+        elif resp.status >= 500:
+            raise APIError500(
+                "The server returned a 500 error. "
+                "The developers have been notified of the issue."
+            )
+        else:
+            details = (await resp.json())["details"]
+            raise APIError(details)