feat(contract): ContractBuilder can use as a ContractProcedure (#90)

Author: unnoq

packages/contract/src/builder.test-d.ts

@@ -17,7 +17,11 @@ const baseErrorMap = {
   },
 }
 
-const builder = new ContractBuilder({ errorMap: baseErrorMap })
+const builder = new ContractBuilder({ errorMap: baseErrorMap, OutputSchema: undefined, InputSchema: undefined })
+
+it('also is a contract procedure', () => {
+  expectTypeOf(builder).toMatchTypeOf<ContractProcedure<undefined, undefined, typeof baseErrorMap>>()
+})
 
 describe('self chainable', () => {
   describe('errors', () => {

packages/contract/src/builder.test.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod'
 import { ContractBuilder } from './builder'
-import { ContractProcedure } from './procedure'
+import { ContractProcedure, isContractProcedure } from './procedure'
 import { DecoratedContractProcedure } from './procedure-decorated'
 import { ContractRouterBuilder } from './router-builder'
 
@@ -25,10 +25,14 @@ const baseErrorMap = {
   },
 }
 
-const builder = new ContractBuilder({ errorMap: baseErrorMap })
+const builder = new ContractBuilder({ errorMap: baseErrorMap, OutputSchema: undefined, InputSchema: undefined })
 
 const schema = z.object({ val: z.string().transform(val => Number(val)) })
 
+it('also is a contract procedure', () => {
+  expect(builder).toSatisfy(isContractProcedure)
+})
+
 describe('self chainable', () => {
   it('errors', () => {
     const errors = {

packages/contract/src/builder.ts

@@ -1,24 +1,17 @@
 import type { ErrorMap, ErrorMapGuard, ErrorMapSuggestions, StrictErrorMap } from './error-map'
-import type { RouteOptions } from './procedure'
 import type { ContractRouter } from './router'
 import type { AdaptedContractRouter } from './router-builder'
-
 import type { HTTPPath, Schema, SchemaInput, SchemaOutput } from './types'
+
+import { ContractProcedure, type RouteOptions } from './procedure'
 import { DecoratedContractProcedure } from './procedure-decorated'
 import { ContractRouterBuilder } from './router-builder'
 
 export type ContractBuilderDef<TErrorMap extends ErrorMap> = {
   errorMap: TErrorMap
 }
 
-export class ContractBuilder<TErrorMap extends ErrorMap> {
-  '~type' = 'ContractBuilder' as const
-  '~orpc': ContractBuilderDef<TErrorMap>
-
-  constructor(def: ContractBuilderDef<TErrorMap>) {
-    this['~orpc'] = def
-  }
-
+export class ContractBuilder<TErrorMap extends ErrorMap> extends ContractProcedure<undefined, undefined, TErrorMap> {
   errors<const U extends ErrorMap & ErrorMapGuard<TErrorMap> & ErrorMapSuggestions>(errors: U): ContractBuilder<U & TErrorMap> {
     return new ContractBuilder({
       ...this['~orpc'],

packages/contract/src/index.ts

@@ -19,4 +19,6 @@ export * from './types'
 
 export const oc = new ContractBuilder<Record<never, never>>({
   errorMap: {},
+  InputSchema: undefined,
+  OutputSchema: undefined,
 })