Install.py

import subprocess
import urllib.request
import os
import shutil
from colorama import init, Fore, Style

# Initialize colorama
init(autoreset=True)

# Function to display the KIRMKIT banner
def display_banner():
    print(f"{Fore.GREEN}K   K  III  RRRR   M   M  K   K  III  TTTTT\nK  K    I   R   R  MM MM  K  K    I     T\nKKK     I   RRRR   M M M  KKK     I     T\nK  K    I   R  R   M   M  K  K    I     T\nK   K  III  R   RR M   M  K   K  III    T    http://github.com/kirm1/KirmKit.git\n================================================================================================================================{Style.RESET_ALL}")

# Function to check if Nmap is installed
def check_nmap_installation():
    try:
        # Choose the appropriate command based on the platform
        if os.name == 'nt':  # Windows
            subprocess.run(['nmap', '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
        elif os.name == 'posix':  # Linux
            subprocess.run(['nmap', '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
        return True
    except (subprocess.CalledProcessError, FileNotFoundError):
        return False

# Function to download and run the Nmap installer with elevated privileges
def install_nmap():
    print(f"\nDownloading and installing {Fore.RED}[{Fore.GREEN}nmap{Style.RESET_ALL}{Fore.RED}]{Style.RESET_ALL}...")

    # Download the Nmap installer
    nmap_installer_url = "https://nmap.org/dist/nmap-7.94-setup.exe" if os.name == 'nt' else None

    if nmap_installer_url:
        installer_path = "nmap_setup.exe"
        urllib.request.urlretrieve(nmap_installer_url, installer_path)

        # Run the installer with elevated privileges
        result = subprocess.run(['nmap_setup.exe'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

        if result.returncode == 0:
            print(f"{Fore.RED}[{Fore.GREEN}KirmKit{Style.RESET_ALL}{Fore.RED}]{Style.RESET_ALL} Installation Successful!")
        else:
            print(f"[{Fore.RED}ERROR{Style.RESET_ALL}] KirmKit installation failed.")
            print(result.stderr.decode())
            exit(1)
    else:
        # Linux platform, use package manager for installation
        result = subprocess.run(['sudo', 'apt-get', 'install', 'nmap'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)

        if result.returncode == 0:
            print(f"{Fore.RED}[{Fore.GREEN}KirmKit{Style.RESET_ALL}{Fore.RED}]{Style.RESET_ALL} Installation Successful!")
        else:
            print(f"[{Fore.RED}ERROR{Style.RESET_ALL}] KirmKit installation failed.")
            print(result.stderr.decode())
            exit(1)

# Function to clone GitHub repositories
def clone_github_repositories():
    repos = ["https://github.com/palahsu/DDoS-Ripper.git"]

    for repo in repos:
        repo_name = repo.split("/")[-1].split(".")[0]

        if os.path.exists(repo_name):
            replace_existing = input(f"\nFound Existing Directory '{repo_name}'. Would you like to replace it with a newer version? [Y/n]: ")
            if replace_existing.lower() != 'y':
                print(f"[{Fore.RED}ERROR{Style.RESET_ALL}] KirmKit installation has failed.")
                exit(1)
            else:
                print(f"Replacing existing '{repo_name}'...")

                # Remove existing directory in a cross-platform manner
                shutil.rmtree(repo_name, ignore_errors=True)

        print(f"\nInstalling dependencies...")

        # Run the git clone command with progress
        cmd = ['git', 'clone', repo]
        process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, universal_newlines=True)

        # Display progress bar
        progress = 0
        while True:
            output = process.stdout.readline()
            if output == '' and process.poll() is not None:
                break
            if 'Receiving objects' in output:
                progress = int(output.split()[3][:-1])
                print(f"\rProgress: {'=' * progress}{'>'}{' ' * (100 - progress)}", end='', flush=True)

        if process.returncode != 0:
            print(f'\n[{Fore.RED}ERROR{Style.RESET_ALL}] Failed to install dependencies.')
            exit(1)

    print(f"\n[{Fore.GREEN}KirmKit{Style.RESET_ALL}] Installation has finished!")

# Display the installation menu with color
print("\n" + Fore.GREEN + "Choose an option:" + Style.RESET_ALL)
print(f"{Fore.RED}[{Fore.GREEN}1{Style.RESET_ALL}{Fore.RED}]{Style.RESET_ALL} Install KIRMKIT")
print(f"{Fore.GREEN}[{Fore.RED}2{Style.RESET_ALL}{Fore.GREEN}]{Style.RESET_ALL} Exit")

# Prompt the user for input
choice = input("Enter your choice (1 or 2): ")

# Case statement to handle user choice
if choice == '1':
    if check_nmap_installation():
        print(f"[{Fore.RED}nmap{Style.RESET_ALL}] Is Already Installed.")
    else:
        install_nmap()

    # Clone GitHub repositories regardless of Nmap installation status
    clone_github_repositories()

elif choice == '2':
    print("\nExiting script.")
    exit(0)
else:
    print("\nInvalid choice. Please enter 1 or 2.")

input('\nPress enter to exit...')