learnhouse/apps/web/components/Contexts/AI/AIChatBotContext.tsx

'use client'
import { AIMessage } from '@components/Objects/Activities/AI/AIActivityAsk'
import React, { createContext, useContext, useReducer } from 'react'
export const AIChatBotContext = createContext(null) as any
export const AIChatBotDispatchContext = createContext(null) as any

export type AIChatBotStateTypes = {
  messages: AIMessage[]
  isModalOpen: boolean
  aichat_uuid: string
  isWaitingForResponse: boolean
  chatInputValue: string
  error: AIError
}

type AIError = {
  isError: boolean
  status: number
  error_message: string
}

function AIChatBotProvider({ children }: { children: React.ReactNode }) {
  const [aiChatBotState, dispatchAIChatBot] = useReducer(aiChatBotReducer, {
    messages: [] as AIMessage[],
    isModalOpen: false,
    aichat_uuid: null,
    isWaitingForResponse: false,
    chatInputValue: '',
    error: { isError: false, status: 0, error_message: ' ' } as AIError,
  })
  return (
    <AIChatBotContext.Provider value={aiChatBotState}>
      <AIChatBotDispatchContext.Provider value={dispatchAIChatBot}>
        {children}
      </AIChatBotDispatchContext.Provider>
    </AIChatBotContext.Provider>
  )
}

export default AIChatBotProvider

export function useAIChatBot() {
  return useContext(AIChatBotContext)
}

export function useAIChatBotDispatch() {
  return useContext(AIChatBotDispatchContext)
}

function aiChatBotReducer(state: any, action: any) {
  switch (action.type) {
    case 'setMessages':
      return { ...state, messages: action.payload }
    case 'addMessage':
      return { ...state, messages: [...state.messages, action.payload] }
    case 'setIsModalOpen':
      return { ...state, isModalOpen: true }
    case 'setIsModalClose':
      return { ...state, isModalOpen: false }
    case 'setAichat_uuid':
      return { ...state, aichat_uuid: action.payload }
    case 'setIsWaitingForResponse':
      return { ...state, isWaitingForResponse: true }
    case 'setIsNoLongerWaitingForResponse':
      return { ...state, isWaitingForResponse: false }
    case 'setChatInputValue':
      return { ...state, chatInputValue: action.payload }
    case 'setError':
      return { ...state, error: action.payload }

    default:
      throw new Error(`Unhandled action type: ${action.type}`)
  }
}