From 1fbb63c39751543edce641e92f6190a540959ce4 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Wed, 23 Jul 2025 22:05:37 +0200
Subject: [PATCH] Add auto restart configuration fields to InstanceModal and
 zodFormUtils

---
 ui/src/components/InstanceModal.tsx | 74 +++++++++++++++++++++--------
 ui/src/lib/zodFormUtils.ts          | 10 ++++
 2 files changed, 65 insertions(+), 19 deletions(-)

diff --git a/ui/src/components/InstanceModal.tsx b/ui/src/components/InstanceModal.tsx
index 54317a7..7c64af0 100644
--- a/ui/src/components/InstanceModal.tsx
+++ b/ui/src/components/InstanceModal.tsx
@@ -109,6 +109,9 @@ const InstanceModal: React.FC<InstanceModalProps> = ({
     setShowAdvanced(!showAdvanced)
   }
 
+  // Check if auto_restart is enabled
+  const isAutoRestartEnabled = formData.auto_restart === true
+
   return (
     <Dialog open={open} onOpenChange={onOpenChange}>
       <DialogContent className="sm:max-w-[600px] max-h-[80vh] overflow-hidden flex flex-col">
@@ -146,17 +149,47 @@ const InstanceModal: React.FC<InstanceModalProps> = ({
               </p>
             </div>
 
-            {/* Basic Fields - Automatically generated from type */}
+            {/* Auto Restart Configuration Section */}
+            <div className="space-y-4">
+              <h3 className="text-lg font-medium">Auto Restart Configuration</h3>
+              
+              {/* Auto Restart Toggle */}
+              <ZodFormField
+                fieldKey="auto_restart"
+                value={formData.auto_restart}
+                onChange={handleFieldChange}
+              />
+
+              {/* Show restart options only when auto restart is enabled */}
+              {isAutoRestartEnabled && (
+                <div className="ml-6 space-y-4 border-l-2 border-muted pl-4">
+                  <ZodFormField
+                    fieldKey="max_restarts"
+                    value={formData.max_restarts}
+                    onChange={handleFieldChange}
+                  />
+                  <ZodFormField
+                    fieldKey="restart_delay"
+                    value={formData.restart_delay}
+                    onChange={handleFieldChange}
+                  />
+                </div>
+              )}
+            </div>
+
+            {/* Basic Fields - Automatically generated from type (excluding auto restart options) */}
             <div className="space-y-4">
               <h3 className="text-lg font-medium">Basic Configuration</h3>
-              {basicFields.map((fieldKey) => (
-                <ZodFormField
-                  key={fieldKey}
-                  fieldKey={fieldKey}
-                  value={formData[fieldKey]}
-                  onChange={handleFieldChange}
-                />
-              ))}
+              {basicFields
+                .filter(fieldKey => fieldKey !== 'auto_restart') // Exclude auto_restart as it's handled above
+                .map((fieldKey) => (
+                  <ZodFormField
+                    key={fieldKey}
+                    fieldKey={fieldKey}
+                    value={formData[fieldKey]}
+                    onChange={handleFieldChange}
+                  />
+                ))}
             </div>
 
             {/* Advanced Fields Toggle */}
@@ -173,23 +206,26 @@ const InstanceModal: React.FC<InstanceModalProps> = ({
                 )}
                 Advanced Configuration
                 <span className="text-muted-foreground text-sm font-normal">
-                  ({advancedFields.length} options)
+                  ({advancedFields.filter(f => !['max_restarts', 'restart_delay'].includes(f as string)).length} options)
                 </span>
               </Button>
             </div>
 
-            {/* Advanced Fields - Automatically generated from type */}
+            {/* Advanced Fields - Automatically generated from type (excluding restart options) */}
             {showAdvanced && (
               <div className="space-y-4 pl-6 border-l-2 border-muted">
                 <div className="space-y-4">
-                  {advancedFields.sort().map((fieldKey) => (
-                    <ZodFormField
-                      key={fieldKey}
-                      fieldKey={fieldKey}
-                      value={formData[fieldKey]}
-                      onChange={handleFieldChange}
-                    />
-                  ))}
+                  {advancedFields
+                    .filter(fieldKey => !['max_restarts', 'restart_delay'].includes(fieldKey as string)) // Exclude restart options as they're handled above
+                    .sort()
+                    .map((fieldKey) => (
+                      <ZodFormField
+                        key={fieldKey}
+                        fieldKey={fieldKey}
+                        value={formData[fieldKey]}
+                        onChange={handleFieldChange}
+                      />
+                    ))}
                 </div>
               </div>
             )}
diff --git a/ui/src/lib/zodFormUtils.ts b/ui/src/lib/zodFormUtils.ts
index 1f01edc..be2afd7 100644
--- a/ui/src/lib/zodFormUtils.ts
+++ b/ui/src/lib/zodFormUtils.ts
@@ -11,6 +11,16 @@ export const basicFieldsConfig: Record<string, {
     label: 'Auto Restart',
     description: 'Automatically restart the instance on failure'
   },
+  max_restarts: {
+    label: 'Max Restarts',
+    placeholder: '3',
+    description: 'Maximum number of restart attempts (0 = unlimited)'
+  },
+  restart_delay: {
+    label: 'Restart Delay (seconds)',
+    placeholder: '5',
+    description: 'Delay in seconds before attempting restart'
+  },
   model: {
     label: 'Model Path',
     placeholder: '/path/to/model.gguf',