ditto/src/cron.ts

import { Stickynotes } from '@soapbox/stickynotes';

import { Conf } from '@/config.ts';
import { DittoDB } from '@/db/DittoDB.ts';
import { handleEvent } from '@/pipeline.ts';
import { AdminSigner } from '@/signers/AdminSigner.ts';
import { getTrendingTagValues } from '@/trends/trending-tag-values.ts';
import { Time } from '@/utils/time.ts';

const console = new Stickynotes('ditto:trends');

async function updateTrendingTags(
  l: string,
  tagName: string,
  kinds: number[],
  limit: number,
  extra = '',
  aliases?: string[],
) {
  console.info(`Updating trending ${l}...`);
  const kysely = await DittoDB.getInstance();
  const signal = AbortSignal.timeout(1000);

  const yesterday = Math.floor((Date.now() - Time.days(1)) / 1000);
  const now = Math.floor(Date.now() / 1000);

  const tagNames = aliases ? [tagName, ...aliases] : [tagName];

  const trends = await getTrendingTagValues(kysely, tagNames, {
    kinds,
    since: yesterday,
    until: now,
    limit,
  });

  if (!trends.length) {
    return;
  }

  const signer = new AdminSigner();

  const label = await signer.signEvent({
    kind: 1985,
    content: '',
    tags: [
      ['L', 'pub.ditto.trends'],
      ['l', l, 'pub.ditto.trends'],
      ...trends.map(({ value, authors, uses }) => [tagName, value, extra, authors.toString(), uses.toString()]),
    ],
    created_at: Math.floor(Date.now() / 1000),
  });

  await handleEvent(label, signal);
  console.info(`Trending ${l} updated.`);
}

/** Start cron jobs for the application. */
export function cron() {
  Deno.cron(
    'update trending pubkeys',
    '0 * * * *',
    () => updateTrendingTags('#p', 'p', [1, 6, 7, 9735], 40, Conf.relay),
  );
  Deno.cron(
    'update trending zapped events',
    '7 * * * *',
    () => updateTrendingTags('zapped', 'e', [9735], 40, Conf.relay, ['q']),
  );
  Deno.cron(
    'update trending events',
    '15 * * * *',
    () => updateTrendingTags('#e', 'e', [1, 6, 7, 9735], 40, Conf.relay, ['q']),
  );
  Deno.cron(
    'update trending hashtags',
    '30 * * * *',
    () => updateTrendingTags('#t', 't', [1], 20),
  );
  Deno.cron(
    'update trending links',
    '45 * * * *',
    () => updateTrendingTags('#r', 'r', [1], 20),
  );
}