From 59cb3f4919eb64c8afad1545f67330fee453e839 Mon Sep 17 00:00:00 2001
From: Patrick Arminio <patrick.arminio@gmail.com>
Date: Wed, 24 Nov 2021 15:10:58 +0000
Subject: [PATCH] Add untest poetry cache implementation

---
 src/cache-distributions/poetry-cache.ts | 75 +++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 src/cache-distributions/poetry-cache.ts

diff --git a/src/cache-distributions/poetry-cache.ts b/src/cache-distributions/poetry-cache.ts
new file mode 100644
index 0000000..7eeaa3e
--- /dev/null
+++ b/src/cache-distributions/poetry-cache.ts
@@ -0,0 +1,75 @@
+import * as glob from '@actions/glob';
+import * as os from 'os';
+import * as path from 'path';
+import * as exec from '@actions/exec';
+
+import CacheDistributor from './cache-distributor';
+
+class PoetryCache extends CacheDistributor {
+  constructor(
+    private pythonVersion: string,
+    protected patterns: string = '**/poetry.lock'
+  ) {
+    super('poetry', patterns);
+  }
+
+  protected async getCacheGlobalDirectories() {
+    const poetryConfig = await this.getPoetryConfiguration();
+
+    const cacheDir = poetryConfig['cache-dir'];
+    const virtualenvsPath = poetryConfig['virtualenvs.path'].replace(
+      '{cache-dir}',
+      cacheDir
+    );
+
+    const paths = [virtualenvsPath];
+
+    if (poetryConfig['virtualenvs.in-project'] === 'true') {
+      paths.push(path.join(process.cwd(), '.venv'));
+    }
+
+    return paths;
+  }
+
+  protected async computeKeys() {
+    const hash = await glob.hashFiles(this.patterns);
+    const primaryKey = `${this.CACHE_KEY_PREFIX}-${process.env['RUNNER_OS']}-python-${this.pythonVersion}-${this.packageManager}-${hash}`;
+    const restoreKey = undefined;
+    return {
+      primaryKey,
+      restoreKey
+    };
+  }
+
+  private async getPoetryConfiguration() {
+    const {stdout, stderr, exitCode} = await exec.getExecOutput(
+      'poetry config --list'
+    );
+
+    if (exitCode && stderr) {
+      throw new Error(
+        `Could not get cache folder path for poetry package manager`
+      );
+    }
+
+    const lines = stdout.split(os.EOL);
+
+    const config = {} as {
+      'cache-dir': string;
+      'virtualenvs.in-project': string;
+      'virtualenvs.path': string;
+    };
+
+    for (let line of lines) {
+      line = line.replace(/#.*$/, '');
+
+      const [key, value] = line.split('=').map(part => part.trim());
+
+      config[key as keyof typeof config] = value;
+    }
+
+    return config;
+  }
+}
+
+export default PoetryCache;