forked from akai/readest
Huang Xin
70 lines
2.2 KiB
TypeScript
70 lines
2.2 KiB
TypeScript
import { NextApiRequest, NextApiResponse } from 'next';
|
|
import { corsAllMethods, runMiddleware } from '@/utils/cors';
|
|
import { supabase } from '@/utils/supabase';
|
|
import { getUserPlan } from '@/utils/access';
|
|
import { query as deeplQuery } from '@/utils/deepl';
|
|
|
|
const DEEPL_FREE_API = 'https://api-free.deepl.com/v2/translate';
|
|
const DEEPL_PRO_API = 'https://api.deepl.com/v2/translate';
|
|
|
|
const getUserAndToken = async (authHeader: string | undefined) => {
|
|
if (!authHeader) return {};
|
|
|
|
const token = authHeader.replace('Bearer ', '');
|
|
const {
|
|
data: { user },
|
|
error,
|
|
} = await supabase.auth.getUser(token);
|
|
|
|
if (error || !user) return {};
|
|
return { user, token };
|
|
};
|
|
|
|
const getDeepLAPIKey = (keys: string | undefined) => {
|
|
const keyArray = keys?.split(',') ?? [];
|
|
return keyArray.length ? keyArray[Math.floor(Math.random() * keyArray.length)] : '';
|
|
};
|
|
|
|
const handler = async (req: NextApiRequest, res: NextApiResponse) => {
|
|
const { user, token } = await getUserAndToken(req.headers['authorization']);
|
|
let deeplApiUrl = DEEPL_FREE_API;
|
|
let userPlan = 'free';
|
|
if (user && token) {
|
|
userPlan = getUserPlan(token);
|
|
if (userPlan !== 'free') deeplApiUrl = DEEPL_PRO_API;
|
|
}
|
|
const deeplAuthKey =
|
|
deeplApiUrl === DEEPL_PRO_API
|
|
? getDeepLAPIKey(process.env['DEEPL_PRO_API_KEYS'])
|
|
: getDeepLAPIKey(process.env['DEEPL_FREE_API_KEYS']);
|
|
|
|
await runMiddleware(req, res, corsAllMethods);
|
|
|
|
const { text, source_lang = 'auto', target_lang = 'en' } = req.body;
|
|
try {
|
|
if (userPlan !== 'pro') {
|
|
const result = await deeplQuery({
|
|
text: text[0],
|
|
sourceLang: source_lang,
|
|
targetLang: target_lang,
|
|
});
|
|
return res.status(200).json(result);
|
|
}
|
|
const response = await fetch(deeplApiUrl, {
|
|
method: 'POST',
|
|
headers: {
|
|
Authorization: `DeepL-Auth-Key ${deeplAuthKey}`,
|
|
'Content-Type': 'application/json',
|
|
},
|
|
body: req.method === 'POST' ? JSON.stringify(req.body) : undefined,
|
|
});
|
|
res.status(response.status);
|
|
res.json(await response.json());
|
|
} catch (error) {
|
|
console.error('Error proxying DeepL request:', error);
|
|
res.status(500).json({ error: 'Internal Server Error' });
|
|
}
|
|
};
|
|
|
|
export default handler;
|