mirror of
https://code.forgejo.org/actions/cache.git
synced 2024-08-25 22:11:38 +02:00
127 lines
3.8 KiB
TypeScript
127 lines
3.8 KiB
TypeScript
import * as core from "@actions/core";
|
|
import * as fs from "fs";
|
|
|
|
import { BearerCredentialHandler } from "typed-rest-client/Handlers";
|
|
import { HttpClient } from "typed-rest-client/HttpClient";
|
|
import { IHttpClientResponse } from "typed-rest-client/Interfaces";
|
|
import { RestClient, IRequestOptions } from "typed-rest-client/RestClient";
|
|
|
|
import { ArtifactCacheEntry } from "./contracts";
|
|
|
|
export async function getCacheEntry(
|
|
keys: string[]
|
|
): Promise<ArtifactCacheEntry> {
|
|
const cacheUrl = getCacheUrl();
|
|
const token = process.env["ACTIONS_RUNTIME_TOKEN"] || "";
|
|
const bearerCredentialHandler = new BearerCredentialHandler(token);
|
|
|
|
const resource = `_apis/artifactcache/cache?keys=${encodeURIComponent(
|
|
keys.join(",")
|
|
)}`;
|
|
|
|
const restClient = new RestClient("actions/cache", cacheUrl, [
|
|
bearerCredentialHandler
|
|
]);
|
|
|
|
const response = await restClient.get<ArtifactCacheEntry>(
|
|
resource,
|
|
getRequestOptions()
|
|
);
|
|
if (response.statusCode === 204) {
|
|
throw new Error(
|
|
`Cache not found for input keys: ${JSON.stringify(keys)}.`
|
|
);
|
|
}
|
|
if (response.statusCode !== 200) {
|
|
throw new Error(`Cache service responded with ${response.statusCode}`);
|
|
}
|
|
const cacheResult = response.result;
|
|
core.debug(`Cache Result:`);
|
|
core.debug(JSON.stringify(cacheResult));
|
|
if (!cacheResult || !cacheResult.archiveLocation) {
|
|
throw new Error("Cache not found.");
|
|
}
|
|
|
|
return cacheResult;
|
|
}
|
|
|
|
export async function downloadCache(
|
|
cacheEntry: ArtifactCacheEntry,
|
|
archivePath: string
|
|
): Promise<void> {
|
|
const stream = fs.createWriteStream(archivePath);
|
|
const httpClient = new HttpClient("actions/cache");
|
|
const downloadResponse = await httpClient.get(cacheEntry.archiveLocation!);
|
|
await pipeResponseToStream(downloadResponse, stream);
|
|
}
|
|
|
|
async function pipeResponseToStream(
|
|
response: IHttpClientResponse,
|
|
stream: NodeJS.WritableStream
|
|
): Promise<void> {
|
|
return new Promise(resolve => {
|
|
response.message.pipe(stream).on("close", () => {
|
|
resolve();
|
|
});
|
|
});
|
|
}
|
|
|
|
export async function saveCache(stream: NodeJS.ReadableStream, key: string) {
|
|
const cacheUrl = getCacheUrl();
|
|
const token = process.env["ACTIONS_RUNTIME_TOKEN"] || "";
|
|
const bearerCredentialHandler = new BearerCredentialHandler(token);
|
|
|
|
const resource = `_apis/artifactcache/cache/${encodeURIComponent(key)}`;
|
|
const postUrl = cacheUrl + resource;
|
|
|
|
const restClient = new RestClient("actions/cache", undefined, [
|
|
bearerCredentialHandler
|
|
]);
|
|
|
|
const requestOptions = getRequestOptions();
|
|
requestOptions.additionalHeaders = {
|
|
"Content-Type": "application/octet-stream"
|
|
};
|
|
|
|
const response = await restClient.uploadStream<void>(
|
|
"POST",
|
|
postUrl,
|
|
stream,
|
|
requestOptions
|
|
);
|
|
if (response.statusCode !== 200) {
|
|
throw new Error(`Cache service responded with ${response.statusCode}`);
|
|
}
|
|
|
|
core.info("Cache saved successfully");
|
|
}
|
|
|
|
function getRequestOptions(): IRequestOptions {
|
|
const requestOptions: IRequestOptions = {
|
|
acceptHeader: createAcceptHeader("application/json", "5.2-preview.1")
|
|
};
|
|
|
|
return requestOptions;
|
|
}
|
|
|
|
function createAcceptHeader(type: string, apiVersion: string): string {
|
|
return `${type};api-version=${apiVersion}`;
|
|
}
|
|
|
|
function getCacheUrl(): string {
|
|
// Ideally we just use ACTIONS_CACHE_URL
|
|
let cacheUrl: string = (
|
|
process.env["ACTIONS_CACHE_URL"] ||
|
|
process.env["ACTIONS_RUNTIME_URL"] ||
|
|
""
|
|
).replace("pipelines", "artifactcache");
|
|
if (!cacheUrl) {
|
|
throw new Error(
|
|
"Cache Service Url not found, unable to restore cache."
|
|
);
|
|
}
|
|
|
|
core.debug(`Cache Url: ${cacheUrl}`);
|
|
return cacheUrl;
|
|
}
|