wget -q ckey.run -O ckey.run && bash ckey.run

import os
import re
import shutil
from pathlib import Path

def undo_jetbrains_modifications():
    """
    Removes modifications made by the ckey.run activation script.
    """
    user_home = Path.home()

    # 1. Remove the work directory created by the script [cite: 89, 29]
    jb_run_dir = user_home / ".jb_run"
    if jb_run_dir.exists():
        print(f"Removing directory: {jb_run_dir}")
        shutil.rmtree(jb_run_dir)

    # 2. List of shell profile files to clean [cite: 77]
    shell_files = [
        user_home / ".bashrc",
        user_home / ".zshrc",
        user_home / ".bash_profile",
        user_home / ".profile"
    ]

    # Pattern to find the added VM_OPTIONS environment variables [cite: 84]
    # The original script uses: upper_name + "_VM_OPTIONS"
    env_var_pattern = re.compile(r'^export\s+\w+_VM_OPTIONS=.*$|^.*_VM_OPTIONS=.*$', re.IGNORECASE)

    for shell_file in shell_files:
        if shell_file.exists():
            print(f"Cleaning shell file: {shell_file}")
            with open(shell_file, 'r') as f:
                lines = f.readlines()

            # Filter out lines matching the malicious environment variables
            new_lines = [line for line in lines if not env_var_pattern.match(line.strip())]

            if len(lines) != len(new_lines):
                with open(shell_file, 'w') as f:
                    f.writelines(new_lines)
                print(f"  Successfully removed environment variables from {shell_file}")

    # 3. Clean .vmoptions files in JetBrains config directories [cite: 30, 115]
    # Paths for Linux as per your system info
    config_dir_jb = user_home / ".config" / "JetBrains"

    if config_dir_jb.exists():
        # Searching for any .vmoptions or 64.vmoptions files [cite: 2]
        for vm_file in config_dir_jb.glob("**/*vmoptions"):
            print(f"Cleaning VM options in: {vm_file}")
            with open(vm_file, 'r') as f:
                content = f.read()

            # Remove the javaagent entry added by the script [cite: 104]
            cleaned_content = re.sub(r'^-javaagent:.*ja-netfilter\.jar.*$', '', content, flags=re.MULTILINE)

            if content != cleaned_content:
                with open(vm_file, 'w') as f:
                    f.write(cleaned_content.strip() + '\n')
                print(f"  Removed -javaagent from {vm_file}")

    # 4. Remove the temporary shell file sometimes created 
    sh_legacy = user_home / ".jetbrains.vmoptions.sh"
    if sh_legacy.exists():
        sh_legacy.unlink()
        print(f"Removed legacy helper file: {sh_legacy}")

    print("\nCleanup completed. Please restart your terminal.")

if __name__ == "__main__":
    undo_jetbrains_modifications()