Default languages with env & langcodes typed (#94)

pages/[[...slug]].tsx

@@ -9,7 +9,7 @@ import { useHotkeys } from "react-hotkeys-hook";
 import { CustomHead, LangSelect, TranslationArea } from "@components";
 import { useToastOnLoad } from "@hooks";
 import { googleScrape, extractSlug, textToSpeechScrape } from "@utils/translate";
-import { retrieveFromType, replaceBoth } from "@utils/language";
+import { retrieveFromType, replaceBoth, isValid } from "@utils/language";
 import langReducer, { Actions, initialState } from "@utils/reducer";
 import { localGetItem, localSetItem } from "@utils/storage";
 
@@ -46,18 +46,22 @@ const Page: FC<InferGetStaticPropsType<typeof getStaticProps>> = ({ home, transl
     }, [isLoading, source, target, home, initial]);
 
     useEffect(() => {
-        if (home)
+        if (home) {
+            const localSource = localGetItem("source");
+            const localTarget = localGetItem("target");
             return dispatch({
                 type: Actions.SET_ALL,
                 payload: {
                     state: {
                         ...initialState,
-                        source: localGetItem("source") || initialState.source,
-                        target: localGetItem("target") || initialState.target,
+                        source: isValid(localSource) ? localSource : initialState.source,
+                        target: isValid(localTarget) ? localTarget : initialState.target,
                         isLoading: false
                     }
                 }
             });
+        }
+
         if (!initial)
             return;
 
@@ -218,6 +222,11 @@ export const getStaticProps: GetStaticProps = async ({ params }) => {
             }
         }
 
+    if (!isValid(source) || !isValid(target))
+        return {
+            notFound: true
+        };
+
     const textScrape = await googleScrape(source, target, query);
 
     const [sourceAudio, targetAudio] = await Promise.all([

pages/api/graphql.ts

@@ -2,7 +2,7 @@ import { ApolloServer, gql, IResolvers, ApolloError, UserInputError } from "apol
 import { NextApiHandler } from "next";
 import NextCors from "nextjs-cors";
 import { googleScrape, textToSpeechScrape } from "@utils/translate";
-import { retrieveFromType, getName } from "@utils/language";
+import { retrieveFromType, getName, isValid } from "@utils/language";
 
 export const typeDefs = gql`
     enum LangType {
@@ -33,6 +33,10 @@ export const resolvers: IResolvers = {
     Query: {
         translation(_, args) {
             const { source, target, query } = args;
+
+            if (!isValid(source) || !isValid(target))
+                throw new UserInputError("Invalid language code");
+
             return {
                 source: {
                     lang: {
@@ -49,6 +53,10 @@ export const resolvers: IResolvers = {
         },
         audio(_, args) {
             const { lang, query } = args;
+
+            if (!isValid(lang))
+                throw new UserInputError("Invalid language code");
+
             return {
                 lang: {
                     code: lang
@@ -87,13 +95,7 @@ export const resolvers: IResolvers = {
     Language: {
         name(parent) {
             const { code, name } = parent;
-            if (name)
-                return name;
-
-            const newName = getName(code);
-            if (!newName)
-                throw new UserInputError("Invalid language code");
-            return newName;
+            return name || getName(code);
         }
     }
 };

pages/api/v1/[[...slug]].ts

@@ -1,6 +1,7 @@
 import { NextApiHandler } from "next";
 import NextCors from "nextjs-cors";
 import { googleScrape, textToSpeechScrape } from "@utils/translate";
+import { isValid } from "@utils/language";
 
 type Data = {
     translation: string,
@@ -34,6 +35,9 @@ const handler: NextApiHandler<Data> = async (req, res) => {
 
     const [source, target, query] = slug;
 
+    if (!isValid(target))
+        return res.status(400).json({ error: "Invalid target language" });
+
     if (source === "audio") {
         const audio = await textToSpeechScrape(target, query);
         return audio
@@ -41,6 +45,9 @@ const handler: NextApiHandler<Data> = async (req, res) => {
             : res.status(500).json({ error: "An error occurred while retrieving the audio" });
     }
 
+    if (!isValid(source))
+        return res.status(400).json({ error: "Invalid source language" });
+
     const textScrape = await googleScrape(source, target, query);
 
     if ("errorMsg" in textScrape)

pages/api/v1/languages/[[...slug]].ts

@@ -1,10 +1,10 @@
 import { NextApiHandler } from "next";
 import NextCors from "nextjs-cors";
-import { retrieveFromType } from "@utils/language";
+import { retrieveFromType, LangCode } from "@utils/language";
 
 type Data = {
     languages: {
-        code: string,
+        code: LangCode,
         name: string
     }[]
 } | {