#!/usr/bin/env python3
import re
from sys import argv, exit as s_exit

# read arguments
arguments = argv[1:]

# define replacement text depending on arguments
if len(arguments) > 0:
    if arguments[0] == 'WSL':
        replacement = """run(['pwsh.exe', '-c', 'Set-Clipboard', _copy_line[_index].rstrip()])
        Popen("sleep 30; pwsh.exe -c 'echo \\"\\" | Set-Clipboard'", shell=True)"""
    elif arguments[0] == 'MAC':
        replacement = """run(['pbcopy'], stdin=Popen(['printf', _copy_line[_index].rstrip().replace('\\\\\\', '\\\\\\\\\\\\\\').replace('%', '%%')],
                                    stdout=PIPE).stdout)
        Popen("sleep 30; printf '' | pbcopy", shell=True)"""
    elif arguments[0] == 'HAIKU':
        replacement = """run(['clipboard', '-c', _copy_line[_index].rstrip()])
        Popen('sleep 30; clipboard -r', shell=True)"""
    elif arguments[0] == 'TERMUX':
        replacement = """run(['termux-clipboard-set', _copy_line[_index].rstrip()])
        Popen("sleep 30; termux-clipboard-set ''", shell=True)"""
    elif arguments[0] == 'BSD':
        replacement = """run(['xclip', '-sel', 'c'], stdin=Popen(['printf', _copy_line[_index].rstrip().replace('\\', '\\\\')
                                            .replace('%', '%%')], stdout=PIPE).stdout)
    Popen("sleep 30; printf '' | xclip -sel c", shell=True)"""
    elif arguments[0] == 'LINUX':
        replacement = """if 'WAYLAND_DISPLAY' in environ:
        run(['wl-copy', _copy_line[_index].rstrip()])
        Popen('sleep 30; wl-copy -c', shell=True)
    else:
        run(['xclip', '-sel', 'c'], stdin=Popen(['printf', _copy_line[_index].rstrip().replace('\\\\\\', '\\\\\\\\\\\\\\')
                                                .replace('%', '%%')], stdout=PIPE).stdout)
        Popen("sleep 30; printf '' | xclip -sel c", shell=True)"""
else:
    s_exit()

# define PORT target
string1 = '# PORT START CLIPBOARD'
string2 = '# PORT END CLIPBOARD'

# read input file
text = open('working/sshyp.py', 'r').read()

# find and replace the defined PORT target
regex = re.compile(f"{string1}.*?{string2}", re.DOTALL)
new_text = re.sub(regex, replacement, text)

# write updated text
open('working/sshyp.py', 'w').write(new_text)