import obspython as obs
import tkinter as tk
import ctypes
import threading

GWL_EXSTYLE = -20
WS_EX_LAYERED = 0x80000
WS_EX_TRANSPARENT = 0x20
LWA_COLORKEY = 0x00000001
RED_DOT_PAUSE = 5
WS_EX_TOOLWINDOW = 0x80

OFF = False

B64_RECORD = "iVBORw0KGgoAAAANSUhEUgAAAB4AAAAeCAYAAAA7MK6iAAAACXBIWXMAAAsTAAALEwEAmpwYAAAFTElEQVRIia2WXYxkRRXHf6fqdt/+mo+ebXZ2JsAqyOyuLAkMGpQsQUKMD0TfjDExxg0JL5r44gPxCZ6MD75h/OLFRE10HzZqxESNEBCWjQIPZAdGcSPi7KjLTu+O0933q+r4cLtn+nbfnriJlVR33apT538+/nWq5Ds/OMf8wgK1Wq322h8vfm1j49JjcRyfstYaEeFmmg7/ZTRWxTmn1Wp188SJk79+4GMPfitN091ut4v85Ge/wXt/6sc/+uFzF1995QOIUKs1MCK5ChnTWICQIooUp4fIqCpRNMB7x/0f+ej2F7549tO1sP5aUK1WG9//3rcvvPDC8wtrayep1erICHTUhvjoUN8E2KRtRWxFUeI45g8vvbiSptnLX/7KV2+1xz944qlf/vz8J++48y7q9QbWVrDWYozB2LxbY5Cx79GaNQa7L2P3943mclmLDOfa7SU2Lr0ZtJeW2ra9dOzZ7StX2p3O0SHoUFjMTXZBhJK50b9gxNDr7WGMWQ2iaLBSqzfyBZML3lybIsBUGwEjQrPZIoqixcAY66Zyuq+wkC2K2RyXL2O/TsmIyChlPhARL4Ds/5R5U6SPSwWX5JGxVcVWlHJqj3t9oMmI8Wa2tUVvxELSE3a3qvjM0OhkNI5kaAa7WxXSvkWsluyfMgGAoBCKicVcgSIW9rYr1BYc6196n2P39qktOgAG1y3/fKPB5q/a7G1XaK2kqBuPUnkRCmbM71ttLOxuV1g8nvDw17fpnBoQ37Ck/TzUrWMptz7Q4/hD/+HFb6xy4+9V5lZSvBvXMw1SoHAxQHnVGHQDmp2MR5/eYunOiJ2/hPTfD0j7hrRv6F8N2HknpLMW8chTV6gvOQZdC1JGwlLgkryokOwZTn9uh/YdMdffDZk6bSYnTvdvNY58KOKez18j6RnQspNSCjwtlPaFxeMJq+t9elcDxMwmjRildzVg5b4+C7en+6n4H4CnoZO+Zf72hPpSlntR1sZsSQeGcN7RWk5Jo8NvtkO1qRNsxWOsTmeirH4oGKOYQMFPFp9DgA9E8qMQ1Bz9qwHJnsVWSm0rjE2gpJEhvmGxVc9hx+mQRAjVlufaOzV2Loc0OinqZSZf1AuNTsaNd0N2LodUm77EwlLgaQEbgIsNb/9iEbFQnXN5cZggrHohnM8A2DjfxiWCCZgWLADPJKqiCq3llPcuNLn4zFFaR1PmVtL8SA33iVXmVhOatzguPrPMexeatJYzVA+/tQols9iGNdoozY7j0rklBjsBd3+2y8JtMTbMFbvY0P1ryJs/PcLl5+doLaeI5EaPHCjzOjgIcbmFqoINPa1jKZd/P8f2Gw2O3h3RWkkB2NsO+PelBoOuZW4lwdg89LMd0gmPx2+2SVEviIH51ZQsEv7xahOX5YI2UMIFx/xqgqqgvrCzXCEQqKoZYZZZNm517r3SuCWlcHhhIrRlno5WBVWVwHtncyJMws+qPOVcKP8+GKvqsHucc9ZUq+G/4miAKqh6FM//u+nwfY0q/d4eYRjumpMnP3w+SWKSJMI5h3ce7x2qDq8eVY8fdr2Jnu9xeO/wzuGcI81Sev0ea2snfms+/uCZp++9bz398+bbxHFMlmV5TzNclpFlKS4bjUt6WjZOp+SSJGZzc4PTp+/hE488+mSg6q6fffyJMyLmuddf/9ORaiWkVq8jyMHbr8CjsYefMLx3x7gx9qnD3MZRRBxHrK/fv3f28Sc+Y6zZku8+e452u40Y23nl5Zee3HzrrU/FSXybtdaIkVzxrOfTrDkAyfPqnNNqUN26a23td2ceevibiGzt7Fzjv0X5pGt3iiPfAAAAAElFTkSuQmCC"

B64_BUFFER = "iVBORw0KGgoAAAANSUhEUgAAAB4AAAAeCAYAAAA7MK6iAAAACXBIWXMAAAsTAAALEwEAmpwYAAAEnklEQVRIieWXXWxURRSAv7l3t912W9qtlC7FkEqBEn9aUEMiJhgRRIIhJYUEItGAQUHUKARBo8ZEkASNGgwaE0EwCiYg8qAJETE2rU2QIAVewCI/bdndKpTtttvu9s69x4fdLf2D3RojD56XOzNn5nwz52fuvUpEuBVi3BLqrQS7Bg80njzTr6eSz75w3AkUAfXXpygYEi4FCP1Hp1VNGTBjpCfWQB1wb7KfBzwFVI3QzojAXmBeoql+BrKALkR2AY1ALbD83wRnA5uAi6A+RADDyFfZngvK47kL00xFYiaoncBR4P50RofEeJBUghwAyrFt1KgCzBL/98Tj9fa1q3/hOJ3mmJLFRq53qQ4FH5dIRxamOR04RiIEX/wTcAXIcURcoHBNmLjfuXrl2c5Pt7fHGuqwgwGwbQz/2GbPgzP359UszjPLJ32iL/yxDASU2q0gAurgcMbV4AskmdUGqAAiJShwl098o+enw5vaN6wlfubMOmCJAeUocITzAnuyJ01+/7Z3PyBn9tz1VlPT1kRBKEBKgeDUQVl9I/AWYCO2xjVpyq7YkR+WhxbOHw/Uuvz+sgEllGzrUOi8wMNjv/2u2TNrzg59rmkFpguQb4BF6cGnzhYgcg1Qhs8XdcLhvNDchwwr1NbuHjeuANvumytcr3RlurAut7a7/P7RY3+sF+X1djgd4VFJbenUqopgf87QrBaZB0ohgllcsr3z88/oDbVtc5eWDoCmHNm3zNa4SkuLekOhbZ27d2CWlOxMeEUBLByMGa6cZgAojwcdDHwZb6jDgFUp1970lSKCAatj9bXYbW27yfYkV8j0TMAVIKhcL3bzxdNWS3Op4fWaKbAaZsEAcK7H1K0tPvtya6ORk5PSlGUC9iY0BmJp0LoAcwQXnGGC4+SIYycSL7HVIQaGsaiuAEgshllcPNosKjpHT0/GXInFMQsLg0ahzy+9vanh5gzAcgpAerpx3VH+qPueKsu29AlMMz3VNLG1PuGunCausgnV0tNNMisaMwBzGBTYDth6rXfxUoBVTmcnGDdxuWHghMMYsDJvyRMQj63BcVLag+nBStWBtKJAt7Tc552/YF7hCy/9akUir0l39/Bww0C6urCi0Q0F6145nvPInBp9ufXuRIypA86lBwuAWg0KHBs7FDzge2vzeN/KVVt0OLzICgSapKcb0RqxLCTahRUInNWRSHXRiy9v9b359mQdCOzrO61SzyU3MPB8N7gyAfaDqkFrDJ+v3Sges6B7395fIrt2YJ0+OcWJdj2gTFfUyM8/5q6adiH/yRV4q2vm6rbQAYl05GK6QLEeeA9gamVFxmCAI8AsbAey3LjGl30l0a6P4yd+a7Cv/IlyZ2EWjyG7smo22Z41+tLFamydCsdmlHo9ZWhkYAWI+gjkeQAcQWVlYRQWHlWenKfFsQ9JLHa709EB2krWLZeAjcDX/V08QnDfW+gxUK8CMxOKRCKg1FKEPcl+LbAX2AlYfetvAE73BZKSQyCHQM0AlgETAXcSVAa8AzwD/J6hvYzBKWkAaQAQMFXCG1uAYmB0Apz4tE0nQ1z9X8n/7xfmb0gn8Knb3EhiAAAAAElFTkSuQmCC"

B64_PAUSE = "iVBORw0KGgoAAAANSUhEUgAAAB4AAAAeCAYAAAA7MK6iAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAAJcEhZcwAADsMAAA7DAcdvqGQAAAAYdEVYdFNvZnR3YXJlAFBhaW50Lk5FVCA1LjEuN4vW9zkAAAC2ZVhJZklJKgAIAAAABQAaAQUAAQAAAEoAAAAbAQUAAQAAAFIAAAAoAQMAAQAAAAIAAAAxAQIAEAAAAFoAAABphwQAAQAAAGoAAAAAAAAAYAAAAAEAAABgAAAAAQAAAFBhaW50Lk5FVCA1LjEuNwADAACQBwAEAAAAMDIzMAGgAwABAAAAAQAAAAWgBAABAAAAlAAAAAAAAAACAAEAAgAEAAAAUjk4AAIABwAEAAAAMDEwMAAAAAAlR56NozS1xQAAA+ZJREFUSEvtlE1vG1UUht9rz4fHM7bH8diOPbYc22rSpBIoRVHLJiJCQkgFFXUDEqVZBAIsYcV/YMGKBUJd8CFAYhUJBAWxqKBqKiHxIbVFojFEDcJp2iQ4DlCPPS+LwcYzaRzTBWLRR5rFfd575swc6V7gHv8RIigOol2txkLNZhGOYyEUAlz3lmua16VarRHcO4ihG7u53GPY2ZkX7fY4hGih3W5CCEAIHUKolKRrkOW3Q1tbS8Hau4K2Pc3R0fOMRi8xGn3RNc2qc/iw0s2d8XHZTSTK1PVFqupFGsZXbiYz43/Lv4SFwhnG46tMpV4IZvvBaHSBhrHKXO7ZYDYULJfnWSyusFy+P5gdhFssTrFQ+JGVymIwGwjz+QeYSKzy0KH7gtmwcGJikvH4KtPpB4PZvjCf/5Km+VxvPTYmU1FepySdoySdo6q+xtnZEABQUV79239K03yDR49GenW6/gx1fZknToS6bl+YzT5Bw7joc8lkkqp6jfn8aeZyC5SkH5hMajxyRKaiXGYm8zxLpadoWSs0zbSvVlXPM5F4st8BwN4vaTTm4bpnfY4EOp0G6vUlbGx8BMA7Sq4rQP6OGzc+hmUtYXf3N2xv+0ohSW+i1Trjl4HGtO0EXLeKSOTzft8jEolA01QAAiTgOIAkAamUinpdRbsNhMP+GkX5Aq5bZD5v9Wv/HzvOGIA/MDq65vNBQiHvkWWg0wG2t731HRBbW78C2EGrVen3/t3NZhpkQ1y50vH5Lt54vYf0HOmtB6EoDezsDPhj0hvhMJDeh4ghbl0hiE7Ht9Hf2DBuIRw2ukflQIT4Z+yDuH07jkRis1/5K1T1Zwih4+rVnM+jb6T9TbpjHwDT6QyAOGS51u99jcXa2iaEuI5mc67fw3EAXQdsu4V22wHA3oi9o+ZA05zeup/d3VnI8rqo19f79d4ZKcq7IBd8zrKAaFTH5uZxuO4MgGhvzEAEsjyD9fXjEMLYc5xUdRGm+Z5f3gHOzUk0jK8Zi/VuG46Pq9S0DyhJy5SkS1SUtzg15V2ZmnaWknSJkrRMWf6Qpqn16rLZk7Ttb3nsmNp1A6FlPcRo9CfmcuVgNiwslQqMxVZo248Es4GwVHqJpdJlTkyMBbOD4ORkgZXK9yyXXwlmQ8Fq9WXGYjUaxqlgth/MZB5nPL7CSuXumnbhyMjD1PVvGIl8xljsFLPZ1J49ljVCTTvJVOoTFovfMZN5NLgnyBDXDsDpaQW12mm020/DddMgtwB4FwKZBDmCcPgmUqn3USq9Iy5c+DP4jiBDNe6Htp1Hs1lFp5OGogCuexOyvCI2Nn4J7r3H/4K/AO99fuVP9rpsAAAAAElFTkSuQmCC"


STATUS = {"record": False, "buffer": False, "paused": False}
CORNER = "top-left"
MARGIN = 5
SIZE = (30, 30)
UPDATE_MS = 100
overlay = None


def set_clickthrough(hwnd):
    styles = ctypes.windll.user32.GetWindowLongW(hwnd, GWL_EXSTYLE)
    styles |= WS_EX_LAYERED | WS_EX_TRANSPARENT | WS_EX_TOOLWINDOW
    ctypes.windll.user32.SetWindowLongW(hwnd, GWL_EXSTYLE, styles)
    ctypes.windll.user32.SetLayeredWindowAttributes(hwnd, 0, 255, LWA_COLORKEY)


def _hide_buffer():
    STATUS["buffer"] = False


class Overlay:
    def __init__(self):
        self.root = tk.Tk()
        self.root.overrideredirect(True)
        self.root.config(bg="#000000")
        self.root.attributes("-alpha", 1)
        self.root.wm_attributes("-topmost", 1)
        self.root.attributes("-transparentcolor", "#000000")
        set_clickthrough(self.root.winfo_id())
        self.record_img = tk.PhotoImage(data=B64_RECORD)
        self.buffer_img = tk.PhotoImage(data=B64_BUFFER)
        self.pause_img = tk.PhotoImage(data=B64_PAUSE)
        self.label = tk.Label(self.root, bg="black", bd=0)
        self.label.pack()
        self._loop()

    def _update_position(self):
        global OFF
        w, h = SIZE
        sw = self.root.winfo_screenwidth()
        sh = self.root.winfo_screenheight()
        if CORNER == "off":
            OFF = True
            return
        elif CORNER == "top-left":
            x, y = MARGIN, MARGIN
        elif CORNER == "top-right":
            x, y = sw - w - MARGIN, MARGIN
        elif CORNER == "bottom-left":
            x, y = MARGIN, sh - h - MARGIN
        else:
            x, y = sw - w - MARGIN, sh - h - MARGIN
        self.root.geometry(f"{w}x{h}+{x}+{y}")

    def _loop(self):
        if OFF:
            self.label.config(image="")
            self.root.after(UPDATE_MS, self._loop)
            return
        self._update_position()
        if STATUS["buffer"]:
            self.label.config(image=self.buffer_img)
            self.root.after(RED_DOT_PAUSE * 1000, _hide_buffer)
        elif STATUS["paused"] and STATUS["record"]:
            self.label.config(image=self.pause_img)
        elif STATUS["record"]:
            self.label.config(image=self.record_img)
        else:
            self.label.config(image="")
        self.root.after(UPDATE_MS, self._loop)


def event_handler(evt):
    if evt == obs.OBS_FRONTEND_EVENT_RECORDING_STARTING:
        STATUS["record"] = True
    elif evt == obs.OBS_FRONTEND_EVENT_RECORDING_STOPPED:
        STATUS["record"] = False
        STATUS["paused"] = False
    elif evt == obs.OBS_FRONTEND_EVENT_RECORDING_PAUSED:
        STATUS["paused"] = True
    elif evt == obs.OBS_FRONTEND_EVENT_RECORDING_UNPAUSED:
        STATUS["paused"] = False
    elif evt == obs.OBS_FRONTEND_EVENT_REPLAY_BUFFER_SAVED:
        STATUS["buffer"] = True


def start_app():
    global overlay
    overlay = Overlay()
    overlay.root.mainloop()


def script_description():
    return "Recording and replay buffer indicator with screen corner selection. © Tap1x, 2025 - Until the End of Time."


def script_properties():
    props = obs.obs_properties_create()
    corners = obs.obs_properties_add_list(
        props,
        "corner",
        "Screen Corner",
        obs.OBS_COMBO_TYPE_LIST,
        obs.OBS_COMBO_FORMAT_STRING
    )
    for key, label in [
        ("top-left", "Top Left"),
        ("top-right", "Top Right"),
        ("bottom-left", "Bottom Left"),
        ("bottom-right", "Bottom Right"),
        ("off", "Off"),
    ]:
        obs.obs_property_list_add_string(corners, label, key)

    return props


def script_update(settings):
    global CORNER, OFF
    c = obs.obs_data_get_string(settings, "corner")
    if c in ("top-left", "top-right", "bottom-left", "bottom-right", "off"):
        CORNER = c
        if c == "off":
            OFF = True
        else:
            OFF = False


obs.obs_frontend_add_event_callback(event_handler)
threading.Thread(target=start_app, daemon=True).start()