papercrate/frontend/src/asset_manager.ts

import type { AxiosInstance } from 'axios';

export type Identifier = string | number;

type Nullable<T> = T | null;

export interface AssetObject {
  ordinal?: number;
  url?: string | null;
  metadata?: Record<string, unknown> | null;
  expires_at?: number | null;
  [key: string]: unknown;
}

export interface AssetLike {
  id?: Identifier;
  asset_type?: string;
  cardinality?: number | null;
  url?: string | null;
  metadata?: Record<string, unknown> | null;
  expiresAt?: number | null;
  assets?: Record<string, AssetLike> | AssetLike[] | null;
  objects?: AssetObject[] | null;
  [key: string]: unknown;
}

export interface DocumentVersionLike {
  assets?: Record<string, AssetLike> | AssetLike[] | null;
  metadata?: Record<string, unknown> & { page_count?: number } | null;
  [key: string]: unknown;
}

export interface DocumentLike {
  id?: Identifier;
  current_version?: DocumentVersionLike | null;
  [key: string]: unknown;
}

export type EnsureAssetUrl = (
  documentId: Identifier,
  asset: AssetLike,
  options?: { start?: number | null; limit?: number | null; force?: boolean; [key: string]: unknown },
) => Promise<unknown>;

export type GetAsset = (document: DocumentLike, assetType: string) => Nullable<AssetLike>;

export const getAssetFromGroup = (
  assets?: AssetLike[] | Record<string, AssetLike> | null,
  assetType: string = '',
): Nullable<AssetLike> => {
  if (!assetType || !assets) {
    return null;
  }

  if (Array.isArray(assets)) {
    return assets.find((entry) => entry?.asset_type === assetType) || null;
  }

  return assets?.[assetType] || null;
};

export const getAssetFromVersion = (currentVersion: Nullable<DocumentVersionLike>, assetType: string) => {
  if (!currentVersion) {
    return null;
  }
  return getAssetFromGroup(currentVersion.assets ?? null, assetType);
};

const normalizeAssetObjects = (objects?: AssetObject[] | null): AssetObject[] => {
  if (!Array.isArray(objects)) {
    return [];
  }
  return objects
    .filter((entry) => Number.isInteger(entry?.ordinal))
    .slice()
    .sort((a, b) => a.ordinal - b.ordinal);
};

const mergeAssetObjects = (
  existingObjects?: AssetObject[] | null,
  incomingObjects?: AssetObject[] | null,
): AssetObject[] => {
  const merged = new Map<number, AssetObject>();

  normalizeAssetObjects(existingObjects).forEach((entry) => {
    if (Number.isFinite(entry.ordinal)) {
      merged.set(entry.ordinal as number, { ...entry });
    }
  });

  normalizeAssetObjects(incomingObjects).forEach((entry) => {
    if (Number.isFinite(entry.ordinal)) {
      const ordinal = entry.ordinal as number;
      const current = merged.get(ordinal) || {};
      merged.set(ordinal, { ...current, ...entry });
    }
  });

  return Array.from(merged.entries())
    .sort((a, b) => a[0] - b[0])
    .map(([, value]) => value);
};

export class AssetView {
  asset: AssetLike | null;
  private _objectsRef: AssetObject[] | null;
  private _sortedObjects: AssetObject[];

  constructor(asset?: AssetLike | null) {
    this.asset = asset || null;
    this._objectsRef = null;
    this._sortedObjects = [];
  }

  getCardinality(): number {
    if (!this.asset) {
      return 0;
    }

    const reported = Number(this.asset.cardinality);
    if (Number.isFinite(reported) && reported > 0) {
      return reported;
    }

    const objectsCount = this.getObjects().length;
    if (objectsCount > 0) {
      return objectsCount;
    }

    return this.asset.metadata ? 1 : 0;
  }

  getObjects(): AssetObject[] {
    if (!this.asset || !Array.isArray(this.asset.objects) || this.asset.objects.length === 0) {
      return [];
    }

    if (this._objectsRef === this.asset.objects) {
      return this._sortedObjects;
    }

    this._objectsRef = this.asset.objects;
    this._sortedObjects = normalizeAssetObjects(this.asset.objects);
    return this._sortedObjects;
  }

  getObject(ordinal = 1): AssetObject | null {
    const fromObjects = this.getObjects().find((entry) => entry.ordinal === ordinal);
    if (fromObjects) {
      return fromObjects;
    }

    if (ordinal === 1 && this.asset) {
      if (this.asset.url || this.asset.metadata) {
        return {
          ordinal: 1,
          url: this.asset.url || null,
          metadata: this.asset.metadata || null,
          expires_at: this.asset.expiresAt ?? null,
        };
      }
    }

    return null;
  }

  getPrimaryObject(): AssetObject | null {
    return this.getObject(1);
  }

  getPrimaryMetadata(): Record<string, unknown> | null {
    return this.getPrimaryObject()?.metadata || null;
  }

  getPrimaryUrl(): string | null {
    return this.getPrimaryObject()?.url || null;
  }

  hasObject(ordinal: number): boolean {
    return Boolean(this.getObject(ordinal));
  }
}

export const createAssetView = (asset?: AssetLike | null): AssetView => new AssetView(asset);

export const resolveDocumentAssetUrl = (
  doc: Nullable<DocumentLike>,
  type: string,
  {
    ensureAssetUrl,
    getAsset,
    ensureOptions,
    objectOrdinal = 1,
  }: {
    ensureAssetUrl?: EnsureAssetUrl;
    getAsset?: GetAsset;
    ensureOptions?: { start?: number; limit?: number; [key: string]: unknown };
    objectOrdinal?: number;
  } = {},
): Nullable<string> => {
  if (!doc || !type) {
    return null;
  }
  const asset = getAsset ? getAsset(doc, type) : null;
  if (!asset) {
    return null;
  }
  const view = createAssetView(asset);
  const object = view.getObject(objectOrdinal);
  const url = object?.url || (objectOrdinal === 1 ? view.getPrimaryUrl() : null);
  const expiresAt = Number.isFinite(object?.expires_at)
    ? Number(object?.expires_at)
    : objectOrdinal === 1 && Number.isFinite(asset.expiresAt)
      ? Number(asset.expiresAt)
      : null;
  const now = Date.now();
  if (url && (!expiresAt || expiresAt > now)) {
    return url;
  }
  if (doc.id && asset.id && ensureAssetUrl) {
    const force = Boolean(url && expiresAt && expiresAt <= now);
    const options: { force: boolean; start?: number | null; limit?: number | null; [key: string]: unknown } = {
      force,
      start: objectOrdinal,
      limit: 1,
      ...(ensureOptions || {}),
    };
    if (!options.start) {
      options.start = objectOrdinal;
    }
    if (!options.limit) {
      options.limit = 1;
    }
    ensureAssetUrl(doc.id, asset, options).catch(() => {});
  }
  return null;
};

class AssetManager {
  api: AxiosInstance | null;
  assetPresignTtlMs: number;
  assetCache: Map<Identifier, AssetLike>;
  assetInflight: Map<string, Promise<AssetLike | null>>;

  constructor({ api, assetPresignTtlMs }: { api: AxiosInstance | null; assetPresignTtlMs: number }) {
    this.api = api;
    this.assetPresignTtlMs = assetPresignTtlMs;
    this.assetCache = new Map();
    this.assetInflight = new Map();
  }

  setApi(api: AxiosInstance | null) {
    this.api = api;
  }

  rememberAsset(entry?: Nullable<AssetLike>) {
    if (entry?.id) {
      this.assetCache.set(entry.id, entry);
    }
  }

  ensureAsset(
    documentId?: Identifier | null,
    asset?: Nullable<AssetLike>,
    { force = false, start = null, limit = null }: { force?: boolean; start?: number | null; limit?: number | null } = {},
  ): Promise<Nullable<AssetLike>> {
    if (!documentId || !asset?.id) {
      return Promise.resolve(asset ?? null);
    }

    const requestedStart = Number.isInteger(start) && start > 0 ? start : 1;
    const requestedLimit = Number.isInteger(limit) && limit > 0 ? limit : 1;
    const requestedEnd = requestedStart + requestedLimit - 1;

    const baseAsset = this.assetCache.get(asset.id) || asset;
    const view = createAssetView(baseAsset);
    const assetExpiresAt = Number.isFinite(baseAsset.expiresAt) ? Number(baseAsset.expiresAt) : null;
    const now = Date.now();

    const isOrdinalSatisfied = (ordinal) => {
      const object = view.getObject(ordinal);
      if (!object) {
        return false;
      }
      if (!object.url) {
        return false;
      }
      if (Number.isFinite(object.expires_at)) {
        return Number(object.expires_at) > now;
      }
      if (ordinal === 1 && baseAsset.url && (!assetExpiresAt || assetExpiresAt > now)) {
        return true;
      }
      return true;
    };

    let needsFetch = force;
    if (!needsFetch) {
      for (let ordinal = requestedStart; ordinal <= requestedEnd; ordinal += 1) {
        if (!isOrdinalSatisfied(ordinal)) {
          needsFetch = true;
          break;
        }
      }
    }

    if (!needsFetch) {
      this.rememberAsset(baseAsset);
      return Promise.resolve(baseAsset);
    }

    const inflightKey = `${documentId}:${asset.id}:${start ?? 'd'}:${limit ?? 'd'}`;
    if (!force && this.assetInflight.has(inflightKey)) {
      return this.assetInflight.get(inflightKey);
    }

    if (!this.api) {
      return Promise.reject(new Error('AssetManager API client is not configured.'));
    }

    const params: Record<string, number> = {};
    if (Number.isInteger(start) && start > 0) {
      params.start = start;
    }
    if (Number.isInteger(limit) && limit > 0) {
      params.limit = limit;
    }

    const requestConfig = Object.keys(params).length ? { params } : undefined;

    const request: Promise<AssetLike | null> = this.api
      .get(`/assets/${asset.id}`, requestConfig)
      .then(({ data }) => {
        const incomingObjects = Array.isArray(data.objects) ? data.objects : [];
        const cachedEntry = this.assetCache.get(asset.id) || baseAsset;
        const mergedObjects = mergeAssetObjects(cachedEntry?.objects, incomingObjects);
        const combined = { ...cachedEntry, ...asset, ...data, objects: mergedObjects };
        const view = createAssetView(combined);
        const primaryObject = view.getPrimaryObject();
        const expiresAt = Number.isFinite(primaryObject?.expires_at)
          ? Number(primaryObject?.expires_at)
          : Date.now() + this.assetPresignTtlMs;
        const cardinality = (() => {
          const reported = Number(data.cardinality ?? asset.cardinality ?? cachedEntry?.cardinality);
          const objectsCount = mergedObjects.length;
          if (Number.isFinite(reported) && reported > 0) {
            return Math.max(reported, objectsCount) || null;
          }
          return objectsCount || null;
        })();

        const entry = {
          ...combined,
          cardinality,
          url: view.getPrimaryUrl(),
          expiresAt,
        };

        this.rememberAsset(entry);
        return entry;
      })
      .finally(() => {
        this.assetInflight.delete(inflightKey);
      });

    this.assetInflight.set(inflightKey, request);
    return request;
  }

  reset() {
    this.assetCache.clear();
    this.assetInflight.clear();
  }
}

export default AssetManager;