ditto/src/db/unattached-media.ts

import { Kysely } from 'kysely';

import { DittoDB } from '@/db/DittoDB.ts';
import { DittoTables } from '@/db/DittoTables.ts';

interface UnattachedMedia {
  id: string;
  pubkey: string;
  url: string;
  /** NIP-94 tags. */
  data: string[][];
  uploaded_at: number;
}

/** Add unattached media into the database. */
async function insertUnattachedMedia(media: UnattachedMedia) {
  const kysely = await DittoDB.getInstance();
  await kysely.insertInto('unattached_media')
    .values({ ...media, data: JSON.stringify(media.data) })
    .execute();

  return media;
}

/** Select query for unattached media. */
function selectUnattachedMediaQuery(kysely: Kysely<DittoTables>) {
  return kysely.selectFrom('unattached_media')
    .select([
      'unattached_media.id',
      'unattached_media.pubkey',
      'unattached_media.url',
      'unattached_media.data',
      'unattached_media.uploaded_at',
    ]);
}

/** Find attachments that exist but aren't attached to any events. */
function getUnattachedMedia(kysely: Kysely<DittoTables>, until: Date) {
  return selectUnattachedMediaQuery(kysely)
    .leftJoin('nostr_tags', 'unattached_media.url', 'nostr_tags.value')
    .where('uploaded_at', '<', until.getTime())
    .execute();
}

/** Delete unattached media by URL. */
async function deleteUnattachedMediaByUrl(url: string) {
  const kysely = await DittoDB.getInstance();
  return kysely.deleteFrom('unattached_media')
    .where('url', '=', url)
    .execute();
}

/** Get unattached media by IDs. */
async function getUnattachedMediaByIds(kysely: Kysely<DittoTables>, ids: string[]): Promise<UnattachedMedia[]> {
  if (!ids.length) return [];

  const results = await selectUnattachedMediaQuery(kysely)
    .where('id', 'in', ids)
    .execute();

  return results.map((row) => ({
    ...row,
    data: JSON.parse(row.data),
  }));
}

/** Delete rows as an event with media is being created. */
async function deleteAttachedMedia(pubkey: string, urls: string[]): Promise<void> {
  if (!urls.length) return;
  const kysely = await DittoDB.getInstance();
  await kysely.deleteFrom('unattached_media')
    .where('pubkey', '=', pubkey)
    .where('url', 'in', urls)
    .execute();
}

export {
  deleteAttachedMedia,
  deleteUnattachedMediaByUrl,
  getUnattachedMedia,
  getUnattachedMediaByIds,
  insertUnattachedMedia,
  type UnattachedMedia,
};