k6-node
Version:
CLI tool that enables k6 installation via npm packages
87 lines (86 loc) • 3.73 kB
JavaScript
;
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 () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.extractFile = void 0;
const node_fs_1 = __importDefault(require("node:fs"));
const node_path_1 = __importDefault(require("node:path"));
const extractFile = async (filePath, outputDir, platform) => {
try {
console.log('Extracting artifact...');
if (!node_fs_1.default.existsSync(outputDir))
node_fs_1.default.mkdirSync(outputDir, { recursive: true });
if (filePath.endsWith('.zip')) {
await (await Promise.resolve().then(() => __importStar(require('zip-lib')))).extract(filePath, outputDir);
}
else if (filePath.endsWith('.tar.gz')) {
(await Promise.resolve().then(() => __importStar(require('tar')))).extract({
sync: true,
file: filePath,
cwd: outputDir,
});
}
// Find k6 binary in the extracted directory
const findBinary = (dir) => {
const files = node_fs_1.default.readdirSync(dir, { recursive: true });
console.log(files);
const binaryName = platform === 'windows' ? 'k6.exe' : 'k6';
const binaryPath = files.find((file) => typeof file === 'string' && file.endsWith(binaryName));
if (!binaryPath)
throw new Error('k6 executable not found! Try reinstall');
return node_path_1.default.join(dir, binaryPath);
};
const binaryPath = findBinary(outputDir);
if (platform !== 'windows') {
node_fs_1.default.chmodSync(binaryPath, 0o755);
}
// move binary to the correct path
const finalPath = node_path_1.default.join(node_path_1.default.dirname(outputDir), platform === 'windows' ? 'k6.exe' : 'k6');
node_fs_1.default.renameSync(binaryPath, finalPath);
//clear temp
node_fs_1.default.unlinkSync(filePath);
node_fs_1.default.rmSync(outputDir, { recursive: true, force: true });
return finalPath;
}
catch (error) {
if (error instanceof Error) {
throw new Error(`Extraction error: ${error.message}`);
}
throw error;
}
};
exports.extractFile = extractFile;