#    Copyright (C) 2025 James Brotosky, Brandon Wickline
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as published
#    by the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.


import logging
import os
import platform
import re
import subprocess
import tempfile
import tkinter as tk
from tkinter import filedialog

import pandas as pd

from utils.configmanager import load_env

logger = logging.getLogger(__name__)




def import_to_dataframe(file_path: str) -> pd.DataFrame:
    df = pd.DataFrame()

    try:
        if not os.path.exists(file_path):
            print(colorText(f"Error: File '{file_path}' does not exist.", "red"))
            return df

        ext = os.path.splitext(file_path)[1].lower()

        if ext == ".csv":
            df = pd.read_csv(file_path)
        elif ext == ".parquet":
            df = pd.read_parquet(file_path)
        else:
            print(colorText(f"Error: Unsupported file extension '{ext}'.", "red"))
            return df

        if df.empty:
            print(colorText("Error: File has headers but no data rows.", "red"))
        else:
            print(colorText(f"Data loaded successfully from {file_path}", "green"))

        return df

    except pd.errors.EmptyDataError:
        print(
            colorText(
                "Notice: CSV file is completely empty, falling back to empty frame",
                "white",
            )
        )
        return pd.DataFrame()

    except Exception as e:
        print(colorText(f"Error reading file: {e}", "red"))
        return pd.DataFrame()


def choose_directory():
    root = tk.Tk()
    root.withdraw()  # Hide the main window
    directory = filedialog.askdirectory(title="Select a Directory")
    print("Selected directory:", directory)
    return directory


def choose_file(initial_directory=None, required_substring=None):
    """Open a file dialog and ensure the selected file contains a required substring."""
    while True:
        root = tk.Tk()
        root.withdraw()  # Hide the main window
        file_path = filedialog.askopenfilename(initialdir=initial_directory)

        if not file_path:
            print("No file selected.")
            return None

        if required_substring and required_substring not in file_path:
            print(
                f"The selected file must contain '{required_substring}' in its path or name. Please try again."
            )
        else:
            return file_path




def get_sanitized_input(prompt: str) -> str:
    while True:
        user_input = input(prompt)
        if user_input.strip() == "":
            return user_input  # Allow blank lines
        if re.match(r'^[a-zA-Z0-9_\- .]+$', user_input.strip()):
            return user_input
        else:
            print("Invalid input. Only letters, numbers, underscores, spaces, hyphens, and periods are allowed.")


def regulator(paths, case_insensitive=True):
    """
    Build a regex pattern that matches any of the given Windows path fragments.
    """
    escaped = [re.escape(p) for p in paths]
    pattern = "(?:" + "|".join(escaped) + ")"
    if case_insensitive:
        pattern = "(?i)" + pattern  # Add inline case-insensitive flag
    print(f"Regulator is providing: {pattern}")
    return pattern
def irtang():
        print(
        colorText(
            r"""
                                                           ███                  
                ████                                  ░████████                 
               █████████████                      ███████████████               
              █████████████████████          █████████████████████              
              ███████████████████           ██████████████████████▓             
             ███████████████████             ██████████████████████             
            █████████████████████            ███████████████████████            
            ████████████████████████████████████████████████████████            
            █████████      ██              ██              █████████            
            █████████      ██      ███      █              █████████            
            █████████      ██      ████     █████      █████████████            
            █████████      ██              ██████      █████████████            
             ████████      ██             ███████      ████████████░            
              ███████      ██      ██▓     ██████      ████████████             
               ██████      ██      ████     █████      ███████████              
                █████████████████████████████████████████████████               
                  ▒████████████████████      ██████████████████                 
                     ███████████████████      ███████████████▒                  
                         ███████████████      █████████████                     
                                ██████████    ███████████                       
                                             ████████                           
                                             ████                               
""",
            "yellow",
        )
    )
def displayIntro():

    print(
        colorText(
            r"""
   _____  .__       .__                 __     ___________           .__          
  /  _  \ |__|______|  |   ____   ____ |  | __ \__    ___/___   ____ |  |   ______
 /  /_\  \|  \_  __ \  |  /  _ \_/ ___\|  |/ /   |    | /  _ \ /  _ \|  |  /  ___/
/    |    \  ||  | \/  |_(  <_> )  \___|    <    |    |(  <_> |  <_> )  |__\___ \ 
\____|__  /__||__|  |____/\____/ \___  >__|_ \   |____| \____/ \____/|____/____  >
        \/                           \/     \/                                 \/ 
""",
            "cyan",
        )
    )
def welcome():
    print(
        colorText(
            "=================================================================================",
            "cyan",
        )
    )
    print(
        colorText(
            "======================== Welcome to the Airlock API Tool ========================",
            "cyan",
        )
    )
    print(
        colorText(
            "=================================================================================",
            "cyan",
        )
    )

def section_header(title):
    print(colorText("\n --------------------------------------------------------------------", "cyan"))
    print(colorText(f" ------------- {title} -------------", "cyan"))
    print(colorText(" --------------------------------------------------------------------", "cyan"))

def printEnforceChecklist(selected_policies, destination_policy, destination_allowlist):
    working_dir = load_env("WORKING_DIR")
    section_header("🛠️ 🔒 Prepare to Enforce Policy 🛠️ 🔒")
    print(colorText("\nSequentially follow these steps to prepare a policy for enforcement:", "white"))

    # Step 1: Originating Policies
    print(colorText("\n1. Choose which policy or policies to gather execution info from", "cyan"))
    if not selected_policies:
        print(colorText("  [✗]  No policies have been chosen", "red"))
    else:
        print(colorText("The following policies have been chosen:", "green"))
        for policy in selected_policies:
            print(colorText(f"  [✓] {policy.name}", "green"))

    # Step 2: Destination Policy and Allowlist
    print(colorText("2. Choose the destination policy and associated allowlist", "cyan"))
    if destination_policy:
        print(colorText(f"  [✓] {destination_policy[0].name} has been selected as the destination policy", "green"))
    else:
        print(colorText("  [✗]  No destination policy has been chosen", "red"))

    if destination_allowlist:
        print(colorText(f"  [✓] {destination_allowlist[0].name} has been selected as allowlist", "green"))
    else:
        print(colorText("  [✗]  No allowlist has been chosen", "red"))

    # Step 3: Data Preparation
    print(colorText(f"3. Select to begin pulling execution history. The executions will be sorted and placed in {working_dir}\\data\\Needs_Review", "cyan"))
    if selected_policies:
        policy_id = selected_policies[0].name
        review_path = f"{working_dir}\\Needs_Review\\Review_First\\{policy_id}_approved_executions.csv"
        print(colorText("  [✓] Data has been fetched" if os.path.exists(review_path) else "  [✗] Data has not been fetched", "green" if os.path.exists(review_path) else "red"))
    else:
        print(colorText("  [✗]  No policies selected, cannot check data fetch status", "red"))

    # Step 4: Manual Review
    print(colorText("4. Manually review the files:", "cyan"))
    print(colorText("     Remove the rows containing hashes you do not approve of", "cyan"))
    print(colorText(f"     When complete, save both csv files to {working_dir}\\data\\Approved and choose this option.", "cyan"))
    print(colorText("     This will start the process to generate possible filepath approvals", "cyan"))

    if selected_policies:
        policy_id = selected_policies[0].name
        approved_path = f"{working_dir}\\Approved\\{policy_id}_approved_executions.csv"
        second_review_path = f"{working_dir}\\Needs_Review\\Review_Second\\{policy_id}_primary_Paths.csv"
        print(colorText("  [✓]  Reviewed hashes have been loaded" if os.path.exists(approved_path) else "  [✗]  Reviewed hashes have not been loaded", "green" if os.path.exists(approved_path) else "red"))
        print(colorText("  [✓]  Path review list created" if os.path.exists(second_review_path) else "  [✗]  Path review list has not been created", "green" if os.path.exists(second_review_path) else "red"))
    else:
        print(colorText("  [✗]  No policies selected, cannot check reviewed hashes or path list", "red"))

    # Step 5: Path Review
    print(colorText(f"5. Manually review the files in {working_dir}\\Needs_Review\\Review_Second\\", "cyan"))
    print(colorText("     Remove the rows containing path exclusions or publishers you do not approve of.", "cyan"))
    print(colorText(f"     When complete, save the files to {working_dir}\\data\\Approved", "cyan"))
    print(colorText("     Choose this option when done to build your preflights", "cyan"))

    if selected_policies:
        policy_id = selected_policies[0].name
        reviewed_path = f"{working_dir}\\Approved\\{policy_id}_primary_Paths.csv"
        preflight_paths = f"{working_dir}\\Preflight\\{policy_id}_approved_paths.csv"
        preflight_hashes = f"{working_dir}\\Preflight\\{policy_id}_approved_hashes.csv"
        print(colorText("  [✓]  Reviewed path list detected" if os.path.exists(reviewed_path) else "  [✗]  Path review list has not been detected", "green" if os.path.exists(reviewed_path) else "red"))
        preflight_ready = os.path.exists(preflight_paths) and os.path.exists(preflight_hashes)
        print(colorText("  [✓]  Preflight Path Exclusion List has been generated" if preflight_ready else "  [✗]  Preflight Path Exclusion List has not been generated", "green" if preflight_ready else "red"))
    else:
        print(colorText("  [✗]  No policies selected, cannot check preflight status", "red"))

    # Final Steps
    print(colorText("6. Test ------------------------------------------------------", "cyan"))
    print(colorText("     Prints to console the changes that would be made, must be done to proceed. ", "cyan"))

    print(colorText("7. Liftoff ------------------------------------------------------", "cyan"))
    print(colorText("     Apply path exclusions and approved publishers to selected policy", "cyan"))
    print(colorText("     Apply approved hashes to allowlist", "cyan"))


    # Utility Options
    print(colorText("R. Remove/Reset Generated data - will prompt to allow keeping execution history", "cyan"))
    print(colorText("F. 📂  - Open Working Directory", "cyan"))
    print(colorText("B. 🔚  - Back", "cyan"))


def areYouSure():
    print(
        colorText(
            "🛑****************************************************************************************************************************************🛑",
            "red",
        )
    )
    print(
        colorText(
            "⚠️=========================================================================================================================================⚠️",
            "yellow",
        )
    )
    print(
        colorText(
            "🛑========================================================================================================================================🛑",
            "red",
        )
    )
    print(
        colorText(
            "⚠️-------------This program will now begin to make changes to the Airlock Console. Do you understand and agree to proceed? ----------------⚠️",
            "yellow",
        )
    )
    print(
        colorText(
            "🛑========================================================================================================================================🛑",
            "red",
        )
    )
    print(
        colorText(
            "⚠️=========================================================================================================================================⚠️",
            "yellow",
        )
    )
    print(
        colorText(
            "🛑****************************************************************************************************************************************🛑",
            "red",
        )
    )


def locked():
    print(
        colorText(
            r"""
        ████████████████████████████████████████████████████████████████        
      ███                                                              ██       
      ██                             ██████                            ███      
      ██                          ████████████                         ███      
      ██                        ████        ███                        ███      
      ██                        ███          ███                       ███      
      ██                        ███          ███                       ███      
      ██                     ▒████████████████████                     ███      
      ██                     ██████████████████████                    ███      
      ██                     ██████████████████████                    ███      
      ██                     ██████████████████████                    ███      
      ██                     ██████████████████████                    ███      
      ██                     ██████████████████████                    ███      
      ██                                                               ███      
      ███                                                              ███      
      ████████████████████████████████████████████████████████████████████            
      ▒██████████████████████████████████████████████████████████████████▒  
                                     ▒████                                      
                                     ▒████                                      
                  ▓██████████████████████████████████████████                   
                 █████████████████████████████████████████████░  
""",
            "yellow",
        )
    )


def printDeviceEnforceChecklist():
    print(
        colorText(
            "\n --------------------------------------------------------------------",
            "cyan",
        )
    )
    print(
        colorText(
            " ------------- 🛠️ 🔒 Prepare to Enforce Policy 🛠️ 🔒 ------------------",
            "cyan",
        )
    )
    print(
        colorText(
            " --------------------------------------------------------------------",
            "cyan",
        )
    )
    print(
        colorText(
            "\nSequentually follow these steps to prepare a policy for enforcement:",
            "white",
        )
    )

    print(
        colorText(
            "\n1. Choose which originating policy or policies to move to enforcement",
            "cyan",
        )
    )
    print(
        colorText(
            "2. Pull and stage event history, combine the histories, add hash info, then categorize the hashes",
            "cyan",
        )
    )
    print(colorText("3. Manually review the files:", "cyan"))
    print(
        colorText(
            "     'needs_approved\\good_{first_policy}_{second_policy}.csv' and 'needs_approved\\unknown_{first_policy}_{second_policy}.csv'",
            "cyan",
        )
    )
    print(
        colorText(
            "     Remove the rows containing hashes you do not approve of, and those you would not approve of without metarules.",
            "cyan",
        )
    )
    print(
        colorText(
            "     If metarules need to be created, please make note of them, and remove the row from the csv.",
            "cyan",
        )
    )
    print(
        colorText(
            "     When complete, save both csv files to the directory 'approved' and choose this option.",
            "cyan",
        )
    )
    print(
        colorText(
            "     This will combine these approved hashes with the automatically approved hashes and generate a list of paths to be reviewed",
            "cyan",
        )
    )
    print(
        colorText(
            "4. Manually review the file 'needs_approved\\paths_needing_review.csv'",
            "cyan",
        )
    )
    print(
        colorText(
            "     Remove the rows containing path exclusions you do not approve of",
            "cyan",
        )
    )
    print(colorText("     When complete, save the csv file to the directory 'approved'", "cyan"))
    print(
        colorText(
            "     Do the same process with the list of publishers forthe same directories",
            "cyan",
        )
    )
    print(colorText("     Preflight Lists will be generated", "cyan"))

    print(colorText("5. Choose the destination policy and parent and child allow list", "cyan"))

    print(colorText("6. Test ------------------------------------------------------", "cyan"))
    print(colorText("     Print rather than apply selected data.", "cyan"))

    print(colorText("7. Liftoff ------------------------------------------------------", "cyan"))
    print(
        colorText(
            "     Apply path exclusions according to allowed and approved paths",
            "cyan",
        )
    )
    print(colorText("     Apply signed or attested hashes to Parent Allow List", "cyan"))
    print(colorText("     Apply approved, but unsigned hashes to the Child Allow List", "cyan"))

    print(
        colorText(
            "R. Remove/Reset Generated data - will prompt to allow keeping execution history",
            "cyan",
        )
    )

    print(colorText("B. Back", "cyan"))


def colorText(text, color):
    colors = {
        "red": "\033[91m",
        "green": "\033[92m",
        "yellow": "\033[93m",
        "blue": "\033[94m",
        "magenta": "\033[95m",
        "cyan": "\033[96m",
        "white": "\033[97m",
        "reset": "\033[0m",
    }
    return f"{colors.get(color, colors['reset'])}{text}{colors['reset']}"


def formatHTML(df, output_html_path=None, overwrite=True):
    from datetime import datetime

    # Get current date and filename for subtitle
    today = datetime.now().strftime("%d %B %Y")  # Changed to "Day Month Year"
    filename = output_html_path.replace(".html", "") if output_html_path else "Report"

    dark_css = """
    <style>
    body {
        background-color: #000000;
        margin: 0;
        padding: 0;
        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
        color: #f8f8f2;
    }
    .header {
        text-align: center;
        margin: 20px auto;
        padding: 10px;
        border-bottom: 2px solid #ffd700;
        max-width: 95%;
    }
    .header h1 {
        color: #ffd700;
        margin: 0;
        font-size: 32px;
    }
    .header p {
        color: #00bfff;
        margin: 5px 0 0 0;
        font-size: 18px;
    }
    .table-container {
        overflow-y: scroll;
        margin: 0 auto;
        width: 95%;
        max-height: calc(80vh - 100px);
        display: block;
        border: 1px solid #3a3a4d;
        margin-bottom: 0;
    }
    table {
        border-collapse: collapse;
        font-size: 14px;
        background-color: #1e1e2f;
        color: #f8f8f2;
        width: max-content;
    }
    th, td {
        border: 1px solid #3a3a4d;
        text-align: left;
        padding: 10px;
        max-width: 300px;
        word-wrap: break-word;
        overflow-wrap: break-word;
    }
    /* First column: no wrap */
    td:nth-child(1), th:nth-child(1) {
        white-space: nowrap;
        max-width: none !important;
        word-wrap: normal !important;
    }
    th {
        background-color: #2e2e40;
        color: #ffd700;
        position: sticky;
        top: 0;
        z-index: 10;
    }
    tr:nth-child(even) {
        background-color: #262638;
    }
    tr:hover {
        background-color: #33334d;
        color: #00bfff;
    }
    /* Custom scrollbar styling */
    .table-container::-webkit-scrollbar {
        width: 12px;
    }
    .table-container::-webkit-scrollbar-track {
        background: #1e1e2f;
    }
    .table-container::-webkit-scrollbar-thumb {
        background-color: #3a3a4d;
        border-radius: 6px;
    }
    </style>
    """

    header = f"""
    <div class="header">
        <h1>Airlock Tools</h1>
        <p>{filename} - {today}</p>
    </div>
    """

    html_table = df.to_html(index=False, escape=False)
    styled_html = (
        f"<html>\n"
        f"<head><title>Airlock Tools Report</title></head>\n"
        f"<body>\n"
        f"{dark_css}\n"
        f"{header}\n"
        f"<div class='table-container'>\n"
        f"  {html_table}\n"
        f"</div>\n"
        f"</body>\n"
        f"</html>"
    )
    if output_html_path:
        with open(output_html_path, "w", encoding="utf-8") as f:
            f.write(styled_html)
        print(f"✅ Styled table saved to '{output_html_path}'")
    elif overwrite:
        with tempfile.NamedTemporaryFile(
            suffix=".html", delete=False, mode="w", encoding="utf-8"
        ) as f:
            f.write(styled_html)
            temp_path = f.name

            print(f"✅ Styled table saved to temporary file: {temp_path}")
    else:
        return styled_html



def open_directory(path):
    system = platform.system()
    
    if system == "Windows":
        os.startfile(path)
    elif system == "Linux":
        subprocess.run(["xdg-open", path])
    else:
        raise OSError(f"Unsupported operating system: {system}")


def print_x_wide(items: list, width: int):
    for i in range(0, len(items), width):
        row = items[i:i+width]
        print(" | ".join(row))

        
def clear_screen():
    os.system('cls' if os.name == 'nt' else 'clear')