AlgerMusicPlayer/src/main/modules/update.ts

import { app, BrowserWindow, ipcMain, shell } from 'electron';
import electronUpdater, {
  type ProgressInfo,
  type UpdateDownloadedEvent,
  type UpdateInfo
} from 'electron-updater';

import {
  APP_UPDATE_RELEASE_URL,
  APP_UPDATE_STATUS,
  type AppUpdateState,
  createDefaultAppUpdateState
} from '../../shared/appUpdate';

const { autoUpdater } = electronUpdater;

type CheckUpdateOptions = {
  manual?: boolean;
};

let updateState: AppUpdateState = createDefaultAppUpdateState(app.getVersion());
let isInitialized = false;
let checkForUpdatesPromise: Promise<AppUpdateState> | null = null;
let downloadUpdatePromise: Promise<AppUpdateState> | null = null;

const isAutoUpdateSupported = (): boolean => {
  // if (!app.isPackaged) {
  //   return false;
  // }

  if (process.platform === 'linux') {
    return Boolean(process.env.APPIMAGE);
  }

  return true;
};

const normalizeReleaseNotes = (releaseNotes: UpdateInfo['releaseNotes']): string => {
  if (typeof releaseNotes === 'string') {
    return releaseNotes;
  }

  if (Array.isArray(releaseNotes)) {
    return releaseNotes
      .map((item) => {
        const version = item.version ? `## ${item.version}` : '';
        return [version, item.note].filter(Boolean).join('\n');
      })
      .join('\n\n');
  }

  return '';
};

const broadcastUpdateState = () => {
  for (const window of BrowserWindow.getAllWindows()) {
    window.webContents.send('app-update:state', updateState);
  }
};

const setUpdateState = (partial: Partial<AppUpdateState>) => {
  updateState = {
    ...updateState,
    ...partial
  };
  broadcastUpdateState();
};

const resetUpdateState = () => {
  updateState = {
    ...createDefaultAppUpdateState(app.getVersion()),
    supported: isAutoUpdateSupported()
  };
};

const getUnsupportedMessage = () => {
  if (!app.isPackaged) {
    return '当前环境为开发模式，自动更新仅在打包后的应用内可用';
  }

  if (process.platform === 'linux') {
    return '当前 Linux 安装方式不支持自动更新，请前往官网下载安装包更新';
  }

  return '当前环境不支持自动更新，请前往官网下载安装包更新';
};

const applyUpdateInfo = (
  status: AppUpdateState['status'],
  info?: Pick<UpdateInfo, 'version' | 'releaseDate' | 'releaseNotes'>
) => {
  setUpdateState({
    status,
    availableVersion: info?.version ?? null,
    releaseDate: info?.releaseDate ?? null,
    releaseNotes: info ? normalizeReleaseNotes(info.releaseNotes) : '',
    releasePageUrl: APP_UPDATE_RELEASE_URL,
    errorMessage: null,
    checkedAt: Date.now()
  });
};

const checkForUpdates = async (options: CheckUpdateOptions = {}): Promise<AppUpdateState> => {
  if (!updateState.supported) {
    const errorMessage = options.manual ? getUnsupportedMessage() : null;
    setUpdateState({
      status: options.manual ? APP_UPDATE_STATUS.error : APP_UPDATE_STATUS.idle,
      errorMessage
    });
    return updateState;
  }

  if (
    updateState.status === APP_UPDATE_STATUS.available ||
    updateState.status === APP_UPDATE_STATUS.downloading ||
    updateState.status === APP_UPDATE_STATUS.downloaded
  ) {
    return updateState;
  }

  if (checkForUpdatesPromise) {
    return await checkForUpdatesPromise;
  }

  checkForUpdatesPromise = (async () => {
    try {
      setUpdateState({
        status: APP_UPDATE_STATUS.checking,
        errorMessage: null,
        checkedAt: Date.now()
      });
      await autoUpdater.checkForUpdates();
      return updateState;
    } catch (error) {
      const errorMessage = error instanceof Error ? error.message : '检查更新失败';
      setUpdateState({
        status: APP_UPDATE_STATUS.error,
        errorMessage,
        checkedAt: Date.now()
      });
      return updateState;
    } finally {
      checkForUpdatesPromise = null;
    }
  })();

  return await checkForUpdatesPromise;
};

const downloadUpdate = async (): Promise<AppUpdateState> => {
  if (!updateState.supported) {
    setUpdateState({
      status: APP_UPDATE_STATUS.error,
      errorMessage: getUnsupportedMessage()
    });
    return updateState;
  }

  if (updateState.status === APP_UPDATE_STATUS.downloaded) {
    return updateState;
  }

  if (!hasDownloadableUpdate()) {
    setUpdateState({
      status: APP_UPDATE_STATUS.error,
      errorMessage: '当前没有可下载的更新'
    });
    return updateState;
  }

  if (downloadUpdatePromise) {
    return await downloadUpdatePromise;
  }

  downloadUpdatePromise = (async () => {
    try {
      await autoUpdater.downloadUpdate();
      return updateState;
    } catch (error) {
      const errorMessage = error instanceof Error ? error.message : '下载更新失败';
      setUpdateState({
        status: APP_UPDATE_STATUS.error,
        errorMessage
      });
      return updateState;
    } finally {
      downloadUpdatePromise = null;
    }
  })();

  return await downloadUpdatePromise;
};

const hasDownloadableUpdate = () => {
  return updateState.status === APP_UPDATE_STATUS.available;
};

const openReleasePage = async (): Promise<boolean> => {
  await shell.openExternal(updateState.releasePageUrl || APP_UPDATE_RELEASE_URL);
  return true;
};

export function setupUpdateHandlers(mainWindow: BrowserWindow) {
  if (isInitialized) {
    mainWindow.webContents.once('did-finish-load', () => {
      mainWindow.webContents.send('app-update:state', updateState);
    });
    return;
  }

  isInitialized = true;
  resetUpdateState();

  autoUpdater.autoDownload = false;
  autoUpdater.autoInstallOnAppQuit = true;

  autoUpdater.on('checking-for-update', () => {
    setUpdateState({
      status: APP_UPDATE_STATUS.checking,
      errorMessage: null,
      checkedAt: Date.now()
    });
  });

  autoUpdater.on('update-available', (info) => {
    applyUpdateInfo(APP_UPDATE_STATUS.available, info);
  });

  autoUpdater.on('update-not-available', () => {
    setUpdateState({
      status: APP_UPDATE_STATUS.notAvailable,
      availableVersion: null,
      releaseNotes: '',
      releaseDate: null,
      errorMessage: null,
      checkedAt: Date.now()
    });
  });

  autoUpdater.on('download-progress', (progress: ProgressInfo) => {
    setUpdateState({
      status: APP_UPDATE_STATUS.downloading,
      downloadProgress: progress.percent,
      downloadedBytes: progress.transferred,
      totalBytes: progress.total,
      bytesPerSecond: progress.bytesPerSecond,
      errorMessage: null
    });
  });

  autoUpdater.on('update-downloaded', (info: UpdateDownloadedEvent) => {
    setUpdateState({
      status: APP_UPDATE_STATUS.downloaded,
      availableVersion: info.version,
      releaseNotes: normalizeReleaseNotes(info.releaseNotes),
      releaseDate: info.releaseDate,
      downloadProgress: 100,
      downloadedBytes: info.files.reduce((total, file) => total + (file.size ?? 0), 0),
      totalBytes: info.files.reduce((total, file) => total + (file.size ?? 0), 0),
      bytesPerSecond: 0,
      errorMessage: null
    });
  });

  autoUpdater.on('error', (error) => {
    setUpdateState({
      status: APP_UPDATE_STATUS.error,
      errorMessage: error?.message ?? '自动更新失败'
    });
  });

  ipcMain.handle('app-update:get-state', async () => {
    return updateState;
  });

  ipcMain.handle('app-update:check', async (_event, options?: CheckUpdateOptions) => {
    return await checkForUpdates(options);
  });

  ipcMain.handle('app-update:download', async () => {
    return await downloadUpdate();
  });

  ipcMain.handle('app-update:quit-and-install', async () => {
    autoUpdater.quitAndInstall(false, true);
    return true;
  });

  ipcMain.handle('app-update:open-release-page', async () => {
    return await openReleasePage();
  });

  mainWindow.webContents.once('did-finish-load', () => {
    mainWindow.webContents.send('app-update:state', updateState);
  });
}