Fix get input script

nguyenhuuit · nguyenhuuit · commit 761f6543c60a · 2023-12-02T13:54:40.000+07:00
diff --git a/src/constants.ts b/src/constants.ts
@@ -1,3 +1,14 @@
-export const VALID_YEARS = ['2015','2016','2017','2018','2019','2020','2021','2022'];
+export const VALID_YEARS = [
+  '2015',
+  '2016',
+  '2017',
+  '2018',
+  '2019',
+  '2020',
+  '2021',
+  '2022',
+  '2023',
+  '2024'
+];
 
 export const HOST = 'https://adventofcode.com';
diff --git a/src/hooks/useInputFile.ts b/src/hooks/useInputFile.ts
@@ -3,14 +3,27 @@ import fs from 'fs';
 import { useEffect, useState } from "react";
 import { HOST, VALID_YEARS } from '../constants.js';
 
-const SAMPLE_REGEX = /\(your puzzle input\)[\s\S]*?<code>([\s\S]+?)<\/code>/;
-
 const decode = (str: string) => {
   return str
     .replaceAll('&gt;', '>')
-    .replaceAll('&lt;', '<');
+    .replaceAll('&lt;', '<')
+    .replaceAll('<code>', '')
+    .replaceAll('</code>', '')
+    .replaceAll('<em>', '')
+    .replaceAll('</em>', '')
+    ;
 };
 
+const findLongest = (str: string[]) => {
+  let longest = ''
+  str.forEach((s: string) => {
+    if (s.length > longest.length) {
+      longest = s
+    }
+  })
+  return longest
+}
+
 export const useInputFile = (year: string, day: string, inp: string, ts: number): AppFile => {
   const [name, setName] = useState<string>('')
   const [size, setSize] = useState<number>(0)
@@ -56,9 +69,10 @@ export const useInputFile = (year: string, day: string, inp: string, ts: number)
         }
       }).then(res => {
         if (res.data) {
+          const SAMPLE_REGEX = /<code>(<em>)?([\s\S]+?)(<\/em>)?<\/code>/g;
           const matches = res.data.match(SAMPLE_REGEX);
           if (matches) {
-            fs.writeFileSync(file, decode(matches[1]).trim());
+            fs.writeFileSync(file, decode(findLongest(matches)).trim());
             const stats = fs.statSync(file)
             setSize(stats.size);
           }
