added dependency resolving

1 files changed, 87 insertions, 28 deletions

diff --git a/src/verbs/install.py b/src/verbs/install.py
index 11a2183..0847b1d 100644
--- a/src/verbs/install.py
+++ b/src/verbs/install.py
@@ -4,11 +4,8 @@ import util
 import colors
 import time
 
-def find_package(query, repos, packages_dir):
-    sources = []
-    checksum = None
-    requested_repo = None
 
+def find_package(query, repos, packages_dir, sources):
     for repo in repos:
         repo_dir = os.path.join(packages_dir, repo)
         files = os.listdir(repo_dir)
@@ -17,13 +14,21 @@ def find_package(query, repos, packages_dir):
             requested_repo = repo
             with open(os.path.join(repo_dir, query)) as file:
                 checksum = file.readline().strip().split("=")[-1]
-                sources = file.readline().strip().split("=")[-1].split()
-                return checksum, sources, requested_repo
+                listed_sources = file.readline().strip().split("=")[-1].split()
+                found_sources = {
+                            source: util.add_path(url, repo) 
+                            for source, url in sources.items() 
+                            if source in listed_sources
+                        }
+                return checksum, found_sources, requested_repo
 
     return None, [], None
 
 def retrieve_package_info(sources, checksum, package_name, 
                             verbose=False, skip_verification=False):
+    
+    # TODO we may potentially do this a few times while resolving deps, might want to cache things here
+    # TODO actually use the ping times we made earlier to decide which source to pick
     for source,url in sources.items():
         package_info_url = util.add_path(url, package_name + ".xipkg.info")
         status, response = util.curl(package_info_url)
@@ -43,14 +48,18 @@ def retrieve_package_info(sources, checksum, package_name,
 
 def retrieve_package(sources, checksum, package_name, 
                             verbose=False, skip_verification=False):
+
+    # TODO actually use the ping times we made earlier to decide which source to pick
+    # TODO actually save tar file, and add loading bar
     for source,url in sources.items():
-        package_info_url = util.add_path(url, package_name + ".xipkg.info")
+        package_info_url = util.add_path(url, package_name + ".xipkg")
         status, response = util.curl(package_info_url)
    
         if status == 200:
-            info = parse_package_info(response)
-            if info["CHECKSUM"] == checksum or skip_verification:
-                return info
+            downloaded_checksum = util.md5sum(response)
+            print(downloaded_checksum, "compared to requested", checksum)
+            if downloaded_checksum == checksum or skip_verification:
+                return reponse
             else:
                 if verbose:
                     print(colors.RED 
@@ -67,11 +76,63 @@ def parse_package_info(packageinfo):
         split = line.split("=")
         if len(split) > 1:
             info[split[0]] = "=".join(split[1:])
+
     return info
 
-def resolve_dependencies(package_info):
-    getpkgs = lambda deps: re.findall("[\(\s](\w)[\)\s]")
-    package_info[""]
+def resolve_dependencies(package_info, config):
+    getpkgs = lambda deps: re.findall("\w*", deps)
+    deps = getpkgs(package_info["DEPS"])
+
+    deps = [
+                dep for dep in deps if len(dep) > 0
+            ]
+
+    return deps
+
+def find_all_dependencies(package_names, options, config):
+    # this is all assuming that the order of deps installed doesn't matter
+    to_check = [p for p in package_names]
+    all_deps = []
+
+    while len(to_check) > 0:
+        util.loading_bar(len(all_deps), len(all_deps) + len(to_check), "Resolving dependencies...")
+        dep = to_check.pop()
+
+        dep_checksum, dep_sources, dep_repo = find_package(dep, config["repos"], config["dir"]["packages"], config["sources"])
+        if dep_checksum is not None:
+            info = retrieve_package_info(
+                        dep_sources, dep_checksum, dep,
+                        verbose=options["v"], skip_verification=options["u"]
+                    )
+
+            if len(info) > 0:
+                all_deps.append(dep)
+                deps = resolve_dependencies(info, config)
+                for dep in deps:
+                    if not dep in all_deps:
+
+                        if is_installed(dep, config):
+                            print(colors.YELLOW + f"Package {query} has already been installed")
+                        else:
+                            to_check.append(dep)
+            else:
+                if options["v"]:
+                    util.print_reset(colors.CLEAR_LINE + colors.RED + f"Failed to retrieve info for {query}")
+        else:
+            util.print_reset(colors.CLEAR_LINE + colors.RED + f"Failed to find package {dep}")
+
+    if len(all_deps) > 0:
+        util.loading_bar(len(all_deps), len(all_deps) + len(to_check), "Resolved dependencies")
+        print(colors.RESET)
+
+    # assuming that the latter packages are core dependencies
+    # we can reverse the array to reflect the more important packages to install
+    all_deps.reverse()
+    return all_deps
+
+def is_installed(package_name, config):
+    # TODO actually find out if its installed
+    return False
 
 def install(args, options, config):
     sources = config["sources"]
@@ -81,23 +142,21 @@ def install(args, options, config):
     unsafe = options["u"]
 
     packages_dir = config["dir"]["packages"]
-    for query in args:
 
-        # TODO FIRST CHECK IF ALREADY INSTALLED
-        checksum, listed_sources, repo = find_package(query, repos, packages_dir)
+    to_install = args if options["n"] else find_all_dependencies(args, options, config)
 
-        if checksum is not None:
-            info = retrieve_package_info(
-                        {
-                        source: util.add_path(url, repo) 
-                        for source, url in sources.items() 
-                        if source in listed_sources
-                    }, checksum, query,
-                        verbose=v, skip_verification=unsafe
-                    )
+    if len(to_install) > 0:
+        print(colors.BLUE + "The following packages will be installed:")
+        print(end="\t")
+        for d in to_install:
+            print(colors.BLUE if d in args else colors.LIGHT_BLUE, d, end="")
+        print()
 
-            print(info)
+        if util.ask_confirmation(colors.BLUE + "Continue?", no_confirm=options["y"]):
+            print("installed")
         else:
-            print(colors.RED + "Package not found")
-        print(colors.RESET, end="")
+            print(colors.RED + "Action cancelled by user")
+    else:
+        print(colors.LIGHT_RED + "Nothing to do")
+