js: add monorepo support

javascript.py

@@ -1,5 +1,6 @@
 "javascript specific code for stage 3."
 
+import glob
 import json
 import os
 import sys
@@ -13,32 +14,61 @@ def load_package_json(package_json_path):
 
 def local_dependencies(dirs):
     """Return a dictionary of local dependencies with the name as key
-    and the directory as value."""
+    and the directory as value. It also looks for workspaces in package.json."""
     deps = {}
     for local_dir in dirs:
-        local_path = dirs[local_dir]["path"]
+        if "subdir" in dirs[local_dir]:
+            local_path = os.path.join(
+                dirs[local_dir]["path"], dirs[local_dir]["subdir"]
+            )
+        else:
+            local_path = dirs[local_dir]["path"]
         package_json_path = os.path.join(local_path, "package.json")
         if os.path.exists(package_json_path):
-            deps[load_package_json(package_json_path)["name"]] = dirs[local_dir]
+            package = load_package_json(package_json_path)
+            deps[package["name"]] = dirs[local_dir]
+            if "workspaces" in package and "packages" in package["workspaces"]:
+                print("detected workspaces", file=sys.stderr)
+                for workspace_glob in package["workspaces"]["packages"]:
+                    print(f"processing {workspace_glob}", file=sys.stderr)
+                    for workspace_dir in glob.glob(
+                        os.path.join(local_path, workspace_glob)
+                    ):
+                        print(f"found subdir={workspace_dir}", file=sys.stderr)
+                        workspace_path = os.path.join(local_path, workspace_dir)
+                        print(
+                            f"checking workspace {workspace_path} for package.json",
+                            file=sys.stderr,
+                        )
+                        workspace_package_json_path = os.path.join(
+                            workspace_path, "package.json"
+                        )
+                        if os.path.exists(workspace_package_json_path):
+                            workspace_package = load_package_json(
+                                workspace_package_json_path
+                            )
+                            print(
+                                f"found package {workspace_package['name']}",
+                                file=sys.stderr,
+                            )
+                            deps[workspace_package["name"]] = {
+                                "path": workspace_path,
+                                "subdir": workspace_dir[len(local_path) + 1 :],
+                                "fork_url": dirs[local_dir]["fork_url"],
+                                "branch": dirs[local_dir]["branch"],
+                            }
     return deps
 
 
-def process_javascript(main_dir, dirs, container_mode):
-    """Replace directive in package.json if present"""
-    package_json_path = os.path.join(main_dir, "package.json")
-    if not os.path.exists(package_json_path):
-        return False
-    package = load_package_json(package_json_path)
-    if "dependencies" not in package:
-        return False
-    dependencies = package["dependencies"]
+def process_dependencies(dependencies, dirs, container_mode, package_json_path):
+    """Process dependencies in package.json and replace local dependencies"""
     local_deps = local_dependencies(dirs)
+    print(f"Found {len(local_deps)} local dependencies: {local_deps=}", file=sys.stderr)
     count = 0
     for dependency in dependencies:
         if dependency in local_deps:
             if container_mode:
                 info = local_deps[dependency]
-                # remove https://github.com/ at the beginning and .git at the end
                 repo = "git+" + info["fork_url"] + "#" + info["branch"]
                 print(
                     f"Replacing {dependency} with remote version from {repo} in package.json",
@@ -52,6 +82,27 @@ def process_javascript(main_dir, dirs, container_mode):
                 )
                 dependencies[dependency] = "file:" + local_deps[dependency]["path"]
             count += 1
+    return count
+
+
+def process_javascript(main_dir, dirs, container_mode):
+    """Use changes from PR in package.json if present"""
+    package_json_path = os.path.join(main_dir, "package.json")
+    if not os.path.exists(package_json_path):
+        return False
+    print(
+        "Detected package.json file, checking for local dependencies", file=sys.stderr
+    )
+    package = load_package_json(package_json_path)
+    if "dependencies" not in package:
+        return False
+    dependencies = package["dependencies"]
+    count = process_dependencies(dependencies, dirs, container_mode, package_json_path)
+    if "devDependencies" in package:
+        dependencies = package["devDependencies"]
+        count += process_dependencies(
+            dependencies, dirs, container_mode, package_json_path
+        )
     with open(package_json_path, "w", encoding="UTF-8") as package_json:
         json.dump(package, package_json, indent=2)
     return count > 0