import { z } from 'zod'
import type { ZodSchema } from 'zod'

import { baseResponseSchema, commonResponseDataSchema } from '../response'
import messageNodeSchema, { nodeMessage } from './node'

const oneBotHttpApiMessageGroup: Record<
  string,
  {
    description?: string
    request: ZodSchema
    response: ZodSchema
  }
> = {
  '/send_group_msg': {
    description: '发送群消息',
    request: z
      .object({
        group_id: z
          .union([z.string(), z.number()])
          .describe('群号')
          .describe('群号'),
        message: z.array(messageNodeSchema).describe('消息内容')
      })
      .refine(
        (data) => {
          const hasReply = data.message.some((item) => item.type === 'reply')

          if (hasReply) {
            return data.message[0].type === 'reply'
          }

          return true
        },
        {
          message:
            '如果 message 包含 reply 类型的消息，那么只能包含一个，而且排在最前面'
        }
      ),
    response: baseResponseSchema.extend({
      data: commonResponseDataSchema
    })
  },
  '/send_group_forward_msg': {
    description: '发送群合并转发消息',
    request: z.object({
      group_id: z.union([z.string(), z.number()]).describe('群号'),
      messages: z.array(nodeMessage).describe('消息内容'),
      news: z
        .array(
          z.object({
            text: z.string()
          })
        )
        .describe('?'),
      prompt: z.string().describe('外显'),
      summary: z.string().describe('底下文本'),
      source: z.string().describe('内容')
    }),
    response: baseResponseSchema.extend({
      data: commonResponseDataSchema
    })
  },
  '/forward_group_single_msg': {
    description: '消息转发到群',
    request: z.object({
      group_id: z.union([z.string(), z.number()]).describe('群号'),
      message_id: z.union([z.string(), z.number()]).describe('消息 ID')
    }),
    response: baseResponseSchema.extend({
      data: commonResponseDataSchema
    })
  },
  '/group_poke': {
    description: '发送戳一戳',
    request: z.object({
      group_id: z.union([z.string(), z.number()]).describe('群号'),
      user_id: z.union([z.string(), z.number()]).describe('对方QQ号')
    }),
    response: baseResponseSchema
  }
}

export default oneBotHttpApiMessageGroup