bot/packages/models/src/features/blocks/inputs/rating.ts

import { z } from 'zod'
import {
  defaultButtonLabel,
  InputBlockType,
  optionBaseSchema,
  blockBaseSchema,
} from '../shared'

export const defaultRatingInputOptions: RatingInputOptions = {
  buttonType: 'Numbers',
  length: 10,
  labels: { button: defaultButtonLabel },
  customIcon: { isEnabled: false },
}

export const ratingInputOptionsSchema = optionBaseSchema.and(
  z.object({
    buttonType: z.literal('Icons').or(z.literal('Numbers')),
    length: z.number(),
    labels: z.object({
      left: z.string().optional(),
      right: z.string().optional(),
      button: z.string(),
    }),
    customIcon: z.object({
      isEnabled: z.boolean(),
      svg: z.string().optional(),
    }),
  })
)

export const ratingInputBlockSchema = blockBaseSchema.and(
  z.object({
    type: z.literal(InputBlockType.RATING),
    options: ratingInputOptionsSchema,
  })
)

export type RatingInputBlock = z.infer<typeof ratingInputBlockSchema>
export type RatingInputOptions = z.infer<typeof ratingInputOptionsSchema>