import { create } from 'zustand';

export interface ActiveMoodComponent {
  id: string;         // matches MoodComponentDefinition.id
  intensity: number; // 1–3
}

interface MoodEntryState {
  // The mood selected in the main screen (e.g. 'good')
  selectedMood: string | null;

  // The active components the user has added + their intensities
  components: ActiveMoodComponent[];

  // Actions
  setSelectedMood: (mood: string) => void;
  setComponents: (components: ActiveMoodComponent[]) => void;
  addComponent: (id: string) => void;
  removeComponent: (id: string) => void;
  setComponentIntensity: (id: string, intensity: number) => void;
  reset: () => void;
}

export const useMoodEntryStore = create<MoodEntryState>((set) => ({
  selectedMood: null,
  components: [],

  setSelectedMood: (mood) => set({ selectedMood: mood }),

  // Used by the modal to write back the full list at once
  setComponents: (components) => set({ components }),

  addComponent: (id) =>
    set((state) => {
      if (state.components.find((c) => c.id === id)) return state;
      return { components: [...state.components, { id, intensity: 1 }] };
    }),

  removeComponent: (id) =>
    set((state) => ({
      components: state.components.filter((c) => c.id !== id),
    })),

  setComponentIntensity: (id, intensity) =>
    set((state) => ({
      components: state.components.map((c) =>
        c.id === id ? { ...c, intensity } : c
      ),
    })),

  reset: () => set({ selectedMood: null, components: [] }),
}));