const hyprland = await Service.import("hyprland");
import options from "options";

const { workspaces, monitorSpecific } = options.bar.workspaces;

function range(length, start = 1) {
  return Array.from({ length }, (_, i) => i + start);
}

const Workspaces = (monitor = -1, ws = 8) => {
  const getWorkspacesForMonitor = (curWs, wsRules) => {
    if (!wsRules || !wsRules.length) {
      return true;
    }

    const monitorMap = {};
    hyprland.monitors.forEach((m) => (monitorMap[m.id] = m.name));

    const currentMonitorName = monitorMap[monitor];
    return wsRules[currentMonitorName].includes(curWs);
  };

  const getWorkspaceRules = () => {
    try {
      const rules = Utils.exec("hyprctl workspacerules -j");

      const workspaceRules = {};

      JSON.parse(rules).forEach((rule, index) => {
        if (Object.hasOwnProperty.call(workspaceRules, rule.monitor)) {
          workspaceRules[rule.monitor].push(index + 1);
        } else {
          workspaceRules[rule.monitor] = [index + 1];
        }
      });

      return workspaceRules;
    } catch (err) {
      console.error(err);
    }
  };

  return {
    component: Widget.Box({
      class_name: "workspaces",
      children: Utils.merge(
        [workspaces.bind(), monitorSpecific.bind()],
        (workspaces, monitorSpecific) => {
          return range(workspaces || 8)
            .filter((i) => {
              if (!monitorSpecific) {
                return true;
              }
              const workspaceRules = getWorkspaceRules();
              return getWorkspacesForMonitor(i, workspaceRules);
            })
            .map((i) => {
              return Widget.Label({
                attribute: i,
                vpack: "center",
                label: `${i}`,
                setup: (self) =>
                  self.hook(hyprland, () => {
                    self.toggleClassName(
                      "active",
                      hyprland.active.workspace.id === i,
                    );
                    self.toggleClassName(
                      "occupied",
                      (hyprland.getWorkspace(i)?.windows || 0) > 0,
                    );

                    const isCurrentMonitor =
                      monitor !== -1 &&
                      hyprland.getWorkspace(i)?.monitorID !== monitor;

                    self.toggleClassName("hidden", isCurrentMonitor);
                  }),
              });
            });
        },
      ),
      setup: (box) => {
        if (ws === 0) {
          box.hook(hyprland.active.workspace, () =>
            box.children.map((btn) => {
              btn.visible = hyprland.workspaces.some(
                (ws) => ws.id === btn.attribute,
              );
            }),
          );
        }
      },
    }),
    isVisible: true,
    boxClass: "workspaces",
  };
};
export { Workspaces };