From 206a40baedcef06dc97f654e67f8b10435596e5b Mon Sep 17 00:00:00 2001
From: Markus Thielker <mail.markus.thielker@gmail.com>
Date: Fri, 13 Dec 2024 20:20:11 +0100
Subject: [PATCH] NORY-22: add missing UI for boolean traits

---
 .../src/components/forms/IdentityTraitForm.tsx   | 16 ++++++++++++++++
 dashboard/src/lib/forms/identity-form.ts         |  4 ++++
 2 files changed, 20 insertions(+)

diff --git a/dashboard/src/components/forms/IdentityTraitForm.tsx b/dashboard/src/components/forms/IdentityTraitForm.tsx
index 0c91720..8f863d8 100644
--- a/dashboard/src/components/forms/IdentityTraitForm.tsx
+++ b/dashboard/src/components/forms/IdentityTraitForm.tsx
@@ -8,6 +8,7 @@ import { z } from 'zod';
 import { zodResolver } from '@hookform/resolvers/zod';
 import { toast } from 'sonner';
 import { Identity } from '@ory/client';
+import { Checkbox } from '@/components/ui/checkbox';
 
 interface IdentityTraitFormProps {
     schema: KratosSchema;
@@ -21,6 +22,21 @@ function renderUiNodes(form: UseFormReturn, properties: KratosSchemaProperties,
     return Object.entries(properties).map(([key, value]) => {
             if (value.type === 'object') {
                 return renderUiNodes(form, value.properties!, key);
+            } else if (value.type === 'boolean') {
+                return (
+                    <FormField
+                        control={form.control}
+                        name={keyPrefix + key}
+                        key={key}
+                        className="space-y-0"
+                        render={({ field }) => (
+                            <FormItem className="flex items-center space-x-2 space-y-0">
+                                <Checkbox {...field} checked={field.value}/>
+                                <FormLabel>{value.title}</FormLabel>
+                            </FormItem>
+                        )}
+                    />
+                );
             } else {
                 return (
                     <FormField
diff --git a/dashboard/src/lib/forms/identity-form.ts b/dashboard/src/lib/forms/identity-form.ts
index bed74aa..c891884 100644
--- a/dashboard/src/lib/forms/identity-form.ts
+++ b/dashboard/src/lib/forms/identity-form.ts
@@ -51,6 +51,7 @@ export function generateZodSchema(properties: KratosSchemaProperties) {
                     zodType = zodType.max(value.maxLength);
                 }
                 break;
+            case 'integer':
             case 'number':
                 zodType = z.number();
                 if (value.minimum) {
@@ -60,6 +61,9 @@ export function generateZodSchema(properties: KratosSchemaProperties) {
                     zodType = zodType.max(value.maximum);
                 }
                 break;
+            case 'boolean':
+                zodType = z.boolean();
+                break;
             case 'object':
                 const schemaCopy = structuredClone(schema);
                 schemaCopy.properties.traits.properties = value.properties!;