Sweetmantech/myc 3555 apiimagegenerate use credits

app/api/image/generate/route.ts

@@ -5,7 +5,7 @@ import { z } from "zod";
 
 const queryParamsSchema = z.object({
  prompt: z.string().min(1, "The provided prompt is invalid or empty"),
-artist_account_id: z.string().min(1, "The provided artist_account_id is invalid or not found"),
+account_id: z.string().min(1, "The provided account_id is invalid or not found"),
 });
 
 /**
@@ -22,7 +22,7 @@ export async function OPTIONS() {
 
 /**
  * GET handler for image generation endpoint.
- * Accepts prompt and artist_account_id, and fetches from the x402-protected endpoint.
+ * Accepts prompt and account_id, and fetches from the x402-protected endpoint.
  *
  * @param request - The request object containing query parameters.
  * @returns {Promise<NextResponse>} JSON response matching the Recoup API format.
@@ -54,10 +54,10 @@ export async function GET(request: NextRequest) {
  );
  }
 
- const { prompt } = validationResult.data;
+ const { prompt, account_id } = validationResult.data;
 
  const baseUrl = request.nextUrl.origin;
- const data = await x402GenerateImage(prompt, baseUrl);
+ const data = await x402GenerateImage(prompt, baseUrl, account_id);
 
  return NextResponse.json(data, {
  status: 200,

lib/credits/deductCredits.ts

@@ -0,0 +1,58 @@
+import { selectCreditsUsage } from "@/lib/supabase/credits_usage/selectCreditsUsage";
+import { updateCreditsUsage } from "@/lib/supabase/credits_usage/updateCreditsUsage";
+
+interface DeductCreditsParams {
+ accountId: string;
+ creditsToDeduct: number;
+}
+
+interface DeductCreditsResult {
+ success: boolean;
+ newBalance?: number;
+ message?: string;
+}
+
+/**
+ * Deducts credits from an account's credit balance.
+ *
+ * @param params - The parameters for deducting credits.
+ * @param params.accountId - The account ID to deduct credits from.
+ * @param params.creditsToDeduct - The number of credits to deduct.
+ * @returns Result object indicating success/failure and new balance.
+ */
+export const deductCredits = async ({
+ accountId,
+ creditsToDeduct,
+}: DeductCreditsParams): Promise<DeductCreditsResult> => {
+ try {
+ const response = await selectCreditsUsage({ account_id: accountId });
+
+ if (!response || response.length === 0) {
+ return {
+ success: false,
+ message: "No credits usage found for this account",
+ };
+ }
+
+ const currentCredits = response[0];
+ const newBalance = currentCredits.remaining_credits - creditsToDeduct;
+
+ await updateCreditsUsage({
+ account_id: accountId,
+ updates: {
+ remaining_credits: newBalance,
+ },
+ });
+
+ return {
+ success: true,
+ newBalance,
+ };
+ } catch (error) {
+ console.error("Failed to deduct credits:", error);
+ return {
+ success: false,
+ message: error instanceof Error ? error.message : "Unknown error occurred",
+ };
+ }
+};

lib/supabase/credits_usage/selectCreditsUsage.ts

@@ -0,0 +1,27 @@
+import serverClient from "../serverClient";
+import { Tables } from "@/types/database.types";
+
+export type CreditsUsage = Tables<"credits_usage">;
+
+interface SelectCreditsUsageParams {
+ account_id?: string;
+}
+
+export const selectCreditsUsage = async (
+ params?: SelectCreditsUsageParams,
+): Promise<CreditsUsage[]> => {
+ let query = serverClient.from("credits_usage").select();
+
+ if (params?.account_id) {
+ query = query.eq("account_id", params.account_id);
+ }
+
+ const { data, error } = await query;
+
+ if (error) {
+ console.error("Error selecting credits usage:", error);
+ throw error;
+ }
+
+ return data;
+};

lib/supabase/credits_usage/updateCreditsUsage.ts

@@ -0,0 +1,30 @@
+import serverClient from "../serverClient";
+import { CreditsUsage } from "./selectCreditsUsage";
+
+interface UpdateCreditsUsageParams {
+ account_id: string;
+ updates: Partial<Omit<CreditsUsage, "account_id">>;
+}
+
+export const updateCreditsUsage = async ({
+ account_id,
+ updates,
+}: UpdateCreditsUsageParams): Promise<CreditsUsage> => {
+ const { data, error } = await serverClient
+ .from("credits_usage")
+ .update(updates)
+ .eq("account_id", account_id)
+ .select()
+ .single();
+
+ if (error) {
+ console.error("Error updating credits usage:", error);
+ throw error;
+ }
+
+ if (!data) {
+ throw new Error(`No credits usage found for account_id: ${account_id}`);
+ }
+
+ return data;
+};

lib/supabase/serverClient.ts

@@ -0,0 +1,12 @@
+import { createClient } from "@supabase/supabase-js";
+
+const SUPABASE_URL = process.env.SUPABASE_URL as string;
+const SUPABASE_KEY = process.env.SUPABASE_KEY as string;
+
+if (!SUPABASE_URL || !SUPABASE_KEY) {
+ throw new Error("SUPABASE_URL and SUPABASE_KEY must be set");
+}
+
+const supabase = createClient(SUPABASE_URL, SUPABASE_KEY);
+
+export default supabase;

lib/x402/fetchWithPayment.ts

@@ -1,17 +1,20 @@
 import { wrapFetchWithPayment } from "x402-fetch";
 import { toAccount } from "viem/accounts";
 import { getAccount } from "@/lib/coinbase/getAccount";
+import { deductCredits } from "../credits/deductCredits";
 
 const ACCOUNT_ADDRESS = "0x7AfB9872Ea382B7Eb01c67B6884dD99A744eA64f" as const;
 
 /**
  * Fetches a URL with x402 payment handling.
  *
  * @param url - The URL to fetch.
+ * @param accountId - The account ID.
  * @returns Promise resolving to the Response.
  */
-export async function fetchWithPayment(url: string): Promise<Response> {
+export async function fetchWithPayment(url: string, accountId: string): Promise<Response> {
  const account = await getAccount(ACCOUNT_ADDRESS);
+ await deductCredits({ accountId, creditsToDeduct: 1 });
  const fetchWithPaymentWrapper = wrapFetchWithPayment(fetch, toAccount(account));
  return fetchWithPaymentWrapper(url, {
  method: "GET",

lib/x402/recoup/x402GenerateImage.ts

@@ -5,12 +5,17 @@ import { fetchWithPayment } from "../fetchWithPayment";
  *
  * @param prompt - The text prompt describing the image to generate.
  * @param baseUrl - The base URL for the API.
+ * @param accountId - The account ID.
  * @returns Promise resolving to the generated image data.
  */
-export async function x402GenerateImage(prompt: string, baseUrl: string): Promise<unknown> {
+export async function x402GenerateImage(
+ prompt: string,
+ baseUrl: string,
+ accountId: string,
+): Promise<unknown> {
  const x402Url = new URL("/api/x402/image/generate", baseUrl);
  x402Url.searchParams.set("prompt", prompt);
 
- const response = await fetchWithPayment(x402Url.toString());
+ const response = await fetchWithPayment(x402Url.toString(), accountId);
  return response.json();
 }

package.json

@@ -18,6 +18,7 @@
  "@coinbase/x402": "^0.7.3",
  "@composio/core": "^0.2.6",
  "@composio/vercel": "^0.2.18",
+ "@supabase/supabase-js": "^2.86.0",
  "ai": "6.0.0-beta.122",
  "arweave": "^1.15.7",
  "multiformats": "^13.4.1",