Class ViewerServer

packages/memos-core/src/viewer/server.ts:118–4991

}

export class ViewerServer {
  private server: http.Server | null = null;
  private readonly store: SqliteStore;
  private readonly embedder: Embedder;
  private readonly port: number;
  private readonly log: Logger;
  private readonly dataDir: string;
  private readonly authFile: string;
  private readonly auth: AuthState;
  private readonly ctx?: PluginContext;
  private readonly cookieName: string;
  private readonly defaultHubPort: number;
  private readonly branding?: ViewerBranding;

  private static readonly SESSION_TTL = 24 * 60 * 60 * 1000;
  private static readonly PLUGIN_VERSION: string = (() => {
    try {
      const pkgPath = path.resolve(__dirname, "../../package.json");
      return JSON.parse(fs.readFileSync(pkgPath, "utf-8")).version ?? "unknown";
    } catch {
      return "unknown";
    }
  })();
  private resetToken: string;
  private migrationRunning = false;
  private migrationAbort = false;
  private migrationState: MigrationStateSnapshot = createInitialMigrationState();
  private migrationSSEClients: http.ServerResponse[] = [];

  private ppRunning = false;
  private ppAbort = false;
  private ppState: { running: boolean; done: boolean; stopped: boolean; processed: number; total: number; tasksCreated: number; skillsCreated: number; errors: number; skippedSessions: number; totalSessions: number } =
    { running: false, done: false, stopped: false, processed: 0, total: 0, tasksCreated: 0, skillsCreated: 0, errors: 0, skippedSessions: 0, totalSessions: 0 };
  private ppSSEClients: http.ServerResponse[] = [];

  private notifSSEClients: http.ServerResponse[] = [];
  private notifPollTimer?: ReturnType<typeof setInterval>;
  private lastKnownNotifCount = 0;
  private hubHeartbeatTimer?: ReturnType<typeof setInterval>;
  private static readonly HUB_HEARTBEAT_INTERVAL_MS = 45_000;
  private static readonly STALE_TASK_TIMEOUT_MS = 4 * 60 * 60 * 1000;
  private staleFinalizeRunning = false;

  constructor(opts: ViewerServerOptions) {
    this.store = opts.store;
    this.embedder = opts.embedder;
    this.port = opts.port;
    this.log = opts.log;
    this.dataDir = opts.dataDir;
    this.ctx = opts.ctx;
    this.authFile = path.join(opts.dataDir, "viewer-auth.json");
    this.auth = { passwordHash: null, sessions: new Map() };
    this.cookieName = `memos_token_${opts.port}`;
    this.defaultHubPort = opts.defaultHubPort ?? 18800;
    this.branding = opts.branding;
    this.resetToken = crypto.randomBytes(16).toString("hex");
    this.loadAuth();
  }