init

1 files changed, 142 insertions, 0 deletions

diff --git a/src/__main__.py b/src/__main__.py
new file mode 100755
index 0000000..d38d550
--- /dev/null
+++ b/src/__main__.py
@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+import copy
+from common import *
+from sync import *
+from add import *
+from rm import *
+
+if len(sys.argv) == 1:
+    print("gentoo portage overlay steal (gpo-steal), written by amelia (https://github.com/ameliasquires)")
+    print("\tlicensed under 3-clause BSD, see license for more info")
+    print("")
+    print("utility for copying packages from a overlay to yours")
+    print("")
+    print("usage:")
+    print(" sync - selects repository to update from source, also updates tracked packages")
+    print(" add [repository] - adds a overlay, accepts git links, github account/name (ex: gentoo/gentoo),")
+    print("\tand between everything listed on https://overlays.gentoo.org/")
+    print(" delete - remove a repository")
+    print(" copy [package] - searches all added repositories for package, accepts package-name,")
+    print("\tpackage-location/package-name, do not use : to specify repository")
+    print(" rm [package] - removes or untracks a package, accepts the same params as copy, but allows : to specify repository")
+    print(" ls - lists all tracked packages (not untracked, yet installed ones)")
+    print("")
+    exit(0)
+
+match sys.argv[1]:
+    case "sync" | "s":
+        options = list(repos.keys())
+        name = questionary.select("which repository", options).ask()
+        sync_repos = [repos[name]]
+
+        sync(sync_repos)
+
+    case "add" | "a":
+        assert len(sys.argv) >= 3
+
+        obj = get_repo(sys.argv[2])
+ 
+        repos[obj["name"]] = obj
+    
+        save()
+
+        if questionary.confirm("sync repository?").ask():
+            sync([obj])
+
+    case "delete" | "del" | "d":
+        options = list(repos.keys())
+        name = questionary.select("which repository", options).ask()
+        del_repos = [repos[name]]
+
+        oper = questionary.select("delete", ["repository", "entry & repository"]).ask()
+
+        if not questionary.confirm("are you sure?").ask():
+            print("exiting")
+            exit(0)
+        
+        for r in del_repos:
+            full_path = os.path.join(REPO_LOC, r["name"])
+            print("del "+ full_path)
+            
+            if os.path.isdir(full_path):
+                shutil.rmtree(full_path)
+            if oper == "entry & repository":
+                del repos[name]
+
+        save()
+
+    case "copy" | "c":
+        assert len(sys.argv) >= 3
+
+        search = sys.argv[2]
+        full = '/' in search
+
+        options = []
+
+        for r in repos:
+            path = os.path.join(REPO_LOC, r)
+
+            if full:
+                full_path = os.path.join(path, search)
+                if os.path.isdir(full_path):
+                    options.append({"path": search, "repo": r, "full_name": f"{search}:{r}"})
+                    print(options)
+
+            else:
+                if not os.path.isdir(path):
+                    continue
+                repo = os.listdir(path) 
+                for dir in repo:
+                    dir_path = os.path.join(path, dir)
+
+                    if os.path.isdir(dir_path):
+                        for f in os.listdir(dir_path):
+                            if f == search:
+
+                                full_path = os.path.join(dir_path, f)
+                                name = os.path.join(dir, search)
+                                options.append({"path": name, "full_path": full_path, "repo": r, "full_name": f"{name}:{r}" })
+                   
+        use = None
+        if len(options) > 1:
+            choices = [i["full_name"] for i in options]
+            choice = questionary.select("select an option", choices).ask()
+            use = [x for x in options if x["full_name"] == choice][0]
+        elif len(options) == 1:
+            use = options[0]
+        else:
+            print("no matches")
+            exit(0)
+        
+        copy.copy([use])
+
+    case "r" | "rm":
+        assert len(sys.argv) >= 3
+
+        search = sys.argv[2]
+        has_repo = ":" in search
+
+        if has_repo:
+            [search, repo] = search.split(":")
+
+            if not repo in tracked:
+                print("repository has no tracked packages")
+                exit(1)
+            
+            rm(search, [repo])
+        else:
+            rm(search, tracked.keys())
+
+    case "ls" | "l" | "list":
+
+        if len(sys.argv) >= 3:
+            if not sys.argv[2] in tracked:
+                print("nothing in repo")
+                exit(1)
+
+            for i in tracked[sys.argv[2]]:
+                print(i)
+        else:
+            for r in tracked.keys():
+                for p in tracked[r]:
+                    print(f"{p}:{r}")