Class WorkerImporter

apps/desktop/main/api/adapters.ts:81–153

// ==================== WorkerImporter ====================

export class WorkerImporter implements DataImporter {
  private logger: SyncLogger

  constructor(logger?: SyncLogger) {
    this.logger = logger ?? NOOP_LOGGER
  }

  sessionExists(sessionId: string): boolean {
    assertDesktopDataDirCompatible(getPathProvider(), getDesktopAppVersion(app.getVersion()))

    const dbPath = path.join(worker.getDbDirectory(), `${sessionId}.db`)
    if (!fs.existsSync(dbPath)) return false
    try {
      const db = new Database(dbPath, { readonly: true })
      const row = db
        .prepare("SELECT COUNT(*) as cnt FROM sqlite_master WHERE type='table' AND name='message'")
        .get() as { cnt: number }
      db.close()
      if (row.cnt === 0) {
        this.logger.warn(`[Pull] DB file exists but has no message table: ${sessionId}, removing`)
        try {
          fs.unlinkSync(dbPath)
        } catch {
          /* ignore */
        }
        return false
      }
      return true
    } catch {
      this.logger.warn(`[Pull] Cannot validate DB file: ${sessionId}, removing`)
      try {
        fs.unlinkSync(dbPath)
      } catch {
        /* ignore */
      }
      return false
    }
  }

  async importFile(tempFile: string, targetSessionId: string | undefined, externalId: string): Promise<ImportResult> {
    if (targetSessionId && this.sessionExists(targetSessionId)) {
      return this.incrementalImportFile(targetSessionId, tempFile)
    }
    return this.fullImportFile(tempFile, externalId)
  }

  private async incrementalImportFile(sessionId: string, tempFile: string): Promise<ImportResult> {
    this.logger.info(`[Pull] Incremental import to session ${sessionId}`)
    const result = await worker.incrementalImport(sessionId, tempFile)
    if (result.success) {
      this.logger.info(`[Pull] Incremental OK: +${result.newMessageCount} messages`)
      return { success: true, newMessageCount: result.newMessageCount, sessionId }
    }
    if (result.error === 'error.session_not_found' || result.error?.includes('no such table')) {
      this.logger.warn(`[Pull] Session ${sessionId} not found or schema invalid, need full resync`)
      return { success: false, newMessageCount: 0, sessionId, needFullResync: true }
    }
    this.logger.error(`[Pull] Incremental import failed: ${result.error}`)