From 75fd6b26cb128a82716e21588aec6ecc2a8db606 Mon Sep 17 00:00:00 2001
From: Tommy Chen <tommy351@gmail.com>
Date: Mon, 22 Mar 2021 16:06:59 +0800
Subject: [PATCH] fix(docker-build): Fix transient dependencies

Fix protocol files of transient dependencies are not copied.
---
 packages/docker-build/src/commands/build.ts   |  2 +-
 .../src/utils/copyProtocolFiles.ts            | 56 +++++++++----------
 2 files changed, 28 insertions(+), 30 deletions(-)

diff --git a/packages/docker-build/src/commands/build.ts b/packages/docker-build/src/commands/build.ts
index 2189860..940b0c2 100644
--- a/packages/docker-build/src/commands/build.ts
+++ b/packages/docker-build/src/commands/build.ts
@@ -132,8 +132,8 @@ export default class DockerBuildCommand extends BaseCommand {
 
             await copyProtocolFiles({
               destination: manifestDir,
-              workspaces: project.workspaces,
               report,
+              project,
               parseDescriptor: (descriptor) => {
                 if (descriptor.range.startsWith('exec:')) {
                   const parsed = parseSpec(descriptor.range);
diff --git a/packages/docker-build/src/utils/copyProtocolFiles.ts b/packages/docker-build/src/utils/copyProtocolFiles.ts
index 0612dea..dd21cee 100644
--- a/packages/docker-build/src/utils/copyProtocolFiles.ts
+++ b/packages/docker-build/src/utils/copyProtocolFiles.ts
@@ -1,9 +1,9 @@
 import {
   Descriptor,
   Locator,
+  Project,
   Report,
   structUtils,
-  Workspace,
 } from '@yarnpkg/core';
 import { PortablePath, ppath, xfs } from '@yarnpkg/fslib';
 
@@ -12,55 +12,53 @@ const BUILTIN_REGEXP = /^builtin<([^>]+)>$/;
 
 export default async function copyProtocolFiles({
   destination,
-  workspaces,
   report,
+  project,
   parseDescriptor,
 }: {
   destination: PortablePath;
-  workspaces: Workspace[];
   report: Report;
+  project: Project;
   parseDescriptor: (
     descriptor: Descriptor,
   ) => { parentLocator: Locator; paths: PortablePath[] } | undefined;
 }): Promise<void> {
   const copiedPaths = new Set<string>();
 
-  for (const ws of workspaces) {
-    for (const descriptor of ws.dependencies.values()) {
-      const patchDescriptor = structUtils.isVirtualDescriptor(descriptor)
-        ? structUtils.devirtualizeDescriptor(descriptor)
-        : descriptor;
+  for (const descriptor of project.storedDescriptors.values()) {
+    const resolvedDescriptor = structUtils.isVirtualDescriptor(descriptor)
+      ? structUtils.devirtualizeDescriptor(descriptor)
+      : descriptor;
 
-      const parsed = parseDescriptor(patchDescriptor);
-      if (!parsed) continue;
+    const parsed = parseDescriptor(resolvedDescriptor);
+    if (!parsed) continue;
 
-      const { parentLocator, paths } = parsed;
+    const { parentLocator, paths } = parsed;
 
-      for (const path of paths) {
-        // Ignore builtin modules
-        if (BUILTIN_REGEXP.test(path)) continue;
+    for (const path of paths) {
+      // Ignore builtin modules
+      if (BUILTIN_REGEXP.test(path)) continue;
 
-        // TODO: Handle absolute path
-        if (ppath.isAbsolute(path)) continue;
+      // TODO: Handle absolute path
+      if (ppath.isAbsolute(path)) continue;
 
-        // Get the workspace by parentLocator
-        const parentWorkspace = ws.project.getWorkspaceByLocator(parentLocator);
+      // Get the workspace by parentLocator
+      const parentWorkspace = project.getWorkspaceByLocator(parentLocator);
 
-        // The path relative to the project CWD
-        const relativePath = ppath.join(parentWorkspace.relativeCwd, path);
+      // The path relative to the project CWD
+      const relativePath = ppath.join(parentWorkspace.relativeCwd, path);
 
-        // Skip if the path has been copied already
-        if (copiedPaths.has(relativePath)) continue;
+      // Skip if the path has been copied already
+      if (copiedPaths.has(relativePath)) continue;
 
-        copiedPaths.add(relativePath);
+      copiedPaths.add(relativePath);
 
-        const src = ppath.join(parentWorkspace.cwd, path);
-        const dest = ppath.join(destination, relativePath);
+      const src = ppath.join(parentWorkspace.cwd, path);
+      const dest = ppath.join(destination, relativePath);
 
-        report.reportInfo(null, relativePath);
-        await xfs.mkdirpPromise(ppath.dirname(dest));
-        await xfs.copyFilePromise(src, dest);
-      }
+      report.reportInfo(null, relativePath);
+      await xfs.mkdirpPromise(ppath.dirname(dest));
+      await xfs.copyFilePromise(src, dest);
     }
   }
 }