gui/modules/desktop/src/libs/stores/notifications.ts

import { writable } from "svelte/store";
import { nanoid } from "nanoid";

import { NotificationType } from "@tea/ui/types";
import type { Notification } from "@tea/ui/types";

import { listenToChannel, relaunch } from "@native";

export default function initNotificationStore() {
  const notifications: Notification[] = [];
  const { update, subscribe } = writable<Notification[]>([]);

  const remove = (id: string) => {
    update((notifications) => notifications.filter((n) => n.id != id));
  };

  listenToChannel("message", (data: any) => {
    const { message, params }: { message: string; params: { [key: string]: string } } = data;

    update((value) => {
      const newNotification: Notification = {
        id: nanoid(4),
        message,
        i18n_key: params["i18n_key"] || "",
        type: NotificationType.ACTION_BANNER,
        params
      };
      if (params.action) {
        newNotification.callback_label = params.action.toUpperCase();
        newNotification.callback = () => {
          relaunch();
          remove(newNotification.id); // not sure yet
        };
      }
      return [...value, newNotification];
    });
  });

  return {
    notifications,
    subscribe,
    remove,
    add: (partialNotification: Partial<Notification>) => {
      if (!partialNotification.message) throw new Error("message is required");

      const notification: Notification = {
        id: nanoid(4),
        i18n_key: partialNotification.i18n_key || "",
        type: NotificationType.MESSAGE,
        message: partialNotification.message || "",
        ...partialNotification
      };

      update((values) => [notification, ...values]);
    }
  };
}