#!/usr/bin/env python3

import sys

import gi

gi.require_version('AyatanaAppIndicatorGlib', '2.0')

from gi.repository import Gio  # noqa
from gi.repository import GLib  # noqa
from gi.repository import AyatanaAppIndicatorGlib as AppIndicator3  # noqa

__version__ = '0.0.0'

IGNORE = []

NODE_INFO = Gio.DBusNodeInfo.new_for_xml("""
<?xml version="1.0" encoding="UTF-8"?>
<node name="/org/freedesktop/Notifications">
    <interface name="org.freedesktop.Notifications">
        <method name="GetCapabilities">
            <arg direction="out" name="capabilities"    type="as"/>
        </method>
        <method name="Notify">
            <arg direction="in"  name="app_name"        type="s"/>
            <arg direction="in"  name="replaces_id"     type="u"/>
            <arg direction="in"  name="app_icon"        type="s"/>
            <arg direction="in"  name="summary"         type="s"/>
            <arg direction="in"  name="body"            type="s"/>
            <arg direction="in"  name="actions"         type="as"/>
            <arg direction="in"  name="hints"           type="a{sv}"/>
            <arg direction="in"  name="expire_timeout"  type="i"/>
            <arg direction="out" name="id"              type="u"/>
        </method>
        <method name="CloseNotification">
            <arg direction="in"  name="id"              type="u"/>
        </method>
        <method name="GetServerInformation">
            <arg direction="out" name="name"            type="s"/>
            <arg direction="out" name="vendor"          type="s"/>
            <arg direction="out" name="version"         type="s"/>
            <arg direction="out" name="spec_version"    type="s"/>
        </method>
   </interface>
</node>""")

next_id = 1
indicator = None
menu = None
threads = {}


def remove_menu_item(menu, key):
    # menu items can only be removed by position
    for i in range(menu.get_n_items()):
        if menu.get_item_attribute_value(0, 'app-name').unpack() == key:
            menu.remove(i)
            return


def clear_thread(_action, _, key):
    del threads[key]
    remove_menu_item(menu, key)
    actions.remove_action(key)
    if menu.get_n_items() == 0:
        indicator.set_status(AppIndicator3.IndicatorStatus.PASSIVE)


def matches_rule(params, rule):
    for key, pattern in rule.items():
        if key == 'hints':
            for k, p in pattern.items():
                if p != params['hints'].get(k):
                    return False
        elif pattern != params[key]:
            return False
    return True


def on_add_notification(params, id):
    if any(matches_rule(params, rule) for rule in IGNORE):
        return

    app_name = (
        str(params['hints'].get('desktop-entry', ''))
        or params['app_name']
        or f'${id}'
    )
    thread = threads.get(app_name)
    if params['summary'] == app_name:
        label = f'{app_name}: {params["body"]}'
    else:
        label = f'{app_name}: {params["summary"]}'

    if thread:
        remove_menu_item(menu, app_name)

    item = Gio.MenuItem.new(label, app_name)
    actions.add_action_entries([(app_name, clear_thread)], app_name)
    item.set_attribute([('app-name', 's', app_name)])
    menu.append_item(item)
    indicator.set_status(AppIndicator3.IndicatorStatus.ATTENTION)

    threads[app_name] = {
        'menuitem': item,
        'id': id,
    }


def on_close_notification(id):
    for key, thread in list(threads.items()):
        if id == thread['id']:
            clear_thread(None, None, key)


def on_call(
    conn, sender, path, interface, method, params, invocation, user_data=None
):
    global next_id

    if method == 'GetCapabilities':
        # announce fake capabilities to avoid firefox fallback
        reply = GLib.Variant('(as)', [['actions', 'body', 'body-hyperlinks']])
        invocation.return_value(reply)
        conn.flush()
    elif method == 'Notify':
        on_add_notification({
            'app_name': params[0],
            'replaces_id': params[1],
            'app_icon': params[2],
            'summary': params[3],
            'body': params[4],
            'actions': params[5],
            'hints': params[6],
            'expire_timeout': params[7],
        }, next_id)
        invocation.return_value(GLib.Variant('(u)', [next_id]))
        conn.flush()
        next_id += 1
    elif method == 'CloseNotification':
        on_close_notification(params[0])
        invocation.return_value(None)
        conn.flush()
    elif method == 'GetServerInformation':
        info = ['notification-hub', 'xi', __version__, '1.2']
        invocation.return_value(GLib.Variant('(ssss)', info))
        conn.flush()


def on_bus_acquired(conn, name, user_data=None):
    conn.register_object(
        '/org/freedesktop/Notifications',
        NODE_INFO.interfaces[0],
        on_call,
    )


def on_name_lost(conn, name, user_data=None):
    sys.exit(
        f'Could not aquire name {name}. '
        f'Is some other service blocking it?'
    )


if __name__ == '__main__':
    owner_id = Gio.bus_own_name(
        Gio.BusType.SESSION,
        NODE_INFO.interfaces[0].name,
        Gio.BusNameOwnerFlags.NONE,
        on_bus_acquired,
        None,
        on_name_lost,
    )

    indicator = AppIndicator3.Indicator.new(
        'notifications',
        'user-available',
        AppIndicator3.IndicatorCategory.APPLICATION_STATUS,
    )
    indicator.set_status(AppIndicator3.IndicatorStatus.PASSIVE)
    indicator.set_title('Notifications')
    menu = Gio.Menu()
    indicator.set_menu(menu)
    actions = Gio.SimpleActionGroup()
    indicator.set_actions(actions)

    try:
        loop = GLib.MainLoop()
        loop.run()
    finally:
        Gio.bus_unown_name(owner_id)