From d243fa241a2240304d4fc8f62dd2695195eac5aa Mon Sep 17 00:00:00 2001
From: Zachary <hello@zach.ie>
Date: Mon, 11 Nov 2024 19:06:06 +0100
Subject: [PATCH] Move to trace endpoint

---
 src/routes/+page.svelte | 33 +++++++++++++++++----------------
 1 file changed, 17 insertions(+), 16 deletions(-)

diff --git a/src/routes/+page.svelte b/src/routes/+page.svelte
index b5e1ee3..fc13859 100644
--- a/src/routes/+page.svelte
+++ b/src/routes/+page.svelte
@@ -1,15 +1,3 @@
-<script context="module">
-    export async function load({ request }) {
-        // Access the CF-IPCountry header
-        const countryCode = request.headers.get('cf-ipcountry') || 'IE';
-        return {
-            props: {
-                countryCode
-            }
-        };
-    }
-</script>
-
 <script>
     import { onMount } from 'svelte';
     import countries from 'i18n-iso-countries';
@@ -18,21 +6,33 @@
     import { getHolidaysForYear, optimizeDaysOff, calculateConsecutiveDaysOff } from '../lib/holidayUtils.js';
     import { ptoData } from '../lib/ptoData.js';
 
-    export let countryCode; // Receive the country code as a prop
-
     countries.registerLocale(enLocale);
     let countriesList = countries.getNames('en');
 
     let year = new Date().getFullYear();
     let months = Array.from({ length: 12 }, (_, i) => i);
-    let selectedCountry = countriesList[countryCode] || 'Ireland';
+    let selectedCountry = 'Belgium';
     let holidays = [];
-    let daysOff = ptoData[countryCode] || ptoData['IE'];
+    let daysOff = ptoData['BE'];
     let optimizedDaysOff = [];
     let consecutiveDaysOff = [];
     let placeholder = "Country";
     let inputElement;
 
+    async function fetchCountryCode() {
+        try {
+            const response = await fetch('/cdn-cgi/trace');
+            const text = await response.text();
+            const countryCodeMatch = text.match(/cf-ipcountry=(\w+)/);
+            const countryCode = countryCodeMatch ? countryCodeMatch[1] : 'BE';
+            selectedCountry = countriesList[countryCode] || 'Belgium';
+            daysOff = ptoData[countryCode] || ptoData['BE'];
+            updateHolidays();
+        } catch (error) {
+            console.error('Error fetching country code:', error);
+        }
+    }
+
     function handleCountryChange(event) {
         const fullValue = event.target.value;
         const countryCode = Object.keys(countriesList).find(code => countriesList[code] === fullValue);
@@ -102,6 +102,7 @@
     }
 
     onMount(() => {
+        fetchCountryCode(); // Fetch the country code on mount
         adjustInputWidth(inputElement);
         inputElement.addEventListener('input', () => {
             adjustInputWidth(inputElement);