bot/packages/models/features/blocks/bubbles/schemas.ts

import { z } from 'zod'
import { audioBubbleBlockSchema, audioBubbleContentSchema } from './audio'
import { embedBubbleContentSchema, embedBubbleBlockSchema } from './embed'
import { imageBubbleContentSchema, imageBubbleBlockSchema } from './image'
import { textBubbleContentSchema, textBubbleBlockSchema } from './text'
import { videoBubbleContentSchema, videoBubbleBlockSchema } from './video'

export const bubbleBlockContentSchema = textBubbleContentSchema
  .or(imageBubbleContentSchema)
  .or(videoBubbleContentSchema)
  .or(embedBubbleContentSchema)
  .or(audioBubbleContentSchema)

export const bubbleBlockSchema = textBubbleBlockSchema
  .or(imageBubbleBlockSchema)
  .or(videoBubbleBlockSchema)
  .or(embedBubbleBlockSchema)
  .or(audioBubbleBlockSchema)

export type BubbleBlock = z.infer<typeof bubbleBlockSchema>
export type BubbleBlockContent = z.infer<typeof bubbleBlockContentSchema>