105 lines
3.1 KiB
TypeScript
105 lines
3.1 KiB
TypeScript
import { NextResponse } from 'next/server';
|
|
import { createClient } from '@/lib/supabase/server';
|
|
import { createAdminClient } from '@/lib/supabase/admin';
|
|
import { getTodayUsage, recordUsage, MUSIC_DAILY_LIMIT } from '@/lib/ai-usage';
|
|
|
|
export const runtime = 'nodejs';
|
|
|
|
type GenerateBody = {
|
|
mode: 'simple' | 'custom';
|
|
prompt?: string;
|
|
title?: string;
|
|
lyrics?: string;
|
|
tags?: string;
|
|
make_instrumental?: boolean;
|
|
model?: string;
|
|
};
|
|
|
|
export async function POST(request: Request) {
|
|
// 1) 인증 — 로그인 사용자만 (Suno API 무단 호출 방지)
|
|
const supabase = await createClient();
|
|
const { data: { user } } = await supabase.auth.getUser();
|
|
if (!user) {
|
|
return NextResponse.json({ error: '로그인이 필요합니다.' }, { status: 401 });
|
|
}
|
|
|
|
// 2) 일일 제한
|
|
const admin = createAdminClient();
|
|
const used = await getTodayUsage(admin, user.id, 'music');
|
|
if (used >= MUSIC_DAILY_LIMIT) {
|
|
return NextResponse.json(
|
|
{ error: `오늘 음악 생성을 모두 사용했습니다. (${MUSIC_DAILY_LIMIT}회/일)` },
|
|
{ status: 429 }
|
|
);
|
|
}
|
|
|
|
const apiUrl = process.env.SUNO_API_URL ?? 'https://api.sunoapi.org';
|
|
const apiKey = process.env.SUNO_API_KEY;
|
|
|
|
if (!apiKey) {
|
|
return NextResponse.json(
|
|
{ error: 'Suno API 미설정 (SUNO_API_KEY 환경변수 필요)' },
|
|
{ status: 503 },
|
|
);
|
|
}
|
|
|
|
let body: GenerateBody;
|
|
try {
|
|
body = (await request.json()) as GenerateBody;
|
|
} catch {
|
|
return NextResponse.json({ error: 'Invalid JSON' }, { status: 400 });
|
|
}
|
|
|
|
const origin = new URL(request.url).origin;
|
|
const callBackUrl = `${origin}/api/studio/callback`;
|
|
|
|
const isCustom = body.mode === 'custom';
|
|
const payload = isCustom
|
|
? {
|
|
prompt: body.lyrics ?? '',
|
|
style: body.tags ?? '',
|
|
title: body.title ?? 'Untitled',
|
|
customMode: true,
|
|
instrumental: !!body.make_instrumental,
|
|
model: body.model ?? 'V4',
|
|
callBackUrl,
|
|
}
|
|
: {
|
|
prompt: body.prompt ?? '',
|
|
customMode: false,
|
|
instrumental: !!body.make_instrumental,
|
|
model: body.model ?? 'V4',
|
|
callBackUrl,
|
|
};
|
|
|
|
try {
|
|
const res = await fetch(`${apiUrl.replace(/\/$/, '')}/api/v1/generate`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
Authorization: `Bearer ${apiKey}`,
|
|
},
|
|
body: JSON.stringify(payload),
|
|
});
|
|
|
|
const data = await res.json().catch(() => null);
|
|
if (!res.ok || (data && typeof data === 'object' && 'code' in data && data.code !== 200)) {
|
|
return NextResponse.json(
|
|
{ error: '생성 실패', detail: data ?? (await res.text().catch(() => '')) },
|
|
{ status: res.ok ? 502 : res.status },
|
|
);
|
|
}
|
|
try {
|
|
await recordUsage(admin, user.id, 'music');
|
|
} catch {
|
|
/* 집계 실패는 무시 — 생성은 이미 성공 */
|
|
}
|
|
return NextResponse.json({ ok: true, data });
|
|
} catch (e) {
|
|
return NextResponse.json(
|
|
{ error: 'Suno API 호출 오류', detail: e instanceof Error ? e.message : String(e) },
|
|
{ status: 502 },
|
|
);
|
|
}
|
|
}
|