#!/usr/bin/env python3

import os
import gi
gi.require_version('Gtk', '3.0')
gi.require_version('AyatanaAppIndicator3', '0.1')
gi.require_version('Notify', '0.7')
from gi.repository import Gtk, Gdk, GLib, AyatanaAppIndicator3

desktop = os.environ.get("XDG_CURRENT_DESKTOP", "")

if desktop == "KDE":
    APPINDICATOR_ID = ""
else:
    APPINDICATOR_ID = "recordmydesktop"

print("APPINDICATOR_ID =", APPINDICATOR_ID)

def quit(source):
    Gtk.main_quit()

def script1(widget):
    os.popen("~/.config/recordmydesktop/settings/set")

def script2(widget):
    os.popen("/usr/share/recordmydesktop/scripts/settings")

def script3(widget):
    os.popen("/usr/share/recordmydesktop/about/RecordMyDesktop")

def script4(widget):
    os.popen("pkill -f /usr/share/recordmydesktop/RecordMyDesktop")
    os.popen("/usr/share/recordmydesktop/scripts/killall")

def build_menu():
    menu = Gtk.Menu()

    # Add tooltips to each menu item
    item_recording = Gtk.MenuItem(label="🔴️ Record")
    item_recording.connect('activate', script1)
    item_recording.set_tooltip_text("Start Recording")
    menu.append(item_recording)

    item_setings = Gtk.MenuItem(label="⚙️ Settings")
    item_setings.connect('activate', script2)
    item_setings.set_tooltip_text("Recording Setting")
    menu.append(item_setings)

    item_about = Gtk.MenuItem(label="💬️ About")
    item_about.connect('activate', script3)
    item_about.set_tooltip_text("About RecordMy Desktop")
    menu.append(item_about)

    item_quit = Gtk.MenuItem(label="❌️ Quit")
    item_quit.connect('activate', script4)
    menu.append(item_quit)

    menu.show_all()
    return menu

def on_status_icon_click(icon, event):
    if event.type == Gdk.EventType.BUTTON_PRESS and (event.button == 1 or event.button == 3):
        # Show the menu
        menu.popup(None, None, Gtk.StatusIcon.position_menu, icon, event.button, event.time)
        return True

# Create the indicator using libayatana-appindicator
indicator = AyatanaAppIndicator3.Indicator.new(
    APPINDICATOR_ID,
    "/usr/share/recordmydesktop/icons/rmd.png",
    AyatanaAppIndicator3.IndicatorCategory.APPLICATION_STATUS
)

# Set the menu
menu = build_menu()
indicator.set_menu(menu)

# Create a Gtk.StatusIcon for mouse events
status_icon = Gtk.StatusIcon()
status_icon.set_from_file("/usr/share/recordmydesktop/icons/rmd.png")
status_icon.set_visible(True)

# Connect the button press event
status_icon.connect('button-press-event', on_status_icon_click)

Gtk.main()

