From fcd553012bc7eedc78b03fb96beb2c9acc393cf0 Mon Sep 17 00:00:00 2001
From: thorstenspille <thorsten@spille-edv.de>
Date: Sun, 1 Oct 2023 22:16:03 +0200
Subject: [PATCH] Add repo config

---
 postinstall | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/postinstall b/postinstall
index 87435f4..667b1b4 100755
--- a/postinstall
+++ b/postinstall
@@ -259,7 +259,22 @@ set_locales(){
 }
 
 set_pve_repo(){
-    echo ""
+    nosub=$(grep pve-no-subscription /etc/apt/sources.list)
+    enterprise=$(grep pve-enterprise /etc/apt/sources.list.d/pve-enterprise.list)
+    test=$(grep pvetest /etc/apt/sources.list)
+    if [[ $repo_selection == "pve-enterprise" ]]; then
+        pvesh set nodes/px1/apt/repositories --handle enterprise
+        sed -i "s/$nosub/# $nosub/g" /etc/apt/sources.list
+        sed -i "s/$test/# $test/g" /etc/apt/sources.list
+    elif [[ $repo_selection == "pve-no-subscription" ]]; then
+        pvesh set nodes/px1/apt/repositories --handle no-subscription
+        sed -i "s/$enterprise/# $enterprise/g" /etc/apt/sources.list.d/pve-enterprise.list
+        sed -i "s/$test/# $test/g" /etc/apt/sources.list
+    elif [[ $repo_selection == "pvetest" ]]; then
+        pvesh set nodes/px1/apt/repositories --handle test
+        sed -i "s/$nosub/# $nosub/g" /etc/apt/sources.list
+        sed -i "s/$enterprise/# $enterprise/g" /etc/apt/sources.list.d/pve-enterprise.list
+    fi
 }
 
 update_system(){