tls.go

//Author:github.com/Azumi67
//This script is for educational use and for my own learning, but I'd be happy if you find it useful too.
//This script simplifies the configuration of WS + WSS Reverse tunnel.
//You can send me feedback so I can use it to learn more.
//This script comes without any warranty
//Thank you.
package main

import (
    "time"
	"strconv"
	"bufio"
	"fmt"
	"os"
	"os/exec"
	"strings"
	"github.com/fatih/color"
	"log"
	"github.com/AlecAivazis/survey/v2"
	"net"
	"io/ioutil"
)
func readInputs(reader *bufio.Reader) string {
	input, _ := reader.ReadString('\n')
	return strings.TrimSpace(input)
}

func enableResetKharej() {
	deleteCron()
	deleteCron2()
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("\033[93mQuestion time !\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mDo you want to enable/edit \033[96mReverseTLS \033[92mreset time\033[93m? (\033[92myes\033[93m/\033[91mno\033[93m): \033[0m")
	enableReset := readInputs(reader)

	if enableReset == "yes" || enableReset == "y" {
		fmt.Println("\033[93m╭───────────────────────────────────────╮\033[0m")
		fmt.Println("1. \033[92mHour\033[0m")
		fmt.Println("2. \033[93mMinute\033[0m")
		fmt.Println("\033[93m╰───────────────────────────────────────╯\033[0m")

		fmt.Print("\033[93mEnter your choice: \033[0m")
		timeUnitChoice := readInputs(reader)

		var timeUnit string
		if timeUnitChoice == "1" {
			timeUnit = "hour"
		} else if timeUnitChoice == "2" {
			timeUnit = "minute"
		} else {
			fmt.Println("\033[91mWrong choice\033[0m")
			return
		}

		fmt.Print("\033[93mEnter the \033[92mdesired input\033[93m: \033[0m")
		timeValue := readInputs(reader)
		timeInt, err := strconv.Atoi(timeValue)
		if err != nil {
			fmt.Println("\033[91mInvalid input for time value\033[0m")
			return
		}

		var intervalSeconds int
		if timeUnit == "hour" {
			intervalSeconds = timeInt * 3600
		} else {
			intervalSeconds = timeInt * 60
		}

		resetRatKharej(intervalSeconds)
		fmt.Println("\033[93m────────────────────────────────────────\033[0m")
	}
}

func enableResetIran() {
	deleteCron()
	deleteCron2()
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("\033[93mQuestion time !\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mDo you want to enable/edit \033[96mReverseTLS \033[92mreset time\033[93m? (\033[92myes\033[93m/\033[91mno\033[93m): \033[0m")
	enableReset := readInputs(reader)

	if enableReset == "yes" || enableReset == "y" {
		fmt.Println("\033[93m╭───────────────────────────────────────╮\033[0m")
		fmt.Println("1. \033[92mHour\033[0m")
		fmt.Println("2. \033[93mMinute\033[0m")
		fmt.Println("\033[93m╰───────────────────────────────────────╯\033[0m")

		fmt.Print("\033[93mEnter your choice: \033[0m")
		timeUnitChoice := readInputs(reader)

		var timeUnit string
		if timeUnitChoice == "1" {
			timeUnit = "hour"
		} else if timeUnitChoice == "2" {
			timeUnit = "minute"
		} else {
			fmt.Println("\033[91mWrong choice\033[0m")
			return
		}

		fmt.Print("\033[93mEnter the \033[92mdesired input\033[93m: \033[0m")
		timeValue := readInputs(reader)
		timeInt, err := strconv.Atoi(timeValue)
		if err != nil {
			fmt.Println("\033[91mInvalid input for time value\033[0m")
			return
		}

		var intervalSeconds int
		if timeUnit == "hour" {
			intervalSeconds = timeInt * 3600
		} else {
			intervalSeconds = timeInt * 60
		}

		resetRatIran(intervalSeconds)
		fmt.Println("\033[93m────────────────────────────────────────\033[0m")
	}
}

func resetRatKharej(interval int) {
    daemonScript := fmt.Sprintf(`#!/bin/bash
INTERVAL=%d

while true; do
    /bin/bash /etc/tls.sh
    sleep $INTERVAL
done
`, interval)
	err := os.WriteFile("/usr/local/bin/tls_daemon.sh", []byte(daemonScript), 0755)
	if err != nil {
		fmt.Println("Error writing daemon script:", err)
		return
	}

	exec.Command("chmod", "+x", "/usr/local/bin/tls_daemon.sh").Run()

	serviceContent := `[Unit]
Description=Custom Daemon

[Service]
ExecStart=/usr/local/bin/tls_daemon.sh
Restart=always

[Install]
WantedBy=multi-user.target
`

	err = os.WriteFile("/etc/systemd/system/tls_reset.service", []byte(serviceContent), 0644)
	if err != nil {
		fmt.Println("Error writing service file:", err)
		return
	}

	ipsecScript := `#!/bin/bash
systemctl daemon-reload
pids=$(pgrep rtun-client)
sudo kill -9 $pids
sudo systemctl restart rtun-kharej
sudo journalctl --vacuum-size=1M
`
	err = os.WriteFile("/etc/tls.sh", []byte(ipsecScript), 0755)
	if err != nil {
		fmt.Println("Error writing TLS reset script:", err)
		return
	}

	exec.Command("chmod", "+x", "/etc/tls.sh").Run()
	exec.Command("systemctl", "daemon-reload").Run()
	exec.Command("systemctl", "enable", "tls_reset.service").Run()
	exec.Command("systemctl", "restart", "tls_reset.service").Run()
}

func resetRatIran(interval int) {
    daemonScript := fmt.Sprintf(`#!/bin/bash
INTERVAL=%d

while true; do
    /bin/bash /etc/tls.sh
    sleep $INTERVAL
done
`, interval)
	err := os.WriteFile("/usr/local/bin/tls_daemon.sh", []byte(daemonScript), 0755)
	if err != nil {
		fmt.Println("Error writing daemon script:", err)
		return
	}

	exec.Command("chmod", "+x", "/usr/local/bin/tls_daemon.sh").Run()

	serviceContent := `[Unit]
Description=Custom Daemon

[Service]
ExecStart=/usr/local/bin/tls_daemon.sh
Restart=always

[Install]
WantedBy=multi-user.target
`

	err = os.WriteFile("/etc/systemd/system/tls_reset.service", []byte(serviceContent), 0644)
	if err != nil {
		fmt.Println("Error writing service file:", err)
		return
	}

	ipsecScript := `#!/bin/bash
systemctl daemon-reload
pids=$(pgrep rtun-server)
sudo kill -9 $pids
sudo systemctl restart rtun-iran
sudo journalctl --vacuum-size=1M
`
	err = os.WriteFile("/etc/tls.sh", []byte(ipsecScript), 0755)
	if err != nil {
		fmt.Println("Error writing IPSec reset script:", err)
		return
	}

	exec.Command("chmod", "+x", "/etc/tls.sh").Run()
	exec.Command("systemctl", "daemon-reload").Run()
	exec.Command("systemctl", "enable", "tls_reset.service").Run()
	exec.Command("systemctl", "restart", "tls_reset.service").Run()
}
func getIPv4() string {
	interfaces, err := net.Interfaces()
	if err != nil {
		return ""
	}

	for _, iface := range interfaces {
		name := iface.Name
		if strings.HasPrefix(name, "eth") || strings.HasPrefix(name, "en") {
			addresses, err := iface.Addrs()
			if err != nil {
				continue
			}

			for _, addr := range addresses {
				ipnet, ok := addr.(*net.IPNet)
				if ok && !ipnet.IP.IsLoopback() && ipnet.IP.To4() != nil {
					return ipnet.IP.String()
				}
			}
		}
	}

	return ""
}
func displayProgress(total, current int) {
	width := 40
	percentage := current * 100 / total
	completed := width * current / total
	remaining := width - completed

	fmt.Printf("\r[%s>%s] %d%%", strings.Repeat("=", completed), strings.Repeat(" ", remaining), percentage)
}

func displayError(message string) {
	fmt.Printf("\u2718 Error: %s\n", message)
}

func displayNotification(message string) {
	fmt.Printf("\033[93m%s\033[0m\n", message)
}

func displayCheckmark(message string) {
	fmt.Printf("\033[92m\u2714 \033[0m%s\n", message)
}

func displayLoading() {
    frames := []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}
    delay := 100 * time.Millisecond
    duration := 5 * time.Second

    endTime := time.Now().Add(duration)

    for time.Now().Before(endTime) {
        for _, frame := range frames {
            fmt.Printf("\r[%s] Loading... ", frame)
            time.Sleep(delay)
        }
    }
    fmt.Println()
}
func displayLogo2() error {
	cmd := exec.Command("bash", "-c", "/etc/./logo.sh")
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	err := cmd.Run()
	if err != nil {
		return err
	}

	return nil
}
func displayLogo() {
	logo := `
   ______    _______    __      _______          __       _____  ___  
  /    " \  |   __ "\  |" \    /"      \        /""\      (\"   \|" \ 
 // ____  \ (. |__) :) ||  |  |:        |      /    \     |.\\   \   |
/  /    ) :)|:  ____/  |:  |  |_____/   )     /' /\  \    |: \.   \\ |
(: (____/ // (|  /     |.  |   //       /    //  __'  \   |.  \    \ |
\        // |__/ \     /\  |\  |:  __   \   /   /  \\  \  |    \    \|
 \"_____ / (_______)  (__\_|_) |__|  \___) (___/    \___) \___|\____\)
`
	
    cyan := color.New(color.FgCyan, color.Bold).SprintFunc()
    blue := color.New(color.FgBlue, color.Bold).SprintFunc()
	green := color.New(color.FgHiGreen, color.Bold).SprintFunc()      
    yellow := color.New(color.FgHiYellow, color.Bold).SprintFunc()   
    red := color.New(color.FgHiRed, color.Bold).SprintFunc()        


	

	    logo = cyan("  ______   ") + blue(" _______  ") + green("  __    ") + yellow("   ______   ") + red("     __      ") + cyan("  _____  ___  \n") +
		cyan(" /     \" \\  ") + blue("|   __ \" ") + green(" |\" \\  ") + yellow("   /\"     \\   ") + red("   /\"\"\\     ") + cyan(" (\\\"   \\|\"  \\ \n") +
		cyan("//  ____  \\ ")  + blue("(. |__) :)") + green("||  |  ") + yellow(" |:       |  ") + red("  /    \\   ") + cyan("  |.\\\\   \\   |\n") +
		cyan("/  /    ) :)") + blue("|:  ____/ ") + green("|:  |  ") + yellow(" |_____/  )  ") + red(" /' /\\  \\   ") + cyan(" |: \\.   \\\\ |\n") +
		cyan("(: (____/ / ") + blue("(|  /     ") + green("|.  | ") + yellow("  //      /  ") + red("//   __'  \\  ") + cyan(" |.  \\    \\ |\n") +
		cyan("\\        / ") + blue("/|__/ \\   ") + green(" /\\  |\\ ") + yellow(" |:  __  \\ ") + red(" /   /  \\\\   ") + cyan ("  |    \\    \\|\n") +
		cyan(" \"_____ / ") + blue("(_______)") + green("  (__\\_|_)") + yellow(" (__) \\___)") + red("(___/    \\___)") + cyan(" \\___|\\____\\)\n")


	fmt.Println(logo)
}
func main() {
	if os.Geteuid() != 0 {
		fmt.Println("\033[91mThis script must be run as root. Please use sudo -i.\033[0m")
		os.Exit(1)
	}

	mainMenu()
}
func readInput() {
	fmt.Print("Press Enter to continue..")
	fmt.Scanln()
	mainMenu()
}
func clearScreen() {
	cmd := exec.Command("clear")
	cmd.Stdout = os.Stdout
	cmd.Run()
}
func mainMenu() {
	for {
		err := displayLogo2()
		if err != nil {
			log.Fatalf("failed to display logo: %v", err)
		}
		displayLogo()
		border := "\033[93m+" + strings.Repeat("=", 70) + "+\033[0m"
		content := "\033[93m║            ▌║█║▌│║▌│║▌║▌█║ \033[92mMain Menu\033[93m  ▌│║▌║▌│║║▌█║▌                  ║"
		footer := " \033[92m            Join Opiran Telegram \033[34m@https://t.me/OPIranClub\033[0m "

		borderLength := len(border) - 2
		centeredContent := fmt.Sprintf("%[1]*s", -borderLength, content)

		fmt.Println(border)
		fmt.Println(centeredContent)
		fmt.Println(border)

		fmt.Println(border)
		fmt.Println(footer)
		fmt.Println(border)
		prompt := &survey.Select{
			Message: "Enter your choice Please:",
			Options: []string{"0. \033[91mSTATUS Menu\033[0m", "1. \033[93mEdit \033[92mResetTimer\033[0m", "2. \033[96mWS \033[92mTCP \033[0m", "3. \033[93mWS \033[92mUDP\033[0m", "4. \033[96mWSS + TLS \033[92mTCP\033[0m", "5. \033[93mWSS + TLS \033[92mUDP\033[0m", "6. \033[92mStop | Restart Service\033[0m", "7. \033[91mUninstall\033[0m", "q. Exit"},
		
		}
		fmt.Println("\033[93m╰─────────────────────────────────────────────────────────────────────╯\033[0m")

		var choice string
		err = survey.AskOne(prompt, &choice)
		if err != nil {
			log.Fatalf("\033[91muser input is wrong:\033[0m %v", err)
		}
		switch choice {
		case "0. \033[91mSTATUS Menu\033[0m":
			status()
		case "1. \033[93mEdit \033[92mResetTimer\033[0m":
			cronMenu()
		case "2. \033[96mWS \033[92mTCP \033[0m":
			wsMenu1()
		case "3. \033[93mWS \033[92mUDP\033[0m":
			wsMenu2()
		case "4. \033[96mWSS + TLS \033[92mTCP\033[0m":
			wsMenu3()
		case "5. \033[93mWSS + TLS \033[92mUDP\033[0m":
			wsMenu4()
		case "6. \033[92mStop | Restart Service\033[0m":
			startMain()
		case "7. \033[91mUninstall\033[0m":
			UniMenu()
		case "q. Exit":
			fmt.Println("Exiting...")
			return
		default:
			fmt.Println("Invalid choice.")
		}

		
		readInput()
	}
}
func rmv() error {
	file := "/etc/tls.sh"
	if _, err := os.Stat(file); err == nil {
		err := os.Remove(file)
		if err != nil {
			return fmt.Errorf("\033[91mbash file doesn't exists:\033[0m %v", err)
		}
		fmt.Println("\033[92mbash file removed successfully!\033[0m")
	}
	return nil
}
func rmv2() error {
	file := "/usr/local/bin/tls_daemon.sh"
	if _, err := os.Stat(file); err == nil {
		err := os.Remove(file)
		if err != nil {
			return fmt.Errorf("\033[91mbash file doesn't exists:\033[0m %v", err)
		}
		fmt.Println("\033[92mbash file removed successfully!\033[0m")
	}
	return nil
}
func deleteCron2() {
	entriesToDelete := []string{
		"*/1 * * * * /bin/bash /etc/tls.sh",
		"*/2 * * * * /bin/bash /etc/tls.sh",
		"*/3 * * * * /bin/bash /etc/tls.sh",
		"*/4 * * * * /bin/bash /etc/tls.sh",
		"*/5 * * * * /bin/bash /etc/tls.sh",
		"*/6 * * * * /bin/bash /etc/tls.sh",
		"*/7 * * * * /bin/bash /etc/tls.sh",
		"*/8 * * * * /bin/bash /etc/tls.sh",
		"*/9 * * * * /bin/bash /etc/tls.sh",
		"*/10 * * * * /bin/bash /etc/tls.sh",
		"*/11 * * * * /bin/bash /etc/tls.sh",
		"*/12 * * * * /bin/bash /etc/tls.sh",
		"*/13 * * * * /bin/bash /etc/tls.sh",
		"*/14 * * * * /bin/bash /etc/tls.sh",
		"*/15 * * * * /bin/bash /etc/tls.sh",
		"*/16 * * * * /bin/bash /etc/tls.sh",
		"*/17 * * * * /bin/bash /etc/tls.sh",
		"*/18 * * * * /bin/bash /etc/tls.sh",
		"*/19 * * * * /bin/bash /etc/tls.sh",
		"*/20 * * * * /bin/bash /etc/tls.sh",
		"*/21 * * * * /bin/bash /etc/tls.sh",
		"*/22 * * * * /bin/bash /etc/tls.sh",
		"*/23 * * * * /bin/bash /etc/tls.sh",
		"*/24 * * * * /bin/bash /etc/tls.sh",
		"*/25 * * * * /bin/bash /etc/tls.sh",
		"*/26 * * * * /bin/bash /etc/tls.sh",
		"*/27 * * * * /bin/bash /etc/tls.sh",
		"*/28 * * * * /bin/bash /etc/tls.sh",
		"*/29 * * * * /bin/bash /etc/tls.sh",
		"*/30 * * * * /bin/bash /etc/tls.sh",
		"*/31 * * * * /bin/bash /etc/tls.sh",
		"*/32 * * * * /bin/bash /etc/tls.sh",
		"*/33 * * * * /bin/bash /etc/tls.sh",
		"*/34 * * * * /bin/bash /etc/tls.sh",
		"*/35 * * * * /bin/bash /etc/tls.sh",
		"*/36 * * * * /bin/bash /etc/tls.sh",
		"*/37 * * * * /bin/bash /etc/tls.sh",
		"*/38 * * * * /bin/bash /etc/tls.sh",
		"*/39 * * * * /bin/bash /etc/tls.sh",
		"*/40 * * * * /bin/bash /etc/tls.sh",
		"*/41 * * * * /bin/bash /etc/tls.sh",
		"*/42 * * * * /bin/bash /etc/tls.sh",
		"*/43 * * * * /bin/bash /etc/tls.sh",
		"*/44 * * * * /bin/bash /etc/tls.sh",
		"*/45 * * * * /bin/bash /etc/tls.sh",
		"*/46 * * * * /bin/bash /etc/tls.sh",
		"*/47 * * * * /bin/bash /etc/tls.sh",
		"*/48 * * * * /bin/bash /etc/tls.sh",
		"*/49 * * * * /bin/bash /etc/tls.sh",
		"*/50 * * * * /bin/bash /etc/tls.sh",
		"*/51 * * * * /bin/bash /etc/tls.sh",
		"*/52 * * * * /bin/bash /etc/tls.sh",
		"*/53 * * * * /bin/bash /etc/tls.sh",
		"*/54 * * * * /bin/bash /etc/tls.sh",
		"*/55 * * * * /bin/bash /etc/tls.sh",
		"*/56 * * * * /bin/bash /etc/tls.sh",
		"*/57 * * * * /bin/bash /etc/tls.sh",
		"*/58 * * * * /bin/bash /etc/tls.sh",
		"*/59 * * * * /bin/bash /etc/tls.sh",
	}

	existingCrontab, err := exec.Command("crontab", "-l").Output()
	if err != nil {
		fmt.Println("\033[91mNo existing cron found!\033[0m")
		return
	}

	newCrontab := string(existingCrontab)
	for _, entry := range entriesToDelete {
		if strings.Contains(newCrontab, entry) {
			newCrontab = strings.Replace(newCrontab, entry, "", -1)
		}
	}

	if newCrontab != string(existingCrontab) {
		cmd := exec.Command("crontab")
		cmd.Stdin = strings.NewReader(newCrontab)

		_, err = cmd.CombinedOutput()
        if err != nil {
            fmt.Printf("\033[91mfailed to delete some cron entries. don't worry about it \033[0m\n")
		} else {
			displayNotification("\033[92mDeleting Previous Crons..\033[0m")
		}
	} else {
		fmt.Println("\033[91mCron doesn't exist, moving on..!\033[0m")
	}
}
func deleteCron4() {
	entriesToDelete := []string{
		"*/1 * * * * /etc/tls.sh",
		"*/2 * * * * /etc/tls.sh",
		"*/3 * * * * /etc/tls.sh",
		"*/4 * * * * /etc/tls.sh",
		"*/5 * * * * /etc/tls.sh",
		"*/6 * * * * /etc/tls.sh",
		"*/7 * * * * /etc/tls.sh",
		"*/8 * * * * /etc/tls.sh",
		"*/9 * * * * /etc/tls.sh",
		"*/10 * * * * /etc/tls.sh",
		"*/11 * * * * /etc/tls.sh",
		"*/12 * * * * /etc/tls.sh",
		"*/13 * * * * /etc/tls.sh",
		"*/14 * * * * /etc/tls.sh",
		"*/15 * * * * /etc/tls.sh",
		"*/16 * * * * /etc/tls.sh",
		"*/17 * * * * /etc/tls.sh",
		"*/18 * * * * /etc/tls.sh",
		"*/19 * * * * /etc/tls.sh",
		"*/20 * * * * /etc/tls.sh",
		"*/21 * * * * /etc/tls.sh",
		"*/22 * * * * /etc/tls.sh",
		"*/23 * * * * /etc/tls.sh",
		"*/24 * * * * /etc/tls.sh",
		"*/25 * * * * /etc/tls.sh",
		"*/26 * * * * /etc/tls.sh",
		"*/27 * * * * /etc/tls.sh",
		"*/28 * * * * /etc/tls.sh",
		"*/29 * * * * /etc/tls.sh",
		"*/30 * * * * /etc/tls.sh",
		"*/31 * * * * /etc/tls.sh",
		"*/32 * * * * /etc/tls.sh",
		"*/33 * * * * /etc/tls.sh",
		"*/34 * * * * /etc/tls.sh",
		"*/35 * * * * /etc/tls.sh",
		"*/36 * * * * /etc/tls.sh",
		"*/37 * * * * /etc/tls.sh",
		"*/38 * * * * /etc/tls.sh",
		"*/39 * * * * /etc/tls.sh",
		"*/40 * * * * /etc/tls.sh",
		"*/41 * * * * /etc/tls.sh",
		"*/42 * * * * /etc/tls.sh",
		"*/43 * * * * /etc/tls.sh",
		"*/44 * * * * /etc/tls.sh",
		"*/45 * * * * /etc/tls.sh",
		"*/46 * * * * /etc/tls.sh",
		"*/47 * * * * /etc/tls.sh",
		"*/48 * * * * /etc/tls.sh",
		"*/49 * * * * /etc/tls.sh",
		"*/50 * * * * /etc/tls.sh",
		"*/51 * * * * /etc/tls.sh",
		"*/52 * * * * /etc/tls.sh",
		"*/53 * * * * /etc/tls.sh",
		"*/54 * * * * /etc/tls.sh",
		"*/55 * * * * /etc/tls.sh",
		"*/56 * * * * /etc/tls.sh",
		"*/57 * * * * /etc/tls.sh",
		"*/58 * * * * /etc/tls.sh",
		"*/59 * * * * /etc/tls.sh",
	}

	existingCrontab, err := exec.Command("crontab", "-l").Output()
	if err != nil {
		fmt.Println("\033[91mNo existing cron found!\033[0m")
		return
	}

	newCrontab := string(existingCrontab)
	for _, entry := range entriesToDelete {
		if strings.Contains(newCrontab, entry) {
			newCrontab = strings.Replace(newCrontab, entry, "", -1)
		}
	}

	if newCrontab != string(existingCrontab) {
		cmd := exec.Command("crontab")
		cmd.Stdin = strings.NewReader(newCrontab)

		_, err = cmd.CombinedOutput()
        if err != nil {
            fmt.Printf("\033[91mfailed to delete some cron entries. don't worry about it \033[0m\n")
		} else {
			displayNotification("\033[92mDeleting Previous Crons..\033[0m")
		}
	} else {
		fmt.Println("\033[91mCron doesn't exist, moving on..!\033[0m")
	}
}
func deleteCron3() {
	entriesToDelete := []string{
		"0 * * * * /etc/tls.sh",
		"0 */2 * * * /etc/tls.sh",
		"0 */3 * * * /etc/tls.sh",
		"0 */4 * * * /etc/tls.sh",
		"0 */5 * * * /etc/tls.sh",
		"0 */6 * * * /etc/tls.sh",
		"0 */7 * * * /etc/tls.sh",
		"0 */8 * * * /etc/tls.sh",
		"0 */9 * * * /etc/tls.sh",
		"0 */10 * * * /etc/tls.sh",
		"0 */11 * * * /etc/tls.sh",
		"0 */12 * * * /etc/tls.sh",
		"0 */13 * * * /etc/tls.sh",
		"0 */14 * * * /etc/tls.sh",
		"0 */15 * * * /etc/tls.sh",
		"0 */16 * * * /etc/tls.sh",
		"0 */17 * * * /etc/tls.sh",
		"0 */18 * * * /etc/tls.sh",
		"0 */19 * * * /etc/tls.sh",
		"0 */20 * * * /etc/tls.sh",
		"0 */21 * * * /etc/tls.sh",
		"0 */22 * * * /etc/tls.sh",
		"0 */23 * * * /etc/tls.sh",
	}

	existingCrontab, err := exec.Command("crontab", "-l").Output()
	if err != nil {
		fmt.Println("\033[91mNo existing cron found!\033[0m")
		return
	}

	newCrontab := string(existingCrontab)
	for _, entry := range entriesToDelete {
		if strings.Contains(newCrontab, entry) {
			newCrontab = strings.Replace(newCrontab, entry, "", -1)
		}
	}

	if newCrontab != string(existingCrontab) {
		cmd := exec.Command("crontab")
		cmd.Stdin = strings.NewReader(newCrontab)

		_, err = cmd.CombinedOutput()
        if err != nil {
            fmt.Printf("\033[91mfailed to delete some cron entries. don't worry about it \033[0m\n")
		} else {
			displayNotification("\033[92mDeleting Previous Crons..\033[0m")
		}
	} else {
		fmt.Println("\033[91mCron doesn't exist, moving on..!\033[0m")
	}
}
func deleteCron() {
	entriesToDelete := []string{
		"0 * * * * /bin/bash /etc/tls.sh",
		"0 */2 * * * /bin/bash /etc/tls.sh",
		"0 */3 * * * /bin/bash /etc/tls.sh",
		"0 */4 * * * /bin/bash /etc/tls.sh",
		"0 */5 * * * /bin/bash /etc/tls.sh",
		"0 */6 * * * /bin/bash /etc/tls.sh",
		"0 */7 * * * /bin/bash /etc/tls.sh",
		"0 */8 * * * /bin/bash /etc/tls.sh",
		"0 */9 * * * /bin/bash /etc/tls.sh",
		"0 */10 * * * /bin/bash /etc/tls.sh",
		"0 */11 * * * /bin/bash /etc/tls.sh",
		"0 */12 * * * /bin/bash /etc/tls.sh",
		"0 */13 * * * /bin/bash /etc/tls.sh",
		"0 */14 * * * /bin/bash /etc/tls.sh",
		"0 */15 * * * /bin/bash /etc/tls.sh",
		"0 */16 * * * /bin/bash /etc/tls.sh",
		"0 */17 * * * /bin/bash /etc/tls.sh",
		"0 */18 * * * /bin/bash /etc/tls.sh",
		"0 */19 * * * /bin/bash /etc/tls.sh",
		"0 */20 * * * /bin/bash /etc/tls.sh",
		"0 */21 * * * /bin/bash /etc/tls.sh",
		"0 */22 * * * /bin/bash /etc/tls.sh",
		"0 */23 * * * /bin/bash /etc/tls.sh",
	}

	existingCrontab, err := exec.Command("crontab", "-l").Output()
	if err != nil {
		fmt.Println("\033[91mNo existing cron found!\033[0m")
		return
	}

	newCrontab := string(existingCrontab)
	for _, entry := range entriesToDelete {
		if strings.Contains(newCrontab, entry) {
			newCrontab = strings.Replace(newCrontab, entry, "", -1)
		}
	}

	if newCrontab != string(existingCrontab) {
		cmd := exec.Command("crontab")
		cmd.Stdin = strings.NewReader(newCrontab)

		_, err = cmd.CombinedOutput()
        if err != nil {
            fmt.Printf("\033[91mfailed to delete some cron entries. don't worry about it \033[0m\n")
		} else {
			displayNotification("\033[92mDeleting Previous Crons..\033[0m")
		}
	} else {
		fmt.Println("\033[91mCron doesn't exist, moving on..!\033[0m")
	}
}

func cronMenu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reset \033[92mTimer \033[93mMenu\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKharej\033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}

	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		enableResetIran()
	case "2. \033[93mKharej\033[0m":
		enableResetKharej()
	case "0. \033[94mBack to the main menu\033[0m":
		clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}
	readInput()
}

func resHourz() {
	deleteCron()
	deleteCron2()
    
	fmt.Println("╭──────────────────────────────────────╮")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mEnter \033[92mReset timer\033[93m (hours):\033[0m ")
	hoursStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
	hoursStr = strings.TrimSpace(hoursStr)
	fmt.Println("╰──────────────────────────────────────╯")

	hours, err := strconv.Atoi(hoursStr)
	if err != nil {
		log.Fatalf("\033[91mInvalid input for reset timer:\033[0m %v", err)
	}

	var cronEntry string
	if hours == 1 {
		cronEntry = "0 * * * * /bin/bash /etc/tls.sh"
	} else if hours >= 2 {
		cronEntry = fmt.Sprintf("0 */%d * * * /bin/bash /etc/tls.sh", hours)
	}

	crontabFile, err := os.OpenFile(crontabFilePath, os.O_RDWR|os.O_CREATE, 0644)
	if err != nil {
		log.Fatalf("\033[91mCouldn't open Cron:\033[0m %v", err)
	}
	defer crontabFile.Close()

	var crontabContent strings.Builder
	scanner := bufio.NewScanner(crontabFile)
	for scanner.Scan() {
		line := scanner.Text()
		if line == cronEntry {
			fmt.Println("\033[92mOh... Cron entry already exists!\033[0m")
			return
		}
		crontabContent.WriteString(line)
		crontabContent.WriteString("\n")
	}

	crontabContent.WriteString(cronEntry)
	crontabContent.WriteString("\n")

	if err := scanner.Err(); err != nil {
		log.Fatalf("\033[91mcrontab Reading error:\033[0m %v", err)
	}

	if err := crontabFile.Truncate(0); err != nil {
		log.Fatalf("\033[91mcouldn't truncate cron file:\033[0m %v", err)
	}

	if _, err := crontabFile.Seek(0, 0); err != nil {
		log.Fatalf("\033[91mcouldn't find cron file: \033[0m%v", err)
	}

	if _, err := crontabFile.WriteString(crontabContent.String()); err != nil {
		log.Fatalf("\033[91mCouldn't write cron file:\033[0m %v", err)
	}

	fmt.Println("\033[92mCron entry added successfully!\033[0m")
}

func resMins() {
	deleteCron()
	deleteCron2()

	fmt.Println("╭──────────────────────────────────────╮")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mEnter \033[92mReset timer\033[93m (minutes):\033[0m ")
	minutesStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
	minutesStr = strings.TrimSpace(minutesStr)
	fmt.Println("╰──────────────────────────────────────╯")

	minutes, err := strconv.Atoi(minutesStr)
	if err != nil {
		log.Fatalf("\033[91mInvalid input for reset timer:\033[0m %v", err)
	}

	cronEntry := fmt.Sprintf("*/%d * * * * /bin/bash /etc/tls.sh", minutes)

	crontabFile, err := os.OpenFile(crontabFilePath, os.O_RDWR|os.O_CREATE, 0644)
	if err != nil {
		log.Fatalf("\033[91mCouldn't open Cron:\033[0m %v", err)
	}
	defer crontabFile.Close()

	var crontabContent strings.Builder
	scanner := bufio.NewScanner(crontabFile)
	for scanner.Scan() {
		line := scanner.Text()
		if line == cronEntry {
			fmt.Println("\033[92mOh... Cron entry already exists!\033[0m")
			return
		}
		crontabContent.WriteString(line)
		crontabContent.WriteString("\n")
	}

	crontabContent.WriteString(cronEntry)
	crontabContent.WriteString("\n")

	if err := scanner.Err(); err != nil {
		log.Fatalf("\033[91mcrontab Reading error:\033[0m %v", err)
	}

	if err := crontabFile.Truncate(0); err != nil {
		log.Fatalf("\033[91mcouldn't truncate cron file:\033[0m %v", err)
	}

	if _, err := crontabFile.Seek(0, 0); err != nil {
		log.Fatalf("\033[91mcouldn't find cron file: \033[0m%v", err)
	}

	if _, err := crontabFile.WriteString(crontabContent.String()); err != nil {
		log.Fatalf("\033[91mCouldn't write cron file:\033[0m %v", err)
	}

	fmt.Println("\033[92mCron entry added successfully!\033[0m")
}

const crontabFilePath = "/var/spool/cron/crontabs/root"

func resKharej() {
	deleteCron()
	deleteCron2()

	if _, err := os.Stat("/etc/tls.sh"); err == nil {
		os.Remove("/etc/tls.sh")
	}

	file, err := os.Create("/etc/tls.sh")
	if err != nil {
		log.Fatalf("\033[91mbash creation error:\033[0m %v", err)
	}
	defer file.Close()

	file.WriteString("#!/bin/bash\n")
	file.WriteString("sudo systemctl daemon-reload\n")
    file.WriteString("pids=$(pgrep rtun-client)\n")
	file.WriteString("sudo kill -9 $pids\n")
	file.WriteString("sudo systemctl restart rtun-kharej\n")
	file.WriteString("sudo journalctl --vacuum-size=1M\n")

	cmd := exec.Command("chmod", "+x", "/etc/tls.sh")
	if err := cmd.Run(); err != nil {
		log.Fatalf("\033[91mchmod cmd error:\033[0m %v", err)
	}

	fmt.Println("╭──────────────────────────────────────╮")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mChoose an option:\n1. \033[96mReset timer \033[92m(hours)\033[93m\n2. \033[96mReset timer \033[92m(minutes)\n\033[0m")
    fmt.Print("Choose option: ")
	optionStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
    
	optionStr = strings.TrimSpace(optionStr)
	fmt.Println("╰──────────────────────────────────────╯")

	option, err := strconv.Atoi(optionStr)
	if err != nil || (option != 1 && option != 2) {
		log.Fatalf("\033[91mInvalid input for option:\033[0m %v", err)
	}

	var timeUnit string
	var cronEntry string

	if option == 1 {
		timeUnit = "hours"
	} else if option == 2 {
		timeUnit = "minutes"
	}

	fmt.Printf("\033[93mEnter reset time (%s):\033[0m ", timeUnit)
	timeStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
	timeStr = strings.TrimSpace(timeStr)

	timeValue, err := strconv.Atoi(timeStr)
	if err != nil {
		log.Fatalf("\033[91mInvalid input for reset time:\033[0m %v", err)
	}

	if option == 1 {
		if timeValue == 1 {
			cronEntry = "0 * * * * /etc/tls.sh"
		} else if timeValue >= 2 {
			cronEntry = fmt.Sprintf("0 */%d * * * /bin/bash /etc/tls.sh", timeValue)
		}
	} else if option == 2 {
		cronEntry = fmt.Sprintf("*/%d * * * * /bin/bash /etc/tls.sh", timeValue)
	}

	crontabFile, err := os.OpenFile(crontabFilePath, os.O_RDWR|os.O_CREATE, 0644)
	if err != nil {
		log.Fatalf("\033[91mCouldn't open Cron:\033[0m %v", err)
	}
	defer crontabFile.Close()

	var crontabContent strings.Builder
	scanner := bufio.NewScanner(crontabFile)
	for scanner.Scan() {
		line := scanner.Text()
		if line == cronEntry {
			fmt.Println("\033[92mOh... Cron entry already exists!\033[0m")
			return
		}
		crontabContent.WriteString(line)
		crontabContent.WriteString("\n")
	}

	crontabContent.WriteString(cronEntry)
	crontabContent.WriteString("\n")

	if err := scanner.Err(); err != nil {
		log.Fatalf("\033[91mcrontab Reading error:\033[0m %v", err)
	}

	if err := crontabFile.Truncate(0); err != nil {
		log.Fatalf("\033[91mcouldn't truncate cron file:\033[0m %v", err)
	}

	if _, err := crontabFile.Seek(0, 0); err != nil {
		log.Fatalf("\033[91mcouldn't find cron file: \033[0m%v", err)
	}

	if _, err := crontabFile.WriteString(crontabContent.String()); err != nil {
		log.Fatalf("\033[91mCouldn't write cron file:\033[0m %v", err)
	}

	fmt.Println("\033[92mCron entry added successfully!\033[0m")
}
func resIran() {
	deleteCron()
	deleteCron2()
	if _, err := os.Stat("/etc/tls.sh"); err == nil {
		os.Remove("/etc/tls.sh")
	}

	file, err := os.Create("/etc/tls.sh")
	if err != nil {
		log.Fatalf("\033[91mbash creation error:\033[0m %v", err)
	}
	defer file.Close()

    file.WriteString("#!/bin/bash\n")
    file.WriteString("sudo systemctl daemon-reload\n")
    file.WriteString("pids=$(pgrep rtun-server)\n")
	file.WriteString("sudo kill -9 $pids\n")
	file.WriteString("sudo systemctl restart rtun-iran\n")
    file.WriteString("sudo journalctl --vacuum-size=1M\n")

	cmd := exec.Command("chmod", "+x", "/etc/tls.sh")
	if err := cmd.Run(); err != nil {
		log.Fatalf("\033[91mchmod cmd error:\033[0m %v", err)
	}

	fmt.Println("╭──────────────────────────────────────╮")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mChoose an option:\n1. \033[96mReset timer \033[92m(hours)\033[93m\n2. \033[96mReset timer \033[92m(minutes)\n\033[0m")
    fmt.Print("Choose option: ")
	optionStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
	optionStr = strings.TrimSpace(optionStr)
	fmt.Println("╰──────────────────────────────────────╯")

	option, err := strconv.Atoi(optionStr)
	if err != nil || (option != 1 && option != 2) {
		log.Fatalf("\033[91mInvalid input for option:\033[0m %v", err)
	}

	var timeUnit string
	var cronEntry string

	if option == 1 {
		timeUnit = "hours"
	} else if option == 2 {
		timeUnit = "minutes"
	}

	fmt.Printf("\033[93mEnter reset time (%s):\033[0m ", timeUnit)
	timeStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatalf("Error reading input: %v", err)
	}
	timeStr = strings.TrimSpace(timeStr)

	timeValue, err := strconv.Atoi(timeStr)
	if err != nil {
		log.Fatalf("\033[91mInvalid input for reset time:\033[0m %v", err)
	}

	if option == 1 {
		if timeValue == 1 {
			cronEntry = "0 * * * * /etc/tls.sh"
		} else if timeValue >= 2 {
			cronEntry = fmt.Sprintf("0 */%d * * * /bin/bash /etc/tls.sh", timeValue)
		}
	} else if option == 2 {
		cronEntry = fmt.Sprintf("*/%d * * * * /bin/bash /etc/tls.sh", timeValue)
	}

	crontabFile, err := os.OpenFile(crontabFilePath, os.O_RDWR|os.O_CREATE, 0644)
	if err != nil {
		log.Fatalf("\033[91mCouldn't open Cron:\033[0m %v", err)
	}
	defer crontabFile.Close()

	var crontabContent strings.Builder
	scanner := bufio.NewScanner(crontabFile)
	for scanner.Scan() {
		line := scanner.Text()
		if line == cronEntry {
			fmt.Println("\033[92mOh... Cron entry already exists!\033[0m")
			return
		}
		crontabContent.WriteString(line)
		crontabContent.WriteString("\n")
	}

	crontabContent.WriteString(cronEntry)
	crontabContent.WriteString("\n")

	if err := scanner.Err(); err != nil {
		log.Fatalf("\033[91mcrontab Reading error:\033[0m %v", err)
	}

	if err := crontabFile.Truncate(0); err != nil {
		log.Fatalf("\033[91mcouldn't truncate cron file:\033[0m %v", err)
	}

	if _, err := crontabFile.Seek(0, 0); err != nil {
		log.Fatalf("\033[91mcouldn't find cron file: \033[0m%v", err)
	}

	if _, err := crontabFile.WriteString(crontabContent.String()); err != nil {
		log.Fatalf("\033[91mCouldn't write cron file:\033[0m %v", err)
	}

	fmt.Println("\033[92mCron entry added successfully!\033[0m")
}
func runCmd(command string, args ...string) {
	cmd := exec.Command(command, args...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	err := cmd.Run()
	if err != nil {
		log.Fatalf("Couldn't Run Command '%s': %v", command, err)
	}
}
func startMain() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[92m Service \033[93mMenu\033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mRestart\033[0m", "2. \033[93mStop \033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mRestart\033[0m":
		start()
	case "2. \033[93mStop \033[0m":
		stop()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func start() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[92m Restart \033[93mMenu\033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mWS\033[0m", "2. \033[93mWSS \033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mWS\033[0m":
		restartws()
	case "2. \033[93mWS \033[0m":
		restartws()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		startMain()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func restartws() {
	cmd := exec.Command("clear")
	cmd.Stdout = os.Stdout
	cmd.Run()

	displayNotification("\033[93mRestarting WS Reverse \033[93m..\033[0m")
	fmt.Println("\033[93m╭─────────────────────────────────────────────╮\033[0m")

	cmd = exec.Command("systemctl", "restart", "rtun-kharej")
	cmd.Run()
	time.Sleep(1 * time.Second)

	cmd = exec.Command("systemctl", "restart", "rtun-iran")
	cmd.Run()
	time.Sleep(1 * time.Second)

	fmt.Print("Progress: ")

	frames := []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}
	delay := 0.1
	duration := 1.0
	endTime := time.Now().Add(time.Duration(duration) * time.Second)

	for time.Now().Before(endTime) {
		for _, frame := range frames {
			fmt.Printf("\r[%s] Loading...  ", frame)
			time.Sleep(time.Duration(delay * float64(time.Second)))
			fmt.Printf("\r[%s]             ", frame)
			time.Sleep(time.Duration(delay * float64(time.Second)))
		}
	}

	displayCheckmark("\033[92mRestart completed!\033[0m")
}
func stop() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[92m Stop \033[93mMenu\033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mWS\033[0m", "2. \033[93mWSS \033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mWS\033[0m":
		stopws()
	case "2. \033[93mWSS \033[0m":
		stopws()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		startMain()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func stopws() {
	cmd := exec.Command("clear")
	cmd.Stdout = os.Stdout
	cmd.Run()

	displayNotification("\033[93mStopping WS Reverse \033[93m..\033[0m")
	fmt.Println("\033[93m╭─────────────────────────────────────────────╮\033[0m")

	cmd = exec.Command("systemctl", "stop", "rtun-kharej")
	cmd.Run()
	time.Sleep(1 * time.Second)

	cmd = exec.Command("systemctl", "stop", "rtun-iran")
	cmd.Run()
	time.Sleep(1 * time.Second)

	fmt.Print("Progress: ")

	frames := []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}
	delay := 0.1
	duration := 1.0
	endTime := time.Now().Add(time.Duration(duration) * time.Second)

	for time.Now().Before(endTime) {
		for _, frame := range frames {
			fmt.Printf("\r[%s] Loading...  ", frame)
			time.Sleep(time.Duration(delay * float64(time.Second)))
			fmt.Printf("\r[%s]             ", frame)
			time.Sleep(time.Duration(delay * float64(time.Second)))
		}
	}

	displayCheckmark("\033[92mService Stopped!\033[0m")
}
func status() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[92m Status \033[93mMenu\033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mWS\033[0m", "2. \033[93mWSS \033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mWS\033[0m":
		wsStatus()
	case "2. \033[93mWSS \033[0m":
		wsStatus()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func wsStatus() {
	services := []string{"rtun-iran", "rtun-kharej"}

	fmt.Println("\033[93m            ╔════════════════════════════════════════════╗\033[0m")
	fmt.Println("\033[93m            ║               \033[92mReverse Status\033[93m               ║\033[0m")
	fmt.Println("\033[93m            ╠════════════════════════════════════════════╣\033[0m")

	for _, service := range services {
		cmd := exec.Command("systemctl", "is-active", "--quiet", service)
		err := cmd.Run()
		if err != nil {
			continue
		}

		status := "\033[92m✓ Active      \033[0m"
		displayName := ""
		switch service {
		case "rtun-iran":
			displayName = "\033[93mIRAN Server   \033[0m"
		case "rtun-kharej":
			displayName = "\033[93mKharej Server \033[0m"
		default:
			displayName = service
		}

		fmt.Printf("           \033[93m ║\033[0m    %s   |    %s\033[93m    ║\033[0m\n", displayName, status)
	}

	fmt.Println("\033[93m            ╚════════════════════════════════════════════╝\033[0m")
}
func UniMenu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Uninstallation \033[96mMenu\033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mWS\033[0m", "2. \033[93mWSS \033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mWS\033[0m":
		removews()
	case "2. \033[93mWSS \033[0m":
		removews()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func removews() {
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("\033[93mRemoving Config ..\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	deleteCron()
	deleteCron2()
	deleteCron3()
	deleteCron4()
	rmv()
	rmv2()


	if _, err := os.Stat("/root/reverse-tunnel"); err == nil {
		os.RemoveAll("/root/reverse-tunnel")
	}

	azumiServices := []string{
		"rtun-iran", "rtun-kharej", "tls_reset",
	}

	for _, serviceName := range azumiServices {
		hideCmd("systemctl", "disable", serviceName+".service")
		hideCmd("systemctl", "stop", serviceName+".service")
		hideCmd("rm", "/etc/systemd/system/"+serviceName+".service")
	}

	runCmd("systemctl", "daemon-reload")

	fmt.Print("Progress: ")

	frames := []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}
	delay := 100 * time.Millisecond
	duration := 1 * time.Second
	endTime := time.Now().Add(duration)

	for time.Now().Before(endTime) {
		for _, frame := range frames {
			fmt.Printf("\r[%s] Loading...  ", frame)
			time.Sleep(delay)
			fmt.Printf("\r[%s]             ", frame)
			time.Sleep(delay)
		}
	}

	displayCheckmark("\033[92m Uninstallation completed!\033[0m")
}
func hideCmd(cmd string, args ...string) error {
	command := exec.Command(cmd, args...)

	nullDevice, err := os.OpenFile(os.DevNull, os.O_WRONLY, 0666)
	if err != nil {
		return err
	}
	command.Stdout = nullDevice
	command.Stderr = nullDevice

	return command.Run()
}
func install1() {
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Installing Reverse Tunnel")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	hideCmd("apt", "install", "git", "-y")
	
	dir := "/root/reverse-tunnel"
	if _, err := os.Stat(dir); os.IsNotExist(err) {
		runCmd("git", "clone", "https://github.com/snsinfu/reverse-tunnel", dir)
	} else {
		err := os.Chdir(dir)
		if err != nil {
			log.Fatalf("\033[91mCouldn't Change Dir:\033[0m %v", err)
		}
		runCmd("git", "pull")
	}
	
	err := os.Chdir(dir)
	if err != nil {
		log.Fatalf("\033[91mCouldn't Change Dir:\033[0m %v", err)
	}
	
	runCmd("go", "build", "-o", "rtun-server", "github.com/snsinfu/reverse-tunnel/server/cmd")
	
	displayCheckmark(fmt.Sprintf("\033[92mInstallation was Successful\033[0m"))
}
func install2() {
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Installing Reverse Tunnel")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	hideCmd("apt", "install", "git", "-y")
	
	dir := "/root/reverse-tunnel"
	if _, err := os.Stat(dir); os.IsNotExist(err) {
		runCmd("git", "clone", "https://github.com/snsinfu/reverse-tunnel", dir)
	} else {
		err := os.Chdir(dir)
		if err != nil {
			log.Fatalf("\033[91mCouldn't Change Dir:\033[0m %v", err)
		}
		runCmd("git", "pull")
	}
	
	err := os.Chdir(dir)
	if err != nil {
		log.Fatalf("\033[91mCouldn't Change Dir:\033[0m %v", err)
	}
	
	runCmd("go", "build", "-o", "rtun-client", "github.com/snsinfu/reverse-tunnel/agent/cmd")
	
	displayCheckmark(fmt.Sprintf("\033[92mInstallation was Successful\033[0m"))
}

func wsMenu1() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mTCP\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIPV4\033[0m", "2. \033[93mIPV6\033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIPV4\033[0m":
		wsIP4()
	case "2. \033[93mIPV6\033[0m":
		wsIP6()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func wsIP4() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mTCP IPV4\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWst()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWst()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWst()
	case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWst()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWst()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		wsMenu1()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func wsIP6() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mTCP IPV6\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWst()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWst6()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWst6()
    case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWst6()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWst6()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		wsMenu1()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
const service1 = `[Unit]
Description=Rtun server

[Service]
Type=simple
ExecStart=/root/reverse-tunnel/./rtun-server -f /root/reverse-tunnel/rtun-server.yml
Restart=always
RestartSec=5
LimitNOFILE=1048576


[Install]
WantedBy=default.target
`
func iranWst() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m IRAN\033[92m TCP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install1()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Iran")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var numServers int
	for {
		fmt.Print("\033[93mHow many \033[92mKharej Servers\033[93m do you have?\033[0m ")
		var numServersStr string
		fmt.Scanln(&numServersStr)
		numServersStr = strings.TrimSpace(numServersStr)
		var err error
		numServers, err = strconv.Atoi(numServersStr)
		if err == nil && numServers >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	agents := make([]string, numServers)
	for i := 0; i < numServers; i++ {
		authKey := genAuthKey()
		ports := genPorts(i + 1)

		serverInfo := fmt.Sprintf("- auth_key: %s\n  ports: %s", authKey, ports)
		agents[i] = serverInfo

		box := color.New(color.FgYellow)
		key := color.New(color.FgWhite, color.BgGreen)
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("│                      %s                         │\n", box.Sprint("   Server Key ", i+1))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("%s\n", key.Sprint(authKey))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println()
	}

	configContent := fmt.Sprintf("control_address: 0.0.0.0:%d\n\nagents:\n%s", tunnelPort, strings.Join(agents, "\n"))
	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-server.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mfailed to write config:\033[0m %v\n", err)
		return
	}
	err = iranService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create iran service:\033[0m %v\n", err)
		return
	}
    enableResetIran()
	fmt.Println("\033[92mIran Service Created Successfully.\033[0m")
}


func genPorts(serverNum int) string {
	fmt.Printf("\033[93mEnter \033[92mKharej \033[93m%d \033[96mPorts \033[93m(use comma between ports):\033[0m ", serverNum)
	var portsStr string
	fmt.Scanln(&portsStr)
	portsStr = strings.TrimSpace(portsStr)
	portsSlice := strings.Split(portsStr, ",")

	ports := make([]string, len(portsSlice))
	for i, port := range portsSlice {
		port = strings.TrimSpace(port)
		ports[i] = port + "/tcp"
	}

	return fmt.Sprintf("[%s]", strings.Join(ports, ", "))
}

func iranService() error {
	content := []byte(service1)

	err := ioutil.WriteFile("/etc/systemd/system/rtun-iran.service", content, 0644)
	if err != nil {
		return err
	}
	err = exec.Command("systemctl", "daemon-reload").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to reload:\033[0m %v", err)
	}
	err = exec.Command("systemctl", "enable", "rtun-iran").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to enable Service:\033[0m %v", err)
	}
	err = exec.Command("sudo", "chmod", "u+x", "/etc/systemd/system/rtun-iran.service").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to enablin Service:\033[0m %v", err)
	}
	err = exec.Command("systemctl", "restart", "rtun-iran").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to restart Service!:\033[0m %v", err)
	}

	return nil
}
const service2 = `[Unit]
Description=Rtun client

[Service]
Type=simple
ExecStart=/root/reverse-tunnel/./rtun-client -f /root/reverse-tunnel/rtun-client.yml
Restart=always
RestartSec=5
LimitNOFILE=1048576

[Install]
WantedBy=default.target
`

func kharejWst() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m TCP IPV4 \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Kharej")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranIP string
	fmt.Print("\033[93mEnter \033[92mIran IPV4\033[93m address:\033[0m ")
	fmt.Scanln(&iranIP)
	iranIP = strings.TrimSpace(iranIP)

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var authKey string
	fmt.Print("\033[93mEnter \033[92mIRAN \033[93mkey:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)

	var configPorts []string
	for {
		fmt.Print("\033[93mEnter \033[92mconfig ports \033[93m(use comma between ports):\033[0m ")
		var configPortsStr string
		fmt.Scanln(&configPortsStr)
		configPortsStr = strings.TrimSpace(configPortsStr)
		configPortsSlice := strings.Split(configPortsStr, ",")

		configPorts = make([]string, len(configPortsSlice))
		for i, port := range configPortsSlice {
			port = strings.TrimSpace(port)
			configPorts[i] = port + "/tcp"
		}

		if len(configPorts) >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: ws://%s:%d\nauth_key: %s\n\nforwards:\n", iranIP, tunnelPort, authKey)
	for _, port := range configPorts {
		configContent += fmt.Sprintf("  - port: %s\n    destination: 127.0.0.1:%s\n", port, strings.Split(port, "/")[0])
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}

func KharejService() error {
	content := []byte(service2)

	err := ioutil.WriteFile("/etc/systemd/system/rtun-kharej.service", content, 0644)
	if err != nil {
		return err
	}
	err = exec.Command("systemctl", "daemon-reload").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to reload:\033[0m %v", err)
	}
	err = exec.Command("systemctl", "enable", "rtun-kharej").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to enable Service:\033[0m %v", err)
	}
	err = exec.Command("sudo", "chmod", "u+x", "/etc/systemd/system/rtun-kharej.service").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to enablin Service:\033[0m %v", err)
	}
	err = exec.Command("systemctl", "restart", "rtun-kharej").Run()
	if err != nil {
		return fmt.Errorf("\033[91mFailed to restart Service!:\033[0m %v", err)
	}

	return nil
}
func kharejWst6() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m TCP IPV6 \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Kharej")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranIP string
	fmt.Print("\033[93mEnter \033[92mIran IPV6\033[93m address:\033[0m ")
	fmt.Scanln(&iranIP)
	iranIP = strings.TrimSpace(iranIP)

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var authKey string
	fmt.Print("\033[93mEnter \033[92mIRAN \033[93mkey:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)

	var configPorts []string
	for {
		fmt.Print("\033[93mEnter \033[92mconfig ports \033[93m(use comma between ports):\033[0m ")
		var configPortsStr string
		fmt.Scanln(&configPortsStr)
		configPortsStr = strings.TrimSpace(configPortsStr)
		configPortsSlice := strings.Split(configPortsStr, ",")

		configPorts = make([]string, len(configPortsSlice))
		for i, port := range configPortsSlice {
			port = strings.TrimSpace(port)
			configPorts[i] = port + "/tcp"
		}

		if len(configPorts) >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: ws://[%s]:%d\nauth_key: %s\n\nforwards:\n", iranIP, tunnelPort, authKey)
	for _, port := range configPorts {
		configContent += fmt.Sprintf("  - port: %s\n    destination: 127.0.0.1:%s\n", port, strings.Split(port, "/")[0])
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}
func wsMenu2() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mUDP\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIPV4\033[0m", "2. \033[93mIPV6\033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIPV4\033[0m":
		wsIpu4()
	case "2. \033[93mIPV6\033[0m":
		wsIpu6()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func wsIpu4() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mUDP IPV4\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWsu()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWsu()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWsu()
	case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWsu()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWsu()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		wsMenu2()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func wsIpu6() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWS \033[96mUDP IPV6\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the previous menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWsu()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWsu6()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWsu6()
	case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWsu6()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWsu6()
	case "0. \033[94mBack to the previous menu\033[0m":
	    clearScreen()
		wsMenu2()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func iranWsu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m IRAN\033[92m UDP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install1()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Iran")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var numServers int
	for {
		fmt.Print("\033[93mHow many \033[92mKharej Servers\033[93m do you have?\033[0m ")
		var numServersStr string
		fmt.Scanln(&numServersStr)
		numServersStr = strings.TrimSpace(numServersStr)
		var err error
		numServers, err = strconv.Atoi(numServersStr)
		if err == nil && numServers >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	agents := make([]string, numServers)
	for i := 0; i < numServers; i++ {
		authKey := genAuthKey()
		ports := genPortu(i + 1)

		serverInfo := fmt.Sprintf("- auth_key: %s\n  ports: %s", authKey, ports)
		agents[i] = serverInfo

		box := color.New(color.FgYellow)
		key := color.New(color.FgWhite, color.BgGreen)
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("│                      %s                         │\n", box.Sprint("   Server Key ", i+1))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("%s\n", key.Sprint(authKey))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println()
	}

	configContent := fmt.Sprintf("control_address: 0.0.0.0:%d\n\nagents:\n%s", tunnelPort, strings.Join(agents, "\n"))
	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-server.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mfailed to write config:\033[0m %v\n", err)
		return
	}
	err = iranService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create iran service:\033[0m %v\n", err)
		return
	}
    enableResetIran()
	fmt.Println("\033[92mIran Service Created Successfully.\033[0m")
}
func genPortu(serverNum int) string {
	fmt.Printf("\033[93mEnter \033[92mKharej \033[93m%d \033[96mPorts \033[93m(use comma between ports):\033[0m ", serverNum)
	var portsStr string
	fmt.Scanln(&portsStr)
	portsStr = strings.TrimSpace(portsStr)
	portsSlice := strings.Split(portsStr, ",")

	ports := make([]string, len(portsSlice))
	for i, port := range portsSlice {
		port = strings.TrimSpace(port)
		ports[i] = port + "/udp"
	}

	return fmt.Sprintf("[%s]", strings.Join(ports, ", "))
}
func kharejWsu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m UDP IPV4 \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Kharej")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranIP string
	fmt.Print("\033[93mEnter \033[92mIran IPV4\033[93m address:\033[0m ")
	fmt.Scanln(&iranIP)
	iranIP = strings.TrimSpace(iranIP)

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var authKey string
	fmt.Print("\033[93mEnter \033[92mIRAN \033[93mkey:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)

	var configPorts []string
	for {
		fmt.Print("\033[93mEnter \033[92mconfig ports \033[93m(use comma between ports):\033[0m ")
		var configPortsStr string
		fmt.Scanln(&configPortsStr)
		configPortsStr = strings.TrimSpace(configPortsStr)
		configPortsSlice := strings.Split(configPortsStr, ",")

		configPorts = make([]string, len(configPortsSlice))
		for i, port := range configPortsSlice {
			port = strings.TrimSpace(port)
			configPorts[i] = port + "/udp"
		}

		if len(configPorts) >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: ws://%s:%d\nauth_key: %s\n\nforwards:\n", iranIP, tunnelPort, authKey)
	for _, port := range configPorts {
		configContent += fmt.Sprintf("  - port: %s\n    destination: 127.0.0.1:%s\n", port, strings.Split(port, "/")[0])
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}
func kharejWsu6() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m UDP IPV6 \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}

	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring Kharej")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranIP string
	fmt.Print("\033[93mEnter \033[92mIran IPV6\033[93m address:\033[0m ")
	fmt.Scanln(&iranIP)
	iranIP = strings.TrimSpace(iranIP)

	var tunnelPort int
	for {
		fmt.Print("\033[93mEnter \033[92mTunnel port: \033[0m")
		var tunnelPortStr string
		fmt.Scanln(&tunnelPortStr)
		tunnelPortStr = strings.TrimSpace(tunnelPortStr)
		var err error
		tunnelPort, err = strconv.Atoi(tunnelPortStr)
		if err == nil && tunnelPort >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	var authKey string
	fmt.Print("\033[93mEnter \033[92mIRAN \033[93mkey:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)

	var configPorts []string
	for {
		fmt.Print("\033[93mEnter \033[92mconfig ports \033[93m(use comma between ports):\033[0m ")
		var configPortsStr string
		fmt.Scanln(&configPortsStr)
		configPortsStr = strings.TrimSpace(configPortsStr)
		configPortsSlice := strings.Split(configPortsStr, ",")

		configPorts = make([]string, len(configPortsSlice))
		for i, port := range configPortsSlice {
			port = strings.TrimSpace(port)
			configPorts[i] = port + "/udp"
		}

		if len(configPorts) >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: ws://[%s]:%d\nauth_key: %s\n\nforwards:\n", iranIP, tunnelPort, authKey)
	for _, port := range configPorts {
		configContent += fmt.Sprintf("  - port: %s\n    destination: 127.0.0.1:%s\n", port, strings.Split(port, "/")[0])
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}
func cmd(name string, args ...string) error {
	cmd := exec.Command(name, args...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	err := cmd.Run()
	if err != nil {
		errorMsg := fmt.Sprintf("\033[31mfailed to run '%s %s': %v\033[0m", name, strings.Join(args, " "), err)
		return fmt.Errorf(errorMsg)
	}

	return nil
}

func hidden(name string, args ...string) error {
	cmd := exec.Command(name, args...)
	cmd.Stdout = nil
	cmd.Stderr = nil

	err := cmd.Run()
	if err != nil {
		errorMsg := fmt.Sprintf("\033[31mfailed to run '%s %s': %v\033[0m", name, strings.Join(args, " "), err)
		return fmt.Errorf(errorMsg)
	}

	return nil
}
func input(prompt string) (string, error) {
	fmt.Print(prompt)
	reader := bufio.NewReader(os.Stdin)
	input, err := reader.ReadString('\n')
	if err != nil {
		return "", err
	}

	input = strings.TrimSpace(input)

	return input, nil
}
func acme() {
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Installing acme & Getting Cert")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	err := requirments()
	if err != nil {
		log.Println("\033[91mfailed to install requirements:\033[0m", err)
	}

	err = dnlAcme()
	if err != nil {
		log.Println("\033[91mfailed to download acme:\033[0m", err)
	}
    fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	email, err := input("\033[93mEnter your \033[92mEmail\033[93m address:\033[0m ")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	if err != nil {
		log.Println("\033[91mThere was sth wrong with your email address:\033[0m", err)
	}

	err = registerAcc(email)
	if err != nil {
		log.Println("\033[91mCouldn't register account:\033[0m", err)
	}
    fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	subdomain, err := input("\033[93mEnter your \033[92mSubdomain\033[93m:\033[0m ")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	if err != nil {
		log.Println("\033[91mThere was sth wrong with your subdomain:\033[0m", err)
	}
	
	err = cert(subdomain)
	if err != nil {
		log.Println("\033[91mfailed to issue certificate:\033[0m", err)
	}

	displayCheckmark("\033[92mGetting Cert was Successful!\033[0m")
}

func requirments() error {
	err := hidden("apt", "install", "curl", "-y")
	if err != nil {
		return err
	}
	err = hidden("apt", "install", "socat", "-y")
	if err != nil {
		return err
	}

	return nil
}

func dnlAcme() error {
	err := cmd("curl", "-L", "-o", "acme.sh.tar.gz", "https://github.com/acmesh-official/acme.sh/archive/master.tar.gz")
	if err != nil {
		return err
	}
	err = cmd("tar", "xzf", "acme.sh.tar.gz")
	if err != nil {
		return err
	}
	err = os.Chdir("acme.sh-master")
	if err != nil {
		return err
	}

	err = cmd("sh", "acme.sh", "--install")
	if err != nil {
		return err
	}

	return nil
}

func registerAcc(email string) error {
	err := os.Setenv("PATH", fmt.Sprintf("%s:%s", os.Getenv("PATH"), "/root/.acme.sh"))
	if err != nil {
		return err
	}
	err = cmd("acme.sh", "--register-account", "--accountemail", email)
	if err != nil {
		return err
	}

	return nil
}

func cert(subdomain string) error {
	err := os.Setenv("PATH", fmt.Sprintf("%s:%s", os.Getenv("PATH"), "/root/.acme.sh"))
	if err != nil {
		return err
	}

	err = cmd("acme.sh", "--issue", "--standalone", "-d", subdomain)
	if err != nil {
		return err
	}

	return nil
}
func wsMenu3() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWSS \033[96mTCP\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWsst()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWsst()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWsst()
	case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWsst()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWsst()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func iranWsst() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m IRAN WSS\033[92m TCP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mHave you already obtained a cert for Iran? (\033[92myes/\033[91mno)\033[93m:\033[0m ")
	answer, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	answer = strings.ToLower(strings.TrimSpace(answer))

	if answer == "yes" || answer == "y" {
		fmt.Println("\033[91mlemme skip real quick..\033[0m")
		goto SkipCert
	}

	acme()
SkipCert:
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install1()
	} else {
		fmt.Println("\033[93m───────────────────────────────────────\033[0m")
		fmt.Println("\033[93mSkipping download..\033[0m")
	}


	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring IRAN")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var numServers int
	for {
		fmt.Print("\033[93mEnter \033[92mthe number\033[93m of \033[96mkharej servers\033[93m:\033[0m ")
		numServersStr, err := reader.ReadString('\n')
		if err != nil {
			log.Fatal(err)
		}
		numServersStr = strings.TrimSpace(numServersStr)
		numServers, err = strconv.Atoi(numServersStr)
		if err == nil {
			break 
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input!\033[0m")
	}
	fmt.Print("\033[93mEnter \033[92mTunnel port\033[93m:\033[0m ")
	tunnelPort, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	tunnelPort = strings.TrimSpace(tunnelPort)
	fmt.Print("\033[93mEnter your \033[92mSub domain\033[93m:\033[0m ")
	domain, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	domain = strings.TrimSpace(domain)
	config := fmt.Sprintf("control_address: :%s\n\nlets_encrypt:\n  domain: %s\n\nagents:\n", tunnelPort, domain)
	authKeys := genAuthK(numServers)
	for i := 0; i < numServers; i++ {
		ports := genPortzs(i + 1)
		serverInfo := fmt.Sprintf("  - auth_key: %s\n    ports: [%s]\n", authKeys[i], ports)
		config += serverInfo
		box := color.New(color.FgYellow)
		key := color.New(color.FgWhite, color.BgGreen)
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("│                      %s                         │\n", box.Sprint("   Server Key ", i+1))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("%s\n", key.Sprint(authKeys[i]))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println()
	}
	err = os.WriteFile("/root/reverse-tunnel/rtun-server.yml", []byte(config), 0644)
	if err != nil {
		fmt.Printf("\033[91mFailed to write the config: %v\n\033[0m", err)
		return
	}
	err = iranService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create iran service:\033[0m %v\n", err)
		return
	}
	enableResetIran()
	fmt.Println("\033[92mIran Service Created Successfully.\033[0m")
}

func genAuthK(numKeys int) []string {
	authKeys := make([]string, numKeys)
	for i := 0; i < numKeys; i++ {
		authKeys[i] = genAuthKey()
	}
	return authKeys
}

func genAuthKey() string {
	cmd := exec.Command("openssl", "rand", "-hex", "32")
	output, err := cmd.Output()
	if err != nil {
		fmt.Printf("\033[91mCouldn't generate the key:\033[0m %v\n", err)
		os.Exit(1)
	}

	return strings.TrimSpace(string(output))
}

func genPortzs(serverNum int) string {
	reader := bufio.NewReader(os.Stdin)

	fmt.Printf("\033[93mEnter \033[92mKharej %d\033[96m Config ports \033[93m(use comma between ports):\033[0m ", serverNum)
	portsStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	portsStr = strings.TrimSpace(portsStr)
	ports := strings.Split(portsStr, ",")
	formatPorts := make([]string, len(ports))
	for i, port := range ports {
		formatPorts[i] = strings.TrimSpace(fmt.Sprintf("%s/tcp", port))
	}

	return strings.Join(formatPorts, ",")
}
func kharejWsst() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m TCP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranDomain string
	fmt.Print("\033[93mEnter \033[92mIran subdomain\033[93m:\033[0m ")
	fmt.Scanln(&iranDomain)
	iranDomain = strings.TrimSpace(iranDomain)

	var authKey string
	fmt.Print("\033[93mEnter \033[92miran key\033[93m:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)
	var numConfigs int
	for {
		fmt.Print("\033[93mEnter the \033[92mnumber\033[93m of\033[96m configs:\033[0m ")
		var numConfigsStr string
		fmt.Scanln(&numConfigsStr)
		numConfigsStr = strings.TrimSpace(numConfigsStr)
		var err error
		numConfigs, err = strconv.Atoi(numConfigsStr)
		if err == nil && numConfigs >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: wss://%s\nauth_key: %s\n\nforwards:\n", iranDomain, authKey)
	for i := 0; i < numConfigs; i++ {
		var configPort string
		fmt.Printf("\033[93mEnter \033[92mconfig port %d: \033[0m", i+1)
		fmt.Scanln(&configPort)
		configPort = strings.TrimSpace(configPort)

		configContent += fmt.Sprintf("  - port: %s/tcp\n    destination: 127.0.0.1:%s\n", configPort, configPort)
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}


func wsMenu4() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Reverse \033[92mWSS \033[96mUDP\033[0m ")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")

	prompt := &survey.Select{
		Message: "Enter your choice Please:",
		Options: []string{"1. \033[92mIRAN\033[0m", "2. \033[93mKHAREJ\033[92m[1]\033[0m", "3. \033[93mKHAREJ\033[92m[2]\033[0m", "4. \033[93mKHAREJ\033[92m[3]\033[0m", "5. \033[93mKHAREJ\033[92m[4]\033[0m", "0. \033[94mBack to the main menu\033[0m"},
	}
    
	var choice string
	err := survey.AskOne(prompt, &choice)
	if err != nil {
		log.Fatalf("\033[91mCan't read user input, sry!:\033[0m %v", err)
	}

	switch choice {
	case "1. \033[92mIRAN\033[0m":
		iranWssu()
	case "2. \033[93mKHAREJ\033[92m[1]\033[0m":
		kharejWssu()
	case "3. \033[93mKHAREJ\033[92m[2]\033[0m":
		kharejWssu()
	case "4. \033[93mKHAREJ\033[92m[3]\033[0m":
		kharejWssu()
	case "5. \033[93mKHAREJ\033[92m[4]\033[0m":
		kharejWssu()
	case "0. \033[94mBack to the main menu\033[0m":
	    clearScreen()
		mainMenu()
	default:
		fmt.Println("\033[91mInvalid choice\033[0m")
	}

	readInput()
}
func iranWssu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m IRAN WSS\033[92m UDP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	reader := bufio.NewReader(os.Stdin)
	fmt.Print("\033[93mHave you already obtained a cert for Iran? (\033[92myes/\033[91mno)\033[93m:\033[0m ")
	answer, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	answer = strings.ToLower(strings.TrimSpace(answer))

	if answer == "yes" || answer == "y" {
		fmt.Println("\033[91mlemme skip real quick..\033[0m")
		goto SkipCert
	}

	acme()
SkipCert:
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install1()
	} else {
		fmt.Println("\033[93m───────────────────────────────────────\033[0m")
		fmt.Println("\033[93mSkipping download..\033[0m")
	}


	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	displayNotification("Configuring IRAN")
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var numServers int
	for {
		fmt.Print("\033[93mEnter \033[92mthe number\033[93m of \033[96mkharej servers\033[93m:\033[0m ")
		numServersStr, err := reader.ReadString('\n')
		if err != nil {
			log.Fatal(err)
		}
		numServersStr = strings.TrimSpace(numServersStr)
		numServers, err = strconv.Atoi(numServersStr)
		if err == nil {
			break 
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input!\033[0m")
	}

	fmt.Print("\033[93mEnter \033[92mTunnel port\033[93m:\033[0m ")
	tunnelPort, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	tunnelPort = strings.TrimSpace(tunnelPort)

	fmt.Print("\033[93mEnter your \033[92mSub domain\033[93m:\033[0m ")
	domain, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	domain = strings.TrimSpace(domain)

	config := fmt.Sprintf("control_address: :%s\n\nlets_encrypt:\n  domain: %s\n\nagents:\n", tunnelPort, domain)
	authKeys := genAuthK(numServers)
	for i := 0; i < numServers; i++ {
		ports := genPortzu(i + 1)
		serverInfo := fmt.Sprintf("  - auth_key: %s\n    ports: [%s]\n", authKeys[i], ports)
		config += serverInfo

		box := color.New(color.FgYellow)
		key := color.New(color.FgWhite, color.BgGreen)
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("│                      %s                         │\n", box.Sprint("   Server Key ", i+1))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println("╭──────────────────────────────────────────────────────────────╮")
		fmt.Printf("%s\n", key.Sprint(authKeys[i]))
		fmt.Println("╰──────────────────────────────────────────────────────────────╯")
		fmt.Println()
	}

	err = os.WriteFile("/root/reverse-tunnel/rtun-server.yml", []byte(config), 0644)
	if err != nil {
		fmt.Printf("\033[91mFailed to write configuration file: %v\n\033[0m", err)
		return
	}
	
	err = iranService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create iran service:\033[0m %v\n", err)
		return
	}
    enableResetIran()
	fmt.Println("\033[92mIran Service Created Successfully.\033[0m")
}

func genPortzu(serverNum int) string {
	reader := bufio.NewReader(os.Stdin)

	fmt.Printf("\033[93mEnter \033[92mKharej %d\033[96m Config ports \033[93m(use comma between ports):\033[0m ", serverNum)
	portsStr, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	portsStr = strings.TrimSpace(portsStr)
	ports := strings.Split(portsStr, ",")
	formatPorts := make([]string, len(ports))
	for i, port := range ports {
		formatPorts[i] = strings.TrimSpace(fmt.Sprintf("%s/udp", port))
	}

	return strings.Join(formatPorts, ",")
}
func kharejWssu() {
	clearScreen()
	fmt.Println("\033[92m ^ ^\033[0m")
	fmt.Println("\033[92m(\033[91mO,O\033[92m)\033[0m")
	fmt.Println("\033[92m(   ) \033[93m Kharej\033[92m UDP \033[0m")
	fmt.Println("\033[92m \"-\" \033[93m════════════════════════════════════\033[0m")
	if _, err := os.Stat("/root/reverse-tunnel"); os.IsNotExist(err) {
		install2()
	} else {
		fmt.Println("\033[93mSkipping download..\033[0m")
	}
	fmt.Println("\033[93m───────────────────────────────────────\033[0m")
	var iranDomain string
	fmt.Print("\033[93mEnter \033[92mIran subdomain\033[93m:\033[0m ")
	fmt.Scanln(&iranDomain)
	iranDomain = strings.TrimSpace(iranDomain)

	var authKey string
	fmt.Print("\033[93mEnter \033[92miran key\033[93m:\033[0m ")
	fmt.Scanln(&authKey)
	authKey = strings.TrimSpace(authKey)
	var numConfigs int
	for {
		fmt.Print("\033[93mEnter the \033[92mnumber\033[93m of\033[96m configs:\033[0m ")
		var numConfigsStr string
		fmt.Scanln(&numConfigsStr)
		numConfigsStr = strings.TrimSpace(numConfigsStr)
		var err error
		numConfigs, err = strconv.Atoi(numConfigsStr)
		if err == nil && numConfigs >= 1 {
			break
		}
		fmt.Println("\033[91mInvalid input. Plz enter a valid input.\033[0m")
	}

	configContent := fmt.Sprintf("gateway_url: wss://%s\nauth_key: %s\n\nforwards:\n", iranDomain, authKey)
	for i := 0; i < numConfigs; i++ {
		var configPort string
		fmt.Printf("\033[93mEnter \033[92mconfig port %d: \033[0m", i+1)
		fmt.Scanln(&configPort)
		configPort = strings.TrimSpace(configPort)

		configContent += fmt.Sprintf("  - port: %s/udp\n    destination: 127.0.0.1:%s\n", configPort, configPort)
	}

	err := ioutil.WriteFile("/root/reverse-tunnel/rtun-client.yml", []byte(configContent), 0644)
	if err != nil {
		fmt.Printf("\033[91mCouldn't write the config:\033[0m %v\n", err)
		return
	}

	err = KharejService()
	if err != nil {
		fmt.Printf("\033[91mCouldn't create kharej service:\033[0m %v\n", err)
		return
	}
    enableResetKharej()
	fmt.Println("\033[92mKharej Service Created Successfully.\033[0m")
}