Merge pull request #592 from components-ai/theme-gen

Fix regen for theme values

Author: johno

.changeset/nasty-meals-fold.md

@@ -0,0 +1,5 @@
+---
+'@compai/css-gui': patch
+---
+
+Fix regen for theme values

packages/gui/src/components/schemas/theme.tsx

@@ -1,9 +1,10 @@
 import { useTheme } from '../providers/ThemeContext'
-import { get, range } from 'lodash-es'
+import { get, range, sample } from 'lodash-es'
 import { ThemeNamedValue, ThemeValue } from '../../types/css'
 import { SelectInput } from '../inputs/SelectInput'
 import { DataTypeSchema } from './types'
 import { joinPath } from '../providers/util'
+import { randomInt } from '../../lib/random'
 
 export function themeScale(path: string): DataTypeSchema<ThemeValue> {
   return {
@@ -19,6 +20,13 @@ export function themeScale(path: string): DataTypeSchema<ThemeValue> {
     stringify(value, theme) {
       return get(theme, joinPath(value.path, value.index))
     },
+    regenerate(options) {
+      const scale = get(options.theme, options.previousValue.path)
+      return {
+        ...options.previousValue,
+        index: randomInt(0, scale.length),
+      }
+    },
     inlineInput(props) {
       const theme = useTheme()
       const numOptions = get(theme, path)?.length || 0
@@ -64,6 +72,15 @@ export function themeRecord(path: string): DataTypeSchema<ThemeNamedValue> {
     stringify(value, theme) {
       return get(theme, `${value.path}.${value.key}`)
     },
+    regenerate(options) {
+      const records = get(options.theme, options.previousValue.path)
+      const paths = Object.keys(records || {})
+
+      return {
+        ...options.previousValue,
+        key: sample(paths) ?? '',
+      }
+    },
     inlineInput(props) {
       const theme = useTheme()
       const options = Object.keys(get(theme, path))