UNPKG

@syngrisi/syngrisi

Version:
1,348 lines (1,340 loc) 32.4 kB
// src/tasks/restore.ts import { input, confirm, select } from "@inquirer/prompts"; import * as fs3 from "fs"; import * as path3 from "path"; // src/server/config.ts import fs from "fs"; import dotenv2 from "dotenv"; // package.json var version = "3.5.0"; var gitHead = "12bfda406cbe5aaccf3f17fdab02a9bd1a9d6343"; // src/server/config.ts import crypto2 from "crypto"; import { execSync } from "child_process"; // src/server/envConfig.ts import { cleanEnv, host, num, port, str, bool } from "envalid"; import crypto from "crypto"; import path from "path"; import dotenv from "dotenv"; dotenv.config({ quiet: true }); if (!process.env.NODE_ENV) { process.env.NODE_ENV = "production"; } var env = cleanEnv(process.env, { NODE_ENV: str({ choices: ["development", "production", "test"] }), SYNGRISI_DB_URI: str({ default: "mongodb://127.0.0.1:27017/SyngrisiDb" }), SYNGRISI_APP_PORT: port({ default: 3e3 }), SYNGRISI_IMAGES_PATH: str({ default: path.join(process.cwd(), "./.snapshots-images") }), SYNGRISI_DOM_SNAPSHOTS_PATH: str({ default: "" }), // If empty, uses SYNGRISI_IMAGES_PATH SYNGRISI_TMP_DIR: str({ default: path.join(process.cwd(), ".tmp") }), SYNGRISI_ADMIN_DATA_JOBS_PATH: str({ default: path.join(process.cwd(), ".tmp", "admin-data-jobs") }), SYNGRISI_ADMIN_DATA_JOBS_TTL_MS: num({ default: 24 * 60 * 60 * 1e3 }), SYNGRISI_ADMIN_DATA_MAX_CONCURRENT_JOBS: num({ default: 1 }), SYNGRISI_ADMIN_DATA_UPLOAD_MAX_SIZE_MB: num({ default: 10240 }), SYNGRISI_HTTP_LOG: bool({ default: false }), SYNGRISI_COVERAGE: bool({ default: false }), SYNGRISI_HOSTNAME: host({ default: "localhost" }), SYNGRISI_AUTH: bool({ default: true }), SYNGRISI_TEST_MODE: bool({ default: false }), SYNGRISI_DISABLE_FIRST_RUN: bool({ default: false }), MONGODB_ROOT_USERNAME: str({ default: "" }), MONGODB_ROOT_PASSWORD: str({ default: "" }), LOGLEVEL: str({ choices: ["error", "warn", "info", "verbose", "debug", "silly"], default: "debug" }), // Legacy tests expect 20 rows per page; keep default aligned for e2e SYNGRISI_PAGINATION_SIZE: num({ default: 20 }), SYNGRISI_DISABLE_DEV_CORS: bool({ default: true, devDefault: true }), SYNGRISI_SESSION_STORE_KEY: str({ default: crypto.randomBytes(64).toString("hex") }), SYNGRISI_LOG_LEVEL: str({ default: "debug" }), SYNGRISI_DISABLE_LOGS: bool({ default: false }), SYNGRISI_AUTO_REMOVE_CHECKS_POLL_INTERVAL_MS: num({ default: 10 * 60 * 1e3 }), // 10 minutes SYNGRISI_AUTO_REMOVE_CHECKS_MIN_INTERVAL_MS: num({ default: 24 * 60 * 60 * 1e3 }), SYNGRISI_ENABLE_SCHEDULERS_IN_TEST_MODE: bool({ default: false }), // RCA SYNGRISI_RCA: bool({ default: false }), // trunk features SYNGRISI_TRUNK_FEATURE_AI_SEVERITY: bool({ default: false }), SYNGRISI_AI_KEY: str({ default: "" }), OPENAI_API_BASE_URL: str({ default: "https://api.openai.com/v1" }), OPENAI_API_KEY: str({ default: "" }), SYNGRISI_V8_COVERAGE_ON_EXIT: bool({ default: false }), // Rate Limiting SYNGRISI_RATE_LIMIT_WINDOW_MS: num({ default: 15 * 60 * 1e3 }), // 15 minutes SYNGRISI_RATE_LIMIT_MAX: num({ default: 5e4 }), SYNGRISI_AUTH_RATE_LIMIT_WINDOW_MS: num({ default: 15 * 60 * 1e3 }), // 15 minutes SYNGRISI_AUTH_RATE_LIMIT_MAX: num({ default: 200 }), // Mongo tuneables for tests/CI flake reduction SYNGRISI_MONGO_SOCKET_TIMEOUT_MS: num({ default: 6e4 }), SYNGRISI_MONGO_MAX_POOL_SIZE: num({ default: 20 }), SYNGRISI_MONGO_MIN_POOL_SIZE: num({ default: 2 }), SYNGRISI_MONGO_MAX_IDLE_TIME_MS: num({ default: 3e4 }), SYNGRISI_MONGO_WAIT_QUEUE_TIMEOUT_MS: num({ default: 3e4 }), SYNGRISI_MONGO_SERVER_SELECTION_TIMEOUT_MS: num({ default: 1e4 }), SYNGRISI_MONGO_CONNECT_TIMEOUT_MS: num({ default: 3e4 }), // SSO Configuration SSO_ENABLED: bool({ default: false }), SSO_PROTOCOL: str({ choices: ["", "oauth2", "saml"], default: "" }), SSO_CLIENT_ID: str({ default: "" }), SSO_CLIENT_SECRET: str({ default: "" }), SSO_AUTHORIZATION_URL: str({ default: "" }), SSO_TOKEN_URL: str({ default: "" }), SSO_USERINFO_URL: str({ default: "" }), SSO_CALLBACK_URL: str({ default: "/v1/auth/sso/oauth/callback" }), // SAML specific SSO_ENTRY_POINT: str({ default: "" }), SSO_ISSUER: str({ default: "" }), SSO_CERT: str({ default: "" }), SSO_IDP_ISSUER: str({ default: "" }), SSO_IDP_METADATA_URL: str({ default: "" }), // URL to fetch IdP metadata XML (alternative to manual SSO_ENTRY_POINT/SSO_CERT) // SSO user settings SSO_DEFAULT_ROLE: str({ choices: ["", "user", "admin", "reviewer"], default: "reviewer" }), SSO_AUTO_CREATE_USERS: bool({ default: true }), SSO_ALLOW_ACCOUNT_LINKING: bool({ default: true }), // Plugin System SYNGRISI_PLUGINS_ENABLED: str({ default: "" }), // Comma-separated list of enabled plugins SYNGRISI_PLUGINS_DIR: str({ default: "" }), // Directory for external plugins // Okta Auth Plugin // Deprecated: Use SYNGRISI_PLUGIN_JWT_AUTH_* variables instead OKTA_JWKS_URL: str({ default: "" }), OKTA_ISSUER: str({ default: "" }), OKTA_SERVICE_USER_ROLE: str({ default: "" }), OKTA_AUTH_HEADER: str({ default: "" }), // Custom Check Validator Plugin CHECK_MISMATCH_THRESHOLD: str({ default: "0" }), // Mismatch % below which checks pass CHECK_VALIDATOR_SCRIPT: str({ default: "" }) // Path to custom validation script }); // src/server/data/devices.json var devices_default = [ { os: "ios", os_version: "16", device: "iPhone 14 Pro Max", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 14 Pro", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 14 Plus", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 14", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 12 Pro Max", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 12 Pro", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 12 Mini", realMobile: true }, { os: "ios", os_version: "16", device: "iPhone 11 Pro Max", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone XS", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 13 Pro Max", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 13 Pro", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 13 Mini", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 13", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 11 Pro", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 11", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone XS", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 12 Pro Max", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 12 Pro", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 12 Mini", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 12", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 11 Pro Max", realMobile: true }, { os: "ios", os_version: "14", device: "iPhone 11", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone XS", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone 11 Pro Max", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone 11 Pro", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone 11", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone XS", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone XS Max", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone XR", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone XR", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone X", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone 8", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone 8", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone 8", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone 8", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone 8 Plus", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone 8 Plus", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone 7", realMobile: true }, { os: "ios", os_version: "10", device: "iPhone 7", realMobile: true }, { os: "ios", os_version: "12", device: "iPhone 6S", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone 6S", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone 6S Plus", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone 6", realMobile: true }, { os: "ios", os_version: "15", device: "iPhone SE 2022", realMobile: true }, { os: "ios", os_version: "13", device: "iPhone SE 2020", realMobile: true }, { os: "ios", os_version: "11", device: "iPhone SE", realMobile: true }, { os: "ios", os_version: "14", device: "iPad Air 4", realMobile: true }, { os: "ios", os_version: "15", device: "iPad 9th", realMobile: true }, { os: "ios", os_version: "16", device: "iPad Pro 12.9 2022", realMobile: true }, { os: "ios", os_version: "16", device: "iPad Pro 12.9 2020", realMobile: true }, { os: "ios", os_version: "16", device: "iPad Pro 11 2022", realMobile: true }, { os: "ios", os_version: "16", device: "iPad 10th", realMobile: true }, { os: "ios", os_version: "15", device: "iPad Air 5", realMobile: true }, { os: "ios", os_version: "14", device: "iPad Pro 12.9 2021", realMobile: true }, { os: "ios", os_version: "14", device: "iPad Pro 12.9 2020", realMobile: true }, { os: "ios", os_version: "14", device: "iPad Pro 11 2021", realMobile: true }, { os: "ios", os_version: "13", device: "iPad Pro 12.9 2020", realMobile: true }, { os: "ios", os_version: "16", device: "iPad 8th", realMobile: true }, { os: "ios", os_version: "15", device: "iPad Pro 12.9 2018", realMobile: true }, { os: "ios", os_version: "15", device: "iPad Mini 2021", realMobile: true }, { os: "ios", os_version: "14", device: "iPad 8th", realMobile: true }, { os: "ios", os_version: "13", device: "iPad Pro 12.9 2018", realMobile: true }, { os: "ios", os_version: "13", device: "iPad Pro 11 2020", realMobile: true }, { os: "ios", os_version: "13", device: "iPad Mini 2019", realMobile: true }, { os: "ios", os_version: "13", device: "iPad Air 2019", realMobile: true }, { os: "ios", os_version: "13", device: "iPad 7th", realMobile: true }, { os: "ios", os_version: "12", device: "iPad Pro 12.9 2018", realMobile: true }, { os: "ios", os_version: "12", device: "iPad Pro 11 2018", realMobile: true }, { os: "ios", os_version: "12", device: "iPad Mini 2019", realMobile: true }, { os: "ios", os_version: "12", device: "iPad Air 2019", realMobile: true }, { os: "ios", os_version: "11", device: "iPad Pro 9.7 2016", realMobile: true }, { os: "ios", os_version: "11", device: "iPad Pro 12.9 2017", realMobile: true }, { os: "ios", os_version: "11", device: "iPad Mini 4", realMobile: true }, { os: "ios", os_version: "11", device: "iPad 6th", realMobile: true }, { os: "ios", os_version: "11", device: "iPad 5th", realMobile: true }, { os: "android", os_version: "12.0", device: "Samsung Galaxy S22 Ultra", realMobile: true }, { os: "android", os_version: "12.0", device: "Samsung Galaxy S22 Plus", realMobile: true }, { os: "android", os_version: "12.0", device: "Samsung Galaxy S22", realMobile: true }, { os: "android", os_version: "12.0", device: "Samsung Galaxy S21", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy S21 Ultra", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy S21", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy S21 Plus", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy S20", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy S20 Plus", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy S20 Ultra", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy M52", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy M32", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy A52", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy Note 20 Ultra", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy Note 20", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy A51", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy A11", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy S9 Plus", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy S10e", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy S10 Plus", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy S10", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy Note 10 Plus", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy Note 10", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy A10", realMobile: true }, { os: "android", os_version: "8.1", device: "Samsung Galaxy Note 9", realMobile: true }, { os: "android", os_version: "8.1", device: "Samsung Galaxy J7 Prime", realMobile: true }, { os: "android", os_version: "8.0", device: "Samsung Galaxy S9 Plus", realMobile: true }, { os: "android", os_version: "8.0", device: "Samsung Galaxy S9", realMobile: true }, { os: "android", os_version: "7.1", device: "Samsung Galaxy Note 8", realMobile: true }, { os: "android", os_version: "7.1", device: "Samsung Galaxy A8", realMobile: true }, { os: "android", os_version: "7.0", device: "Samsung Galaxy S8 Plus", realMobile: true }, { os: "android", os_version: "7.0", device: "Samsung Galaxy S8", realMobile: true }, { os: "android", os_version: "6.0", device: "Samsung Galaxy S7", realMobile: true }, { os: "android", os_version: "5.0", device: "Samsung Galaxy S6", realMobile: true }, { os: "android", os_version: "13.0", device: "Google Pixel 7 Pro", realMobile: true }, { os: "android", os_version: "13.0", device: "Google Pixel 7", realMobile: true }, { os: "android", os_version: "13.0", device: "Google Pixel 6 Pro", realMobile: true }, { os: "android", os_version: "12.0", device: "Google Pixel 6 Pro", realMobile: true }, { os: "android", os_version: "12.0", device: "Google Pixel 6", realMobile: true }, { os: "android", os_version: "12.0", device: "Google Pixel 5", realMobile: true }, { os: "android", os_version: "11.0", device: "Google Pixel 5", realMobile: true }, { os: "android", os_version: "11.0", device: "Google Pixel 4", realMobile: true }, { os: "android", os_version: "10.0", device: "Google Pixel 4 XL", realMobile: true }, { os: "android", os_version: "10.0", device: "Google Pixel 4", realMobile: true }, { os: "android", os_version: "10.0", device: "Google Pixel 3", realMobile: true }, { os: "android", os_version: "9.0", device: "Google Pixel 3a XL", realMobile: true }, { os: "android", os_version: "9.0", device: "Google Pixel 3a", realMobile: true }, { os: "android", os_version: "9.0", device: "Google Pixel 3 XL", realMobile: true }, { os: "android", os_version: "9.0", device: "Google Pixel 3", realMobile: true }, { os: "android", os_version: "9.0", device: "Google Pixel 2", realMobile: true }, { os: "android", os_version: "8.0", device: "Google Pixel 2", realMobile: true }, { os: "android", os_version: "7.1", device: "Google Pixel", realMobile: true }, { os: "android", os_version: "6.0", device: "Google Nexus 6", realMobile: true }, { os: "android", os_version: "4.4", device: "Google Nexus 5", realMobile: true }, { os: "android", os_version: "11.0", device: "OnePlus 9", realMobile: true }, { os: "android", os_version: "10.0", device: "OnePlus 8", realMobile: true }, { os: "android", os_version: "10.0", device: "OnePlus 7T", realMobile: true }, { os: "android", os_version: "9.0", device: "OnePlus 7", realMobile: true }, { os: "android", os_version: "9.0", device: "OnePlus 6T", realMobile: true }, { os: "android", os_version: "11.0", device: "Xiaomi Redmi Note 11", realMobile: true }, { os: "android", os_version: "10.0", device: "Xiaomi Redmi Note 9", realMobile: true }, { os: "android", os_version: "9.0", device: "Xiaomi Redmi Note 8", realMobile: true }, { os: "android", os_version: "9.0", device: "Xiaomi Redmi Note 7", realMobile: true }, { os: "android", os_version: "11.0", device: "Vivo Y21", realMobile: true }, { os: "android", os_version: "11.0", device: "Vivo V21", realMobile: true }, { os: "android", os_version: "10.0", device: "Vivo Y50", realMobile: true }, { os: "android", os_version: "11.0", device: "Oppo Reno 6", realMobile: true }, { os: "android", os_version: "11.0", device: "Oppo A96", realMobile: true }, { os: "android", os_version: "10.0", device: "Oppo Reno 3 Pro", realMobile: true }, { os: "android", os_version: "11.0", device: "Motorola Moto G71 5G", realMobile: true }, { os: "android", os_version: "10.0", device: "Motorola Moto G9 Play", realMobile: true }, { os: "android", os_version: "9.0", device: "Motorola Moto G7 Play", realMobile: true }, { os: "android", os_version: "9.0", device: "Huawei P30", realMobile: true }, { os: "android", os_version: "12.0", device: "Samsung Galaxy Tab S8", realMobile: true }, { os: "android", os_version: "11.0", device: "Samsung Galaxy Tab S7", realMobile: true }, { os: "android", os_version: "10.0", device: "Samsung Galaxy Tab S7", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy Tab S6", realMobile: true }, { os: "android", os_version: "9.0", device: "Samsung Galaxy Tab S5e", realMobile: true }, { os: "android", os_version: "8.1", device: "Samsung Galaxy Tab S4", realMobile: true } ]; // src/server/config.ts var getCommitHash = () => { if (gitHead) { return gitHead.substring(0, 7); } try { return execSync("git rev-parse --short HEAD", { encoding: "utf-8", stdio: ["pipe", "pipe", "pipe"] }).trim(); } catch { return ""; } }; var customDevicesPath = "./server/data/custom_devices.json"; var logsFolder = "./logs"; dotenv2.config(); var CURRENT_VERSION = version; var [major, minor] = CURRENT_VERSION.split(".").map(Number); var minSupportedMinor = Math.max(0, minor - 2); var MIN_SUPPORTED_SDK_VERSION = `${major}.${minSupportedMinor}.0`; var config = { version, commitHash: getCommitHash(), minSupportedSdkVersion: MIN_SUPPORTED_SDK_VERSION, apiVersion: "1", // this isn't used getDevices: async () => { if (fs.existsSync(customDevicesPath)) { return [...devices_default, ...(await import(customDevicesPath)).default]; } return devices_default; }, defaultImagesPath: env.SYNGRISI_IMAGES_PATH, domSnapshotsPath: env.SYNGRISI_DOM_SNAPSHOTS_PATH || env.SYNGRISI_IMAGES_PATH, connectionString: env.SYNGRISI_DB_URI || "mongodb://127.0.0.1:27017/SyngrisiDb", host: env.SYNGRISI_HOSTNAME, port: env.SYNGRISI_APP_PORT || 3e3, backupsFolder: "./backups", enableHttpLogger: env.SYNGRISI_HTTP_LOG, httpLoggerFilePath: `${logsFolder}/http.log`, storeSessionKey: env.SYNGRISI_SESSION_STORE_KEY || crypto2.randomBytes(64).toString("hex"), codeCoverage: env.SYNGRISI_COVERAGE, disableCors: env.SYNGRISI_DISABLE_DEV_CORS, fileUploadMaxSize: 50 * 1024 * 1024, adminDataJobsPath: env.SYNGRISI_ADMIN_DATA_JOBS_PATH, adminDataJobsTtlMs: env.SYNGRISI_ADMIN_DATA_JOBS_TTL_MS, adminDataMaxConcurrentJobs: env.SYNGRISI_ADMIN_DATA_MAX_CONCURRENT_JOBS, adminDataUploadMaxSize: env.SYNGRISI_ADMIN_DATA_UPLOAD_MAX_SIZE_MB * 1024 * 1024, testMode: env.SYNGRISI_TEST_MODE, jsonLimit: "50mb", tmpDir: env.SYNGRISI_TMP_DIR, helmet: { crossOriginEmbedderPolicy: false, crossOriginResourcePolicy: false, crossOriginOpenerPolicy: false, contentSecurityPolicy: { useDefaults: false, directives: { defaultSrc: ["'self'", "*", "'unsafe-inline'", "'unsafe-eval'", "data:", "blob:"], frameAncestors: ["'self'", "*"], frameSrc: ["'self'", "*"], scriptSrc: ["'self'", "*", "'unsafe-inline'", "'unsafe-eval'"], styleSrc: ["'self'", "*", "'unsafe-inline'"], imgSrc: ["'self'", "*", "data:", "blob:"], fontSrc: ["'self'", "*", "data:"], connectSrc: ["'self'", "*"], baseUri: ["'self'"], formAction: ["'self'"], objectSrc: ["'none'"], scriptSrcAttr: ["'none'"] } }, hsts: false }, rateLimit: { windowMs: env.SYNGRISI_RATE_LIMIT_WINDOW_MS, max: env.SYNGRISI_RATE_LIMIT_MAX, standardHeaders: true, legacyHeaders: false }, authRateLimit: { windowMs: env.SYNGRISI_AUTH_RATE_LIMIT_WINDOW_MS, max: env.SYNGRISI_AUTH_RATE_LIMIT_MAX, standardHeaders: true, legacyHeaders: false } }; if (!fs.existsSync(config.defaultImagesPath)) { fs.mkdirSync(config.defaultImagesPath, { recursive: true }); } if (config.domSnapshotsPath !== config.defaultImagesPath && !fs.existsSync(config.domSnapshotsPath)) { fs.mkdirSync(config.domSnapshotsPath, { recursive: true }); } if (!fs.existsSync(config.adminDataJobsPath)) { fs.mkdirSync(config.adminDataJobsPath, { recursive: true }); } if (!fs.existsSync(logsFolder)) { fs.mkdirSync(logsFolder, { recursive: true }); } // src/tasks/lib/dataBackupRestore.ts import fs2 from "fs"; import { promises as fsp } from "fs"; import path2 from "path"; import { createGzip, createGunzip } from "zlib"; import { promisify } from "util"; import { pipeline } from "stream"; import tar from "tar-stream"; import mongoose from "mongoose"; var pipelineAsync = promisify(pipeline); var { BSON } = mongoose.mongo; async function ensureDir(dirPath) { await fsp.mkdir(dirPath, { recursive: true }); } async function extractTarGzArchive(archivePath, destinationDir) { await ensureDir(destinationDir); const extract = tar.extract(); await new Promise((resolve, reject) => { extract.on("entry", (header, stream, next) => { const outputPath = path2.join(destinationDir, header.name); const finish = (error) => { if (error) { reject(error); return; } next(); }; if (header.type === "directory") { void ensureDir(outputPath).then(() => { stream.resume(); finish(); }).catch((error) => finish(error)); return; } void ensureDir(path2.dirname(outputPath)).then(() => pipelineAsync(stream, fs2.createWriteStream(outputPath))).then(() => finish()).catch((error) => finish(error)); }); extract.on("finish", () => resolve()); extract.on("error", reject); fs2.createReadStream(archivePath).on("error", reject).pipe(createGunzip()).on("error", reject).pipe(extract).on("error", reject); }); } async function importCollectionDump(connection, collectionName, dumpPath) { const db = connection.db; if (!db) { throw new Error("MongoDB connection is not available"); } const collection = db.collection(collectionName); const batch = []; const flush = async () => { if (batch.length === 0) return; await collection.insertMany(batch, { ordered: false }); batch.length = 0; }; const input2 = fs2.createReadStream(dumpPath).pipe(createGunzip()); let pending = Buffer.alloc(0); for await (const chunk of input2) { pending = Buffer.concat([pending, chunk]); while (pending.length >= 4) { const documentSize = pending.readInt32LE(0); if (documentSize <= 0) { throw new Error(`Invalid BSON document size ${documentSize} in ${collectionName}`); } if (pending.length < documentSize) { break; } const documentBuffer = pending.subarray(0, documentSize); pending = pending.subarray(documentSize); batch.push(BSON.deserialize(documentBuffer)); if (batch.length >= 1e3) { await flush(); } } } if (pending.length > 0) { throw new Error(`Unexpected trailing BSON bytes in ${collectionName}`); } await flush(); } async function recreateIndexes(connection, collectionName, indexes) { const db = connection.db; if (!db) { throw new Error("MongoDB connection is not available"); } const filtered = indexes.filter((index) => index.name !== "_id_"); if (filtered.length === 0) { return; } const definitions = filtered.map((index) => { const { key, ...options } = index; return { key, ...options }; }); await db.collection(collectionName).createIndexes(definitions); } async function connectToMongo(connectionString) { const connection = await mongoose.createConnection(connectionString).asPromise(); return connection; } async function restoreDatabaseBackupArchive(connectionString, archivePath) { const connection = await connectToMongo(connectionString); const extractDir = path2.join(path2.dirname(archivePath), `restore-${Date.now()}`); try { await extractTarGzArchive(archivePath, extractDir); const manifestPath = path2.join(extractDir, "manifest.json"); const manifest = JSON.parse(await fsp.readFile(manifestPath, "utf8")); if (manifest.format !== "syngrisi-db-backup-v1") { throw new Error("Unsupported database backup format"); } const db = connection.db; if (!db) { throw new Error("MongoDB connection is not available"); } await db.dropDatabase(); for (const collectionInfo of manifest.collections) { const dumpPath = path2.join(extractDir, "collections", collectionInfo.dumpFile); await importCollectionDump(connection, collectionInfo.name, dumpPath); await recreateIndexes(connection, collectionInfo.name, collectionInfo.indexes); } } finally { await connection.close(); await fsp.rm(extractDir, { recursive: true, force: true }); } } async function restoreScreenshotsArchive(archivePath, destinationPath, options = {}) { const { skipExisting = false } = options; const extract = tar.extract(); await new Promise((resolve, reject) => { extract.on("entry", (header, stream, next) => { const outputPath = path2.join(destinationPath, header.name); const finish = (error) => { if (error) { reject(error); return; } next(); }; if (header.type !== "file") { stream.resume(); finish(); return; } void (async () => { await ensureDir(path2.dirname(outputPath)); if (skipExisting) { try { await fsp.access(outputPath); stream.resume(); return; } catch { } } else { await fsp.rm(outputPath, { force: true }); } await pipelineAsync(stream, fs2.createWriteStream(outputPath)); })().then(() => finish()).catch((error) => finish(error)); }); extract.on("finish", () => resolve()); extract.on("error", reject); fs2.createReadStream(archivePath).on("error", reject).pipe(createGunzip()).on("error", reject).pipe(extract).on("error", reject); }); } // src/tasks/restore.ts var run = async () => { const backupFolder = config.backupsFolder; if (!fs3.existsSync(backupFolder)) { fs3.mkdirSync(backupFolder, { recursive: true }); } console.log("Be sure that application is down before restoring data."); const backupsFolders = fs3.readdirSync(backupFolder, { withFileTypes: true }).filter((x) => x.isDirectory()).map((x) => ({ name: x.name, value: x.name })); const answers = { backupSubFolder: await select({ message: "Enter the Backup Folder name Filename", choices: backupsFolders }), destConnectionString: await input({ message: "Enter the Destination Database Connection String URI", default: config.connectionString }), destImagesSubFolder: await input({ message: "Enter the Images Folder Path", default: config.defaultImagesPath }), confirm: await confirm({ message: "\u26A0\uFE0F Caution! All current Application Data will be removed, before the Restoring! Continue?", default: false }) }; if (!answers.confirm) { return "Skipped"; } const fullBackupPath = path3.join(backupFolder, answers.backupSubFolder); console.log({ fullBackupPath }); const fullSourceDatabasePath = path3.join(fullBackupPath, "database.tar.gz"); if (!fs3.existsSync(fullSourceDatabasePath)) { console.log("The Source Database Folder is not exists, please select tha another folder"); return; } const fullSourceImagesPath = path3.join(fullBackupPath, "images.tar.gz"); if (!fs3.existsSync(fullSourceImagesPath)) { console.log("The Source Images Folder is not exists, please select tha another folder"); return; } if (!fs3.existsSync(answers.destImagesSubFolder)) { console.log("The Destination Images Folder is not exists, please select tha another folder"); return; } console.log("Restore the Database"); await restoreDatabaseBackupArchive(answers.destConnectionString, fullSourceDatabasePath); console.log("Database restore completed"); console.log("Restore the Images"); await restoreScreenshotsArchive(fullSourceImagesPath, answers.destImagesSubFolder, { skipExisting: false }); console.log("Images restore completed"); return "\u2705 Success"; }; run().then((result) => console.log(`operation complete: ${result}`)); //# sourceMappingURL=restore.js.map