import re

class StringStripNode:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "input_string": ("STRING", {
                    "forceInput": True,
                    "tooltip": "The string to process"
                }),
                "strings_to_remove": ("STRING", {"multiline": True, "default": "", "placeholder": "Enter strings to remove (one per line)",
                    "tooltip": "Enter strings to remove (one per line). Can include multiple words or strings on one line like: 'This image is'. This will only match this entire string."}),
                "match_case": ("BOOLEAN", {"default": False,
                    "label_on": "enabled", "label_off": "disabled",
                    "tooltip": "When enabled, matching is case-sensitive."}),
                "match_whole_string": ("BOOLEAN", {"default": True,
                    "label_on": "enabled", "label_off": "disabled",
                    "tooltip": "When enabled, only whole-word matches are removed, not substrings inside other words."}),
                "preserve_punctuation": ("BOOLEAN", {"default": True,
                    "label_on": "enabled", "label_off": "disabled",
                    "tooltip": "When enabled, punctuation marks adjacent to matched text are preserved"}),
                "remove_extra_spaces": ("BOOLEAN", {"default": True,
                    "label_on": "enabled", "label_off": "disabled",
                    "tooltip": "When enabled, collapses repeated spaces and trims the result after removal."})
            }
        }

    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("modified_string",)
    FUNCTION = "string_strip"
    CATEGORY = "utils/StringEssentials"

    def cleanup_text(self, text):
        text = re.sub(r'\s+([,;:.])', r'\1', text)  # Remove spaces before punctuation
        text = re.sub(r'\s{2,}', ' ', text)  # Normalize multiple spaces
        return text.strip()

    def string_strip(self, input_string, strings_to_remove, match_case, match_whole_string, preserve_punctuation, remove_extra_spaces):
        # Create a list of strings to remove, sorted by length
        removals = []
        for line in strings_to_remove.splitlines():
            line = line.strip()
            if not line:
                continue

            # Store length with the string for sorting
            removals.append((len(line), line))

        # Sort by length in descending order to handle longer phrases first
        removals.sort(reverse=True)

        result = input_string
        flags = 0 if match_case else re.IGNORECASE

        # Compile patterns once for better performance
        patterns = []
        for _, string_to_remove in removals:
            if match_whole_string:
                if preserve_punctuation:
                    # Match whole words but preserve adjacent punctuation
                    pattern = r'\b' + re.escape(string_to_remove) + r'\b'
                else:
                    # Original behavior: remove punctuation after match
                    pattern = r'(?:\b|(?<=\s)|^)' + re.escape(string_to_remove) + r'(?:[,;:.])?(?=\s|$)'
            else:
                if preserve_punctuation:
                    # Simple literal removal preserving punctuation
                    pattern = re.escape(string_to_remove)
                else:
                    # Original behavior: remove punctuation after match
                    pattern = re.escape(string_to_remove) + r'(?:[,;:.])?'

            patterns.append(re.compile(pattern, flags=flags))

        # Apply all removal patterns
        for pattern in patterns:
            result = pattern.sub('', result)

        if remove_extra_spaces:
            result = self.cleanup_text(result)

        return (result,)