From a957cf8c6c752aa30a3311b3f09bdf2a91408100 Mon Sep 17 00:00:00 2001
From: Oran Avraham <oranav@gmail.com>
Date: Fri, 9 Dec 2022 17:48:00 +0200
Subject: [PATCH] poetry: Run `poetry env use` only after cache is loaded

The virtualenv cache might contain invalid entries, such as virtualenvs
built in previous, buggy versions of this action.  The `poetry env use`
command will recreate virtualenvs in case they are invalid, but it has
to be run only *after* the cache is loaded.

Refactor `CacheDistributor` a bit such that the validation (and possible
recreation) of virtualenvs happens only after the cache is loaded.
---
 src/cache-distributions/cache-distributor.ts |  3 ++
 src/cache-distributions/poetry-cache.ts      | 50 ++++++++++++--------
 2 files changed, 33 insertions(+), 20 deletions(-)

diff --git a/src/cache-distributions/cache-distributor.ts b/src/cache-distributions/cache-distributor.ts
index f24c78dab..2e46c961d 100644
--- a/src/cache-distributions/cache-distributor.ts
+++ b/src/cache-distributions/cache-distributor.ts
@@ -19,6 +19,7 @@ abstract class CacheDistributor {
     primaryKey: string;
     restoreKey: string[] | undefined;
   }>;
+  protected async handleLoadedCache() {}
 
   public async restoreCache() {
     const {primaryKey, restoreKey} = await this.computeKeys();
@@ -41,6 +42,8 @@ abstract class CacheDistributor {
       restoreKey
     );
 
+    await this.handleLoadedCache();
+
     this.handleMatchResult(matchedKey, primaryKey);
   }
 
diff --git a/src/cache-distributions/poetry-cache.ts b/src/cache-distributions/poetry-cache.ts
index 24a96b13b..ebbffbacf 100644
--- a/src/cache-distributions/poetry-cache.ts
+++ b/src/cache-distributions/poetry-cache.ts
@@ -10,7 +10,8 @@ import {logWarning} from '../utils';
 class PoetryCache extends CacheDistributor {
   constructor(
     private pythonVersion: string,
-    protected patterns: string = '**/poetry.lock'
+    protected patterns: string = '**/poetry.lock',
+    protected poetryProjects: Set<string> = new Set<string>()
   ) {
     super('poetry', patterns);
   }
@@ -20,16 +21,10 @@ class PoetryCache extends CacheDistributor {
     const paths = new Set<string>();
     const globber = await glob.create(this.patterns);
 
-    const pythonLocation = await io.which('python');
-    if (pythonLocation) {
-      core.debug(`pythonLocation is ${pythonLocation}`);
-    } else {
-      logWarning('python binaries were not found in PATH');
-    }
-
     for await (const file of globber.globGenerator()) {
       const basedir = path.dirname(file);
       core.debug(`Processing Poetry project at ${basedir}`);
+      this.poetryProjects.add(basedir);
 
       const poetryConfig = await this.getPoetryConfiguration(basedir);
 
@@ -44,18 +39,6 @@ class PoetryCache extends CacheDistributor {
       if (poetryConfig['virtualenvs.in-project']) {
         paths.add(path.join(basedir, '.venv'));
       }
-
-      if (pythonLocation) {
-        const {exitCode, stderr} = await exec.getExecOutput(
-          'poetry',
-          ['env', 'use', pythonLocation],
-          {ignoreReturnCode: true, cwd: basedir}
-        );
-
-        if (exitCode) {
-          logWarning(stderr);
-        }
-      }
     }
 
     return [...paths];
@@ -71,6 +54,33 @@ class PoetryCache extends CacheDistributor {
     };
   }
 
+  protected async handleLoadedCache() {
+    await super.handleLoadedCache();
+
+    // After the cache is loaded -- make sure virtualenvs use the correct Python version (the one that we have just installed).
+    // This will handle invalid caches, recreating virtualenvs if necessary.
+
+    const pythonLocation = await io.which('python');
+    if (pythonLocation) {
+      core.debug(`pythonLocation is ${pythonLocation}`);
+    } else {
+      logWarning('python binaries were not found in PATH');
+      return;
+    }
+
+    for (const poetryProject of this.poetryProjects) {
+      const {exitCode, stderr} = await exec.getExecOutput(
+        'poetry',
+        ['env', 'use', pythonLocation],
+        {ignoreReturnCode: true, cwd: poetryProject}
+      );
+
+      if (exitCode) {
+        logWarning(stderr);
+      }
+    }
+  }
+
   private async getPoetryConfiguration(basedir: string) {
     const {stdout, stderr, exitCode} = await exec.getExecOutput(
       'poetry',