From 3eca4b2baa45d4183b3c0d437f34635282c7460a Mon Sep 17 00:00:00 2001
From: Late Night Defender <pongpeera054@protonmail.com>
Date: Mon, 13 May 2024 14:55:22 +0700
Subject: [PATCH] add check_root function

---
 server_setup.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/server_setup.go b/server_setup.go
index e69de29..59bd081 100644
--- a/server_setup.go
+++ b/server_setup.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/user"
+)
+
+var mongo_base_path = "/opt/my-mongo"
+var replicaset_name = "techtransthai-db"
+var network_name = "mongonet"
+var number_of_containers int = 5
+var max_mem_alloc_percent int = 95
+var mongo_initdb_root_username = "sasha"
+var mongo_initdb_root_password = "12345"
+
+func check_root() {
+	currentUser, err := user.Current()
+	if err != nil {
+        fmt.Println("Error getting current user:", err)
+        os.Exit(1)
+    }
+	if currentUser.Uid != "0" {
+		fmt.Println("You must be root to run this script")
+        os.Exit(2)
+	}
+	fmt.Println("UID:", currentUser.Uid)
+}
+
+func main() {
+	check_root()
+}
\ No newline at end of file