@automattic/vip 4.1.6 → 4.1.8

package/dist/commands/export-sql.js

@@ -1,30 +1,62 @@
 "use strict";
-
-exports.__esModule = true;
-exports.GENERATE_DOWNLOAD_LINK_MUTATION = exports.ExportSQLCommand = exports.CREATE_EXPORT_JOB_MUTATION = void 0;
-var _chalk = _interopRequireDefault(require("chalk"));
-var _fs = _interopRequireDefault(require("fs"));
-var _graphqlTag = _interopRequireDefault(require("graphql-tag"));
-var _path = _interopRequireDefault(require("path"));
-var _backupDb = require("./backup-db");
-var _api = _interopRequireWildcard(require("../lib/api"));
-var _backupStorageAvailability = require("../lib/backup-storage-availability/backup-storage-availability");
-var exit = _interopRequireWildcard(require("../lib/cli/exit"));
-var _format = require("../lib/cli/format");
-var _progress = require("../lib/cli/progress");
-var _downloadFile = require("../lib/http/download-file");
-var _liveBackupCopy = _interopRequireWildcard(require("../lib/live-backup-copy"));
-var liveBackupCopy = _liveBackupCopy;
-var _retry = require("../lib/retry");
-var _userError = _interopRequireDefault(require("../lib/user-error"));
-var _utils = require("../lib/utils");
-function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function (e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (const t in e) "default" !== t && {}.hasOwnProperty.call(e, t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, t)) && (i.get || i.set) ? o(f, t, i) : f[t] = e[t]); return f; })(e, t); }
-function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+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.ExportSQLCommand = exports.CREATE_EXPORT_JOB_MUTATION = exports.GENERATE_DOWNLOAD_LINK_MUTATION = void 0;
+const chalk_1 = __importDefault(require("chalk"));
+const fs_1 = __importDefault(require("fs"));
+const graphql_tag_1 = __importDefault(require("graphql-tag"));
+const path_1 = __importDefault(require("path"));
+const backup_db_1 = require("./backup-db");
+const api_1 = __importStar(require("../lib/api"));
+const backup_storage_availability_1 = require("../lib/backup-storage-availability/backup-storage-availability");
+const exit = __importStar(require("../lib/cli/exit"));
+const format_1 = require("../lib/cli/format");
+const progress_1 = require("../lib/cli/progress");
+const download_file_1 = require("../lib/http/download-file");
+const liveBackupCopy = __importStar(require("../lib/live-backup-copy"));
+const live_backup_copy_1 = require("../lib/live-backup-copy");
+const retry_1 = require("../lib/retry");
+const user_error_1 = __importDefault(require("../lib/user-error"));
+const utils_1 = require("../lib/utils");
 const EXPORT_SQL_PROGRESS_POLL_INTERVAL = 1000;
-
 // Maximum allowed size for a live backup copy configuration file (500 KB).
 const MAX_LIVE_BACKUP_CONFIG_SIZE_BYTES = 500 * 1024;
-const BACKUP_AND_JOB_STATUS_QUERY = (0, _graphqlTag.default)`
+const BACKUP_AND_JOB_STATUS_QUERY = (0, graphql_tag_1.default) `
 	query AppBackupAndJobStatus($appId: Int!, $envId: Int!) {
 		app(id: $appId) {
 			id
@@ -63,9 +95,8 @@ const BACKUP_AND_JOB_STATUS_QUERY = (0, _graphqlTag.default)`
 		}
 	}
 `;
-
 // Exporting for test purposes
-const GENERATE_DOWNLOAD_LINK_MUTATION = exports.GENERATE_DOWNLOAD_LINK_MUTATION = (0, _graphqlTag.default)`
+exports.GENERATE_DOWNLOAD_LINK_MUTATION = (0, graphql_tag_1.default) `
 	mutation GenerateDBBackupCopyUrl($input: AppEnvironmentGenerateDBBackupCopyUrlInput) {
 		generateDBBackupCopyUrl(input: $input) {
 			url
@@ -73,9 +104,8 @@ const GENERATE_DOWNLOAD_LINK_MUTATION = exports.GENERATE_DOWNLOAD_LINK_MUTATION
 		}
 	}
 `;
-
 // Exporting for test purposes
-const CREATE_EXPORT_JOB_MUTATION = exports.CREATE_EXPORT_JOB_MUTATION = (0, _graphqlTag.default)`
+exports.CREATE_EXPORT_JOB_MUTATION = (0, graphql_tag_1.default) `
 	mutation BackupDBCopy($input: AppEnvironmentStartDBBackupCopyInput) {
 		startDBBackupCopy(input: $input) {
 			message
@@ -83,7 +113,6 @@ const CREATE_EXPORT_JOB_MUTATION = exports.CREATE_EXPORT_JOB_MUTATION = (0, _gra
 		}
 	}
 `;
-
 /**
  * Fetches the latest backup and job status for an environment
  *
@@ -92,37 +121,30 @@ const CREATE_EXPORT_JOB_MUTATION = exports.CREATE_EXPORT_JOB_MUTATION = (0, _gra
  * @return {Promise} A promise which resolves to the latest backup and job status
  */
 async function fetchLatestBackupAndJobStatusBase(appId, envId) {
-  const api = (0, _api.default)();
-  const response = await api.query({
-    query: BACKUP_AND_JOB_STATUS_QUERY,
-    variables: {
-      appId,
-      envId
-    },
-    fetchPolicy: 'network-only'
-  });
-  const environments = response.data?.app?.environments;
-  const firstEnvironment = environments?.[0];
-  const envSqlDumpTool = firstEnvironment?.backupsSqlDumpTool;
-  const latestBackup = firstEnvironment?.latestBackup ?? undefined;
-  const jobs = firstEnvironment?.jobs || [];
-  return {
-    latestBackup,
-    jobs,
-    envSqlDumpTool
-  };
+    const api = (0, api_1.default)();
+    const response = await api.query({
+        query: BACKUP_AND_JOB_STATUS_QUERY,
+        variables: { appId, envId },
+        fetchPolicy: 'network-only',
+    });
+    const environments = response.data?.app?.environments;
+    const firstEnvironment = environments?.[0];
+    const envSqlDumpTool = firstEnvironment?.backupsSqlDumpTool;
+    const latestBackup = firstEnvironment?.latestBackup ?? undefined;
+    const jobs = (firstEnvironment?.jobs || []);
+    return { latestBackup, jobs, envSqlDumpTool };
 }
 async function fetchLatestBackupAndJobStatus(appId, envId) {
-  if (!appId || !envId) {
-    throw new Error('App ID and Env ID missing');
-  }
-  return await (0, _retry.retry)({
-    retryOnlyIf: options => {
-      return (options.error.message || '').indexOf('Unexpected token < in JSON at position 0') !== -1;
+    if (!appId || !envId) {
+        throw new Error('App ID and Env ID missing');
     }
-  }, () => fetchLatestBackupAndJobStatusBase(appId, envId));
+    return await (0, retry_1.retry)({
+        retryOnlyIf: options => {
+            return ((options.error.message || '').indexOf('Unexpected token < in JSON at position 0') !==
+                -1);
+        },
+    }, () => fetchLatestBackupAndJobStatusBase(appId, envId));
 }
-
 /**
  * Generates a download link for a backup
  *
@@ -132,23 +154,22 @@ async function fetchLatestBackupAndJobStatus(appId, envId) {
  * @return {Promise} A promise which resolves to the download link
  */
 async function generateDownloadLink(appId, envId, backupId) {
-  if (!appId || !envId || !backupId) {
-    throw new Error('generateDownloadLink: A parameter is missing');
-  }
-  const api = (0, _api.default)();
-  const response = await api.mutate({
-    mutation: GENERATE_DOWNLOAD_LINK_MUTATION,
-    variables: {
-      input: {
-        id: appId,
-        environmentId: envId,
-        backupId
-      }
+    if (!appId || !envId || !backupId) {
+        throw new Error('generateDownloadLink: A parameter is missing');
     }
-  });
-  return response.data?.generateDBBackupCopyUrl?.url ?? '';
+    const api = (0, api_1.default)();
+    const response = await api.mutate({
+        mutation: exports.GENERATE_DOWNLOAD_LINK_MUTATION,
+        variables: {
+            input: {
+                id: appId,
+                environmentId: envId,
+                backupId,
+            },
+        },
+    });
+    return response.data?.generateDBBackupCopyUrl?.url ?? '';
 }
-
 /**
  * Creates an export job for a backup
  *
@@ -159,381 +180,376 @@ async function generateDownloadLink(appId, envId, backupId) {
  * @throws {Error} Throws an error if the job creation fails
  */
 async function createExportJob(appId, envId, backupId) {
-  if (!appId || !envId || !backupId) {
-    throw new Error('createExportJob: Some fields are undefined');
-  }
-
-  // Disable global error handling so that we can handle errors ourselves
-  (0, _api.disableGlobalGraphQLErrorHandling)();
-  const api = (0, _api.default)();
-  await api.mutate({
-    mutation: CREATE_EXPORT_JOB_MUTATION,
-    variables: {
-      input: {
-        id: appId,
-        environmentId: envId,
-        backupId
-      }
+    if (!appId || !envId || !backupId) {
+        throw new Error('createExportJob: Some fields are undefined');
     }
-  });
-
-  // Re-enable global error handling
-  (0, _api.enableGlobalGraphQLErrorHandling)();
+    // Disable global error handling so that we can handle errors ourselves
+    (0, api_1.disableGlobalGraphQLErrorHandling)();
+    const api = (0, api_1.default)();
+    await api.mutate({
+        mutation: exports.CREATE_EXPORT_JOB_MUTATION,
+        variables: {
+            input: {
+                id: appId,
+                environmentId: envId,
+                backupId,
+            },
+        },
+    });
+    // Re-enable global error handling
+    (0, api_1.enableGlobalGraphQLErrorHandling)();
 }
 /**
  * Class representing an export command workflow
  */
 class ExportSQLCommand {
-  progressTracker;
-  outputFile;
-  generateBackup;
-  confirmEnoughStorageHook;
-  steps = {
-    PREPARE: 'prepare',
-    CREATE: 'create',
-    DOWNLOAD_LINK: 'downloadLink',
-    CONFIRM_ENOUGH_STORAGE: 'confirmEnoughStorage',
-    DOWNLOAD: 'download'
-  };
-  track;
-  liveBackupCopyCLIOptions;
-  showMyDumperWarning;
-  skipDownload;
-
-  /**
-   * Creates an instance of SQLExportCommand
-   *
-   * @param {any}      app        The application object
-   * @param {any}      env        The environment object
-   * @param {object}   options 		The optional parameters
-   * @param {Function} trackerFn  The progress tracker function
-   */
-  constructor(app, env, options = {}, trackerFn = () => {}) {
-    this.app = app;
-    this.env = env;
-    this.outputFile = typeof options.outputFile === 'string' ? (0, _utils.getAbsolutePath)(options.outputFile) : null;
-    this.confirmEnoughStorageHook = options.confirmEnoughStorageHook;
-    this.generateBackup = options.generateBackup || false;
-    this.progressTracker = new _progress.ProgressTracker([{
-      id: this.steps.PREPARE,
-      name: 'Preparing for backup download'
-    }, {
-      id: this.steps.CREATE,
-      name: 'Creating backup copy'
-    }, {
-      id: this.steps.DOWNLOAD_LINK,
-      name: 'Requesting download link'
-    }, {
-      id: this.steps.CONFIRM_ENOUGH_STORAGE,
-      name: "Checking if there's enough storage"
-    }, {
-      id: this.steps.DOWNLOAD,
-      name: 'Downloading file'
-    }]);
-    this.track = trackerFn;
-    this.liveBackupCopyCLIOptions = options.liveBackupCopyCLIOptions;
-    this.showMyDumperWarning = options.showMyDumperWarning === undefined ? true : options.showMyDumperWarning;
-    this.skipDownload = options.skipDownload || false;
-  }
-
-  /**
-   * Fetches the export job of the latest backup
-   *
-   * @return {Promise} A promise which resolves to the export job
-   */
-  async getExportJob() {
-    const {
-      latestBackup,
-      jobs
-    } = await fetchLatestBackupAndJobStatus(this.app.id, this.env.id);
-    if (!latestBackup) {
-      return undefined;
-    }
-
-    // Find the job that generates the export for the latest backup
-    return jobs.find(job => {
-      const metadata = (job.metadata || []).find(md => md?.name === 'backupId');
-      return metadata && parseInt(metadata.value, 10) === latestBackup.id;
-    });
-  }
-
-  /**
-   * Fetches the S3 filename of the exported backup
-   *
-   * @return A promise which resolves to the filename
-   */
-  async getExportedFileName() {
-    const job = await this.getExportJob();
-    if (!job) {
-      throw new Error('Job not found');
-    }
-    const metadata = job.metadata?.find(md => md?.name === 'uploadPath');
-    return metadata?.value?.split('/')[1];
-  }
-
-  /**
-   * Checks if the export job's preflight step is successful
-   *
-   * @param job The export job
-   * @return True if the preflight step is successful
-   */
-  isPrepared(job) {
-    const step = job?.progress?.steps?.find(st => st?.id === 'preflight');
-    return step?.status === 'success';
-  }
-
-  /**
-   * Checks if the export job's S3 upload step is successful
-   *
-   * @param job The export job
-   * @return True if the upload step is successful
-   */
-  isCreated(job) {
-    const step = job?.progress?.steps?.find(st => st?.id === 'upload_backup');
-    return step?.status === 'success';
-  }
-
-  /**
-   * Stops the progress tracker
-   */
-  stopProgressTracker() {
-    this.progressTracker.print();
-    this.progressTracker.stopPrinting();
-  }
-  async runBackupJob() {
-    const cmd = new _backupDb.BackupDBCommand(this.app, this.env);
-    let noticeMessage = `\n${_chalk.default.yellow('NOTICE: ')}`;
-    noticeMessage += 'If a recent database backup does not exist, a new one will be generated for this environment. ';
-    noticeMessage += 'Learn more about this: https://docs.wpvip.com/databases/backups/download-a-full-database-backup/ \n';
-    console.log(noticeMessage);
-    await cmd.run(false);
-  }
-  async confirmEnoughStorage(archiveSize) {
-    if (this.confirmEnoughStorageHook) {
-      return await this.confirmEnoughStorageHook(archiveSize);
+    app;
+    env;
+    progressTracker;
+    outputFile;
+    generateBackup;
+    confirmEnoughStorageHook;
+    steps = {
+        PREPARE: 'prepare',
+        CREATE: 'create',
+        DOWNLOAD_LINK: 'downloadLink',
+        CONFIRM_ENOUGH_STORAGE: 'confirmEnoughStorage',
+        DOWNLOAD: 'download',
+    };
+    track;
+    liveBackupCopyCLIOptions;
+    showMyDumperWarning;
+    skipDownload;
+    /**
+     * Creates an instance of SQLExportCommand
+     *
+     * @param {any}      app        The application object
+     * @param {any}      env        The environment object
+     * @param {object}   options 		The optional parameters
+     * @param {Function} trackerFn  The progress tracker function
+     */
+    constructor(app, env, options = {}, trackerFn = () => { }) {
+        this.app = app;
+        this.env = env;
+        this.outputFile =
+            typeof options.outputFile === 'string' ? (0, utils_1.getAbsolutePath)(options.outputFile) : null;
+        this.confirmEnoughStorageHook = options.confirmEnoughStorageHook;
+        this.generateBackup = options.generateBackup || false;
+        this.progressTracker = new progress_1.ProgressTracker([
+            { id: this.steps.PREPARE, name: 'Preparing for backup download' },
+            { id: this.steps.CREATE, name: 'Creating backup copy' },
+            { id: this.steps.DOWNLOAD_LINK, name: 'Requesting download link' },
+            { id: this.steps.CONFIRM_ENOUGH_STORAGE, name: "Checking if there's enough storage" },
+            { id: this.steps.DOWNLOAD, name: 'Downloading file' },
+        ]);
+        this.track = trackerFn;
+        this.liveBackupCopyCLIOptions = options.liveBackupCopyCLIOptions;
+        this.showMyDumperWarning =
+            options.showMyDumperWarning === undefined ? true : options.showMyDumperWarning;
+        this.skipDownload = options.skipDownload || false;
     }
-    const storageAvailability = new _backupStorageAvailability.BackupStorageAvailability();
-    return await storageAvailability.validateAndPromptDiskSpaceWarningForBackupImport(archiveSize);
-  }
-
-  /**
-   * Sequentially runs the steps of the export workflow
-   *
-   */
-  async run() {
-    if (this.outputFile) {
-      try {
-        _fs.default.accessSync(_path.default.parse(this.outputFile).dir, _fs.default.constants.W_OK);
-      } catch (err) {
-        const error = err;
-        await this.track('error', {
-          error_type: 'cannot_write_to_path',
-          error_message: `Cannot write to the specified path: ${error?.message}`,
-          stack: error?.stack
+    /**
+     * Fetches the export job of the latest backup
+     *
+     * @return {Promise} A promise which resolves to the export job
+     */
+    async getExportJob() {
+        const { latestBackup, jobs } = await fetchLatestBackupAndJobStatus(this.app.id, this.env.id);
+        if (!latestBackup) {
+            return undefined;
+        }
+        // Find the job that generates the export for the latest backup
+        return jobs.find(job => {
+            const metadata = (job.metadata || []).find(md => md?.name === 'backupId');
+            return metadata && parseInt(metadata.value, 10) === latestBackup.id;
         });
-        exit.withError(`Cannot write to the specified path: ${error?.message}`);
-      }
     }
-    const filename = this.outputFile || 'exported.sql.gz';
-    this.progressTracker.stepRunning(this.steps.PREPARE);
-    this.progressTracker.startPrinting();
-    let url;
-    let size;
-    if (this.liveBackupCopyCLIOptions?.useLiveBackupCopy) {
-      const result = await this.generateLiveBackupCopy();
-      url = result.url;
-      size = Number(result.size);
-    } else {
-      url = await this.runBackup();
-      const exportJob = await this.getExportJob();
-      if (!exportJob) {
-        throw new Error('Export job not found');
-      }
-      const bytesWrittenMeta = exportJob.metadata?.find(meta => meta?.name === 'bytesWritten');
-      if (!bytesWrittenMeta?.value) {
-        throw new Error('Export job metadata does not contain bytesWritten');
-      }
-      size = Number(bytesWrittenMeta.value);
-    }
-    if (this.skipDownload) {
-      this.progressTracker.stepSkipped(this.steps.CONFIRM_ENOUGH_STORAGE);
-      this.progressTracker.stepSkipped(this.steps.DOWNLOAD);
-      this.progressTracker.print();
-      this.progressTracker.stopPrinting();
-      return;
-    }
-    const storageConfirmed = await this.progressTracker.handleContinuePrompt(async setPromptShown => {
-      const status = await this.confirmEnoughStorage(Number(size));
-      if (status.isPromptShown) {
-        setPromptShown();
-      }
-      return status.continue;
-    });
-    if (storageConfirmed) {
-      this.progressTracker.stepSuccess(this.steps.CONFIRM_ENOUGH_STORAGE);
-    } else {
-      this.progressTracker.stepFailed(this.steps.CONFIRM_ENOUGH_STORAGE);
-      this.stopProgressTracker();
-      exit.withError('Command canceled by user.');
-    }
-
-    // The export file is prepared. Let's download it
-    try {
-      const onProgressCallback = (current, total) => {
-        if (total) {
-          this.progressTracker.setProgress(`- ${(100 * current / total).toFixed(2)}% (${(0, _format.formatBytes)(current)}/${(0, _format.formatBytes)(total)})`);
+    /**
+     * Fetches the S3 filename of the exported backup
+     *
+     * @return A promise which resolves to the filename
+     */
+    async getExportedFileName() {
+        const job = await this.getExportJob();
+        if (!job) {
+            throw new Error('Job not found');
         }
-      };
-      await (0, _downloadFile.downloadFile)(url, filename, onProgressCallback);
-      this.progressTracker.stepSuccess(this.steps.DOWNLOAD);
-      this.stopProgressTracker();
-      console.log(`File saved to ${filename}`);
-    } catch (err) {
-      const error = err;
-      this.progressTracker.stepFailed(this.steps.DOWNLOAD);
-      this.stopProgressTracker();
-      await this.track('error', {
-        error_type: 'download_failed',
-        error_message: error.message,
-        stack: error.stack
-      });
-      exit.withError(`Error downloading exported file: ${error.message}`);
-    }
-  }
-  generateDownloadURLOutputString(url) {
-    return `${_chalk.default.green('Download URL')}: ${url}`;
-  }
-  async runBackup() {
-    if (this.generateBackup) {
-      await this.runBackupJob();
+        const metadata = job.metadata?.find(md => md?.name === 'uploadPath');
+        return metadata?.value?.split('/')[1];
     }
-    const {
-      latestBackup,
-      envSqlDumpTool
-    } = await fetchLatestBackupAndJobStatus(this.app.id, this.env.id);
-    if (!latestBackup) {
-      await this.track('error', {
-        error_type: 'no_backup_found',
-        error_message: 'No backup found for the site'
-      });
-      exit.withError(`No backup found for site ${this.app.name}`);
+    /**
+     * Checks if the export job's preflight step is successful
+     *
+     * @param job The export job
+     * @return True if the preflight step is successful
+     */
+    isPrepared(job) {
+        const step = job?.progress?.steps?.find(st => st?.id === 'preflight');
+        return step?.status === 'success';
     }
-    const prepareAdditionalInfo = [];
-    const showMyDumperWarning = this.showMyDumperWarning && (latestBackup.sqlDumpTool ?? envSqlDumpTool) === 'mydumper';
-    if (showMyDumperWarning) {
-      prepareAdditionalInfo.push(`${_chalk.default.yellow.bold('WARNING:')} This is a large or complex database. The backup file for this database is generated with MyDumper. The file can only be loaded with MyLoader. For more information: https://github.com/mydumper/mydumper`);
+    /**
+     * Checks if the export job's S3 upload step is successful
+     *
+     * @param job The export job
+     * @return True if the upload step is successful
+     */
+    isCreated(job) {
+        const step = job?.progress?.steps?.find(st => st?.id === 'upload_backup');
+        return step?.status === 'success';
     }
-    if (await this.getExportJob()) {
-      prepareAdditionalInfo.push(`Attaching to an existing export for the backup with timestamp ${latestBackup.createdAt}`);
-    } else {
-      prepareAdditionalInfo.push(`Exporting database backup with timestamp ${latestBackup.createdAt}`);
-      try {
-        await createExportJob(this.app.id, this.env.id, latestBackup.id);
-      } catch (err) {
-        const error = err;
-        // Todo: match error code instead of message substring
-        if (error.message.includes('Backup Copy already in progress')) {
-          await this.track('error', {
-            error_type: 'job_already_running',
-            error_message: error.message,
-            stack: error.stack
-          });
-          exit.withError('There is an export job already running for this environment: ' + `https://dashboard.wpvip.com/apps/${this.app.id}/${this.env.uniqueLabel}/database/backups\n` + 'Currently, we allow only one export job at a time, per site. Please try again later.');
-        } else {
-          await this.track('error', {
-            error_type: 'create_export_job',
-            error_message: error.message,
-            stack: error.stack
-          });
-        }
-        exit.withError(`Error creating export job: ${error.message}`);
-      }
+    /**
+     * Stops the progress tracker
+     */
+    stopProgressTracker() {
+        this.progressTracker.print();
+        this.progressTracker.stopPrinting();
     }
-    await (0, _utils.pollUntil)(this.getExportJob.bind(this), EXPORT_SQL_PROGRESS_POLL_INTERVAL, this.isPrepared.bind(this));
-    this.progressTracker.stepSuccess(this.steps.PREPARE, prepareAdditionalInfo);
-    await (0, _utils.pollUntil)(this.getExportJob.bind(this), EXPORT_SQL_PROGRESS_POLL_INTERVAL, this.isCreated.bind(this));
-    this.progressTracker.stepSuccess(this.steps.CREATE);
-    const url = await generateDownloadLink(this.app.id, this.env.id, latestBackup.id);
-    this.progressTracker.stepSuccess(this.steps.DOWNLOAD_LINK, [this.generateDownloadURLOutputString(url)]);
-    return url;
-  }
-  async generateLiveBackupCopy() {
-    if (!this.liveBackupCopyCLIOptions) {
-      throw new _userError.default('Configuration file is required for live backup copy.');
+    async runBackupJob() {
+        const cmd = new backup_db_1.BackupDBCommand(this.app, this.env);
+        let noticeMessage = `\n${chalk_1.default.yellow('NOTICE: ')}`;
+        noticeMessage +=
+            'If a recent database backup does not exist, a new one will be generated for this environment. ';
+        noticeMessage +=
+            'Learn more about this: https://docs.wpvip.com/databases/backups/download-a-full-database-backup/ \n';
+        console.log(noticeMessage);
+        await cmd.run(false);
     }
-    if (!this.app.id || !this.env.id) {
-      throw new _userError.default('App ID and Environment ID are required to start live backup copy.');
+    async confirmEnoughStorage(archiveSize) {
+        if (this.confirmEnoughStorageHook) {
+            return await this.confirmEnoughStorageHook(archiveSize);
+        }
+        const storageAvailability = new backup_storage_availability_1.BackupStorageAvailability();
+        return await storageAvailability.validateAndPromptDiskSpaceWarningForBackupImport(archiveSize);
     }
-    const config = this.getLiveBackupConfigFromCLIOptions();
-    try {
-      const copyId = await liveBackupCopy.startLiveBackupCopy({
-        appId: this.app.id,
-        environmentId: this.env.id,
-        config
-      });
-      this.progressTracker.stepSuccess(this.steps.PREPARE);
-      this.progressTracker.stepRunning(this.steps.CREATE);
-      const result = await liveBackupCopy.getDownloadURL({
-        appId: this.app.id,
-        environmentId: this.env.id,
-        copyId
-      });
-      this.progressTracker.stepSuccess(this.steps.CREATE);
-      this.progressTracker.stepSuccess(this.steps.DOWNLOAD_LINK, [this.generateDownloadURLOutputString(result.url)]);
-      return result;
-    } catch (err) {
-      const message = (0, _utils.parseApiError)(err) ?? 'Unknown error';
-      await this.track('error', {
-        error_type: 'live_backup_copy',
-        error_message: message,
-        stack: err instanceof Error ? err.stack : undefined
-      });
-      exit.withError(`Error creating live backup copy: ${message}`);
+    /**
+     * Sequentially runs the steps of the export workflow
+     *
+     */
+    async run() {
+        if (this.outputFile) {
+            try {
+                fs_1.default.accessSync(path_1.default.parse(this.outputFile).dir, fs_1.default.constants.W_OK);
+            }
+            catch (err) {
+                const error = err;
+                await this.track('error', {
+                    error_type: 'cannot_write_to_path',
+                    error_message: `Cannot write to the specified path: ${error?.message}`,
+                    stack: error?.stack,
+                });
+                exit.withError(`Cannot write to the specified path: ${error?.message}`);
+            }
+        }
+        const filename = this.outputFile || 'exported.sql.gz';
+        this.progressTracker.stepRunning(this.steps.PREPARE);
+        this.progressTracker.startPrinting();
+        let url;
+        let size;
+        if (this.liveBackupCopyCLIOptions?.useLiveBackupCopy) {
+            const result = await this.generateLiveBackupCopy();
+            url = result.url;
+            size = Number(result.size);
+        }
+        else {
+            url = await this.runBackup();
+            const exportJob = await this.getExportJob();
+            if (!exportJob) {
+                throw new Error('Export job not found');
+            }
+            const bytesWrittenMeta = exportJob.metadata?.find(meta => meta?.name === 'bytesWritten');
+            if (!bytesWrittenMeta?.value) {
+                throw new Error('Export job metadata does not contain bytesWritten');
+            }
+            size = Number(bytesWrittenMeta.value);
+        }
+        if (this.skipDownload) {
+            this.progressTracker.stepSkipped(this.steps.CONFIRM_ENOUGH_STORAGE);
+            this.progressTracker.stepSkipped(this.steps.DOWNLOAD);
+            this.progressTracker.print();
+            this.progressTracker.stopPrinting();
+            return;
+        }
+        const storageConfirmed = await this.progressTracker.handleContinuePrompt(async (setPromptShown) => {
+            const status = await this.confirmEnoughStorage(Number(size));
+            if (status.isPromptShown) {
+                setPromptShown();
+            }
+            return status.continue;
+        });
+        if (storageConfirmed) {
+            this.progressTracker.stepSuccess(this.steps.CONFIRM_ENOUGH_STORAGE);
+        }
+        else {
+            this.progressTracker.stepFailed(this.steps.CONFIRM_ENOUGH_STORAGE);
+            this.stopProgressTracker();
+            exit.withError('Command canceled by user.');
+        }
+        // The export file is prepared. Let's download it
+        try {
+            const onProgressCallback = (current, total) => {
+                if (total) {
+                    this.progressTracker.setProgress(`- ${((100 * current) / total).toFixed(2)}% (${(0, format_1.formatBytes)(current)}/${(0, format_1.formatBytes)(total)})`);
+                }
+            };
+            await (0, download_file_1.downloadFile)(url, filename, onProgressCallback);
+            this.progressTracker.stepSuccess(this.steps.DOWNLOAD);
+            this.stopProgressTracker();
+            console.log(`File saved to ${filename}`);
+        }
+        catch (err) {
+            const error = err;
+            this.progressTracker.stepFailed(this.steps.DOWNLOAD);
+            this.stopProgressTracker();
+            await this.track('error', {
+                error_type: 'download_failed',
+                error_message: error.message,
+                stack: error.stack,
+            });
+            exit.withError(`Error downloading exported file: ${error.message}`);
+        }
     }
-  }
-  getLiveBackupConfigFromCLIOptions() {
-    if (this.liveBackupCopyCLIOptions?.configFile) {
-      return this.loadLiveBackupCopyConfig(this.liveBackupCopyCLIOptions?.configFile);
+    generateDownloadURLOutputString(url) {
+        return `${chalk_1.default.green('Download URL')}: ${url}`;
     }
-    let type = _liveBackupCopy.BackupLiveCopyType.TABLES;
-    let tables;
-    let siteIds;
-    if (this.liveBackupCopyCLIOptions?.tables) {
-      tables = Object.fromEntries(this.liveBackupCopyCLIOptions?.tables.map(key => [key, {}]));
+    async runBackup() {
+        if (this.generateBackup) {
+            await this.runBackupJob();
+        }
+        const { latestBackup, envSqlDumpTool } = await fetchLatestBackupAndJobStatus(this.app.id, this.env.id);
+        if (!latestBackup) {
+            await this.track('error', {
+                error_type: 'no_backup_found',
+                error_message: 'No backup found for the site',
+            });
+            exit.withError(`No backup found for site ${this.app.name}`);
+        }
+        const prepareAdditionalInfo = [];
+        const showMyDumperWarning = this.showMyDumperWarning && (latestBackup.sqlDumpTool ?? envSqlDumpTool) === 'mydumper';
+        if (showMyDumperWarning) {
+            prepareAdditionalInfo.push(`${chalk_1.default.yellow.bold('WARNING:')} This is a large or complex database. The backup file for this database is generated with MyDumper. The file can only be loaded with MyLoader. For more information: https://github.com/mydumper/mydumper`);
+        }
+        if (await this.getExportJob()) {
+            prepareAdditionalInfo.push(`Attaching to an existing export for the backup with timestamp ${latestBackup.createdAt}`);
+        }
+        else {
+            prepareAdditionalInfo.push(`Exporting database backup with timestamp ${latestBackup.createdAt}`);
+            try {
+                await createExportJob(this.app.id, this.env.id, latestBackup.id);
+            }
+            catch (err) {
+                const error = err;
+                // Todo: match error code instead of message substring
+                if (error.message.includes('Backup Copy already in progress')) {
+                    await this.track('error', {
+                        error_type: 'job_already_running',
+                        error_message: error.message,
+                        stack: error.stack,
+                    });
+                    exit.withError('There is an export job already running for this environment: ' +
+                        `https://dashboard.wpvip.com/apps/${this.app.id}/${this.env.uniqueLabel}/database/backups\n` +
+                        'Currently, we allow only one export job at a time, per site. Please try again later.');
+                }
+                else {
+                    await this.track('error', {
+                        error_type: 'create_export_job',
+                        error_message: error.message,
+                        stack: error.stack,
+                    });
+                }
+                exit.withError(`Error creating export job: ${error.message}`);
+            }
+        }
+        await (0, utils_1.pollUntil)(this.getExportJob.bind(this), EXPORT_SQL_PROGRESS_POLL_INTERVAL, this.isPrepared.bind(this));
+        this.progressTracker.stepSuccess(this.steps.PREPARE, prepareAdditionalInfo);
+        await (0, utils_1.pollUntil)(this.getExportJob.bind(this), EXPORT_SQL_PROGRESS_POLL_INTERVAL, this.isCreated.bind(this));
+        this.progressTracker.stepSuccess(this.steps.CREATE);
+        const url = await generateDownloadLink(this.app.id, this.env.id, latestBackup.id);
+        this.progressTracker.stepSuccess(this.steps.DOWNLOAD_LINK, [
+            this.generateDownloadURLOutputString(url),
+        ]);
+        return url;
     }
-    if (this.liveBackupCopyCLIOptions?.siteIds) {
-      type = _liveBackupCopy.BackupLiveCopyType.SITE_IDS;
-      siteIds = this.liveBackupCopyCLIOptions?.siteIds.map(id => Number(id));
+    async generateLiveBackupCopy() {
+        if (!this.liveBackupCopyCLIOptions) {
+            throw new user_error_1.default('Configuration file is required for live backup copy.');
+        }
+        if (!this.app.id || !this.env.id) {
+            throw new user_error_1.default('App ID and Environment ID are required to start live backup copy.');
+        }
+        const config = this.getLiveBackupConfigFromCLIOptions();
+        try {
+            const copyId = await liveBackupCopy.startLiveBackupCopy({
+                appId: this.app.id,
+                environmentId: this.env.id,
+                config,
+            });
+            this.progressTracker.stepSuccess(this.steps.PREPARE);
+            this.progressTracker.stepRunning(this.steps.CREATE);
+            const result = await liveBackupCopy.getDownloadURL({
+                appId: this.app.id,
+                environmentId: this.env.id,
+                copyId,
+            });
+            this.progressTracker.stepSuccess(this.steps.CREATE);
+            this.progressTracker.stepSuccess(this.steps.DOWNLOAD_LINK, [
+                this.generateDownloadURLOutputString(result.url),
+            ]);
+            return result;
+        }
+        catch (err) {
+            const message = (0, utils_1.parseApiError)(err) ?? 'Unknown error';
+            await this.track('error', {
+                error_type: 'live_backup_copy',
+                error_message: message,
+                stack: err instanceof Error ? err.stack : undefined,
+            });
+            exit.withError(`Error creating live backup copy: ${message}`);
+        }
     }
-    if (this.liveBackupCopyCLIOptions?.wpcliCommand) {
-      type = _liveBackupCopy.BackupLiveCopyType.WP_CLI_COMMAND;
+    getLiveBackupConfigFromCLIOptions() {
+        if (this.liveBackupCopyCLIOptions?.configFile) {
+            return this.loadLiveBackupCopyConfig(this.liveBackupCopyCLIOptions?.configFile);
+        }
+        let type = live_backup_copy_1.BackupLiveCopyType.TABLES;
+        let tables;
+        let siteIds;
+        if (this.liveBackupCopyCLIOptions?.tables) {
+            tables = Object.fromEntries(this.liveBackupCopyCLIOptions?.tables.map(key => [key, {}]));
+        }
+        if (this.liveBackupCopyCLIOptions?.siteIds) {
+            type = live_backup_copy_1.BackupLiveCopyType.SITE_IDS;
+            siteIds = this.liveBackupCopyCLIOptions?.siteIds.map(id => Number(id));
+        }
+        if (this.liveBackupCopyCLIOptions?.wpcliCommand) {
+            type = live_backup_copy_1.BackupLiveCopyType.WP_CLI_COMMAND;
+        }
+        return {
+            type,
+            tables,
+            site_ids: siteIds,
+            wpcli_command: this.liveBackupCopyCLIOptions?.wpcliCommand,
+        };
     }
-    return {
-      type,
-      tables,
-      site_ids: siteIds,
-      wpcli_command: this.liveBackupCopyCLIOptions?.wpcliCommand
-    };
-  }
-  loadLiveBackupCopyConfig(configFile) {
-    try {
-      // Read the file once to avoid a check-then-use race condition.
-      const fileContents = _fs.default.readFileSync(configFile, 'utf-8');
-      const size = Buffer.byteLength(fileContents);
-      if (size > MAX_LIVE_BACKUP_CONFIG_SIZE_BYTES) {
-        throw new _userError.default(`Configuration file is too large (${(0, _format.formatBytes)(size)}); the maximum allowed size is ${(0, _format.formatBytes)(MAX_LIVE_BACKUP_CONFIG_SIZE_BYTES)}. For large or complex exports, use the \`--wpcli-command\` option instead.`);
-      }
-      return JSON.parse(fileContents);
-    } catch (err) {
-      if (err instanceof _userError.default) {
-        throw err;
-      }
-      const errMessage = err instanceof Error ? err.message : 'Unknown error';
-      if (err instanceof SyntaxError) {
-        throw new _userError.default(`Invalid JSON in configuration file: ${configFile} - ${errMessage}`);
-      }
-      throw new _userError.default(`Error reading configuration file: ${configFile} - ${errMessage}`);
+    loadLiveBackupCopyConfig(configFile) {
+        try {
+            // Read the file once to avoid a check-then-use race condition.
+            const fileContents = fs_1.default.readFileSync(configFile, 'utf-8');
+            const size = Buffer.byteLength(fileContents);
+            if (size > MAX_LIVE_BACKUP_CONFIG_SIZE_BYTES) {
+                throw new user_error_1.default(`Configuration file is too large (${(0, format_1.formatBytes)(size)}); the maximum allowed size is ${(0, format_1.formatBytes)(MAX_LIVE_BACKUP_CONFIG_SIZE_BYTES)}. For large or complex exports, use the \`--wpcli-command\` option instead.`);
+            }
+            return JSON.parse(fileContents);
+        }
+        catch (err) {
+            if (err instanceof user_error_1.default) {
+                throw err;
+            }
+            const errMessage = err instanceof Error ? err.message : 'Unknown error';
+            if (err instanceof SyntaxError) {
+                throw new user_error_1.default(`Invalid JSON in configuration file: ${configFile} - ${errMessage}`);
+            }
+            throw new user_error_1.default(`Error reading configuration file: ${configFile} - ${errMessage}`);
+        }
     }
-  }
 }
-exports.ExportSQLCommand = ExportSQLCommand;
+exports.ExportSQLCommand = ExportSQLCommand;