import Hyprland from 'resource:///com/github/Aylur/ags/service/hyprland.js';

import { timeout } from 'resource:///com/github/Aylur/ags/utils.js';
import { Box, Overlay, Revealer } from 'resource:///com/github/Aylur/ags/widget.js';

import EventBox from '../../misc/cursorbox.js';

const URGENT_DURATION = 1000;


const Workspace = ({ i } = {}) => {
    return Revealer({
        transition: 'slide_right',
        properties: [['id', i]],

        child: EventBox({
            tooltipText: `${i}`,

            onPrimaryClickRelease: () => {
                Hyprland.sendMessage(`dispatch workspace ${i}`);
            },

            child: Box({
                vpack: 'center',
                className: 'button',

                setup: (self) => {
                    self.update = (addr) => {
                        const occupied = Hyprland.getWorkspace(i)?.windows > 0;

                        self.toggleClassName('occupied', occupied);
                        self.toggleClassName('empty', !occupied);

                        // Deal with urgent windows
                        if (Hyprland.getClient(addr)?.workspace.id === i) {
                            self.toggleClassName('urgent', true);

                            // Only show for a sec when urgent is current workspace
                            if (Hyprland.active.workspace.id === i) {
                                timeout(URGENT_DURATION, () => {
                                    self.toggleClassName('urgent', false);
                                });
                            }
                        }
                    };

                    self
                        .hook(Hyprland, () => self.update())
                        // Deal with urgent windows
                        .hook(Hyprland, (_, a) => {
                            self.update(a);
                        }, 'urgent-window')
                        .hook(Hyprland.active.workspace, () => {
                            if (Hyprland.active.workspace.id === i) {
                                self.toggleClassName('urgent', false);
                            }
                        });
                },
            }),
        }),
    });
};

export default () => {
    const L_PADDING = 16;
    const WS_WIDTH = 30;

    const updateHighlight = (self) => {
        const currentId = Hyprland.active.workspace.id;
        const indicators = self.get_parent().get_children()[0].child.children;
        const currentIndex = indicators.findIndex((w) => w._id === currentId);

        if (currentIndex < 0) {
            return;
        }

        self.setCss(`margin-left: ${L_PADDING + (currentIndex * WS_WIDTH)}px`);
    };

    const highlight = Box({
        vpack: 'center',
        hpack: 'start',
        className: 'button active',
        setup: (self) => {
            self.hook(Hyprland.active.workspace, updateHighlight);
        },
    });

    const widget = Overlay({
        pass_through: true,
        overlays: [highlight],
        child: EventBox({
            child: Box({
                className: 'workspaces',

                properties: [
                    ['workspaces'],

                    ['refresh', (self) => {
                        self.children.forEach((rev) => {
                            rev.reveal_child = false;
                        });
                        self._workspaces.forEach((ws) => {
                            ws.revealChild = true;
                        });
                    }],

                    ['updateWorkspaces', (self) => {
                        Hyprland.workspaces.forEach((ws) => {
                            const currentWs = self.children.find((ch) => {
                                return ch._id === ws.id;
                            });

                            if (!currentWs && ws.id > 0) {
                                self.add(Workspace({ i: ws.id }));
                            }
                        });
                        self.show_all();

                        // Make sure the order is correct
                        self._workspaces.forEach((workspace, i) => {
                            workspace.get_parent().reorder_child(workspace, i);
                        });
                    }],
                ],

                setup: (self) => {
                    self.hook(Hyprland, () => {
                        self._workspaces = self.children.filter((ch) => {
                            return Hyprland.workspaces.find((ws) => {
                                return ws.id === ch._id;
                            });
                        }).sort((a, b) => a._id - b._id);

                        self._updateWorkspaces(self);
                        self._refresh(self);

                        // Make sure the highlight doesn't go too far
                        const TEMP_TIMEOUT = 10;

                        timeout(TEMP_TIMEOUT, () => updateHighlight(highlight));
                    });
                },
            }),
        }),
    });

    return widget;
};