ditto/packages/ditto/controllers/api/pleroma.ts

import { z } from 'zod';

import { type AppController } from '@/app.ts';
import { configSchema, elixirTupleSchema } from '@/schemas/pleroma-api.ts';
import { createAdminEvent, updateAdminEvent, updateUser } from '@/utils/api.ts';
import { lookupPubkey } from '@/utils/lookup.ts';
import { getPleromaConfigs } from '@/utils/pleroma.ts';

const frontendConfigController: AppController = async (c) => {
  const { relay, signal } = c.var;

  const configDB = await getPleromaConfigs(relay, signal);
  const frontendConfig = configDB.get(':pleroma', ':frontend_configurations');

  if (frontendConfig) {
    const schema = elixirTupleSchema.transform(({ tuple }) => tuple).array();
    const data = schema.parse(frontendConfig.value).reduce<Record<string, unknown>>((result, [name, data]) => {
      result[name.replace(/^:/, '')] = data;
      return result;
    }, {});
    return c.json(data);
  } else {
    return c.json({});
  }
};

const configController: AppController = async (c) => {
  const { relay, signal } = c.var;

  const configs = await getPleromaConfigs(relay, signal);
  return c.json({ configs, need_reboot: false });
};

/** Pleroma admin config controller. */
const updateConfigController: AppController = async (c) => {
  const { conf, relay, signal } = c.var;

  const configs = await getPleromaConfigs(relay, signal);
  const { configs: newConfigs } = z.object({ configs: z.array(configSchema) }).parse(await c.req.json());

  configs.merge(newConfigs);

  await createAdminEvent({
    kind: 30078,
    content: await conf.signer.nip44.encrypt(await conf.signer.getPublicKey(), JSON.stringify(configs)),
    tags: [
      ['d', 'pub.ditto.pleroma.config'],
      ['encrypted', 'nip44'],
    ],
  }, c);

  return c.json({ configs: newConfigs, need_reboot: false });
};

const pleromaAdminDeleteStatusController: AppController = async (c) => {
  await createAdminEvent({
    kind: 5,
    tags: [['e', c.req.param('id')]],
  }, c);

  return c.json({});
};

const pleromaAdminTagSchema = z.object({
  nicknames: z.string().array(),
  tags: z.string().array(),
});

const pleromaAdminTagController: AppController = async (c) => {
  const { conf } = c.var;
  const params = pleromaAdminTagSchema.parse(await c.req.json());

  for (const nickname of params.nicknames) {
    const pubkey = await lookupPubkey(nickname);
    if (!pubkey) continue;

    await updateAdminEvent(
      { kinds: [30382], authors: [await conf.signer.getPublicKey()], '#d': [pubkey], limit: 1 },
      (prev) => {
        const tags = prev?.tags ?? [['d', pubkey]];

        for (const tag of params.tags) {
          const existing = prev?.tags.some(([name, value]) => name === 't' && value === tag);
          if (!existing) {
            tags.push(['t', tag]);
          }
        }

        return {
          kind: 30382,
          content: prev?.content ?? '',
          tags,
        };
      },
      c,
    );
  }

  return c.newResponse(null, { status: 204 });
};

const pleromaAdminUntagController: AppController = async (c) => {
  const { conf } = c.var;
  const params = pleromaAdminTagSchema.parse(await c.req.json());

  for (const nickname of params.nicknames) {
    const pubkey = await lookupPubkey(nickname);
    if (!pubkey) continue;

    await updateAdminEvent(
      { kinds: [30382], authors: [await conf.signer.getPublicKey()], '#d': [pubkey], limit: 1 },
      (prev) => ({
        kind: 30382,
        content: prev?.content ?? '',
        tags: (prev?.tags ?? [['d', pubkey]])
          .filter(([name, value]) => !(name === 't' && params.tags.includes(value))),
      }),
      c,
    );
  }

  return c.newResponse(null, { status: 204 });
};

const pleromaAdminSuggestSchema = z.object({
  nicknames: z.string().array(),
});

const pleromaAdminSuggestController: AppController = async (c) => {
  const { nicknames } = pleromaAdminSuggestSchema.parse(await c.req.json());

  for (const nickname of nicknames) {
    const pubkey = await lookupPubkey(nickname);
    if (!pubkey) continue;
    await updateUser(pubkey, { suggested: true }, c);
  }

  return c.newResponse(null, { status: 204 });
};

const pleromaAdminUnsuggestController: AppController = async (c) => {
  const { nicknames } = pleromaAdminSuggestSchema.parse(await c.req.json());

  for (const nickname of nicknames) {
    const pubkey = await lookupPubkey(nickname);
    if (!pubkey) continue;
    await updateUser(pubkey, { suggested: false }, c);
  }

  return c.newResponse(null, { status: 204 });
};

export {
  configController,
  frontendConfigController,
  pleromaAdminDeleteStatusController,
  pleromaAdminSuggestController,
  pleromaAdminTagController,
  pleromaAdminUnsuggestController,
  pleromaAdminUntagController,
  updateConfigController,
};