From dbb756c2c6e4882a5dbbe1fdd16840c9bf435d86 Mon Sep 17 00:00:00 2001
From: sneak <sneak@sneak.berlin>
Date: Sat, 29 Mar 2025 00:25:42 -0700
Subject: [PATCH] latest

---
 updatemirror.sh | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/updatemirror.sh b/updatemirror.sh
index 9446955..bcc1bb2 100644
--- a/updatemirror.sh
+++ b/updatemirror.sh
@@ -15,6 +15,18 @@ function main () {
         exit 1
     fi
 
+    if ! grep -qE 'http(s)?://archive\.ubuntu\.com/' /etc/apt/sources.list.d/*; then
+        echo "Not using default ubuntu repo, script will not function" > /dev/stderr
+        exit 1
+    fi
+
+    echo "Changing mirror country to US as a first approximation..." > /dev/stderr
+
+    find /etc/apt/sources.list.d/ -type f -exec \
+      sed -i "s|archive\.ubuntu\.com|us.ubuntu.com|g" {} +
+
+    echo "Detecting actual country..." > /dev/stderr
+
     apt update && apt install -y curl jq
     IPINFO="$(curl -s ipinfo.io)"
     if [ -z "$IPINFO" ]; then
@@ -22,18 +34,25 @@ function main () {
         exit 1
     fi
     COUNTRYCODE="$(jq .country -r <<< "$IPINFO")"
-    echo "Country code: $COUNTRYCODE"
+    echo "IP geolocated as country: $COUNTRYCODE" > /dev/stderr
 
-    if ! grep -qE 'http(s)?://archive\.ubuntu\.com/' /etc/apt/sources.list.d/*; then
+    if ! grep -qE 'http(s)?://us\.archive\.ubuntu\.com/' /etc/apt/sources.list.d/*; then
         echo "Not using default ubuntu repo, script will not function" > /dev/stderr
         exit 1
     fi
 
+    if [[ "$COUNTRYCODE" == "US" ]]; then
+        echo "Mirror already set to  US, nothing to do" > /dev/stderr
+        exit 0
+    fi
+
     MIRRORHOST="$(echo "$COUNTRYCODE" | tr 'A-Z' 'a-z').archive.ubuntu.com"
 
-    echo "IP geolocated as $COUNTRYCODE, pdating mirror to $MIRRORHOST."
+    echo "Changing mirror country to $COUNTRYCODE..." > /dev/stderr
+
     find /etc/apt/sources.list.d/ -type f -exec \
-      sed -i "s|archive\.ubuntu\.com|$MIRRORHOST|g" {} +
+      sed -i "s|us\.archive\.ubuntu\.com|$MIRRORHOST|g" {} +
+    exit 0
 }
 
 main