fix(wb): make optimizeForDockerBuild support any GitHub-hosted packages

WillBooster · Oct 3, 2023 · 540dfa3 · 540dfa3
1 parent 3a6808e
commit 540dfa3
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/packages/wb/src/commands/optimizeForDockerBuild.ts b/packages/wb/src/commands/optimizeForDockerBuild.ts
@@ -41,20 +41,15 @@ export const optimizeForDockerBuildCommand: CommandModule<unknown, InferredOptio
 
     for (const packageJsonPath of packageJsonPaths) {
       const packageJson: PackageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
-      const deps = packageJson.dependencies || {};
-      if (deps['@discord-bot/llm'] && deps['@discord-bot/llm'] !== 'workspace:*') {
-        deps['@discord-bot/llm'] = './@discord-bot/llm';
-      }
-      const depNames = ['@moti-components/go-e-mon', '@online-judge/shared', 'program-executor', 'judge'];
-      for (const depName of depNames) {
-        if (deps[depName]) {
-          deps[depName] = `./${depName}`;
+      const keys = ['dependencies', 'devDependencies'] as const;
+      for (const key of keys) {
+        const deps = packageJson[key] || {};
+        for (const [name, value] of Object.entries(deps)) {
+          if (value?.startsWith('[email protected]:')) {
+            deps[name] = `./${name}`;
+          }
         }
       }
-      // TODO: remove after migrating from online-judge-shared to @online-judge/shared
-      if (deps['online-judge-shared']) {
-        deps['online-judge-shared'] = './online-judge-shared';
-      }
 
       optimizeDevDependencies(argv, packageJson);