1 files changed, 47 insertions, 0 deletions
diff --git a/src/add.py b/src/add.py
new file mode 100644
index 0000000..00485ce
--- /dev/null
+++ b/src/add.py
@@ -0,0 +1,47 @@
+from common import *
+
+def get_name(url: str) -> str:
+    name = url.split("/")[-1]
+
+    if str(name).endswith(".git"):
+        name = name[0:-4]
+    return name
+                
+
+def get_repo(inp: str) -> repo_pair:
+    if str(inp).startswith("http"):
+        obj: repo_pair = {"url" : inp, "name":""}
+        name = get_name(inp)
+
+        obj["name"] = name
+
+        return obj
+    elif '/' in str(inp):
+        return get_repo(f"https://github.com/{inp}")
+    else:
+        data = requests.get("https://qa-reports.gentoo.org/output/repos/repositories.xml")
+        xml = xmltodict.parse(data.text)
+        close = [x for x in xml["repositories"]["repo"] if x["name"] == inp or SequenceMatcher(None, inp, x["name"]).ratio() > 0.6]
+        found = [x for x in close if x["name"] == inp]
+        
+        if len(found) > 0:
+            if len(found) == 1:
+                valid_sources = [x for x in found[0]["source"] if x["@type"] == "git"]
+
+                return get_repo(valid_sources[0]["#text"])
+            else:
+                print("multiple results found (TODO)")
+                exit(1)
+        else:
+            if len(close) == 0:
+                print("no exact or close matches")
+                exit(1)
+
+            names = [x["name"] for x in close]
+            choice = questionary.select("no direct matches found, similar results", names).ask()
+            use = [x for x in close if x["name"] == choice]
+            valid_sources = [x for x in use[0]["source"] if x["@type"] == "git"]
+
+            return get_repo(valid_sources[0]["#text"])
+
+