From 89052adf60080c1838d8d83ae6a9e487f555873c Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Mon, 19 Feb 2024 13:30:09 +0200 Subject: [PATCH] Add support for caching `uv` See https://github.com/astral-sh/uv --- dist/setup/index.js | 86 ++++++++++++++++++++++++ src/cache-distributions/cache-factory.ts | 6 +- src/cache-distributions/uv-cache.ts | 36 ++++++++++ 3 files changed, 127 insertions(+), 1 deletion(-) create mode 100644 src/cache-distributions/uv-cache.ts diff --git a/dist/setup/index.js b/dist/setup/index.js index dddc7eca..598f2a73 100644 --- a/dist/setup/index.js +++ b/dist/setup/index.js @@ -89595,11 +89595,13 @@ exports.getCacheDistributor = exports.PackageManagers = void 0; const pip_cache_1 = __importDefault(__nccwpck_require__(5546)); const pipenv_cache_1 = __importDefault(__nccwpck_require__(238)); const poetry_cache_1 = __importDefault(__nccwpck_require__(1993)); +const uv_cache_1 = __importDefault(__nccwpck_require__(8795)); var PackageManagers; (function (PackageManagers) { PackageManagers["Pip"] = "pip"; PackageManagers["Pipenv"] = "pipenv"; PackageManagers["Poetry"] = "poetry"; + PackageManagers["Uv"] = "uv"; })(PackageManagers || (exports.PackageManagers = PackageManagers = {})); function getCacheDistributor(packageManager, pythonVersion, cacheDependencyPath) { switch (packageManager) { @@ -89609,6 +89611,8 @@ function getCacheDistributor(packageManager, pythonVersion, cacheDependencyPath) return new pipenv_cache_1.default(pythonVersion, cacheDependencyPath); case PackageManagers.Poetry: return new poetry_cache_1.default(pythonVersion, cacheDependencyPath); + case PackageManagers.Uv: + return new uv_cache_1.default(pythonVersion, cacheDependencyPath); default: throw new Error(`Caching for '${packageManager}' is not supported`); } @@ -89983,6 +89987,88 @@ class PoetryCache extends cache_distributor_1.default { exports["default"] = PoetryCache; +/***/ }), + +/***/ 8795: +/***/ (function(__unused_webpack_module, exports, __nccwpck_require__) { + +"use strict"; + +var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) { + if (k2 === undefined) k2 = k; + var desc = Object.getOwnPropertyDescriptor(m, k); + if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) { + desc = { enumerable: true, get: function() { return m[k]; } }; + } + Object.defineProperty(o, k2, desc); +}) : (function(o, m, k, k2) { + if (k2 === undefined) k2 = k; + o[k2] = m[k]; +})); +var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) { + Object.defineProperty(o, "default", { enumerable: true, value: v }); +}) : function(o, v) { + o["default"] = v; +}); +var __importStar = (this && this.__importStar) || function (mod) { + if (mod && mod.__esModule) return mod; + var result = {}; + if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k); + __setModuleDefault(result, mod); + return result; +}; +var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { + function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } + return new (P || (P = Promise))(function (resolve, reject) { + function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } + function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } + function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } + step((generator = generator.apply(thisArg, _arguments || [])).next()); + }); +}; +var __importDefault = (this && this.__importDefault) || function (mod) { + return (mod && mod.__esModule) ? mod : { "default": mod }; +}; +Object.defineProperty(exports, "__esModule", ({ value: true })); +const glob = __importStar(__nccwpck_require__(8090)); +const os = __importStar(__nccwpck_require__(2037)); +const path = __importStar(__nccwpck_require__(1017)); +const cache_distributor_1 = __importDefault(__nccwpck_require__(8953)); +class UvCache extends cache_distributor_1.default { + constructor(pythonVersion, patterns = '**/requirements.txt') { + super('uv', patterns); + this.pythonVersion = pythonVersion; + this.patterns = patterns; + } + getCacheGlobalDirectories() { + var _a; + return __awaiter(this, void 0, void 0, function* () { + if (process.platform === 'win32') { + // `LOCALAPPDATA` should always be defined, + // but we can't just join `undefined` + // into the path in case it's not. + return [ + path.join((_a = process.env['LOCALAPPDATA']) !== null && _a !== void 0 ? _a : os.homedir(), 'uv', 'cache') + ]; + } + return [path.join(os.homedir(), '.cache/uv')]; + }); + } + computeKeys() { + return __awaiter(this, void 0, void 0, function* () { + const hash = yield 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 + }; + }); + } +} +exports["default"] = UvCache; + + /***/ }), /***/ 8040: diff --git a/src/cache-distributions/cache-factory.ts b/src/cache-distributions/cache-factory.ts index 7becf532..66d4b9db 100644 --- a/src/cache-distributions/cache-factory.ts +++ b/src/cache-distributions/cache-factory.ts @@ -1,11 +1,13 @@ import PipCache from './pip-cache'; import PipenvCache from './pipenv-cache'; import PoetryCache from './poetry-cache'; +import UvCache from './uv-cache'; export enum PackageManagers { Pip = 'pip', Pipenv = 'pipenv', - Poetry = 'poetry' + Poetry = 'poetry', + Uv = 'uv' } export function getCacheDistributor( @@ -20,6 +22,8 @@ export function getCacheDistributor( return new PipenvCache(pythonVersion, cacheDependencyPath); case PackageManagers.Poetry: return new PoetryCache(pythonVersion, cacheDependencyPath); + case PackageManagers.Uv: + return new UvCache(pythonVersion, cacheDependencyPath); default: throw new Error(`Caching for '${packageManager}' is not supported`); } diff --git a/src/cache-distributions/uv-cache.ts b/src/cache-distributions/uv-cache.ts new file mode 100644 index 00000000..b9ca6cd9 --- /dev/null +++ b/src/cache-distributions/uv-cache.ts @@ -0,0 +1,36 @@ +import * as glob from '@actions/glob'; +import * as os from 'os'; +import * as path from 'path'; + +import CacheDistributor from './cache-distributor'; + +export default class UvCache extends CacheDistributor { + constructor( + private pythonVersion: string, + protected patterns: string = '**/requirements.txt' + ) { + super('uv', patterns); + } + + protected async getCacheGlobalDirectories() { + if (process.platform === 'win32') { + // `LOCALAPPDATA` should always be defined, + // but we can't just join `undefined` + // into the path in case it's not. + return [ + path.join(process.env['LOCALAPPDATA'] ?? os.homedir(), 'uv', 'cache') + ]; + } + return [path.join(os.homedir(), '.cache/uv')]; + } + + 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 + }; + } +}