File size: 2,324 Bytes
f56a29b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | /**
* Verify Image Provider API
*
* Lightweight endpoint that validates provider credentials without generating images.
*
* POST /api/verify-image-provider
*
* Headers:
* x-image-provider: ImageProviderId
* x-image-model: string (optional)
* x-api-key: string (optional, server fallback)
* x-base-url: string (optional, server fallback)
*
* Response: { success: boolean, message: string }
*/
import { NextRequest } from 'next/server';
import { testImageConnectivity } from '@/lib/media/image-providers';
import { resolveImageApiKey, resolveImageBaseUrl } from '@/lib/server/provider-config';
import type { ImageProviderId } from '@/lib/media/types';
import { apiError, apiSuccess } from '@/lib/server/api-response';
import { createLogger } from '@/lib/logger';
import { validateUrlForSSRF } from '@/lib/server/ssrf-guard';
const log = createLogger('VerifyImageProvider');
export async function POST(request: NextRequest) {
try {
const providerId = (request.headers.get('x-image-provider') || 'seedream') as ImageProviderId;
const model = request.headers.get('x-image-model') || undefined;
const clientApiKey = request.headers.get('x-api-key') || undefined;
const clientBaseUrl = request.headers.get('x-base-url') || undefined;
if (clientBaseUrl && process.env.NODE_ENV === 'production') {
const ssrfError = await validateUrlForSSRF(clientBaseUrl);
if (ssrfError) {
return apiError('INVALID_URL', 403, ssrfError);
}
}
const apiKey = clientBaseUrl
? clientApiKey || ''
: resolveImageApiKey(providerId, clientApiKey);
const baseUrl = clientBaseUrl ? clientBaseUrl : resolveImageBaseUrl(providerId, clientBaseUrl);
if (!apiKey) {
return apiError('MISSING_API_KEY', 400, 'No API key configured');
}
const result = await testImageConnectivity({
providerId,
apiKey,
baseUrl,
model,
});
if (!result.success) {
return apiError('UPSTREAM_ERROR', 500, result.message);
}
return apiSuccess({ message: result.message });
} catch (err) {
log.error(
`Image provider verification failed [provider=${request.headers.get('x-image-provider') ?? 'seedream'}]:`,
err,
);
return apiError('INTERNAL_ERROR', 500, `Connectivity test error: ${err}`);
}
}
|