privacy.sexy/src/presentation/electron/main/Update/ManualUpdater/Downloader.ts

import { existsSync, createWriteStream } from 'fs';
import { unlink, mkdir } from 'fs/promises';
import path from 'path';
import { app } from 'electron';
import { UpdateInfo } from 'electron-updater';
import fetch from 'cross-fetch';
import { ElectronLogger } from '@/infrastructure/Log/ElectronLogger';
import { UpdateProgressBar } from '../UpdateProgressBar';
import { retryFileSystemAccess } from './RetryFileSystemAccess';
import type { WriteStream } from 'fs';
import type { Readable } from 'stream';

const MAX_PROGRESS_LOG_ENTRIES = 10;
const UNKNOWN_SIZE_LOG_INTERVAL_BYTES = 10 * 1024 * 1024; // 10 MB

export type DownloadUpdateResult = {
  readonly success: false;
} | {
  readonly success: true;
  readonly installerPath: string;
};

export async function downloadUpdate(
  info: UpdateInfo,
  remoteFileUrl: string,
  progressBar: UpdateProgressBar,
): Promise<DownloadUpdateResult> {
  ElectronLogger.info('Starting manual update download.');
  progressBar.showIndeterminateState();
  try {
    const { filePath } = await downloadInstallerFile(
      info.version,
      remoteFileUrl,
      (percentage) => { progressBar.showPercentage(percentage); },
    );
    return {
      success: true,
      installerPath: filePath,
    };
  } catch (e) {
    progressBar.showError(e);
    return {
      success: false,
    };
  }
}

async function downloadInstallerFile(
  version: string,
  remoteFileUrl: string,
  progressHandler: ProgressCallback,
): Promise<{ readonly filePath: string; }> {
  const filePath = `${path.dirname(app.getPath('temp'))}/privacy.sexy/${version}-installer.dmg`;
  if (!await ensureFilePathReady(filePath)) {
    throw new Error(`Failed to prepare the file path for the installer: ${filePath}`);
  }
  await downloadFileWithProgress(
    remoteFileUrl,
    filePath,
    progressHandler,
  );
  return { filePath };
}

async function ensureFilePathReady(filePath: string): Promise<boolean> {
  return retryFileSystemAccess(async () => {
    try {
      const parentFolder = path.dirname(filePath);
      if (existsSync(filePath)) {
        ElectronLogger.info(`Existing update file found and will be replaced: ${filePath}`);
        await unlink(filePath);
      } else {
        await mkdir(parentFolder, { recursive: true });
      }
      return true;
    } catch (error) {
      ElectronLogger.error(`Failed to prepare file path for update: ${filePath}`, error);
      return false;
    }
  });
}

type ProgressCallback = (progress: number) => void;

async function downloadFileWithProgress(
  url: string,
  filePath: string,
  progressHandler: ProgressCallback,
) {
  // autoUpdater cannot handle DMG files, requiring manual download management for these file types.
  ElectronLogger.info(`Retrieving update from ${url}.`);
  const response = await fetch(url);
  if (!response.ok) {
    throw Error(`Download failed: Server responded with ${response.status} ${response.statusText}.`);
  }
  const contentLength = getContentLengthFromResponse(response);
  await withWriteStream(filePath, async (writer) => {
    ElectronLogger.info(contentLength.isValid
      ? `Saving file to ${filePath} (Size: ${contentLength.totalLength} bytes).`
      : `Saving file to ${filePath}.`);
    await withReadableStream(response, async (reader) => {
      await streamWithProgress(contentLength, reader, writer, progressHandler);
    });
  });
}

type ResponseContentLength = {
  readonly isValid: true;
  readonly totalLength: number;
} | {
  readonly isValid: false;
};

function getContentLengthFromResponse(response: Response): ResponseContentLength {
  const contentLengthString = response.headers.get('content-length');
  const headersInfo = Array.from(response.headers.entries());
  if (!contentLengthString) {
    ElectronLogger.warn('Missing \'Content-Length\' header in the response.', headersInfo);
    return { isValid: false };
  }
  const contentLength = Number(contentLengthString);
  if (Number.isNaN(contentLength) || contentLength <= 0) {
    ElectronLogger.error('Unable to determine download size from server response.', headersInfo);
    return { isValid: false };
  }
  return { totalLength: contentLength, isValid: true };
}

async function withReadableStream(
  response: Response,
  handler: (readStream: Readable) => Promise<void>,
) {
  const reader = createReader(response);
  try {
    await handler(reader);
  } finally {
    reader.destroy();
  }
}

async function withWriteStream(
  filePath: string,
  handler: (writeStream: WriteStream) => Promise<void>,
) {
  const writer = createWriteStream(filePath);
  try {
    await handler(writer);
  } finally {
    writer.end();
  }
}

async function streamWithProgress(
  contentLength: ResponseContentLength,
  readStream: Readable,
  writeStream: WriteStream,
  progressHandler: ProgressCallback,
): Promise<void> {
  let receivedLength = 0;
  let logThreshold = 0;
  for await (const chunk of readStream) {
    if (!chunk) {
      throw Error('Received empty data chunk during download.');
    }
    writeStream.write(Buffer.from(chunk));
    receivedLength += chunk.length;
    notifyProgress(contentLength, receivedLength, progressHandler);
    const progressLog = logProgress(receivedLength, contentLength, logThreshold);
    logThreshold = progressLog.nextLogThreshold;
  }
  ElectronLogger.info('Update download completed successfully.');
}

function logProgress(
  receivedLength: number,
  contentLength: ResponseContentLength,
  logThreshold: number,
): { readonly nextLogThreshold: number; } {
  const {
    shouldLog, nextLogThreshold,
  } = shouldLogProgress(receivedLength, contentLength, logThreshold);
  if (shouldLog) {
    ElectronLogger.debug(`Download progress: ${receivedLength} bytes received.`);
  }
  return { nextLogThreshold };
}

function notifyProgress(
  contentLength: ResponseContentLength,
  receivedLength: number,
  progressHandler: ProgressCallback,
) {
  if (!contentLength.isValid) {
    return;
  }
  const percentage = Math.floor((receivedLength / contentLength.totalLength) * 100);
  progressHandler(percentage);
}

function shouldLogProgress(
  receivedLength: number,
  contentLength: ResponseContentLength,
  previousLogThreshold: number,
): { shouldLog: boolean, nextLogThreshold: number } {
  const logInterval = contentLength.isValid
    ? Math.ceil(contentLength.totalLength / MAX_PROGRESS_LOG_ENTRIES)
    : UNKNOWN_SIZE_LOG_INTERVAL_BYTES;

  if (receivedLength >= previousLogThreshold + logInterval) {
    return { shouldLog: true, nextLogThreshold: previousLogThreshold + logInterval };
  }
  return { shouldLog: false, nextLogThreshold: previousLogThreshold };
}

function createReader(response: Response): Readable {
  if (!response.body) {
    throw new Error('Response body is empty, cannot proceed with download.');
  }
  // On browser, we could use browser API response.body.getReader()
  // But here, we use cross-fetch that gets node-fetch on a node application
  // This API is node-fetch specific, see https://github.com/node-fetch/node-fetch#streams
  return response.body as unknown as Readable;
}