#!/usr/bin/env python3

import os
import psutil
import re
import subprocess
import sys
import time


BLOCKED_BROWSER_WORDS = ["mogelpower", "nitter", "spon"]


def is_window_blocked(window_name, blocked):
    for b in blocked:
        if type(b) is str and b == window_name:
            return True
        elif type(b) is re.Pattern and b.findall(window_name):
            return True
    return False


def get_active_window_name_and_pid():
    CMD = ["xdotool", "getactivewindow", "getwindowname", "getwindowpid"]
    p = subprocess.run(CMD, capture_output=True)
    if p.returncode != 0:
        return "", ""
    window_name, window_pid, _ = p.stdout.decode().split("\n")
    return window_name, window_pid


def find_window_name(window_name):
    CMD = ["xdotool", "search", window_name, "getwindowpid"]
    p = subprocess.run(CMD, capture_output=True)
    if p.returncode != 0:
        return ""
    l = p.stdout.decode().split("\n")
    print(l)


def init_kill_sequence(blocked):
    count = 5
    while True:
        window_name, window_pid = get_active_window_name_and_pid()
        if not is_window_blocked(window_name, blocked):
            notify(f"{window_name} is okay. Return from kill sequence.")
            return
        notify(f"{window_name} is blocked. Kill in {count} seconds.")
        if count == 0:
            p = psutil.Process(int(window_pid))
            p.kill()
            return
        time.sleep(1)
        count -= 1


def notify(message):
    CMD = ["runuser", "-u", "felixm", "notify-send", message]
    p = subprocess.run(CMD)


def main():
    blocked = [re.compile(f"{word}.*(Firefox|Chromium)", flags=re.IGNORECASE)
               for word in BLOCKED_BROWSER_WORDS]
    while True:
        time.sleep(1)
        window_name, window_pid = get_active_window_name_and_pid()
        if is_window_blocked(window_name, blocked):
            init_kill_sequence(blocked)


if __name__ == "__main__":
    if os.geteuid() == 0:
        newpid = os.fork()
        if newpid == 0:
            main()
    else:
        cmd = ["sudo"] + sys.argv
        subprocess.Popen(cmd, start_new_session=True)