CloudHost/LingvAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor to external scraper and update dependencies (#113)

Browse Source
This commit is contained in:
David
2022-06-15 23:37:15 +02:00
committed by GitHub
parent ff1ad202ae
commit 274e7f1a4b
49 changed files with 6952 additions and 4414 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pages/api/v1/[[...slug]].ts
View File

@@ -1,10 +1,10 @@
import { NextApiHandler } from "next";
import NextCors from "nextjs-cors";
import { googleScrape, textToSpeechScrape } from "@utils/translate";
import { isValid } from "@utils/language";
import { getTranslationInfo, getTranslationText, getAudio, isValidCode, LanguageType, TranslationInfo } from "lingva-scraper";
type Data = {
translation: string,
info?: TranslationInfo
} | {
audio: number[]
} | {
@@ -35,24 +35,29 @@ const handler: NextApiHandler<Data> = async (req, res) => {
const [source, target, query] = slug;
if (!isValid(target))
if (!isValidCode(target, LanguageType.TARGET))
return res.status(400).json({ error: "Invalid target language" });
if (source === "audio") {
const audio = await textToSpeechScrape(target, query);
const audio = await getAudio(target, query);
return audio
? res.status(200).json({ audio })
: res.status(500).json({ error: "An error occurred while retrieving the audio" });
}
if (!isValid(source))
if (!isValidCode(source, LanguageType.SOURCE))
return res.status(400).json({ error: "Invalid source language" });
const textScrape = await googleScrape(source, target, query);
const translation = await getTranslationText(source, target, query);
if ("errorMsg" in textScrape)
return res.status(500).json({ error: textScrape.errorMsg });
res.status(200).json({ translation: textScrape.translationRes });
if (!translation)
return res.status(500).json({ error: "An error occurred while retrieving the translation" });
const info = await getTranslationInfo(source, target, query);
return info
? res.status(200).json({ translation, info })
: res.status(200).json({ translation });
}
export default handler;

4
pages/api/v1/languages/[[...slug]].ts
View File

@@ -1,6 +1,6 @@
import { NextApiHandler } from "next";
import NextCors from "nextjs-cors";
import { retrieveFromType, LangCode } from "@utils/language";
import { languageList, LangCode } from "lingva-scraper";
type Data = {
languages: {
@@ -38,7 +38,7 @@ const handler: NextApiHandler<Data> = async (req, res) => {
if (type !== undefined && type !== "source" && type !== "target")
return res.status(400).json({ error: "Type should be 'source', 'target' or empty" });
const langEntries = retrieveFromType(type);
const langEntries = Object.entries(languageList[type ?? "all"]) as [LangCode, string][];
const languages = langEntries.map(([code, name]) => ({ code, name }));
res.status(200).json({ languages });