@effect-app/infra 2.94.0 → 3.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (54) hide show
  1. package/CHANGELOG.md +11 -0
  2. package/dist/Emailer/service.d.ts +1 -1
  3. package/dist/MainFiberSet.d.ts +1 -1
  4. package/dist/Operations.d.ts +1 -1
  5. package/dist/RequestFiberSet.d.ts +1 -1
  6. package/dist/Store/service.d.ts +2 -2
  7. package/dist/adapters/SQL/Model.d.ts +6 -6
  8. package/dist/adapters/ServiceBus.d.ts +2 -2
  9. package/dist/adapters/ServiceBus.d.ts.map +1 -1
  10. package/dist/adapters/ServiceBus.js +13 -10
  11. package/dist/adapters/memQueue.d.ts +1 -1
  12. package/dist/api/routing/middleware/RouterMiddleware.d.ts +6 -6
  13. package/dist/api/routing/middleware/RouterMiddleware.d.ts.map +1 -1
  14. package/dist/api/routing/middleware/RpcMiddleware.d.ts +4 -15
  15. package/dist/api/routing/middleware/RpcMiddleware.d.ts.map +1 -1
  16. package/dist/api/routing/middleware/RpcMiddleware.js +3 -4
  17. package/dist/api/routing/middleware/RpcMiddlewareX.d.ts +18 -0
  18. package/dist/api/routing/middleware/RpcMiddlewareX.d.ts.map +1 -0
  19. package/dist/api/routing/middleware/RpcMiddlewareX.js +16 -0
  20. package/dist/api/routing/middleware/generic-middleware.d.ts +22 -8
  21. package/dist/api/routing/middleware/generic-middleware.d.ts.map +1 -1
  22. package/dist/api/routing/middleware/generic-middleware.js +59 -62
  23. package/dist/api/routing/middleware/middleware-api.d.ts +51 -32
  24. package/dist/api/routing/middleware/middleware-api.d.ts.map +1 -1
  25. package/dist/api/routing/middleware/middleware-api.js +54 -24
  26. package/dist/api/routing/middleware/middleware-native.d.ts +23 -0
  27. package/dist/api/routing/middleware/middleware-native.d.ts.map +1 -0
  28. package/dist/api/routing/middleware/middleware-native.js +19 -0
  29. package/dist/api/routing/middleware/middleware.d.ts +7 -27
  30. package/dist/api/routing/middleware/middleware.d.ts.map +1 -1
  31. package/dist/api/routing/middleware/middleware.js +73 -88
  32. package/dist/api/routing/middleware.d.ts +1 -0
  33. package/dist/api/routing/middleware.d.ts.map +1 -1
  34. package/dist/api/routing/middleware.js +2 -1
  35. package/package.json +11 -3
  36. package/src/adapters/ServiceBus.ts +7 -5
  37. package/src/api/routing/middleware/RouterMiddleware.ts +9 -8
  38. package/src/api/routing/middleware/RpcMiddleware.ts +4 -51
  39. package/src/api/routing/middleware/RpcMiddlewareX.ts +70 -0
  40. package/src/api/routing/middleware/generic-middleware.ts +142 -107
  41. package/src/api/routing/middleware/middleware-api.ts +239 -61
  42. package/src/api/routing/middleware/middleware-native.ts +23 -0
  43. package/src/api/routing/middleware/middleware.ts +37 -40
  44. package/src/api/routing/middleware.ts +1 -0
  45. package/test/controller.test.ts +39 -14
  46. package/test/dist/controller.test.d.ts.map +1 -1
  47. package/test/dist/fixtures.d.ts +48 -32
  48. package/test/dist/fixtures.d.ts.map +1 -1
  49. package/test/dist/fixtures.js +76 -47
  50. package/test/dist/requires.test.d.ts.map +1 -1
  51. package/test/dist/rpc-multi-middleware.test.d.ts.map +1 -0
  52. package/test/fixtures.ts +70 -16
  53. package/test/requires.test.ts +15 -50
  54. package/test/rpc-multi-middleware.test.ts +134 -0
@@ -1,15 +1,56 @@
1
1
  /* eslint-disable @typescript-eslint/no-explicit-any */
2
2
  import { type RpcMiddleware } from "@effect/rpc"
3
3
  import { Context, Effect, type Layer, type NonEmptyReadonlyArray, Option, type S, type Scope } from "effect-app"
4
- import { type ContextTagArray } from "effect-app/client"
4
+ import { type ContextTagArray, type GetContextConfig, type RPCContextMap } from "effect-app/client"
5
+ import { type Tag } from "effect/Context"
6
+ import { type Simplify } from "effect/Types"
5
7
  import { InfraLogger } from "../../../logger.js"
8
+ import { type MakeTags, type MiddlewareMakerId } from "./middleware-api.js"
6
9
  import { type RpcMiddlewareWrap, type TagClassAny } from "./RpcMiddleware.js"
7
10
 
8
11
  // Effect rpc middleware does not support changing payload or headers, but we do..
9
12
 
10
- export type MiddlewareMaker = TagClassAny & { Default: Layer.Layer.Any } // todo; and Layer..
13
+ export interface MiddlewareMaker<
14
+ RequestContextMap extends Record<string, RPCContextMap.Any>,
15
+ MiddlewareProviders extends ReadonlyArray<MiddlewareMaker.Any>
16
+ > extends
17
+ RpcMiddleware.TagClass<
18
+ MiddlewareMakerId,
19
+ "MiddlewareMaker",
20
+ Simplify<
21
+ & { readonly wrap: true }
22
+ & (Exclude<
23
+ MiddlewareMaker.ManyRequired<MiddlewareProviders>,
24
+ MiddlewareMaker.ManyProvided<MiddlewareProviders>
25
+ > extends never ? {} : {
26
+ readonly requires: MakeTags<
27
+ Exclude<
28
+ MiddlewareMaker.ManyRequired<MiddlewareProviders>,
29
+ MiddlewareMaker.ManyProvided<MiddlewareProviders>
30
+ >
31
+ >
32
+ })
33
+ & (MiddlewareMaker.ManyErrors<MiddlewareProviders> extends never ? {}
34
+ : {
35
+ readonly failure: S.Schema<MiddlewareMaker.ManyErrors<MiddlewareProviders>>
36
+ })
37
+ & (MiddlewareMaker.ManyProvided<MiddlewareProviders> extends never ? {}
38
+ : { readonly provides: MakeTags<MiddlewareMaker.ManyProvided<MiddlewareProviders>> })
39
+ >
40
+ >
41
+ {
42
+ readonly layer: Layer.Layer<MiddlewareMakerId, never, Tag.Identifier<MiddlewareProviders[number]>>
43
+ readonly requestContext: RequestContextTag<RequestContextMap>
44
+ readonly requestContextMap: RequestContextMap
45
+ }
46
+
47
+ export interface RequestContextTag<RequestContextMap extends Record<string, RPCContextMap.Any>>
48
+ extends Context.Tag<"RequestContextConfig", GetContextConfig<RequestContextMap>>
49
+ {}
11
50
 
12
51
  export namespace MiddlewareMaker {
52
+ export type Any = TagClassAny
53
+
13
54
  export type ApplyServices<A extends TagClassAny, R> = Exclude<R, Provided<A>> | Required<A>
14
55
 
15
56
  export type ApplyManyServices<A extends NonEmptyReadonlyArray<TagClassAny>, R> =
@@ -44,120 +85,114 @@ export namespace MiddlewareMaker {
44
85
  }
45
86
 
46
87
  export const middlewareMaker = <
47
- MiddlewareProviders extends ReadonlyArray<MiddlewareMaker>
48
- >(middlewares: MiddlewareProviders): {
49
- dependencies: { [K in keyof MiddlewareProviders]: MiddlewareProviders[K]["Default"] }
50
- effect: Effect.Effect<
51
- RpcMiddlewareWrap<
52
- MiddlewareMaker.ManyProvided<MiddlewareProviders>,
53
- MiddlewareMaker.ManyErrors<MiddlewareProviders>,
54
- Exclude<
55
- MiddlewareMaker.ManyRequired<MiddlewareProviders>,
56
- MiddlewareMaker.ManyProvided<MiddlewareProviders>
57
- > extends never ? never
58
- : Exclude<MiddlewareMaker.ManyRequired<MiddlewareProviders>, MiddlewareMaker.ManyProvided<MiddlewareProviders>>
59
- >
88
+ MiddlewareProviders extends ReadonlyArray<MiddlewareMaker.Any>
89
+ >(middlewares: MiddlewareProviders): Effect.Effect<
90
+ RpcMiddlewareWrap<
91
+ MiddlewareMaker.ManyProvided<MiddlewareProviders>,
92
+ MiddlewareMaker.ManyErrors<MiddlewareProviders>,
93
+ Exclude<
94
+ MiddlewareMaker.ManyRequired<MiddlewareProviders>,
95
+ MiddlewareMaker.ManyProvided<MiddlewareProviders>
96
+ > extends never ? never
97
+ : Exclude<MiddlewareMaker.ManyRequired<MiddlewareProviders>, MiddlewareMaker.ManyProvided<MiddlewareProviders>>
60
98
  >
61
- } => {
99
+ > => {
62
100
  // we want to run them in reverse order because latter middlewares will provide context to former ones
63
101
  middlewares = middlewares.toReversed() as any
64
102
 
65
- return {
66
- dependencies: middlewares.map((_) => _.Default),
67
- effect: Effect.gen(function*() {
68
- const context = yield* Effect.context()
69
-
70
- // returns a Effect/RpcMiddlewareWrap with Scope in requirements
71
- return (
72
- options: Parameters<
73
- RpcMiddlewareWrap<
74
- MiddlewareMaker.ManyProvided<MiddlewareProviders>,
75
- never,
76
- Scope.Scope
77
- >
78
- >[0]
79
- ) => {
80
- // we start with the actual handler
81
- let handler = options.next
82
-
83
- // inspired from Effect/RpcMiddleware
84
- for (const tag of middlewares) {
85
- if (tag.wrap) {
86
- // use the tag to get the middleware from context
87
- const middleware = Context.unsafeGet(context, tag)
88
-
89
- // wrap the current handler, allowing the middleware to run before and after it
90
- handler = InfraLogger.logDebug("Applying middleware wrap " + tag.key).pipe(
91
- Effect.zipRight(middleware({ ...options, next: handler }))
92
- ) as any
93
- } else if (tag.optional) {
94
- // use the tag to get the middleware from context
95
- // if the middleware fails to run, we will ignore the error
96
- const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
97
-
98
- const previous = handler
99
-
100
- // set the previous handler to run after the middleware
101
- // if the middleware is not present, we just return the previous handler
102
- // otherwise the middleware will provide some context to be provided to the previous handler
103
- handler = InfraLogger.logDebug("Applying middleware optional " + tag.key).pipe(
104
- Effect.zipRight(Effect.matchEffect(middleware(options), {
105
- onFailure: () => previous,
106
- onSuccess: tag.provides !== undefined
107
- ? (value) =>
103
+ return Effect.gen(function*() {
104
+ const context = yield* Effect.context()
105
+
106
+ // returns a Effect/RpcMiddlewareWrap with Scope in requirements
107
+ return (
108
+ options: Parameters<
109
+ RpcMiddlewareWrap<
110
+ MiddlewareMaker.ManyProvided<MiddlewareProviders>,
111
+ never,
112
+ Scope.Scope
113
+ >
114
+ >[0]
115
+ ) => {
116
+ // we start with the actual handler
117
+ let handler = options.next
118
+
119
+ // inspired from Effect/RpcMiddleware
120
+ for (const tag of middlewares) {
121
+ if (tag.wrap) {
122
+ // use the tag to get the middleware from context
123
+ const middleware = Context.unsafeGet(context, tag)
124
+
125
+ // wrap the current handler, allowing the middleware to run before and after it
126
+ handler = InfraLogger.logDebug("Applying middleware wrap " + tag.key).pipe(
127
+ Effect.zipRight(middleware({ ...options, next: handler }))
128
+ ) as any
129
+ } else if (tag.optional) {
130
+ // use the tag to get the middleware from context
131
+ // if the middleware fails to run, we will ignore the error
132
+ const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
133
+
134
+ const previous = handler
135
+
136
+ // set the previous handler to run after the middleware
137
+ // if the middleware is not present, we just return the previous handler
138
+ // otherwise the middleware will provide some context to be provided to the previous handler
139
+ handler = InfraLogger.logDebug("Applying middleware optional " + tag.key).pipe(
140
+ Effect.zipRight(Effect.matchEffect(middleware(options), {
141
+ onFailure: () => previous,
142
+ onSuccess: tag.provides !== undefined
143
+ ? (value) =>
144
+ Context.isContext(value)
145
+ ? Effect.provide(previous, value)
146
+ : Effect.provideService(previous, tag.provides as any, value)
147
+ : (_) => previous
148
+ }))
149
+ )
150
+ } else if (tag.dynamic) {
151
+ // use the tag to get the middleware from context
152
+ const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
153
+
154
+ const previous = handler
155
+
156
+ // set the previous handler to run after the middleware
157
+ // we do expect the middleware to be present, but the context might not be available
158
+ // if it is, we provide it to the previous handler
159
+ handler = InfraLogger.logDebug("Applying middleware dynamic " + tag.key, tag.dynamic).pipe(
160
+ Effect.zipRight(
161
+ middleware(options).pipe(
162
+ Effect.flatMap((o) =>
163
+ Option.isSome(o)
164
+ ? Context.isContext(o.value)
165
+ ? Effect.provide(previous, o.value)
166
+ : Effect.provideService(previous, tag.dynamic!.settings.service!, /* TODO */ o.value)
167
+ : previous
168
+ )
169
+ )
170
+ )
171
+ ) as any
172
+ } else {
173
+ // use the tag to get the middleware from context
174
+ const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
175
+
176
+ const previous = handler
177
+
178
+ // set the previous handler to run after the middleware
179
+ // we do expect both the middleware and the context to be present
180
+ handler = InfraLogger.logDebug("Applying middleware " + tag.key).pipe(
181
+ Effect.zipRight(
182
+ tag.provides !== undefined
183
+ ? middleware(options).pipe(
184
+ Effect.flatMap((value) =>
108
185
  Context.isContext(value)
109
186
  ? Effect.provide(previous, value)
110
187
  : Effect.provideService(previous, tag.provides as any, value)
111
- : (_) => previous
112
- }))
113
- )
114
- } else if (tag.dynamic) {
115
- // use the tag to get the middleware from context
116
- const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
117
-
118
- const previous = handler
119
-
120
- // set the previous handler to run after the middleware
121
- // we do expect the middleware to be present, but the context might not be available
122
- // if it is, we provide it to the previous handler
123
- handler = InfraLogger.logDebug("Applying middleware dynamic " + tag.key, tag.dynamic).pipe(
124
- Effect.zipRight(
125
- middleware(options).pipe(
126
- Effect.flatMap((o) =>
127
- Option.isSome(o)
128
- ? Context.isContext(o.value)
129
- ? Effect.provide(previous, o.value)
130
- : Effect.provideService(previous, tag.dynamic!.settings.service!, /* TODO */ o.value)
131
- : previous
132
188
  )
133
189
  )
134
- )
135
- ) as any
136
- } else {
137
- // use the tag to get the middleware from context
138
- const middleware = Context.unsafeGet(context, tag) as RpcMiddleware.RpcMiddleware<any, any>
139
-
140
- const previous = handler
141
-
142
- // set the previous handler to run after the middleware
143
- // we do expect both the middleware and the context to be present
144
- handler = InfraLogger.logDebug("Applying middleware " + tag.key).pipe(
145
- Effect.zipRight(
146
- tag.provides !== undefined
147
- ? middleware(options).pipe(
148
- Effect.flatMap((value) =>
149
- Context.isContext(value)
150
- ? Effect.provide(previous, value)
151
- : Effect.provideService(previous, tag.provides as any, value)
152
- )
153
- )
154
- : Effect.zipRight(middleware(options), previous)
155
- )
156
- ) as any
157
- }
190
+ : Effect.zipRight(middleware(options), previous)
191
+ )
192
+ ) as any
158
193
  }
159
- return handler
160
194
  }
161
- })
162
- } as any
195
+ return handler
196
+ }
197
+ }) as any
163
198
  }
@@ -1,10 +1,15 @@
1
1
  /* eslint-disable @typescript-eslint/no-explicit-any */
2
+ import { Rpc, type RpcGroup, type RpcMiddleware, type RpcSchema } from "@effect/rpc"
2
3
  import { type AnyWithProps } from "@effect/rpc/Rpc"
3
- import { Context, type Effect, type NonEmptyArray, type NonEmptyReadonlyArray, S } from "effect-app"
4
- import { type GetContextConfig, type RPCContextMap } from "effect-app/client"
4
+ import { type HandlersFrom } from "@effect/rpc/RpcGroup"
5
+ import { Tag } from "@effect/rpc/RpcMiddleware"
6
+ import { Context, type Effect, Layer, type NonEmptyArray, type NonEmptyReadonlyArray, S, type Schema, type Scope, type Stream } from "effect-app"
7
+ import { type GetContextConfig, type GetEffectContext, type RPCContextMap } from "effect-app/client"
8
+ import { typedValuesOf } from "effect-app/utils"
9
+ import { type ReadonlyMailbox } from "effect/Mailbox"
5
10
  import { type TypeTestId } from "../../routing.js"
6
- import { type MiddlewareMaker, middlewareMaker } from "./generic-middleware.js"
7
- import { type AnyDynamic, type RpcDynamic, Tag, type TagClassAny } from "./RpcMiddleware.js"
11
+ import { type MiddlewareMaker, middlewareMaker, type RequestContextTag } from "./generic-middleware.js"
12
+ import { type AnyDynamic, type RpcDynamic, type TagClassAny } from "./RpcMiddleware.js"
8
13
 
9
14
  /** Adapter used when setting the dynamic prop on a middleware implementation */
10
15
  export const contextMap = <
@@ -15,26 +20,27 @@ export const contextMap = <
15
20
  settings: { service: rcm[key]!["service"] } as RequestContextMap[Key]
16
21
  })
17
22
 
23
+ const tag = Context.GenericTag("RequestContextConfig")
18
24
  /** Retrieves RequestContextConfig out of the RPC annotations */
19
25
  export const getConfig = <
20
26
  RequestContextMap extends Record<string, RPCContextMap.Any>
21
27
  >() =>
22
28
  (rpc: AnyWithProps): GetContextConfig<RequestContextMap> => {
23
- return Context.unsafeGet(rpc.annotations, Context.GenericTag("RequestContextConfig"))
29
+ return Context.getOrElse(rpc.annotations, tag as any, () => ({}))
24
30
  }
25
31
 
26
32
  // the following implements sort of builder pattern
27
33
  // we support both sideways and upwards elimination of dependencies
28
34
 
29
35
  // it's for dynamic middlewares
30
- type GetDependsOnKeys<MW extends MiddlewareMaker> = MW extends { dependsOn: NonEmptyReadonlyArray<TagClassAny> } ? {
36
+ type GetDependsOnKeys<MW extends MiddlewareMaker.Any> = MW extends { dependsOn: NonEmptyReadonlyArray<TagClassAny> } ? {
31
37
  [K in keyof MW["dependsOn"]]: MW["dependsOn"][K] extends AnyDynamic ? MW["dependsOn"][K]["dynamic"]["key"]
32
38
  : never
33
39
  }[keyof MW["dependsOn"]]
34
40
  : never
35
41
 
36
42
  type FilterInDynamicMiddlewares<
37
- MWs extends ReadonlyArray<MiddlewareMaker>,
43
+ MWs extends ReadonlyArray<MiddlewareMaker.Any>,
38
44
  RequestContextMap extends Record<string, RPCContextMap.Any>
39
45
  > = {
40
46
  [K in keyof MWs]: MWs[K] extends { dynamic: RpcDynamic<any, RequestContextMap[keyof RequestContextMap]> } ? MWs[K]
@@ -46,13 +52,13 @@ type RecursiveHandleMWsSideways<
46
52
  R extends {
47
53
  rcm: Record<string, RPCContextMap.Any>
48
54
  provided: keyof R["rcm"] // that's fine
49
- middlewares: ReadonlyArray<MiddlewareMaker>
55
+ middlewares: ReadonlyArray<MiddlewareMaker.Any>
50
56
  dmp: any
51
57
  middlewareR: any
52
58
  }
53
59
  > = MWs extends [
54
- infer F extends MiddlewareMaker,
55
- ...infer Rest extends ReadonlyArray<MiddlewareMaker>
60
+ infer F extends MiddlewareMaker.Any,
61
+ ...infer Rest extends ReadonlyArray<MiddlewareMaker.Any>
56
62
  ] ? RecursiveHandleMWsSideways<Rest, {
57
63
  rcm: R["rcm"]
58
64
  // when one dynamic middleware depends on another, subtract the key to enforce the dependency to be provided after
@@ -76,11 +82,36 @@ type RecursiveHandleMWsSideways<
76
82
  export interface BuildingMiddleware<
77
83
  RequestContextMap extends Record<string, RPCContextMap.Any>,
78
84
  Provided extends keyof RequestContextMap,
79
- Middlewares extends ReadonlyArray<MiddlewareMaker>,
85
+ Middlewares extends ReadonlyArray<MiddlewareMaker.Any>,
80
86
  DynamicMiddlewareProviders,
81
87
  out MiddlewareR extends { _tag: string } = never
82
88
  > {
83
- middleware<MWs extends NonEmptyArray<MiddlewareMaker>>(
89
+ rpc: <
90
+ const Tag extends string,
91
+ Payload extends Schema.Schema.Any | Schema.Struct.Fields = typeof Schema.Void,
92
+ Success extends Schema.Schema.Any = typeof Schema.Void,
93
+ Error extends Schema.Schema.All = typeof Schema.Never,
94
+ const Stream extends boolean = false,
95
+ Config extends GetContextConfig<RequestContextMap> = {}
96
+ >(tag: Tag, options?: {
97
+ readonly payload?: Payload
98
+ readonly success?: Success
99
+ readonly error?: Error
100
+ readonly stream?: Stream
101
+ readonly config?: Config
102
+ readonly primaryKey?: [Payload] extends [Schema.Struct.Fields]
103
+ ? ((payload: Schema.Simplify<Schema.Struct.Type<NoInfer<Payload>>>) => string)
104
+ : never
105
+ }) =>
106
+ & Rpc.Rpc<
107
+ Tag,
108
+ Payload extends Schema.Struct.Fields ? Schema.Struct<Payload> : Payload,
109
+ Stream extends true ? RpcSchema.Stream<Success, Error> : Success,
110
+ Stream extends true ? typeof Schema.Never : Error
111
+ >
112
+ & { readonly config: Config }
113
+
114
+ middleware<MWs extends NonEmptyArray<MiddlewareMaker.Any>>(
84
115
  ...mw: MWs
85
116
  ): RecursiveHandleMWsSideways<MWs, {
86
117
  rcm: RequestContextMap
@@ -91,7 +122,7 @@ export interface BuildingMiddleware<
91
122
  }> extends infer Res extends {
92
123
  rcm: RequestContextMap
93
124
  provided: keyof RequestContextMap
94
- middlewares: ReadonlyArray<MiddlewareMaker>
125
+ middlewares: ReadonlyArray<MiddlewareMaker.Any>
95
126
  dmp: any
96
127
  middlewareR: any
97
128
  } ? MiddlewaresBuilder<
@@ -113,7 +144,7 @@ export interface BuildingMiddleware<
113
144
  export type MiddlewaresBuilder<
114
145
  RequestContextMap extends Record<string, RPCContextMap.Any>,
115
146
  Provided extends keyof RequestContextMap = never,
116
- Middlewares extends ReadonlyArray<MiddlewareMaker> = [],
147
+ Middlewares extends ReadonlyArray<MiddlewareMaker.Any> = [],
117
148
  DynamicMiddlewareProviders = unknown,
118
149
  MiddlewareR extends { _tag: string } = never
119
150
  > =
@@ -126,50 +157,20 @@ export type MiddlewaresBuilder<
126
157
  >
127
158
  & // keyof Omit<RequestContextMap, Provided> extends never is true when all the dynamic middlewares are provided
128
159
  // MiddlewareR is never when all the required services from generic & dynamic middlewares are provided
129
- (keyof Omit<RequestContextMap, Provided> extends never ? [MiddlewareR] extends [never] ? ReturnType<
130
- typeof makeMiddlewareBasic<
131
- RequestContextMap,
132
- Middlewares
133
- >
160
+ (keyof Omit<RequestContextMap, Provided> extends never ? [MiddlewareR] extends [never] ? MiddlewareMaker<
161
+ RequestContextMap,
162
+ Middlewares
134
163
  >
135
164
  : {}
136
165
  : {})
137
166
 
138
- export const makeMiddleware: <
139
- RequestContextMap extends Record<string, RPCContextMap.Any>
140
- >(rcm: RequestContextMap) => MiddlewaresBuilder<RequestContextMap> = (rcm) => {
141
- let allMiddleware: MiddlewareMaker[] = []
142
- const it = {
143
- middleware: (...middlewares: any[]) => {
144
- for (const mw of middlewares) {
145
- // recall that we run middlewares in reverse order
146
- allMiddleware = [mw, ...allMiddleware]
147
- }
148
- return allMiddleware.filter((m) => !!m.dynamic).length !== Object.keys(rcm).length
149
- // for sure, until all the dynamic middlewares are provided it's non sensical to call makeMiddlewareBasic
150
- ? it
151
- // actually, we don't know yet if MiddlewareR is never, but we can't easily check it at runtime
152
- : Object.assign(makeMiddlewareBasic<any, any>(rcm, ...allMiddleware), it)
153
- }
154
- }
155
- return it as any
156
- }
157
-
158
- //
159
- export interface MiddlewareMakerId {
160
- readonly _id: unique symbol
161
- }
162
-
163
- // TODO: actually end up with [Tag<A, A>, Tag<B, B>, ...]
164
- type MakeTags<A> = Context.Tag<A, A>
165
-
166
167
  const makeMiddlewareBasic =
167
168
  // by setting RequestContextMap beforehand, execute contextual typing does not fuck up itself to anys
168
169
  <
169
170
  RequestContextMap extends Record<string, RPCContextMap.Any>,
170
- MiddlewareProviders extends ReadonlyArray<MiddlewareMaker>
171
+ MiddlewareProviders extends ReadonlyArray<MiddlewareMaker.Any>
171
172
  >(
172
- _rcm: RequestContextMap,
173
+ rcm: RequestContextMap,
173
174
  ...make: MiddlewareProviders
174
175
  ) => {
175
176
  // reverse middlewares and wrap one after the other
@@ -201,26 +202,203 @@ const makeMiddlewareBasic =
201
202
  ],
202
203
  provides: (provides.length > 0
203
204
  ? provides
204
- : undefined) as unknown as MiddlewareMaker.ManyProvided<MiddlewareProviders> extends never ? never : [
205
- MakeTags<MiddlewareMaker.ManyProvided<MiddlewareProviders>>
206
- ],
205
+ : undefined) as unknown as MiddlewareMaker.ManyProvided<MiddlewareProviders> extends never ? never
206
+ : MakeTags<MiddlewareMaker.ManyProvided<MiddlewareProviders>>,
207
207
  wrap: true
208
- })(
209
- middleware as {
210
- dependencies: typeof middleware["dependencies"]
211
- effect: Effect<
208
+ })
209
+
210
+ const layer = Layer
211
+ .scoped(
212
+ MiddlewareMaker,
213
+ middleware as Effect<
212
214
  any, // TODO: why ?
213
- Effect.Error<typeof middleware["effect"]>,
214
- Effect.Context<typeof middleware["effect"]>
215
+ Effect.Error<typeof middleware>,
216
+ Effect.Context<typeof middleware>
215
217
  >
216
- }
217
- )
218
+ )
218
219
 
219
220
  // add to the tag a default implementation
220
221
  return Object.assign(MiddlewareMaker, {
222
+ layer,
221
223
  // tag to be used to retrieve the RequestContextConfig from RPC annotations
222
224
  requestContext: Context.GenericTag<"RequestContextConfig", GetContextConfig<RequestContextMap>>(
223
225
  "RequestContextConfig"
224
- )
226
+ ),
227
+ requestContextMap: rcm
225
228
  })
226
229
  }
230
+
231
+ export const makeMiddleware = <
232
+ RequestContextMap extends Record<string, RPCContextMap.Any>
233
+ >(rcm: RequestContextMap): MiddlewaresBuilder<RequestContextMap> => {
234
+ let allMiddleware: MiddlewareMaker.Any[] = []
235
+ const requestContext = Context.GenericTag<"RequestContextConfig", GetContextConfig<RequestContextMap>>(
236
+ "RequestContextConfig"
237
+ )
238
+ const it = {
239
+ // rpc with config
240
+ rpc: <
241
+ const Tag extends string,
242
+ Payload extends Schema.Schema.Any | Schema.Struct.Fields = typeof Schema.Void,
243
+ Success extends Schema.Schema.Any = typeof Schema.Void,
244
+ Error extends Schema.Schema.All = typeof Schema.Never,
245
+ const Stream extends boolean = false,
246
+ Config extends GetContextConfig<RequestContextMap> = {}
247
+ >(tag: Tag, options?: {
248
+ readonly payload?: Payload
249
+ readonly success?: Success
250
+ readonly error?: Error
251
+ readonly stream?: Stream
252
+ readonly config?: Config
253
+ readonly primaryKey?: [Payload] extends [Schema.Struct.Fields]
254
+ ? ((payload: Schema.Simplify<Schema.Struct.Type<NoInfer<Payload>>>) => string)
255
+ : never
256
+ }):
257
+ & Rpc.Rpc<
258
+ Tag,
259
+ Payload extends Schema.Struct.Fields ? Schema.Struct<Payload> : Payload,
260
+ // TODO: enhance `Error`. type based on middleware config.
261
+ Stream extends true ? RpcSchema.Stream<Success, Error> : Success,
262
+ Stream extends true ? typeof Schema.Never : Error
263
+ >
264
+ & { config: Config } =>
265
+ {
266
+ const config = options?.config ?? {} as Config
267
+
268
+ // based on the config, we must enhance (union) or set failures.
269
+ // TODO: we should only include errors that are relevant based on the middleware config.ks
270
+ const error = options?.error
271
+ const errors = typedValuesOf(rcm).map((_) => _.error).filter((_) => _ && _ !== S.Never) // TODO: only the errors relevant based on config
272
+ const newError = error ? S.Union(error, ...errors) : S.Union(...errors)
273
+
274
+ const rpc = Rpc.make(tag, { ...options, error: newError }) as any
275
+
276
+ return Object.assign(rpc.annotate(requestContext, config), { config })
277
+ },
278
+ middleware: (...middlewares: any[]) => {
279
+ for (const mw of middlewares) {
280
+ // recall that we run middlewares in reverse order
281
+ allMiddleware = [mw, ...allMiddleware]
282
+ }
283
+ return allMiddleware.filter((m) => !!m.dynamic).length !== Object.keys(rcm).length
284
+ // for sure, until all the dynamic middlewares are provided it's non sensical to call makeMiddlewareBasic
285
+ ? it
286
+ // actually, we don't know yet if MiddlewareR is never, but we can't easily check it at runtime
287
+ : Object.assign(makeMiddlewareBasic<any, any>(rcm, ...allMiddleware), it)
288
+ }
289
+ }
290
+ return it as any
291
+ }
292
+
293
+ // customised version of Rpc.AddMiddleware, so that we don't loose the `config`...
294
+ // not needed if there's official support in Rpc.Rpc.
295
+ export type AddMiddleware<R extends Rpc.Any, Middleware extends RpcMiddleware.TagClassAny> = R extends Rpc.Rpc<
296
+ infer _Tag,
297
+ infer _Payload,
298
+ infer _Success,
299
+ infer _Error,
300
+ infer _Middleware
301
+ > ?
302
+ & Rpc.Rpc<
303
+ _Tag,
304
+ _Payload,
305
+ _Success,
306
+ _Error,
307
+ _Middleware | Middleware
308
+ >
309
+ & { readonly config: R extends { readonly config: infer _C } ? _C : never }
310
+ : never
311
+
312
+ // alternatively consider group.serverMiddleware? hmmm
313
+ export const middlewareGroup = <
314
+ RequestContextMap extends Record<string, RPCContextMap.Any>,
315
+ Middleware extends Context.Tag<MiddlewareMakerId, any> & RpcMiddleware.TagClassAny & {
316
+ readonly requestContext: RequestContextTag<RequestContextMap>
317
+ readonly requestContextMap: RequestContextMap
318
+ }
319
+ >(
320
+ middleware: Middleware
321
+ ) =>
322
+ <R extends Rpc.Any>(group: RpcGroup.RpcGroup<R>) => {
323
+ type RN = AddMiddleware<R, typeof middleware>
324
+ const middlewaredGroup = group.middleware(middleware) as unknown as RpcGroup.RpcGroup<RN>
325
+ const toLayerOriginal = middlewaredGroup.toLayer.bind(middlewaredGroup)
326
+ return Object.assign(middlewaredGroup, {
327
+ toLayerDynamic: <
328
+ Handlers extends HandlersFrom<RN>,
329
+ EX = never,
330
+ RX = never
331
+ >(
332
+ build:
333
+ | Handlers
334
+ | Effect.Effect<Handlers, EX, RX>
335
+ ): Layer.Layer<
336
+ Rpc.ToHandler<RN>,
337
+ EX,
338
+ | Exclude<RX, Scope>
339
+ | HandlersContext<RN, Handlers>
340
+ > => {
341
+ return toLayerOriginal(build as any) as any // ??
342
+ }
343
+ })
344
+ }
345
+
346
+ // customized versions to handle dynamically eliminated context.
347
+ export type HandlersContext<Rpcs extends Rpc.Any, Handlers> = keyof Handlers extends infer K
348
+ ? K extends keyof Handlers & string ? HandlerContext<Rpcs, K, Handlers[K]> : never
349
+ : never
350
+
351
+ export type HandlerContext<Rpcs extends Rpc.Any, K extends Rpcs["_tag"], Handler> = [Rpc.IsStream<Rpcs, K>] extends
352
+ [true] ? Handler extends (...args: any) =>
353
+ | Stream.Stream<infer _A, infer _E, infer _R>
354
+ | Rpc.Fork<Stream.Stream<infer _A, infer _E, infer _R>>
355
+ | Effect.Effect<
356
+ ReadonlyMailbox<infer _A, infer _E>,
357
+ infer _EX,
358
+ infer _R
359
+ >
360
+ | Rpc.Fork<
361
+ Effect.Effect<
362
+ ReadonlyMailbox<infer _A, infer _E>,
363
+ infer _EX,
364
+ infer _R
365
+ >
366
+ > ? Exclude<ExcludeProvides<_R, Rpcs, K>, Scope>
367
+ : never
368
+ : Handler extends (
369
+ ...args: any
370
+ ) => Effect.Effect<infer _A, infer _E, infer _R> | Rpc.Fork<Effect.Effect<infer _A, infer _E, infer _R>>
371
+ ? ExcludeProvides<_R, Rpcs, K>
372
+ : never
373
+
374
+ // new
375
+ export type ExtractDynamicallyProvides<R extends Rpc.Any, Tag extends string> = R extends
376
+ Rpc.Rpc<Tag, infer _Payload, infer _Success, infer _Error, infer _Middleware> ? _Middleware extends {
377
+ readonly requestContextMap: infer _RC
378
+ } ? _RC extends Record<string, RPCContextMap.Any> // ? GetEffectContext<_RC, { allowAnonymous: false }>
379
+ ? R extends { readonly config: infer _C } ? GetEffectContext<_RC, _C>
380
+ : GetEffectContext<_RC, {}>
381
+ : never
382
+ : never
383
+ : never
384
+
385
+ export type ExtractProvides<R extends Rpc.Any, Tag extends string> = R extends
386
+ Rpc.Rpc<Tag, infer _Payload, infer _Success, infer _Error, infer _Middleware> ? _Middleware extends {
387
+ readonly provides: Context.Tag<infer _I, infer _S>
388
+ } ? _I
389
+ : never
390
+ : never
391
+
392
+ export type ExcludeProvides<Env, R extends Rpc.Any, Tag extends string> = Exclude<
393
+ Env,
394
+ // customisation is down here.
395
+ ExtractProvides<R, Tag> | ExtractDynamicallyProvides<R, Tag>
396
+ >
397
+
398
+ //
399
+ export interface MiddlewareMakerId {
400
+ readonly _id: unique symbol
401
+ }
402
+
403
+ // TODO: actually end up with [Tag<A, A>, Tag<B, B>, ...] once `provides: []` is implemented
404
+ export type MakeTags<A> = Context.Tag<A, A>