setup.py

#!/usr/bin/env python3
"""
Setup script for Advanced Cloud Penetration Testing Suite
Handles installation, configuration, and dependency management
"""

import os
import sys
import subprocess
import platform
import json
from pathlib import Path

class CloudPentestSetup:
    """Setup and configuration manager for the Cloud Pentest Suite"""

    def __init__(self):
        self.config_dir = Path.home() / '.cloud_pentest_suite'
        self.config_file = self.config_dir / 'config.json'
        self.log_dir = self.config_dir / 'logs'

    def create_directories(self):
        """Create necessary directories"""
        print("Creating application directories...")

        directories = [self.config_dir, self.log_dir]
        for directory in directories:
            directory.mkdir(parents=True, exist_ok=True)
            print(f"✓ Created: {directory}")

    def check_system_requirements(self):
        """Check system requirements and dependencies"""
        print("Checking system requirements...")

        # Check Python version
        python_version = sys.version_info
        if python_version < (3, 8):
            print(f"✗ Python 3.8+ required. Current version: {python_version.major}.{python_version.minor}")
            return False
        print(f"✓ Python version: {python_version.major}.{python_version.minor}.{python_version.micro}")

        # Check operating system
        os_name = platform.system()
        print(f"✓ Operating System: {os_name} {platform.release()}")

        # Check if running on Windows
        if os_name != "Windows":
            print("⚠ This application is optimized for Windows but can run on other platforms")

        return True

    def install_system_dependencies(self):
        """Install system-level dependencies"""
        print("Checking system dependencies...")

        # Check for nmap
        try:
            subprocess.run(['nmap', '--version'], capture_output=True, check=True)
            print("✓ Nmap is installed")
        except (subprocess.CalledProcessError, FileNotFoundError):
            print("✗ Nmap not found. Please install Nmap:")
            print("  Download from: https://nmap.org/download.html")
            print("  Or use package manager: winget install Nmap.Nmap")
            return False

        return True

    def install_python_dependencies(self):
        """Install Python dependencies"""
        print("Installing Python dependencies...")

        try:
            # Upgrade pip first
            subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', 'pip'])

            # Install requirements
            subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt'])
            print("✓ Python dependencies installed successfully")
            return True

        except subprocess.CalledProcessError as e:
            print(f"✗ Failed to install Python dependencies: {e}")
            return False

    def create_default_config(self):
        """Create default configuration file"""
        print("Creating default configuration...")

        default_config = {
            "version": "2.0",
            "settings": {
                "auto_save_results": True,
                "max_threads": 10,
                "timeout_seconds": 30,
                "default_ports": "21,22,23,25,53,80,110,143,443,993,995,8080,8443",
                "log_level": "INFO"
            },
            "aws": {
                "default_region": "us-east-1",
                "max_retries": 3
            },
            "azure": {
                "default_subscription": "",
                "max_retries": 3
            },
            "gcp": {
                "default_project": "",
                "max_retries": 3
            },
            "reporting": {
                "auto_generate": True,
                "format": "excel",
                "include_screenshots": False
            }
        }

        try:
            with open(self.config_file, 'w') as f:
                json.dump(default_config, f, indent=4)
            print(f"✓ Configuration file created: {self.config_file}")
            return True
        except Exception as e:
            print(f"✗ Failed to create configuration: {e}")
            return False

    def verify_installation(self):
        """Verify the installation is working"""
        print("Verifying installation...")

        try:
            # Test imports
            test_imports = [
                'tkinter', 'boto3', 'azure.identity', 'google.cloud.compute_v1',
                'nmap', 'paramiko', 'pandas', 'xlsxwriter'
            ]

            for module in test_imports:
                try:
                    __import__(module)
                    print(f"✓ {module}")
                except ImportError as e:
                    print(f"✗ {module}: {e}")
                    return False

            print("✓ All modules imported successfully")
            return True

        except Exception as e:
            print(f"✗ Verification failed: {e}")
            return False

    def setup_windows_shortcuts(self):
        """Create Windows shortcuts and Start Menu entries"""
        if platform.system() != "Windows":
            return True

        print("Setting up Windows shortcuts...")

        try:
            import winshell
            from win32com.client import Dispatch

            # Create desktop shortcut
            desktop = winshell.desktop()
            shortcut_path = os.path.join(desktop, "Cloud Pentest Suite.lnk")

            shell = Dispatch('WScript.Shell')
            shortcut = shell.CreateShortCut(shortcut_path)
            shortcut.Targetpath = sys.executable
            shortcut.Arguments = os.path.abspath("CloudStrikeX.py")
            shortcut.WorkingDirectory = os.path.dirname(os.path.abspath("CloudStrikeX.py"))
            shortcut.IconLocation = os.path.abspath("assets/CloudStrikeX_icon.ico")
            shortcut.save()

            print("✓ Desktop shortcut created")
            return True

        except ImportError:
            print("⚠ Windows shortcuts not created (missing winshell/pywin32)")
            return True
        except Exception as e:
            print(f"⚠ Failed to create shortcuts: {e}")
            return True

    def run_setup(self):
        """Run the complete setup process"""
        print("=" * 60)
        print("CloudStrikeX - Advanced Cloud Penetration Testing Platform - Setup")
        print("=" * 60)

        steps = [
            ("Checking system requirements", self.check_system_requirements),
            ("Creating directories", self.create_directories),
            ("Installing system dependencies", self.install_system_dependencies),
            ("Installing Python dependencies", self.install_python_dependencies),
            ("Creating configuration", self.create_default_config),
            ("Verifying installation", self.verify_installation),
            ("Setting up shortcuts", self.setup_windows_shortcuts)
        ]

        for step_name, step_func in steps:
            print(f"\n{step_name}...")
            if not step_func():
                print(f"\n✗ Setup failed at: {step_name}")
                return False

        print("\n" + "=" * 60)
        print("SETUP COMPLETED SUCCESSFULLY!")
        print("=" * 60)
        print("\nYou can now run the application using:")
        print("  python CloudStrikeX.py")
        print("\nOr build an executable using:")
        print("  python build_exe.py")
        print("\nConfiguration location:")
        print(f"  {self.config_file}")
        print("=" * 60)

        return True

class CloudPentestLauncher:
    """Application launcher with pre-flight checks"""

    def __init__(self):
        self.setup = CloudPentestSetup()

    def pre_flight_check(self):
        """Perform pre-flight checks before launching"""
        print("Performing pre-flight checks...")

        # Check if configuration exists
        if not self.setup.config_file.exists():
            print("⚠ Configuration not found. Running setup...")
            if not self.setup.run_setup():
                return False

        # Quick dependency check
        critical_modules = ['tkinter', 'boto3']
        for module in critical_modules:
            try:
                __import__(module)
            except ImportError:
                print(f"✗ Critical module missing: {module}")
                print("Please run setup: python setup.py")
                return False

        print("✓ Pre-flight checks passed")
        return True

    def launch_application(self):
        """Launch the main application"""
        if not self.pre_flight_check():
            return False

        print("Launching CloudStrikeX - Advanced Cloud Penetration Testing Platform...")

        try:
            # Import and run the main application
            from CloudStrikeX import CloudPentestGUI
            app = CloudPentestGUI()
            app.run()
            return True

        except Exception as e:
            print(f"✗ Failed to launch application: {e}")
            return False

def main():
    """Main entry point"""
    if len(sys.argv) > 1 and sys.argv[1] == 'launch':
        # Launch mode
        launcher = CloudPentestLauncher()
        if not launcher.launch_application():
            sys.exit(1)
    else:
        # Setup mode
        setup = CloudPentestSetup()
        if not setup.run_setup():
            sys.exit(1)

if __name__ == "__main__":
    main()