ghostui.py

#!/usr/bin/python3

import os
import sys
import requests
import npyscreen
import configparser
from rich import print
from subprocess import Popen, PIPE

# Checking permissions
if int(os.getuid()) != 0:
    print("[bold white on red]You need root permissions to use this tool!!")
    sys.exit(1)

# Checking CyberGhostVPN existence
if os.path.exists("/usr/bin/cyberghostvpn") is False:
    print("[bold white on red]CyberGhostVPN seems not installed on this system!!")
    sys.exit(1)

# Available countries
vpn_country = {
    "Australia": "AU", "Armenia": "AM", "Argentina": "AR",
    "Belgium": "BE", "Bulgaria": "BG", "Brazil": "BR",
    "Canada": "CA", "Switzerland": "CH", "Chile": "CL",
    "China": "CN", "Colombia": "CO", "Czechia": "CZ",
    "Germany": "DE", "Denmark": "DK", "Estonia": "EE",
    "Kazakhstan": "KZ", "Pakistan": "PK", "Qatar": "QA", 
    "Ukraine": "UA", "United States": "US"
}

# Class for main form that handles connections and country selections
class MainForm(npyscreen.FormBaseNew):
    def create(self):
        # Terminal resolution
        self.y, self.x = self.useable_space()

        # Parsing countries
        self.get_country = self.add(
            npyscreen.TitleCombo, name="Select Country:",
            values=[cont for cont in vpn_country]
        )

        # --------------- BUTTONS ------------
        self.add(
            npyscreen.ButtonPress, name="Connect",
            when_pressed_function=self.MakeConnection, relx=20,
            rely=15
        )
        self.add(
            npyscreen.ButtonPress, name="Stop Connection",
            when_pressed_function=self.StopConnect, relx=20
        )
        self.add(
            npyscreen.ButtonPress, name="Auth Information",
            when_pressed_function=self.AuthInfo, relx=20
        )
        self.add(
            npyscreen.ButtonPress, name="Get IP Info",
            when_pressed_function=self.GetIPInfo, relx=20
        )
        self.add(
            npyscreen.ButtonPress, name="Quit",
            when_pressed_function=self.ExitButton, relx=20,
            rely=20
        )

    # ------------------- ACTIONS -------------
    def ExitButton(self):
        # Ask user for exit
        exiting = npyscreen.notify_yes_no(
            "Are you sure to quit?", "WARNING", editw=2
        )
        if exiting:
            self.parentApp.setNextForm(None)
            sys.exit(0)
        else:
            pass
    def MakeConnection(self):
        try:
            target_country = str(self.get_country.values[self.get_country.value].replace('\n', ''))
            npyscreen.notify_wait(
                f"Connecting to: {target_country}", "PROGRESS"
            )
            make_conn = ["cyberghostvpn", "--connect", "--country-code", f"{vpn_country[target_country]}"]
            comm = Popen(make_conn, stderr=PIPE, stdout=PIPE)
            comm.wait()
            npyscreen.notify_wait(
                "Requesting IP information please wait...", "PROGRESS"
            )
            ip_info = requests.get("http://ip-api.com/json")
            api_data = ip_info.json()
            npyscreen.notify_confirm(
                f"IP Address: {api_data['query']}\nCountry: {api_data['country']}\nRegion: {api_data['regionName']}\nISP: {api_data['isp']}",
                "NOTIFICATION"
            )
        except:
            pass
    def StopConnect(self):
        try:
            stops = npyscreen.notify_yes_no(
                "Are you sure to stop VPN connection?", "WARNING",
                editw=2
            )
            if stops:
                npyscreen.notify_wait(
                    "Disabling VPN connection please wait...", "PROGRESS"
                )
                stop_conn = ["cyberghostvpn", "--stop"]
                comm = Popen(stop_conn, stderr=PIPE, stdout=PIPE)
                comm.wait()
                npyscreen.notify_confirm(
                    "Your VPN connection has been stopped.", "NOTIFICATION"
                )
            else:
                pass
        except:
            pass
    def AuthInfo(self):
        auth_conf = configparser.ConfigParser()
        username = os.getenv("SUDO_USER")
        auth_conf.read(f"/home/{username}/.cyberghost/config.ini")
        npyscreen.notify_confirm(
            f"User: {auth_conf['account']['username']}",
            "INFO"
        )
    
    def GetIPInfo(self):
        try:
            npyscreen.notify_wait(
                "Requesting IP information please wait...", "PROGRESS"
            )
            ip_info = requests.get("http://ip-api.com/json")
            api_data = ip_info.json()
            npyscreen.notify_confirm(
                f"IP Address: {api_data['query']}\nCountry: {api_data['country']}\nRegion: {api_data['regionName']}\nISP: {api_data['isp']}",
                "NOTIFICATION"
            )
        except:
            pass

# Class for main application
class MainApp(npyscreen.NPSAppManaged):
    # When app starts
    def onStart(self):
        npyscreen.setTheme(npyscreen.Themes.ColorfulTheme)

        # Our main application
        self.addForm(
            "MAIN", MainForm, name="CyberGhostVPN TUI v0.2", lines=30, columns=90
        )

# Execution area
if __name__ == '__main__':
    try:
        app = MainApp()
        app.run()
    except KeyboardInterrupt:
        print("[+] Goodbye...")
        sys.exit(0)