Update endpoints/antibot.js

endpoints/antibot.js

@@ -26,11 +26,14 @@ function normalize(x) {
   const roman = {i:1,ii:2,iii:3,iv:4,v:5,vi:6,vii:7,viii:8,ix:9,x:10};
   if (roman[t]) return roman[t];
 
-  const word = {one:1,two:2,three:3,four:4,five:5,six:6,seven:7,eight:8,nine:9,ten:10};
+  const word = {
+    zero:0, one:1, two:2, three:3, four:4, five:5,
+    six:6, seven:7, eight:8, nine:9, ten:10
+  };
   if (word[t]) return word[t];
 
   try {
-    if (/[\d+\-*/]/.test(t)) {
+    if (/^[0-9+\-*/]+$/.test(t)) {
       const v = eval(t);
       if (!isNaN(v)) return Number(v);
     }
@@ -43,6 +46,20 @@ function normalize(x) {
     .replace(/1/g,"i");
 }
 
+function solveTokens(tokens) {
+  const n = tokens.map(normalize);
+
+  if (n.length === 3 && typeof n[0] === "number" && typeof n[2] === "number") {
+    const op = String(tokens[1]).replace(/[^+\-*/]/g, "");
+    try {
+      const v = eval(`${n[0]}${op}${n[2]}`);
+      if (!isNaN(v)) return v;
+    } catch {}
+  }
+
+  return n;
+}
+
 async function preprocessSoal(b) {
   return sharp(b)
     .resize({ width: 1000 })
@@ -88,7 +105,8 @@ module.exports = async data => {
   const soalProcessed = await preprocessSoal(soalImg);
   const soalText = await ocr(soalProcessed, "soal");
   const soalRaw = cleanSoal(soalText);
-  const soalNorm = soalRaw.map(normalize);
+
+  let soalNormalized = solveTokens(soalRaw);
 
   const botResults = [];
   for (const b of data.bots) {
@@ -103,14 +121,16 @@ module.exports = async data => {
     });
   }
 
-  const result = soalNorm.map(v => {
+  const target = Array.isArray(soalNormalized) ? soalNormalized : [soalNormalized];
+
+  const result = target.map(v => {
     const f = botResults.find(x => x.value === v);
     return f ? f.id : null;
   });
 
   return {
     soal: soalRaw,
-    soalNormalized: soalNorm,
+    soalNormalized: soalNormalized,
     botResults,
     result
   };