import os

import sys

import pandas as pd

from glob import glob

from .logger import logger

import re

def extract_parameter(text):

    '''

    Extract the parameter number from the text.

    Args:

        text ('str'): Text to search for the parameter number.

    Returns:

        parameter_number ('str'): Parameter number.

    '''

    match = re.search(r'Par(\w+)', text)

    if match:

        return 'Par' + match.group(1)

    else:

        return None

def get_points_paths(base_path, suffix, num_occurrences = 2):

    '''

    Get the paths of the transformed points.

    Args:

        args ('argparse.Namespace'): Command line arguments.

        suffix ('str'): Suffix of the file name.

        num_occurrences ('int'): Number of occurrences of the suffix in the path.

    Returns:

        paths ('list'): List of paths.

    '''

    paths = [path.replace('\\', '/') for path in sorted(glob(os.path.join(base_path, *["***"] * num_occurrences , f"*{suffix}.txt"), recursive=True))]

    return paths

def get_paths(args, suffix):

    '''

    Get the paths of the volumes and segmentations.

    Args:

        args ('argparse.Namespace'): Command line arguments.

        suffix ('str'): Suffix of the file name.

    Returns:

        paths ('list'): List of paths.

    '''

    paths = [path.replace('\\', '/') for path in sorted(glob(os.path.join(args.dataset_path, "***" , f"*{suffix}.nii.gz"), recursive=True))]

    return paths

def check_paths(args, paths, message):

    '''

    Check if the paths exist.

    Args:

        args ('argparse.Namespace'): Command line arguments.

        paths ('list'): List of paths.

        message ('str'): Message to display in the logger.

    Returns:

        None

    '''

    if len(paths) == 0:

        logger.error(f"No {message} found in {args.dataset_path} directory.")

        sys.exit(1)

def create_directory_if_not_exists(path):

    '''

    Create a directory if it does not exist.

    Args:

        path ('str'): Directory path.

    '''

    if not os.path.exists(path):

        os.makedirs(path)

def replace_text_in_file(file_path, search_text, replacement_text):

        '''

        Replace text in a text file.

        Args:

            file_path ('str'): Path to the text file.

            search_text ('str'): Text to search for in the file.

            replacement_text ('str'): Text to replace the searched text with.

        '''

        try:

            # Read the file

            with open(file_path, 'r') as file:

                content = file.read()

            # Replace the search_text with replacement_text

            modified_content = content.replace(search_text, replacement_text)

            # Write the modified content back to the file

            with open(file_path, 'w') as file:

                file.write(modified_content)

            # print(f"Text replaced in {file_path} and saved.")

        except FileNotFoundError:

            print(f"File not found: {file_path}")

        # except Exception as e:

        #     print(f"An error occurred: {e}")

def add_and_delete_rows(file_path, row1, row2):

    # Add two rows at the beginning of the file

    with open(file_path, 'r+') as file:

        content = file.read()

        file.seek(0, 0)

        file.write(row1 + '\n' + row2 + '\n' + content)

def delete_added_rows(file_path):

    # Delete the first two rows from the file

    with open(file_path, 'r') as file:

        lines = file.readlines()

    # Handle different newline characters when writing back to the file

    with open(file_path, 'w', newline=os.linesep) as file:

        file.writelines(lines[2:])