kill-the-news/src/lib/feed-service.ts

import { Context } from "hono";
import { Env, FeedConfig, FeedMetadata } from "../types";
import { generateFeedId } from "../utils/id-generator";
import { bumpCounters } from "../utils/stats";
import { waitUntilSafe } from "../utils/worker";
import { sendUnsubscribes } from "../utils/unsubscribe";
import { getAttachmentBucket } from "../utils/attachments";
import {
  addFeedToList,
  updateFeedInList,
  removeFeedFromList,
  purgeFeedKeysStep,
  collectUnsubscribeUrls,
} from "../routes/admin/helpers";

const HOUR_MS = 3_600_000;

/**
 * Resolve a feed's `expires_at` from a requested lifetime (hours). A server-side
 * `FEED_TTL_HOURS` always overrides the client-supplied value. Returns undefined
 * when no positive lifetime applies (i.e. the feed never expires).
 */
function resolveExpiresAt(
  env: Env,
  lifetimeHours?: number,
): number | undefined {
  const hours = env.FEED_TTL_HOURS
    ? parseInt(env.FEED_TTL_HOURS, 10)
    : (lifetimeHours ?? NaN);
  return Number.isFinite(hours) && hours > 0
    ? Date.now() + hours * HOUR_MS
    : undefined;
}

export interface CreateFeedInput {
  title: string;
  description?: string;
  language: string;
  allowedSenders: string[];
  blockedSenders: string[];
  lifetimeHours?: number;
}

/**
 * Create a feed: write its config + empty metadata, register it in the global
 * list, and bump the `feeds_created` counter. Returns the new feed id + config.
 */
export async function createFeedRecord(
  env: Env,
  input: CreateFeedInput,
): Promise<{ feedId: string; config: FeedConfig }> {
  const emailStorage = env.EMAIL_STORAGE;
  const expiresAt = resolveExpiresAt(env, input.lifetimeHours);
  const feedId = generateFeedId();

  const config: FeedConfig = {
    title: input.title,
    description: input.description,
    language: input.language,
    allowed_senders: input.allowedSenders,
    blocked_senders: input.blockedSenders,
    created_at: Date.now(),
    updated_at: Date.now(),
    ...(expiresAt !== undefined ? { expires_at: expiresAt } : {}),
  };

  const metadata: FeedMetadata = { emails: [] };

  await Promise.all([
    emailStorage.put(`feed:${feedId}:config`, JSON.stringify(config)),
    emailStorage.put(`feed:${feedId}:metadata`, JSON.stringify(metadata)),
  ]);

  await addFeedToList(
    emailStorage,
    feedId,
    input.title,
    input.description,
    expiresAt,
  );

  await bumpCounters(emailStorage, {
    feeds_created: 1,
    last_feed_created_at: new Date().toISOString(),
  });

  return { feedId, config };
}

export interface UpdateFeedInput {
  title?: string;
  description?: string;
  language?: string;
  allowedSenders?: string[];
  blockedSenders?: string[];
  lifetimeHours?: number;
}

export type UpdateFeedResult =
  | { status: "ok"; config: FeedConfig }
  | { status: "not_found" }
  | { status: "expired" };

/**
 * Apply a partial patch to a feed's config and mirror title/description/expiry
 * into the global list. Fields left undefined on `input` are preserved.
 *
 * A full edit (default) rejects expired feeds and recomputes `expires_at` from
 * `FEED_TTL_HOURS`/`lifetimeHours`. `inPlace` skips both — used by the dashboard's
 * minimal title/description edit, which must never touch expiry.
 */
export async function updateFeedRecord(
  env: Env,
  feedId: string,
  input: UpdateFeedInput,
  options: { inPlace?: boolean } = {},
): Promise<UpdateFeedResult> {
  const emailStorage = env.EMAIL_STORAGE;
  const feedConfigKey = `feed:${feedId}:config`;

  const existing = (await emailStorage.get(feedConfigKey, {
    type: "json",
  })) as FeedConfig | null;

  if (!existing) return { status: "not_found" };

  if (
    !options.inPlace &&
    existing.expires_at !== undefined &&
    existing.expires_at <= Date.now()
  ) {
    return { status: "expired" };
  }

  // Full edit recomputes expiry (FEED_TTL_HOURS or a supplied lifetime resets the
  // clock; an absent lifetime preserves it). In-place edits leave expiry alone.
  const expiresAt =
    !options.inPlace &&
    (env.FEED_TTL_HOURS || input.lifetimeHours !== undefined)
      ? resolveExpiresAt(env, input.lifetimeHours)
      : existing.expires_at;

  const config: FeedConfig = {
    ...existing,
    ...(input.title !== undefined ? { title: input.title } : {}),
    ...(input.description !== undefined
      ? { description: input.description }
      : {}),
    ...(input.language !== undefined ? { language: input.language } : {}),
    ...(input.allowedSenders !== undefined
      ? { allowed_senders: input.allowedSenders }
      : {}),
    ...(input.blockedSenders !== undefined
      ? { blocked_senders: input.blockedSenders }
      : {}),
    updated_at: Date.now(),
    expires_at: expiresAt,
  };

  await emailStorage.put(feedConfigKey, JSON.stringify(config));
  await updateFeedInList(
    emailStorage,
    feedId,
    config.title,
    config.description,
    expiresAt,
  );

  return { status: "ok", config };
}

type DeleteFeedFastResult = {
  ok: boolean;
  configDeleted: boolean;
  metadataDeleted: boolean;
  errors: string[];
};

/**
 * Delete a feed's config + metadata keys, reporting per-key outcomes. The
 * larger email/attachment cleanup is handled separately via purgeFeedKeysStep.
 */
export async function deleteFeedFastDetailed(
  emailStorage: KVNamespace,
  feedId: string,
): Promise<DeleteFeedFastResult> {
  const feedConfigKey = `feed:${feedId}:config`;
  const feedMetadataKey = `feed:${feedId}:metadata`;

  const errors: string[] = [];
  let configDeleted = false;
  let metadataDeleted = false;

  try {
    await emailStorage.delete(feedConfigKey);
    configDeleted = true;
  } catch (error) {
    errors.push(`config delete failed: ${String(error)}`);
  }

  try {
    await emailStorage.delete(feedMetadataKey);
    metadataDeleted = true;
  } catch (error) {
    errors.push(`metadata delete failed: ${String(error)}`);
  }

  return { ok: configDeleted, configDeleted, metadataDeleted, errors };
}

/**
 * Delete a single feed end-to-end: capture unsubscribe URLs, drop its config +
 * metadata, remove it from the list, bump the counter, and schedule background
 * unsubscribe requests + key purge via ctx.waitUntil. Returns whether the feed
 * was present in the global list.
 */
export async function deleteFeedRecord(
  c: Context<{ Bindings: Env }>,
  env: Env,
  feedId: string,
): Promise<boolean> {
  const emailStorage = env.EMAIL_STORAGE;

  // Read unsubscribe URLs before the metadata is deleted below.
  const unsubscribeUrls = await collectUnsubscribeUrls(emailStorage, feedId);

  await deleteFeedFastDetailed(emailStorage, feedId);
  const removed = await removeFeedFromList(emailStorage, feedId);
  if (removed) {
    await bumpCounters(emailStorage, { feeds_deleted: 1 });
  }

  if (unsubscribeUrls.length > 0) {
    waitUntilSafe(c, sendUnsubscribes(unsubscribeUrls, env));
  }

  waitUntilSafe(
    c,
    purgeFeedKeysStep(emailStorage, feedId, {
      bucket: getAttachmentBucket(env),
    }),
  );

  return removed;
}