import json
import os
import re
import shutil
import tempfile


def zip_workspace(workspace_path: str, output_filename: str):
    """
    Zips the specified workspace directory and returns the path to the zipped file.
    Ensure the output_filename does not have .zip extension as it's added by make_archive.
    """
    temp_dir = tempfile.mkdtemp()
    # Remove .zip if present in output_filename to avoid duplication
    base_output_path = os.path.join(
        temp_dir, output_filename.replace(".zip", "")
    )
    zip_path = shutil.make_archive(
        base_output_path, "zip", workspace_path
    )
    return zip_path  # make_archive already appends .zip


def sanitize_file_path(file_path: str):
    """
    Cleans and sanitizes the file path to be valid for Windows.
    """
    sanitized_path = file_path.replace("`", "").strip()
    # Replace any invalid characters here with an underscore or remove them
    sanitized_path = re.sub(r'[<>:"/\\|?*]', "_", sanitized_path)
    return sanitized_path


def load_json(json_string: str):
    """
    Loads a JSON string and returns the corresponding Python object.

    Args:
        json_string (str): The JSON string to be loaded.

    Returns:
        object: The Python object representing the JSON data.
    """
    json_data = json.loads(json_string)
    return json_data


# Create file that
def create_file(
    content: str,
    file_path: str,
):
    """
    Creates a file with the specified content at the specified file path.

    Args:
        content (str): The content to be written to the file.
        file_path (str): The path to the file to be created.
    """
    with open(file_path, "w") as file:
        file.write(content)
    return file_path


def create_file_in_folder(
    folder_path: str, file_name: str, content: str
):
    """
    Creates a file in the specified folder with the given file name and content.

    Args:
        folder_path (str): The path of the folder where the file will be created.
        file_name (str): The name of the file to be created.
        content (str): The content to be written to the file.

    Returns:
        str: The path of the created file.
    """
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)

    # Create the file in the folder
    file_path = os.path.join(folder_path, file_name)
    with open(file_path, "w") as file:
        file.write(content)

    return file_path


def zip_folders(folder1_path, folder2_path, zip_file_path):
    """
    Zip two folders into a single zip file.

    Args:
        folder1_path (str): Path to the first folder.
        folder2_path (str): Path to the second folder.
        zip_file_path (str): Path to the output zip file.

    Returns:
        None
    """
    # Create a temporary directory
    with tempfile.TemporaryDirectory() as temp_dir:
        # Copy both folders into the temporary directory
        shutil.copytree(
            folder1_path,
            os.path.join(temp_dir, os.path.basename(folder1_path)),
        )
        shutil.copytree(
            folder2_path,
            os.path.join(temp_dir, os.path.basename(folder2_path)),
        )
        # Create a zip file that contains the temporary directory
        shutil.make_archive(zip_file_path, "zip", temp_dir)
