From 739fea08905a8d98f2c5850cfbbb64cdb826797a Mon Sep 17 00:00:00 2001
From: "P. Reis" <patrickpereirareal1@gmail.com>
Date: Tue, 4 Jun 2024 13:37:04 -0300
Subject: [PATCH] feat: zap profiles directly Rewrite zapController to return
 invoice in body instead of in header

---
 src/app.ts                      |  5 +--
 src/controllers/api/statuses.ts | 63 +++++++++++++++++++++------------
 2 files changed, 43 insertions(+), 25 deletions(-)

diff --git a/src/app.ts b/src/app.ts
index a9509b4..bb1c5b6 100644
--- a/src/app.ts
+++ b/src/app.ts
@@ -139,7 +139,7 @@ app.get('/relay', relayController);
 app.use(
   '*',
   cspMiddleware(),
-  cors({ origin: '*', exposeHeaders: ['link', 'Ln-Invoice'] }),
+  cors({ origin: '*', exposeHeaders: ['link'] }),
   signerMiddleware,
   uploaderMiddleware,
   auth98Middleware(),
@@ -188,7 +188,6 @@ app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/bookmark', requireSigner, bookmarkC
 app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/unbookmark', requireSigner, unbookmarkController);
 app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/pin', requireSigner, pinController);
 app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/unpin', requireSigner, unpinController);
-app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/zap', requireSigner, zapController);
 app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/reblog', requireSigner, reblogStatusController);
 app.post('/api/v1/statuses/:id{[0-9a-f]{64}}/unreblog', requireSigner, unreblogStatusController);
 app.post('/api/v1/statuses', requireSigner, createStatusController);
@@ -240,6 +239,8 @@ app.delete('/api/v1/pleroma/admin/statuses/:id', requireRole('admin'), pleromaAd
 app.get('/api/v1/admin/ditto/relays', requireRole('admin'), adminRelaysController);
 app.put('/api/v1/admin/ditto/relays', requireRole('admin'), adminSetRelaysController);
 
+app.post('/api/v1/ditto/zap', requireSigner, zapController);
+
 app.post('/api/v1/reports', requireSigner, reportController);
 app.get('/api/v1/admin/reports', requireSigner, requireRole('admin'), adminReportsController);
 app.get('/api/v1/admin/reports/:id{[0-9a-f]{64}}', requireSigner, requireRole('admin'), adminReportController);
diff --git a/src/controllers/api/statuses.ts b/src/controllers/api/statuses.ts
index 4666c34..d532d02 100644
--- a/src/controllers/api/statuses.ts
+++ b/src/controllers/api/statuses.ts
@@ -19,6 +19,7 @@ import { getInvoice, getLnurl } from '@/utils/lnurl.ts';
 import { lookupPubkey } from '@/utils/lookup.ts';
 import { addTag, deleteTag } from '@/utils/tags.ts';
 import { asyncReplaceAll } from '@/utils/text.ts';
+import { DittoEvent } from '@/interfaces/DittoEvent.ts';
 
 const createStatusSchema = z.object({
   in_reply_to_id: n.id().nullish(),
@@ -460,47 +461,63 @@ const unpinController: AppController = async (c) => {
 };
 
 const zapSchema = z.object({
+  account_id: n.id(),
+  status_id: n.id().optional().or(z.literal('')),
   amount: z.number().int().positive(),
   comment: z.string().optional(),
 });
 
 const zapController: AppController = async (c) => {
-  const id = c.req.param('id');
   const body = await parseBody(c.req.raw);
-  const params = zapSchema.safeParse(body);
+  const result = zapSchema.safeParse(body);
   const { signal } = c.req.raw;
 
-  if (!params.success) {
-    return c.json({ error: 'Bad request', schema: params.error }, 400);
+  if (!result.success) {
+    return c.json({ error: 'Bad request', schema: result.error }, 400);
   }
 
-  const target = await getEvent(id, { kind: 1, relations: ['author', 'event_stats', 'author_stats'], signal });
-  const author = target?.author;
-  const meta = n.json().pipe(n.metadata()).catch({}).parse(author?.content);
-  const lnurl = getLnurl(meta);
-  const amount = params.data.amount;
+  const { account_id, status_id, amount, comment } = result.data;
 
-  if (target && lnurl) {
-    const nostr = await createEvent({
-      kind: 9734,
-      content: params.data.comment ?? '',
-      tags: [
+  const tags: string[][] = [];
+  let target: undefined | DittoEvent;
+  let lnurl: undefined | string;
+
+  if (status_id) {
+    target = await getEvent(status_id, { kind: 1, relations: ['author'], signal });
+    const author = target?.author;
+    const meta = n.json().pipe(n.metadata()).catch({}).parse(author?.content);
+    lnurl = getLnurl(meta);
+    if (target && lnurl) {
+      tags.push(
         ['e', target.id],
         ['p', target.pubkey],
         ['amount', amount.toString()],
         ['relays', Conf.relay],
         ['lnurl', lnurl],
-      ],
+      );
+    }
+  } else {
+    target = await getEvent(account_id, { kind: 0, signal });
+    const meta = n.json().pipe(n.metadata()).catch({}).parse(target?.content);
+    lnurl = getLnurl(meta);
+    if (target && lnurl) {
+      tags.push(
+        ['p', target.pubkey],
+        ['amount', amount.toString()],
+        ['relays', Conf.relay],
+        ['lnurl', lnurl],
+      );
+    }
+  }
+
+  if (target && lnurl) {
+    const nostr = await createEvent({
+      kind: 9734,
+      content: comment ?? '',
+      tags,
     }, c);
 
-    const status = await renderStatus(target, { viewerPubkey: await c.get('signer')?.getPublicKey() });
-    status.zapped = true;
-
-    return c.json(status, {
-      headers: {
-        'Ln-Invoice': await getInvoice({ amount, nostr: purifyEvent(nostr), lnurl }, signal),
-      },
-    });
+    return c.json({ invoice: await getInvoice({ amount, nostr: purifyEvent(nostr), lnurl }, signal) });
   } else {
     return c.json({ error: 'Event not found.' }, 404);
   }