import { z } from "zod";

export const LoginSchema = z.object({
  username: z.string().min(1, "Uživatelské jméno je povinné"),
  password: z.string().min(1, "Heslo je povinné"),
  remember_me: z.boolean().optional().default(false),
});

export const TotpVerifySchema = z.object({
  login_token: z.string().min(1, "Token je povinný"),
  totp_code: z.string().length(6, "Kód musí mít 6 číslic"),
  remember_me: z.boolean().optional().default(false),
});

export const TotpBackupSchema = z.object({
  login_token: z.string().min(1, "Token je povinný"),
  backup_code: z.string().min(1, "Záložní kód je povinný"),
});

export const TotpEnableSchema = z.object({
  secret: z.string().min(1),
  code: z.string().min(1),
  password: z.string().optional(),
  current_code: z.string().optional(),
});

export const TotpDisableSchema = z.object({
  code: z.string().min(1),
  password: z.string().optional(),
});

export const TotpRequiredSchema = z.object({
  required: z.boolean(),
});

export type LoginInput = z.infer<typeof LoginSchema>;
export type TotpVerifyInput = z.infer<typeof TotpVerifySchema>;
export type TotpBackupInput = z.infer<typeof TotpBackupSchema>;