lambder 2.0.3 → 2.0.4

package/dist/Lambder.d.ts

@@ -115,7 +115,7 @@ export default class Lambder<TSessionData = any, _TContract extends Record<strin
     private handleNoMatchedAction;
     addRoute(condition: Path | ConditionFunction | RegExp, actionFn: ActionFunction): this;
     addSessionRoute(condition: Path | ConditionFunction | RegExp, actionFn: SessionActionFunction<TSessionData>): this;
-    use<_TNewContract extends Record<string, any>>(plugin: (lambder: Lambder<TSessionData, _TContract>) => Lambder<TSessionData, _TNewContract>): Lambder<TSessionData, _TNewContract>;
+    use<_TNewContract extends Record<string, any>>(plugin: (lambder: Lambder<TSessionData, _TContract>) => Lambder<TSessionData, _TNewContract>): Lambder<TSessionData, _TNewContract extends _TContract ? _TNewContract : (_TContract & _TNewContract)>;
     addApi<TName extends string, TInput extends z.ZodTypeAny, TOutput extends z.ZodTypeAny>(name: TName, schema: {
         input: TInput;
         output: TOutput;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lambder",
-  "version": "2.0.3",
+  "version": "2.0.4",
   "description": "",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/Lambder.ts

@@ -285,8 +285,8 @@ export default class Lambder<TSessionData = any, _TContract extends Record<strin
     // Plugin system
     public use<_TNewContract extends Record<string, any>>(
         plugin: (lambder: Lambder<TSessionData, _TContract>) => Lambder<TSessionData, _TNewContract>
-    ): Lambder<TSessionData, _TNewContract> {
-        return plugin(this);
+    ): Lambder<TSessionData, _TNewContract extends _TContract ? _TNewContract : (_TContract & _TNewContract)> {
+        return plugin(this) as any;
     }
 
     // Typed API with Zod

package/tests/use-plugin.test.ts

@@ -4,7 +4,7 @@
  * This file tests the plugin system that allows modular API composition
  */
 
-import { describe, it, expect } from 'vitest';
+import { describe, it, expect, expectTypeOf } from 'vitest';
 import { z } from 'zod';
 import Lambder from '../src/Lambder.js';
 import LambderCaller from '../src/LambderCaller.js';
@@ -435,3 +435,24 @@ describe('Plugin System - Type Safety', () => {
         expect(caller).toBeDefined();
     });
 });
+
+// ============================================================================
+// Test 7: Non-Generic Plugin Type Accumulation
+// ============================================================================
+
+describe('Plugin System - Non-Generic Plugins', () => {
+    it('should accumulate types correctly when using non-generic plugins', () => {
+        const plugin1 = (l: Lambder) => l.addApi('api1', { input: z.void(), output: z.void() }, async (ctx, res) => res.raw({ statusCode: 200, body: '' }));
+        const plugin2 = (l: Lambder) => l.addApi('api2', { input: z.void(), output: z.void() }, async (ctx, res) => res.raw({ statusCode: 200, body: '' }));
+
+        const lambder = new Lambder({ publicPath: '', apiPath: '' })
+            .use(plugin1)
+            .use(plugin2);
+
+        type Contract = typeof lambder.ApiContractType;
+        
+        // Check if both api1 and api2 exist in Contract
+        expectTypeOf<Contract>().toHaveProperty('api1');
+        expectTypeOf<Contract>().toHaveProperty('api2');
+    });
+});