calendar/packages/features/calAIPhone/zod-utils.ts

import { isValidPhoneNumber } from "libphonenumber-js";
import z from "zod";

export const templateTypeEnum = z.enum(["CHECK_IN_APPOINTMENT", "CUSTOM_TEMPLATE"]);

const requiredFields = z.object({
  yourPhoneNumber: z.string().refine((val) => isValidPhoneNumber(val)),
  numberToCall: z.string().refine((val) => isValidPhoneNumber(val)),
  calApiKey: z.string().trim().min(1, {
    message: "Please enter CAL API Key",
  }),
  eventTypeId: z.number(),
  enabled: z.boolean().default(false),
  templateType: templateTypeEnum,
});

export const createPhoneCallSchema = requiredFields.merge(
  z.object({
    schedulerName: z.string().min(1).optional().nullable(),
    guestName: z
      .string()
      .optional()
      .transform((val) => {
        return !!val ? val : undefined;
      }),
    guestEmail: z
      .string()
      .optional()
      .transform((val) => {
        return !!val ? val : undefined;
      }),
    guestCompany: z
      .string()
      .optional()
      .transform((val) => {
        return !!val ? val : undefined;
      }),
    beginMessage: z.string().optional(),
    generalPrompt: z.string().optional(),
  })
);

export type TCreatePhoneCallSchema = z.infer<typeof createPhoneCallSchema>;

export const ZGetPhoneNumberSchema = z
  .object({
    phone_number: z.string(),
    agent_id: z.string().optional(),
    nickname: z.string(),
    inbound_agent_id: z.string(),
    outbound_agent_id: z.string(),
    error_message: z.string().optional(),
  })
  .passthrough();

export type TGetPhoneNumberSchema = z.infer<typeof ZGetPhoneNumberSchema>;

export const ZCreatePhoneSchema = z
  .object({
    call_id: z.string(),
    agent_id: z.string().optional(),
  })
  .passthrough();

export type TCreatePhoneSchema = z.infer<typeof ZCreatePhoneSchema>;

export const fieldSchemaMap = {
  CHECK_IN_APPOINTMENT: requiredFields.merge(
    z.object({
      schedulerName: z.string().min(1),
    })
  ),
  CUSTOM_TEMPLATE: createPhoneCallSchema.omit({ generalPrompt: true }).merge(
    z.object({
      generalPrompt: z.string(),
    })
  ),
};

export type TemplateType = z.infer<typeof templateTypeEnum>;

const fieldTypeEnum = z.enum([
  "name",
  "text",
  "textarea",
  "number",
  "email",
  "phone",
  "address",
  "multiemail",
  "select",
  "multiselect",
  "checkbox",
  "radio",
  "radioInput",
  "boolean",
]);

export const fieldNameEnum = z.enum([
  "schedulerName",
  "generalPrompt",
  "guestName",
  "guestEmail",
  "guestCompany",
  "beginMessage",
]);

export type FieldType = z.infer<typeof fieldTypeEnum>;

const FieldSchema = z.object({
  type: fieldTypeEnum,
  name: fieldNameEnum,
  required: z.boolean(),
  defaultLabel: z.string(),
  placeholder: z.string(),
  variableName: z.string().optional(),
});

const FieldsSchema = z.array(FieldSchema);

export type Fields = z.infer<typeof FieldsSchema>;

export const ZCreateRetellLLMSchema = z
  .object({
    llm_id: z.string(),
    llm_websocket_url: z.string().optional(),
    inbound_dynamic_variables_webhook_url: z.string().optional(),
  })
  .passthrough();

export type TCreateRetellLLMSchema = z.infer<typeof ZCreateRetellLLMSchema>;

export const ZGetRetellLLMSchema = z
  .object({
    general_prompt: z.string(),
    begin_message: z.string().nullable().optional(),
    llm_id: z.string(),
    llm_websocket_url: z.string().optional(),
    inbound_dynamic_variables_webhook_url: z.string().optional(),
    general_tools: z.array(
      z
        .object({
          name: z.string(),
          type: z.string(),
          cal_api_key: z.string().optional(),
          event_type_id: z.number().optional(),
          timezone: z.string().optional(),
        })
        .passthrough()
    ),
    states: z
      .array(
        z
          .object({
            name: z.string(),
            tools: z.array(
              z
                .object({
                  name: z.string(),
                  type: z.string(),
                  cal_api_key: z.string().optional(),
                  event_type_id: z.number().optional(),
                  timezone: z.string().optional(),
                })
                .passthrough()
            ),
          })
          .passthrough()
      )
      .nullable()
      .optional(),
  })
  .passthrough();

export type TGetRetellLLMSchema = z.infer<typeof ZGetRetellLLMSchema>;

export const ZCreatePhoneNumberResponseSchema = z.object({
  phone_number: z.string(),
  phone_number_type: z.string(),
  phone_number_pretty: z.string(),
  inbound_agent_id: z.string().optional().nullable(),
  outbound_agent_id: z.string().optional().nullable(),
  inbound_agent_version: z.number().optional().nullable(),
  outbound_agent_version: z.number().optional().nullable(),
  area_code: z.number().optional().nullable(),
  nickname: z.string(),
  inbound_webhook_url: z.string().url().optional().nullable(),
  last_modification_timestamp: z.number(),
});
export type TCreatePhoneNumberResponseSchema = z.infer<typeof ZCreatePhoneNumberResponseSchema>;

export const ZCreateAgentResponseSchema = z.object({
  agent_id: z.string(),
  agent_name: z.string(),
});

export type TCreateAgentResponseSchema = z.infer<typeof ZCreateAgentResponseSchema>;

export const ZUpdatePhoneNumberResponseSchema = ZCreatePhoneNumberResponseSchema;
export type TUpdatePhoneNumberResponseSchema = TCreatePhoneNumberResponseSchema;