feat: handle multiple stores

Author: okwasniewski

apps/example/.gitignore

@@ -6,3 +6,4 @@
 !.yarn/versions
 
 swift/Generated/**
+kotlin/**/Generated/**

apps/example/kotlin/app/src/main/java/com/callstack/kotlinexample/BrownfieldStore.kt

@@ -34,12 +34,14 @@
     "node": ">=20"
   },
   "brownie": {
-    "stores": {
-      "schema": "./brownfield-store.schema.ts",
-      "typeName": "BrownfieldStore",
-      "swift": "./swift/Generated/BrownfieldStore.swift",
-      "kotlin": "./kotlin/app/src/main/java/com/callstack/kotlinexample/BrownfieldStore.kt",
-      "kotlinPackageName": "com.callstack.kotlinexample"
-    }
+    "stores": [
+      {
+        "schema": "./brownfield-store.schema.ts",
+        "typeName": "BrownfieldStore",
+        "swift": "./swift/Generated/BrownfieldStore.swift",
+        "kotlin": "./kotlin/app/src/main/java/com/callstack/kotlinexample/Generated/BrownfieldStore.kt",
+        "kotlinPackageName": "com.callstack.kotlinexample"
+      }
+    ]
   }
 }

apps/example/package.json

@@ -52,13 +52,40 @@ Add to your app's `package.json`:
 ```json
 {
   "brownie": {
-    "stores": {
-      "schema": "./brownfield-store.schema.ts",
-      "typeName": "BrownfieldStore",
-      "swift": "./ios/Generated/BrownfieldStore.swift",
-      "kotlin": "./android/app/src/main/java/com/example/BrownfieldStore.kt",
-      "kotlinPackageName": "com.example"
-    }
+    "stores": [
+      {
+        "schema": "./brownfield-store.schema.ts",
+        "typeName": "BrownfieldStore",
+        "swift": "./ios/Generated/BrownfieldStore.swift",
+        "kotlin": "./android/app/src/main/java/com/example/BrownfieldStore.kt",
+        "kotlinPackageName": "com.example"
+      }
+    ]
+  }
+}
+```
+
+Multiple stores example:
+
+```json
+{
+  "brownie": {
+    "stores": [
+      {
+        "schema": "./stores/user.schema.ts",
+        "typeName": "UserStore",
+        "swift": "./ios/Generated/UserStore.swift",
+        "kotlin": "./android/app/src/main/java/com/example/UserStore.kt",
+        "kotlinPackageName": "com.example"
+      },
+      {
+        "schema": "./stores/settings.schema.ts",
+        "typeName": "SettingsStore",
+        "swift": "./ios/Generated/SettingsStore.swift",
+        "kotlin": "./android/app/src/main/java/com/example/SettingsStore.kt",
+        "kotlinPackageName": "com.example"
+      }
+    ]
   }
 }
 ```
@@ -234,7 +261,7 @@ preBuild.dependsOn("generateBrownfieldStore")
 
 ### Codegen
 
-- [ ] Support multiple stores in package.json config (array of store configs)
+- [x] Support multiple stores in package.json config (array of store configs)
 - [ ] Generate store keys enum/constants for type safety between JS and Native
 
 ### Native Runtime

brownfield-store-plan.md

@@ -1,10 +1,51 @@
 import { parseArgs, styleText } from 'node:util';
-import { loadConfig } from '../config';
+import { loadConfig, type StoresConfig } from '../config';
 import { generateSwift } from '../generators/swift';
 import { generateKotlin } from '../generators/kotlin';
 
 type Platform = 'swift' | 'kotlin';
 
+async function generateForStore(
+  config: StoresConfig,
+  platforms: Platform[],
+  storeIndex?: number
+): Promise<void> {
+  const storeLabel = storeIndex !== undefined ? ` [${config.typeName}]` : '';
+
+  for (const p of platforms) {
+    const outputPath = config[p];
+    if (!outputPath) {
+      continue;
+    }
+
+    try {
+      if (p === 'swift') {
+        await generateSwift({
+          schemaPath: config.schema,
+          typeName: config.typeName,
+          outputPath,
+        });
+      } else {
+        await generateKotlin({
+          schemaPath: config.schema,
+          typeName: config.typeName,
+          outputPath,
+          packageName: config.kotlinPackageName,
+        });
+      }
+      console.log(styleText('green', `  ✓ ${outputPath}${storeLabel}`));
+    } catch (error) {
+      console.error(
+        styleText(
+          'red',
+          `Error generating ${p}${storeLabel}: ${error instanceof Error ? error.message : error}`
+        )
+      );
+      process.exit(1);
+    }
+  }
+}
+
 const HELP_TEXT = `
 ${styleText('bold', 'brownie codegen')} - Generate native store types from TypeScript schema
 
@@ -50,7 +91,7 @@ export async function runCodegen(
     return;
   }
 
-  const config = loadConfig();
+  const configs = loadConfig();
   const platform = values.platform as Platform | undefined;
 
   if (platform && !['swift', 'kotlin'].includes(platform)) {
@@ -63,55 +104,29 @@ export async function runCodegen(
     process.exit(1);
   }
 
-  const platforms: Platform[] = platform
-    ? [platform]
-    : (['swift', 'kotlin'] as Platform[]).filter((p) => config[p]);
-
-  if (platforms.length === 0) {
-    console.error(
-      styleText('red', 'No output paths configured in brownie.stores')
-    );
-    process.exit(1);
-  }
-
+  const isMultipleStores = configs.length > 1;
+  const schemaList = configs.map((c) => c.schema).join(', ');
   console.log(
-    styleText('cyan', `Generating store types from ${config.schema}...`)
+    styleText('cyan', `Generating store types from ${schemaList}...`)
   );
 
-  for (const p of platforms) {
-    const outputPath = config[p];
-    if (!outputPath) {
-      console.warn(
-        styleText('yellow', `Skipping ${p}: no output path configured`)
-      );
-      continue;
-    }
+  for (let i = 0; i < configs.length; i++) {
+    const config = configs[i];
+    const platforms: Platform[] = platform
+      ? [platform]
+      : (['swift', 'kotlin'] as Platform[]).filter((p) => config[p]);
 
-    try {
-      if (p === 'swift') {
-        await generateSwift({
-          schemaPath: config.schema,
-          typeName: config.typeName,
-          outputPath,
-        });
-      } else {
-        await generateKotlin({
-          schemaPath: config.schema,
-          typeName: config.typeName,
-          outputPath,
-          packageName: config.kotlinPackageName,
-        });
-      }
-      console.log(styleText('green', `  ✓ ${outputPath}`));
-    } catch (error) {
-      console.error(
+    if (platforms.length === 0) {
+      console.warn(
         styleText(
-          'red',
-          `Error generating ${p}: ${error instanceof Error ? error.message : error}`
+          'yellow',
+          `No output paths configured for store ${config.typeName}`
         )
       );
-      process.exit(1);
+      continue;
     }
+
+    await generateForStore(config, platforms, isMultipleStores ? i : undefined);
   }
 
   console.log(styleText('green', '\nDone!'));

packages/brownie/scripts/commands/codegen.ts

@@ -10,17 +10,37 @@ export interface StoresConfig {
 }
 
 export interface BrownieConfig {
-  stores?: StoresConfig;
+  stores?: StoresConfig[];
 }
 
 interface PackageJson {
   brownie?: BrownieConfig;
 }
 
+function validateStoreConfig(config: StoresConfig, index?: number): void {
+  const prefix =
+    index !== undefined ? `brownie.stores[${index}]` : 'brownie.stores';
+
+  if (!config.schema) {
+    throw new Error(`${prefix}.schema is required`);
+  }
+
+  if (!config.typeName) {
+    throw new Error(`${prefix}.typeName is required`);
+  }
+
+  if (!config.swift && !config.kotlin) {
+    throw new Error(
+      `At least one output path is required: ${prefix}.swift or ${prefix}.kotlin`
+    );
+  }
+}
+
 /**
  * Loads brownie config from package.json in the current working directory.
+ * Supports both single store config and array of store configs.
  */
-export function loadConfig(): StoresConfig {
+export function loadConfig(): StoresConfig[] {
   const packageJsonPath = path.resolve(process.cwd(), 'package.json');
 
   if (!fs.existsSync(packageJsonPath)) {
@@ -36,19 +56,15 @@ export function loadConfig(): StoresConfig {
     throw new Error('brownie.stores config not found in package.json');
   }
 
-  if (!config.schema) {
-    throw new Error('brownie.stores.schema is required');
+  if (!Array.isArray(config)) {
+    throw new Error('brownie.stores must be an array');
   }
 
-  if (!config.typeName) {
-    throw new Error('brownie.stores.typeName is required');
+  if (config.length === 0) {
+    throw new Error('brownie.stores array cannot be empty');
   }
 
-  if (!config.swift && !config.kotlin) {
-    throw new Error(
-      'At least one output path is required: brownie.stores.swift or brownie.stores.kotlin'
-    );
-  }
+  config.forEach((c, i) => validateStoreConfig(c, i));
 
   return config;
 }