import { Bot } from '@botpress/sdk'
import { z } from 'zod'

type MessageHandlerProps = Parameters<Parameters<typeof bot.message>[0]>[0]

const bot = new Bot({
  states: {
    dialog: {
      type: 'conversation',
      schema: z.object({
        step: z.number().default(0),
      }),
    },
  },
})

bot.message(async (props) => {
  const dialogState = await props.client
    .getState({ type: 'conversation', id: props.conversation.id, name: 'dialog' })
    .catch(() => ({
      state: {
        payload: {
          step: 0,
        },
      },
    }))

  const step = dialogState.state.payload.step

  switch (step) {
    case 0:
      await sendMessage('Hello!', props)
      break
    case 1:
      await sendMessage('You said: ' + props.message.payload.text, props)
      break
    case 2:
      await sendMessage('Your response was: ' + props.message.payload.text, props)
      break
    default:
      break
  }

  await setState(step + 1, props)
})

async function sendMessage(text: string, { client, conversation, ctx }: MessageHandlerProps) {
  await client.createMessage({
    conversationId: conversation.id,
    userId: ctx.botId,
    tags: {},
    type: 'text',
    payload: {
      text,
    },
  })
}

async function setState(step: number, props: MessageHandlerProps) {
  await props.client.setState({ type: 'conversation', id: props.conversation.id, name: 'dialog', payload: { step } })
}

export default bot