readest/apps/readest-app/src/services/sync/replicaBinaryUpload.ts

import { transferManager } from '@/services/transferManager';
import { getReplicaAdapter } from './replicaRegistry';
import { isSyncCategoryEnabled } from './syncCategories';
import { getAccessToken } from '@/utils/access';
import type { AppService, BaseDir } from '@/types/system';
import type { ReplicaTransferFile } from '@/store/transferStore';
import type { ClosableFile } from '@/utils/file';

/**
 * Open the file just to read its `.size` (Tauri streams metadata; the
 * body isn't read). Used as a fallback when the adapter's
 * enumerateFiles doesn't carry a byteSize — e.g. the dictionary
 * adapter, which doesn't track per-file sizes on its records.
 */
const resolveByteSize = async (
  appService: AppService,
  lfp: string,
  base: BaseDir,
): Promise<number> => {
  const file = await appService.openFile(lfp, base);
  const size = file.size;
  const closable = file as ClosableFile;
  if (closable && closable.close) {
    await closable.close();
  }
  return size;
};

interface ReplicaBinaryRecord {
  contentId?: string;
  name: string;
  reincarnation?: string;
}

/**
 * Queue a replica's binary files for upload via TransferManager.
 * Generic across kinds — uses the adapter registry to dispatch:
 *   - adapter.binary.enumerateFiles to list the per-record files
 *   - adapter.binary.localBaseDir for the upload base
 *
 * Resolves missing byteSize entries via openFile so progress reporting
 * is accurate. The TransferManager fires `replica-transfer-complete`
 * on success; replicaTransferIntegration converts that into a
 * publishReplicaManifest call (binaries first, manifest last).
 *
 * No-op when:
 *   - the user has turned the matching sync category off in Manage Sync
 *     (`dictionary`, `font`, `texture` — i.e. backgrounds). Skipping
 *     here keeps potentially large bundle uploads from leaving the
 *     device when the user has opted out, and matches the metadata-side
 *     gate already enforced by `publishReplicaUpsert`.
 *   - the record lacks contentId (legacy / unsynced)
 *   - the kind isn't registered or has no binary capability
 *   - TransferManager isn't initialized yet (pre-library mount)
 *
 * Caller is responsible for ordering this AFTER publishReplicaUpsert
 * so the metadata row exists before the manifest commit fires.
 */
export const queueReplicaBinaryUpload = async <T extends ReplicaBinaryRecord>(
  kind: string,
  record: T,
  appService: AppService,
): Promise<string | null> => {
  if (!isSyncCategoryEnabled(kind)) return null;
  if (!record.contentId) return null;
  if (!(await getAccessToken())) return null;
  if (!transferManager.isReady()) return null;

  const adapter = getReplicaAdapter<T>(kind);
  if (!adapter?.binary) return null;
  const base = adapter.binary.localBaseDir;

  const enumerated = adapter.binary.enumerateFiles(record);
  if (enumerated.length === 0) return null;

  const files: ReplicaTransferFile[] = await Promise.all(
    enumerated.map(async (f) => ({
      logical: f.logical,
      lfp: f.lfp,
      byteSize: f.byteSize > 0 ? f.byteSize : await resolveByteSize(appService, f.lfp, base),
    })),
  );

  return transferManager.queueReplicaUpload(kind, record.contentId, record.name, files, base, {
    reincarnation: record.reincarnation,
  });
};

/**
 * Backwards-compatible alias for the dictionary-specific helper that
 * call sites used before the kind-agnostic refactor.
 */
export const queueDictionaryBinaryUpload = <T extends ReplicaBinaryRecord>(
  dict: T,
  appService: AppService,
): Promise<string | null> => queueReplicaBinaryUpload('dictionary', dict, appService);