import { create } from "zustand";
import { persist, createJSONStorage } from "zustand/middleware";

export const ARCHIVED_CONVERSATIONS_STORAGE_KEY = "archived-conversations";

interface ArchivedConversationsState {
  archivesByBackendId: Record<string, string[]>;
}

interface ArchivedConversationsActions {
  archiveConversation: (backendId: string, conversationId: string) => void;
  removeArchivedConversation: (
    backendId: string,
    conversationId: string,
  ) => void;
  isArchived: (backendId: string, conversationId: string) => boolean;
}

type ArchivedConversationsStore = ArchivedConversationsState &
  ArchivedConversationsActions;

const initialState: ArchivedConversationsState = {
  archivesByBackendId: {},
};

function getArchivesForBackend(
  archivesByBackendId: Record<string, string[]>,
  backendId: string,
): string[] {
  return archivesByBackendId[backendId] ?? [];
}

export const useArchivedConversationsStore =
  create<ArchivedConversationsStore>()(
    persist(
      (set, get) => ({
        ...initialState,

        archiveConversation: (backendId, conversationId) => {
          const current = getArchivesForBackend(
            get().archivesByBackendId,
            backendId,
          );
          if (current.includes(conversationId)) {
            return;
          }
          set((state) => ({
            archivesByBackendId: {
              ...state.archivesByBackendId,
              [backendId]: [conversationId, ...current],
            },
          }));
        },

        removeArchivedConversation: (backendId, conversationId) => {
          const current = getArchivesForBackend(
            get().archivesByBackendId,
            backendId,
          );
          if (!current.includes(conversationId)) {
            return;
          }
          set((state) => ({
            archivesByBackendId: {
              ...state.archivesByBackendId,
              [backendId]: current.filter((id) => id !== conversationId),
            },
          }));
        },

        isArchived: (backendId, conversationId) =>
          getArchivesForBackend(get().archivesByBackendId, backendId).includes(
            conversationId,
          ),
      }),
      {
        name: ARCHIVED_CONVERSATIONS_STORAGE_KEY,
        storage: createJSONStorage(() => localStorage),
        partialize: (state): ArchivedConversationsState => ({
          archivesByBackendId: state.archivesByBackendId,
        }),
      },
    ),
  );