Move withdraw endpoints

Author: pycanis

apps/server/src/index.ts

@@ -5,9 +5,10 @@ import cors from "cors";
 import express from "express";
 import rateLimit from "express-rate-limit";
 import helmet from "helmet";
-import { appRouter } from "./routers";
 import { subscribeInvoices } from "./subscribeInvoices";
 import { createContext } from "./trpc";
+import { appRouter } from "./trpcRouters";
+import { router as withdrawRouter } from "./withdrawRouter";
 
 (async () => {
 	const isProd = process.env.NODE_ENV === "production";
@@ -47,6 +48,8 @@ import { createContext } from "./trpc";
 
 	subscribeInvoices();
 
+	app.use("/api/withdraw", withdrawRouter);
+
 	app.use(
 		"/api",
 		createExpressMiddleware({

apps/server/src/routers/payments.ts

@@ -1,7 +1,7 @@
 import { initTRPC } from "@trpc/server";
 import type { CreateExpressContextOptions } from "@trpc/server/adapters/express";
 
-export const createContext = ({ req, res: _res }: CreateExpressContextOptions) => ({ req });
+export const createContext = ({ req: _req, res: _res }: CreateExpressContextOptions) => ({});
 
 type Context = Awaited<ReturnType<typeof createContext>>;
 

apps/server/src/trpc.ts

@@ -0,0 +1,22 @@
+import { on } from "node:events";
+import z from "zod";
+import { PAYMENT_SUCCESS_EVENT } from "../constants";
+import { ee } from "../eventEmitter";
+import { publicProcedure, router } from "../trpc";
+
+const paymentsRouter = router({
+	paymentUpdate: publicProcedure.input(z.string()).subscription(async function* ({
+		input: paymentId,
+		signal,
+	}) {
+		for await (const [data] of on(ee, PAYMENT_SUCCESS_EVENT, {
+			signal,
+		})) {
+			if (data === paymentId) {
+				yield data;
+			}
+		}
+	}),
+});
+
+export { paymentsRouter };

apps/server/src/routers/claims.ts apps/server/src/trpcRouters/claims.tsapps/server/src/routers/claims.ts renamed to apps/server/src/trpcRouters/claims.ts

@@ -0,0 +1,122 @@
+import { ClaimStatus, prisma } from "@repo/database";
+import { getRoutingFee } from "@repo/shared";
+import { Mutex, type MutexInterface, withTimeout } from "async-mutex";
+import { Router } from "express";
+import z from "zod";
+import { PAYMENT_SUCCESS_EVENT } from "./constants";
+import { ee } from "./eventEmitter";
+import { lnGrpcClient, promisifyGrpc, routerGrpcClient } from "./lndClient";
+import type { Payment__Output } from "./protos/generated/lnrpc/Payment";
+
+const router = Router();
+
+const mutexes = new Map<string, MutexInterface>();
+
+router.get("/", async (req, res) => {
+	const { k1, pr } = z.object({ k1: z.string(), pr: z.string() }).parse(req.query);
+
+	if (!mutexes.has(k1)) {
+		mutexes.set(k1, withTimeout(new Mutex(), 15 * 1000)); // todo
+	}
+
+	// biome-ignore lint/style/noNonNullAssertion: It is set on the line above
+	const mutex = mutexes.get(k1)!;
+
+	const release = await mutex.acquire();
+
+	const claim = await prisma.claims.findUnique({ where: { id: k1 } });
+
+	if (!claim || claim.status !== ClaimStatus.PAID) {
+		release();
+
+		res.json({
+			status: "ERROR",
+			reason: "Claim not found, already claimed or not paid for yet.",
+		});
+
+		return;
+	}
+
+	const result = await promisifyGrpc(lnGrpcClient.DecodePayReq.bind(lnGrpcClient), {
+		payReq: pr,
+	});
+
+	if (!result) {
+		release();
+
+		res.json({
+			status: "ERROR",
+			reason: "Invalid pr.",
+		});
+
+		return;
+	}
+
+	if (Number(result.numSatoshis) !== claim.receiverSatsAmount) {
+		release();
+
+		res.json({
+			status: "ERROR",
+			reason: `Claim for ${result.numSatoshis} sats doesn't match the expected claim of ${claim.receiverSatsAmount} sats.`,
+		});
+
+		return;
+	}
+
+	res.json({ status: "OK" });
+
+	const stream = routerGrpcClient.SendPaymentV2({
+		paymentRequest: pr,
+		feeLimitSat: getRoutingFee(claim.receiverSatsAmount),
+		timeoutSeconds: 15, // todo
+	});
+
+	stream.on("data", async (data: Payment__Output) => {
+		if (data.status === "FAILED") {
+			release();
+		}
+
+		if (data.status === "SUCCEEDED") {
+			await prisma.claims.update({
+				where: { id: k1 },
+				data: { status: ClaimStatus.CLAIMED },
+			});
+
+			ee.emit(PAYMENT_SUCCESS_EVENT, claim.paymentRequest);
+
+			release();
+		}
+	});
+
+	stream.on("error", (err) => {
+		console.log("Error happened during payment stream: ", err);
+
+		release();
+	});
+});
+
+router.get("/:claimId", async (req, res) => {
+	const claim = await prisma.claims.findUnique({ where: { id: req.params.claimId } });
+
+	if (!claim || claim.status !== ClaimStatus.PAID) {
+		res.json({
+			status: "ERROR",
+			reason: "Claim not found, already claimed or not paid for yet.",
+		});
+
+		return;
+	}
+
+	const receiverMilliSats = claim.receiverSatsAmount * 1000;
+
+	res.json({
+		tag: "withdrawRequest",
+		callback: "https://stackorange.com/api/withdraw",
+		k1: claim.id,
+		defaultDescription: "Orange pill from stackorange.com",
+		minWithdrawable: receiverMilliSats,
+		maxWithdrawable: receiverMilliSats,
+	});
+});
+
+export { router };

apps/server/src/routers/index.ts apps/server/src/trpcRouters/index.tsapps/server/src/routers/index.ts renamed to apps/server/src/trpcRouters/index.ts

@@ -48,7 +48,7 @@ export const ClaimComponent = () => {
 		}),
 	);
 
-	const withdrawLink = `https://stackorange.com/api/payments.getWithdrawInfo?input="${id}"`;
+	const withdrawLink = `https://stackorange.com/api/withdraw/${id}`;
 
 	const withdrawLinkLnurl = bech32
 		.encode("lnurl", bech32.toWords(Buffer.from(withdrawLink, "utf8")), 1023)

apps/server/src/trpcRouters/payments.ts

@@ -7,7 +7,7 @@ import {
 	splitLink,
 } from "@trpc/client";
 import { createTRPCOptionsProxy } from "@trpc/tanstack-react-query";
-import type { AppRouter } from "../../server/src/routers";
+import type { AppRouter } from "../../server/src/trpcRouters";
 import { API_URL } from "./constants";
 
 export const queryClient = new QueryClient();