From b20252f3e1a70b22f2e80e3bcd2f944242a40234 Mon Sep 17 00:00:00 2001
From: Aravinth Manivannan <realaravinth@batsense.net>
Date: Sun, 1 Oct 2023 19:58:34 +0530
Subject: [PATCH] feat: import from ff/nodeinfo-test and ff/webfinger-test

---
 common/__init__.py  |  5 +++++
 common/env.py       | 23 +++++++++++++++++++++
 common/logger.py    | 50 +++++++++++++++++++++++++++++++++++++++++++++
 common/obj.py       | 18 ++++++++++++++++
 common/webfinger.py | 33 ++++++++++++++++++++++++++++++
 5 files changed, 129 insertions(+)
 create mode 100644 common/__init__.py
 create mode 100644 common/env.py
 create mode 100644 common/logger.py
 create mode 100644 common/obj.py
 create mode 100644 common/webfinger.py

diff --git a/common/__init__.py b/common/__init__.py
new file mode 100644
index 0000000..5cfbf09
--- /dev/null
+++ b/common/__init__.py
@@ -0,0 +1,5 @@
+# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+from .logger import logger
diff --git a/common/env.py b/common/env.py
new file mode 100644
index 0000000..8d29811
--- /dev/null
+++ b/common/env.py
@@ -0,0 +1,23 @@
+# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+import os
+
+from .logger import logger
+
+
+def get_env(name) -> str:
+    env = FTEST_AUTH = os.environ.get(name)
+    logger.info(f"Environment: {name}: {env}")
+    if env is None:
+        raise Exception(
+            f"Please set environment variable {name}. See https://git.batsense.net/ForgeFlux/nodeinfo-test#environment-variables"
+        )
+    return env
+
+
+FTEST_AUTH = get_env("FTEST_AUTH")
+FTEST_HOST = get_env("FTEST_HOST")
+FTEST_USER = get_env("FTEST_USER")
+TARGET_HOST = get_env("FTEST_TARGET_HOST")
diff --git a/common/logger.py b/common/logger.py
new file mode 100644
index 0000000..0150ad0
--- /dev/null
+++ b/common/logger.py
@@ -0,0 +1,50 @@
+# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+import logging
+
+LOG_FILE = "ap.log"
+
+
+def configure_logger():
+    logger = logging.getLogger("ap")
+    logger.setLevel(logging.DEBUG)
+    fh = logging.FileHandler(LOG_FILE)
+    fh.setLevel(logging.DEBUG)
+    ch = logging.StreamHandler()
+    ch.setLevel(logging.DEBUG)
+    formatter = logging.Formatter(
+        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    fh.setFormatter(formatter)
+    ch.setFormatter(formatter)
+    logger.addHandler(fh)
+    logger.addHandler(ch)
+    return logger
+
+
+logger = configure_logger()
+
+
+def upload_logs_to_ftest(success: bool, logs: str):
+    parsed_ftest_host = urlparse(FTEST_HOST)
+    ftest = urlunparse(
+        (
+            parsed_ftest_host.scheme,
+            parsed_ftest_host.netloc,
+            f"/api/v1/{FTEST_AUTH}/results",
+            "",
+            "",
+            "",
+        )
+    )
+    logger.info(f"Uploading logs to ftest server {ftest}")
+
+    payload = {"success": success, "logs": logs}
+
+    res = requests.post(ftest, json=payload, headers={"Origin": "http://example.org"})
+    if res.status_code == 200:
+        logger.info("Upload successful")
+    else:
+        print(res)
diff --git a/common/obj.py b/common/obj.py
new file mode 100644
index 0000000..b7f69e7
--- /dev/null
+++ b/common/obj.py
@@ -0,0 +1,18 @@
+# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+import os
+
+import requests
+
+from .logger import logger
+
+
+def get_ap_obj(url):
+    logger.info(f"Getting object: {url}")
+    res = requests.get(url, headers={"Accept": "application/activity+json"})
+    msg = f"AP server response:\n\nSTATUS: {res.status_code}\n\nHEADERS:\n {res.headers}\n\nRESPONSE PAYLOAD:\n{res.text}"
+    logger.debug(msg)
+    assert res.status_code == 200, f"Request FAILURE. {msg}"
+    return res.json()
diff --git a/common/webfinger.py b/common/webfinger.py
new file mode 100644
index 0000000..2fefe98
--- /dev/null
+++ b/common/webfinger.py
@@ -0,0 +1,33 @@
+# SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
+#
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+from urllib.parse import urlparse, urlunparse
+
+import requests
+
+from .logger import logger
+
+
+def get_webfinger():
+    from .env import TARGET_HOST, FTEST_USER
+
+    parsed_target_host = urlparse(TARGET_HOST)
+    webfinger = urlunparse(
+        (
+            parsed_target_host.scheme,
+            parsed_target_host.netloc,
+            "/.well-known/webfinger",
+            "",
+            f"resource=acct:{FTEST_USER}",
+            "",
+        )
+    )
+    logger.info(f"Query webfinger: {webfinger}")
+    res = requests.get(webfinger)
+    logger.debug(
+        f"webfinger response:\n\nSTATUS: {res.status_code}\n\nHEADERS:\n {res.headers}\n\nRESPONSE PAYLOAD:\n{res.json()}"
+    )
+    assert res.status_code == 200
+    logger.info("[SUCCESS] webfinger query response is HTTP 200")
+    return res.json()