feat(server): support chaining after .callable and .actionable (#223)

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Refactor**
- Enhanced procedure operations to return enriched objects with
consistent properties and improved dynamic behavior.
- Updated return types for `callable` and `actionable` methods to ensure
they return specific procedure types.

- **Tests**
- Updated test validations to ensure that the procedure operations
reliably expose all expected attributes.
- Added assertions to confirm the presence of properties on the
`applied` object in test cases.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Author: dinwwwh

packages/server/src/implementer-procedure.test-d.ts

@@ -146,7 +146,7 @@ describe('ImplementedProcedure', () => {
     })
 
     expectTypeOf(applied).toEqualTypeOf<
-      Procedure<
+      ImplementedProcedure<
         InitialContext,
         CurrentContext,
           typeof inputSchema,
@@ -164,7 +164,7 @@ describe('ImplementedProcedure', () => {
     })
 
     expectTypeOf(applied).toEqualTypeOf<
-      Procedure<
+      ImplementedProcedure<
         InitialContext,
         CurrentContext,
           typeof inputSchema,

packages/server/src/implementer-procedure.ts

@@ -73,7 +73,7 @@ export interface ImplementedProcedure<
         TClientContext
       >
     >
-  ): Procedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
+  ): ImplementedProcedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
     & ProcedureClient<TClientContext, TInputSchema, TOutputSchema, TErrorMap >
 
   /**
@@ -90,7 +90,7 @@ export interface ImplementedProcedure<
         TClientContext
       >
     >
-  ): Procedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
+  ): ImplementedProcedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
     & ((...rest: ClientRest<TClientContext, InferSchemaInput<TInputSchema>>) => Promise<InferSchemaOutput<TOutputSchema>>)
 }
 

packages/server/src/procedure-decorated.test-d.ts

@@ -174,7 +174,7 @@ describe('DecoratedProcedure', () => {
     })
 
     expectTypeOf(applied).toEqualTypeOf<
-      Procedure<
+      DecoratedProcedure<
         InitialContext,
         CurrentContext,
         typeof inputSchema,
@@ -192,7 +192,7 @@ describe('DecoratedProcedure', () => {
     })
 
     expectTypeOf(applied).toEqualTypeOf<
-      Procedure<
+      DecoratedProcedure<
         InitialContext,
         CurrentContext,
         typeof inputSchema,

packages/server/src/procedure-decorated.test.ts

@@ -107,19 +107,42 @@ describe('decoratedProcedure', () => {
 
     expect(createProcedureClient).toBeCalledTimes(1)
     expect(createProcedureClient).toBeCalledWith(decorated, options)
-    expect(applied).not.haveOwnPropertyDescriptor('use')
+
+    // can access to function properties
+    expect('name' in applied).toBe(true)
+    expect(typeof applied.name).toBe('string')
+    expect('length' in applied).toBe(true)
+    expect(typeof applied.length).toBe('number')
+
+    expect('use' in applied).toBe(true)
+    expect('route' in applied).toBe(true)
+    expect('meta' in applied).toBe(true)
+
+    expect(applied.route({})).toBeInstanceOf(DecoratedProcedure)
+    expect(applied.route({})).toEqual(decorated)
   })
 
   it('.actionable', () => {
     const options = { context: { db: 'postgres' } }
 
     const applied = decorated.actionable(options)
-
     expect(applied).toBeInstanceOf(Function)
     expect(applied).toSatisfy(isProcedure)
 
     expect(createProcedureClient).toBeCalledTimes(1)
     expect(createProcedureClient).toBeCalledWith(decorated, options)
-    expect(applied).not.haveOwnPropertyDescriptor('use')
+
+    // can access to function properties
+    expect('name' in applied).toBe(true)
+    expect(typeof applied.name).toBe('string')
+    expect('length' in applied).toBe(true)
+    expect(typeof applied.length).toBe('number')
+
+    expect('use' in applied).toBe(true)
+    expect('route' in applied).toBe(true)
+    expect('meta' in applied).toBe(true)
+
+    expect(applied.route({})).toBeInstanceOf(DecoratedProcedure)
+    expect(applied.route({})).toEqual(decorated)
   })
 })

packages/server/src/procedure-decorated.ts

@@ -117,12 +117,16 @@ export class DecoratedProcedure<
         TClientContext
       >
     >
-  ): Procedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
+  ): DecoratedProcedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
     & ProcedureClient<TClientContext, TInputSchema, TOutputSchema, TErrorMap> {
-    return Object.assign(createProcedureClient(this, ...rest as any), {
-      '~type': 'Procedure' as const,
-      '~orpc': this['~orpc'],
-    })
+    return new Proxy(createProcedureClient(this, ...rest as any), {
+      get: (target, key) => {
+        return Reflect.has(this, key) ? Reflect.get(this, key) : Reflect.get(target, key)
+      },
+      has: (target, key) => {
+        return Reflect.has(this, key) || Reflect.has(target, key)
+      },
+    }) as any
   }
 
   /**
@@ -140,7 +144,7 @@ export class DecoratedProcedure<
       >
     >
   ):
-    & Procedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
+    & DecoratedProcedure<TInitialContext, TCurrentContext, TInputSchema, TOutputSchema, TErrorMap, TMeta>
     & ((...rest: ClientRest<TClientContext, InferSchemaInput<TInputSchema>>) => Promise<InferSchemaOutput<TOutputSchema>>) {
     return this.callable(...rest)
   }