Address PR feedback

This commit is contained in:
Aiqiao Yan 2020-04-30 15:28:04 -04:00
parent 97f7baa910
commit a5d9a3b1a6
6 changed files with 73 additions and 35 deletions

View File

@ -12,6 +12,7 @@ jest.mock("@actions/exec");
jest.mock("@actions/io"); jest.mock("@actions/io");
const IS_WINDOWS = process.platform === "win32"; const IS_WINDOWS = process.platform === "win32";
const IS_64OS = process.platform != "win32" || process.arch === "x64";
function getTempDir(): string { function getTempDir(): string {
return path.join(__dirname, "_temp", "tar"); return path.join(__dirname, "_temp", "tar");
@ -46,12 +47,13 @@ test("zstd extract tar", async () => {
const tarPath = IS_WINDOWS const tarPath = IS_WINDOWS
? `${process.env["windir"]}\\System32\\tar.exe` ? `${process.env["windir"]}\\System32\\tar.exe`
: "tar"; : "tar";
const zstdOptions = IS_64OS ? "zstd -d --long=31" : "zstd -d --long=30";
expect(execMock).toHaveBeenCalledTimes(1); expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledWith( expect(execMock).toHaveBeenCalledWith(
`${tarPath}`, `"${tarPath}"`,
[ [
"--use-compress-program", "--use-compress-program",
"zstd -d", zstdOptions,
"-xf", "-xf",
IS_WINDOWS ? archivePath.replace(/\\/g, "/") : archivePath, IS_WINDOWS ? archivePath.replace(/\\/g, "/") : archivePath,
"-P", "-P",
@ -78,7 +80,7 @@ test("gzip extract tar", async () => {
: "tar"; : "tar";
expect(execMock).toHaveBeenCalledTimes(1); expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledWith( expect(execMock).toHaveBeenCalledWith(
`${tarPath}`, `"${tarPath}"`,
[ [
"-z", "-z",
"-xf", "-xf",
@ -107,7 +109,7 @@ test("gzip extract GNU tar on windows", async () => {
expect(isGnuMock).toHaveBeenCalledTimes(1); expect(isGnuMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledTimes(1); expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledWith( expect(execMock).toHaveBeenCalledWith(
`tar`, `"tar"`,
[ [
"-z", "-z",
"-xf", "-xf",
@ -140,13 +142,14 @@ test("zstd create tar", async () => {
const tarPath = IS_WINDOWS const tarPath = IS_WINDOWS
? `${process.env["windir"]}\\System32\\tar.exe` ? `${process.env["windir"]}\\System32\\tar.exe`
: "tar"; : "tar";
const zstdOptions = IS_64OS ? "zstd -T0 --long=31" : "zstd -T0 --long=30";
expect(execMock).toHaveBeenCalledTimes(1); expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledWith( expect(execMock).toHaveBeenCalledWith(
`${tarPath}`, `"${tarPath}"`,
[ [
"--use-compress-program", "--use-compress-program",
"zstd -T0", zstdOptions,
"-cf", "-cf",
IS_WINDOWS IS_WINDOWS
? CacheFilename.Zstd.replace(/\\/g, "/") ? CacheFilename.Zstd.replace(/\\/g, "/")
@ -184,7 +187,7 @@ test("gzip create tar", async () => {
expect(execMock).toHaveBeenCalledTimes(1); expect(execMock).toHaveBeenCalledTimes(1);
expect(execMock).toHaveBeenCalledWith( expect(execMock).toHaveBeenCalledWith(
`${tarPath}`, `"${tarPath}"`,
[ [
"-z", "-z",
"-cf", "-cf",

29
dist/restore/index.js vendored
View File

@ -2237,10 +2237,9 @@ function createHttpClient() {
return new http_client_1.HttpClient("actions/cache", [bearerCredentialHandler], getRequestOptions()); return new http_client_1.HttpClient("actions/cache", [bearerCredentialHandler], getRequestOptions());
} }
function getCacheVersion(compressionMethod) { function getCacheVersion(compressionMethod) {
const components = [core.getInput(constants_1.Inputs.Path, { required: true })].concat(compressionMethod == constants_1.CompressionMethod.Zstd ? [compressionMethod] : []);
// Add salt to cache version to support breaking changes in cache entry // Add salt to cache version to support breaking changes in cache entry
const components = [core.getInput(constants_1.Inputs.Path, { required: true })].concat(compressionMethod == constants_1.CompressionMethod.Zstd components.push(versionSalt);
? [compressionMethod, versionSalt]
: versionSalt);
return crypto return crypto
.createHash("sha256") .createHash("sha256")
.update(components.join("|")) .update(components.join("|"))
@ -3320,7 +3319,7 @@ function unlinkFile(path) {
return util.promisify(fs.unlink)(path); return util.promisify(fs.unlink)(path);
} }
exports.unlinkFile = unlinkFile; exports.unlinkFile = unlinkFile;
function checkVersion(app) { function getVersion(app) {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
core.debug(`Checking ${app} --version`); core.debug(`Checking ${app} --version`);
let versionOutput = ""; let versionOutput = "";
@ -3344,7 +3343,7 @@ function checkVersion(app) {
} }
function getCompressionMethod() { function getCompressionMethod() {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
const versionOutput = yield checkVersion("zstd"); const versionOutput = yield getVersion("zstd");
return versionOutput.toLowerCase().includes("zstd command line interface") return versionOutput.toLowerCase().includes("zstd command line interface")
? constants_1.CompressionMethod.Zstd ? constants_1.CompressionMethod.Zstd
: constants_1.CompressionMethod.Gzip; : constants_1.CompressionMethod.Gzip;
@ -3359,7 +3358,7 @@ function getCacheFileName(compressionMethod) {
exports.getCacheFileName = getCacheFileName; exports.getCacheFileName = getCacheFileName;
function useGnuTar() { function useGnuTar() {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
const versionOutput = yield checkVersion("tar"); const versionOutput = yield getVersion("tar");
return versionOutput.toLowerCase().includes("gnu tar"); return versionOutput.toLowerCase().includes("gnu tar");
}); });
} }
@ -5068,7 +5067,7 @@ function execTar(args, cwd) {
var _a; var _a;
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
try { try {
yield exec_1.exec(`${yield getTarPath(args)}`, args, { cwd: cwd }); yield exec_1.exec(`"${yield getTarPath(args)}"`, args, { cwd: cwd });
} }
catch (error) { catch (error) {
throw new Error(`Tar failed with error: ${(_a = error) === null || _a === void 0 ? void 0 : _a.message}`); throw new Error(`Tar failed with error: ${(_a = error) === null || _a === void 0 ? void 0 : _a.message}`);
@ -5079,14 +5078,22 @@ function getWorkingDirectory() {
var _a; var _a;
return _a = process.env["GITHUB_WORKSPACE"], (_a !== null && _a !== void 0 ? _a : process.cwd()); return _a = process.env["GITHUB_WORKSPACE"], (_a !== null && _a !== void 0 ? _a : process.cwd());
} }
function isOS64() {
return process.platform != "win32" || process.arch === "x64";
}
function extractTar(archivePath, compressionMethod) { function extractTar(archivePath, compressionMethod) {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
// Create directory to extract tar into // Create directory to extract tar into
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
yield io.mkdirP(workingDirectory); yield io.mkdirP(workingDirectory);
// --d: Decompress.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const args = [ const args = [
...(compressionMethod == constants_1.CompressionMethod.Zstd ...(compressionMethod == constants_1.CompressionMethod.Zstd
? ["--use-compress-program", "zstd -d"] ? [
"--use-compress-program",
isOS64() ? "zstd -d --long=31" : "zstd -d --long=30"
]
: ["-z"]), : ["-z"]),
"-xf", "-xf",
archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"), archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"),
@ -5105,10 +5112,14 @@ function createTar(archiveFolder, sourceDirectories, compressionMethod) {
const cacheFileName = utils.getCacheFileName(compressionMethod); const cacheFileName = utils.getCacheFileName(compressionMethod);
fs_1.writeFileSync(path.join(archiveFolder, manifestFilename), sourceDirectories.join("\n")); fs_1.writeFileSync(path.join(archiveFolder, manifestFilename), sourceDirectories.join("\n"));
// -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores. // -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
const args = [ const args = [
...(compressionMethod == constants_1.CompressionMethod.Zstd ...(compressionMethod == constants_1.CompressionMethod.Zstd
? ["--use-compress-program", "zstd -T0"] ? [
"--use-compress-program",
isOS64() ? "zstd -T0 --long=31" : "zstd -T0 --long=30"
]
: ["-z"]), : ["-z"]),
"-cf", "-cf",
cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"), cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"),

29
dist/save/index.js vendored
View File

@ -2237,10 +2237,9 @@ function createHttpClient() {
return new http_client_1.HttpClient("actions/cache", [bearerCredentialHandler], getRequestOptions()); return new http_client_1.HttpClient("actions/cache", [bearerCredentialHandler], getRequestOptions());
} }
function getCacheVersion(compressionMethod) { function getCacheVersion(compressionMethod) {
const components = [core.getInput(constants_1.Inputs.Path, { required: true })].concat(compressionMethod == constants_1.CompressionMethod.Zstd ? [compressionMethod] : []);
// Add salt to cache version to support breaking changes in cache entry // Add salt to cache version to support breaking changes in cache entry
const components = [core.getInput(constants_1.Inputs.Path, { required: true })].concat(compressionMethod == constants_1.CompressionMethod.Zstd components.push(versionSalt);
? [compressionMethod, versionSalt]
: versionSalt);
return crypto return crypto
.createHash("sha256") .createHash("sha256")
.update(components.join("|")) .update(components.join("|"))
@ -3320,7 +3319,7 @@ function unlinkFile(path) {
return util.promisify(fs.unlink)(path); return util.promisify(fs.unlink)(path);
} }
exports.unlinkFile = unlinkFile; exports.unlinkFile = unlinkFile;
function checkVersion(app) { function getVersion(app) {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
core.debug(`Checking ${app} --version`); core.debug(`Checking ${app} --version`);
let versionOutput = ""; let versionOutput = "";
@ -3344,7 +3343,7 @@ function checkVersion(app) {
} }
function getCompressionMethod() { function getCompressionMethod() {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
const versionOutput = yield checkVersion("zstd"); const versionOutput = yield getVersion("zstd");
return versionOutput.toLowerCase().includes("zstd command line interface") return versionOutput.toLowerCase().includes("zstd command line interface")
? constants_1.CompressionMethod.Zstd ? constants_1.CompressionMethod.Zstd
: constants_1.CompressionMethod.Gzip; : constants_1.CompressionMethod.Gzip;
@ -3359,7 +3358,7 @@ function getCacheFileName(compressionMethod) {
exports.getCacheFileName = getCacheFileName; exports.getCacheFileName = getCacheFileName;
function useGnuTar() { function useGnuTar() {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
const versionOutput = yield checkVersion("tar"); const versionOutput = yield getVersion("tar");
return versionOutput.toLowerCase().includes("gnu tar"); return versionOutput.toLowerCase().includes("gnu tar");
}); });
} }
@ -5045,7 +5044,7 @@ function execTar(args, cwd) {
var _a; var _a;
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
try { try {
yield exec_1.exec(`${yield getTarPath(args)}`, args, { cwd: cwd }); yield exec_1.exec(`"${yield getTarPath(args)}"`, args, { cwd: cwd });
} }
catch (error) { catch (error) {
throw new Error(`Tar failed with error: ${(_a = error) === null || _a === void 0 ? void 0 : _a.message}`); throw new Error(`Tar failed with error: ${(_a = error) === null || _a === void 0 ? void 0 : _a.message}`);
@ -5056,14 +5055,22 @@ function getWorkingDirectory() {
var _a; var _a;
return _a = process.env["GITHUB_WORKSPACE"], (_a !== null && _a !== void 0 ? _a : process.cwd()); return _a = process.env["GITHUB_WORKSPACE"], (_a !== null && _a !== void 0 ? _a : process.cwd());
} }
function isOS64() {
return process.platform != "win32" || process.arch === "x64";
}
function extractTar(archivePath, compressionMethod) { function extractTar(archivePath, compressionMethod) {
return __awaiter(this, void 0, void 0, function* () { return __awaiter(this, void 0, void 0, function* () {
// Create directory to extract tar into // Create directory to extract tar into
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
yield io.mkdirP(workingDirectory); yield io.mkdirP(workingDirectory);
// --d: Decompress.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const args = [ const args = [
...(compressionMethod == constants_1.CompressionMethod.Zstd ...(compressionMethod == constants_1.CompressionMethod.Zstd
? ["--use-compress-program", "zstd -d"] ? [
"--use-compress-program",
isOS64() ? "zstd -d --long=31" : "zstd -d --long=30"
]
: ["-z"]), : ["-z"]),
"-xf", "-xf",
archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"), archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"),
@ -5082,10 +5089,14 @@ function createTar(archiveFolder, sourceDirectories, compressionMethod) {
const cacheFileName = utils.getCacheFileName(compressionMethod); const cacheFileName = utils.getCacheFileName(compressionMethod);
fs_1.writeFileSync(path.join(archiveFolder, manifestFilename), sourceDirectories.join("\n")); fs_1.writeFileSync(path.join(archiveFolder, manifestFilename), sourceDirectories.join("\n"));
// -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores. // -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
const args = [ const args = [
...(compressionMethod == constants_1.CompressionMethod.Zstd ...(compressionMethod == constants_1.CompressionMethod.Zstd
? ["--use-compress-program", "zstd -T0"] ? [
"--use-compress-program",
isOS64() ? "zstd -T0 --long=31" : "zstd -T0 --long=30"
]
: ["-z"]), : ["-z"]),
"-cf", "-cf",
cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"), cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"),

View File

@ -86,13 +86,13 @@ function createHttpClient(): HttpClient {
} }
export function getCacheVersion(compressionMethod?: CompressionMethod): string { export function getCacheVersion(compressionMethod?: CompressionMethod): string {
// Add salt to cache version to support breaking changes in cache entry
const components = [core.getInput(Inputs.Path, { required: true })].concat( const components = [core.getInput(Inputs.Path, { required: true })].concat(
compressionMethod == CompressionMethod.Zstd compressionMethod == CompressionMethod.Zstd ? [compressionMethod] : []
? [compressionMethod, versionSalt]
: versionSalt
); );
// Add salt to cache version to support breaking changes in cache entry
components.push(versionSalt);
return crypto return crypto
.createHash("sha256") .createHash("sha256")
.update(components.join("|")) .update(components.join("|"))

View File

@ -22,7 +22,7 @@ async function getTarPath(args: string[]): Promise<string> {
async function execTar(args: string[], cwd?: string): Promise<void> { async function execTar(args: string[], cwd?: string): Promise<void> {
try { try {
await exec(`${await getTarPath(args)}`, args, { cwd: cwd }); await exec(`"${await getTarPath(args)}"`, args, { cwd: cwd });
} catch (error) { } catch (error) {
throw new Error(`Tar failed with error: ${error?.message}`); throw new Error(`Tar failed with error: ${error?.message}`);
} }
@ -32,6 +32,10 @@ function getWorkingDirectory(): string {
return process.env["GITHUB_WORKSPACE"] ?? process.cwd(); return process.env["GITHUB_WORKSPACE"] ?? process.cwd();
} }
function isOS64(): boolean {
return process.platform != "win32" || process.arch === "x64";
}
export async function extractTar( export async function extractTar(
archivePath: string, archivePath: string,
compressionMethod: CompressionMethod compressionMethod: CompressionMethod
@ -39,9 +43,14 @@ export async function extractTar(
// Create directory to extract tar into // Create directory to extract tar into
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
await io.mkdirP(workingDirectory); await io.mkdirP(workingDirectory);
// --d: Decompress.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const args = [ const args = [
...(compressionMethod == CompressionMethod.Zstd ...(compressionMethod == CompressionMethod.Zstd
? ["--use-compress-program", "zstd -d"] ? [
"--use-compress-program",
isOS64() ? "zstd -d --long=31" : "zstd -d --long=30"
]
: ["-z"]), : ["-z"]),
"-xf", "-xf",
archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"), archivePath.replace(new RegExp("\\" + path.sep, "g"), "/"),
@ -65,10 +74,14 @@ export async function createTar(
sourceDirectories.join("\n") sourceDirectories.join("\n")
); );
// -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores. // -T#: Compress using # working thread. If # is 0, attempt to detect and use the number of physical CPU cores.
// --long=#: Enables long distance matching with # bits. Maximum is 30 (1GB) on 32-bit OS and 31 (2GB) on 64-bit.
const workingDirectory = getWorkingDirectory(); const workingDirectory = getWorkingDirectory();
const args = [ const args = [
...(compressionMethod == CompressionMethod.Zstd ...(compressionMethod == CompressionMethod.Zstd
? ["--use-compress-program", "zstd -T0"] ? [
"--use-compress-program",
isOS64() ? "zstd -T0 --long=31" : "zstd -T0 --long=30"
]
: ["-z"]), : ["-z"]),
"-cf", "-cf",
cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"), cacheFileName.replace(new RegExp("\\" + path.sep, "g"), "/"),

View File

@ -124,7 +124,7 @@ export function unlinkFile(path: fs.PathLike): Promise<void> {
return util.promisify(fs.unlink)(path); return util.promisify(fs.unlink)(path);
} }
async function checkVersion(app: string): Promise<string> { async function getVersion(app: string): Promise<string> {
core.debug(`Checking ${app} --version`); core.debug(`Checking ${app} --version`);
let versionOutput = ""; let versionOutput = "";
try { try {
@ -148,7 +148,7 @@ async function checkVersion(app: string): Promise<string> {
} }
export async function getCompressionMethod(): Promise<CompressionMethod> { export async function getCompressionMethod(): Promise<CompressionMethod> {
const versionOutput = await checkVersion("zstd"); const versionOutput = await getVersion("zstd");
return versionOutput.toLowerCase().includes("zstd command line interface") return versionOutput.toLowerCase().includes("zstd command line interface")
? CompressionMethod.Zstd ? CompressionMethod.Zstd
: CompressionMethod.Gzip; : CompressionMethod.Gzip;
@ -161,6 +161,6 @@ export function getCacheFileName(compressionMethod: CompressionMethod): string {
} }
export async function useGnuTar(): Promise<boolean> { export async function useGnuTar(): Promise<boolean> {
const versionOutput = await checkVersion("tar"); const versionOutput = await getVersion("tar");
return versionOutput.toLowerCase().includes("gnu tar"); return versionOutput.toLowerCase().includes("gnu tar");
} }