typespec-rust-emitter 0.1.0 → 0.2.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.
- package/.qwen/settings.json +11 -0
- package/CHANGELOG.md +48 -0
- package/DEV.md +81 -0
- package/QWEN.md +216 -0
- package/README.md +152 -3
- package/dist/src/emitter.js +578 -8
- package/dist/src/emitter.js.map +1 -1
- package/example/output-rust/Cargo.lock +617 -1
- package/example/output-rust/Cargo.toml +8 -1
- package/example/output-rust/src/generated/mod.rs +2 -1
- package/example/output-rust/src/generated/server.rs +712 -0
- package/example/output-rust/src/generated/types.rs +18 -41
- package/justfile +4 -1
- package/package.json +8 -3
- package/src/emitter.ts +758 -8
package/src/emitter.ts
CHANGED
|
@@ -21,6 +21,8 @@ import {
|
|
|
21
21
|
StringLiteral,
|
|
22
22
|
DecoratorContext,
|
|
23
23
|
getNamespaceFullName,
|
|
24
|
+
Operation,
|
|
25
|
+
getTags,
|
|
24
26
|
} from "@typespec/compiler";
|
|
25
27
|
|
|
26
28
|
export interface RustEmitterOptions {
|
|
@@ -73,6 +75,716 @@ export function $rustDerives(
|
|
|
73
75
|
}
|
|
74
76
|
}
|
|
75
77
|
|
|
78
|
+
type HttpMethod = "GET" | "POST" | "PUT" | "PATCH" | "DELETE" | "HEAD";
|
|
79
|
+
|
|
80
|
+
interface OperationInfo {
|
|
81
|
+
name: string;
|
|
82
|
+
method: HttpMethod;
|
|
83
|
+
path: string;
|
|
84
|
+
tags: string[];
|
|
85
|
+
parameters: ParameterInfo[];
|
|
86
|
+
body: ModelProperty | undefined;
|
|
87
|
+
responses: ResponseInfo[];
|
|
88
|
+
doc: string | undefined;
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
interface ParameterInfo {
|
|
92
|
+
name: string;
|
|
93
|
+
rustName: string;
|
|
94
|
+
rustType: string;
|
|
95
|
+
location: "path" | "query" | "header" | "cookie";
|
|
96
|
+
required: boolean;
|
|
97
|
+
optional?: boolean;
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
interface ResponseInfo {
|
|
101
|
+
statusCode: number;
|
|
102
|
+
bodyType: string | undefined;
|
|
103
|
+
bodyDescription: string | undefined;
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
function getDecoratorName(decorator: any): string {
|
|
107
|
+
if (!decorator) return "";
|
|
108
|
+
if (typeof decorator !== "object") return "";
|
|
109
|
+
|
|
110
|
+
if (decorator.node?.target?.sv) {
|
|
111
|
+
return decorator.node.target.sv;
|
|
112
|
+
}
|
|
113
|
+
if (
|
|
114
|
+
decorator.node?.target?.kind === "Identifier" &&
|
|
115
|
+
decorator.node?.target?.sv
|
|
116
|
+
) {
|
|
117
|
+
return decorator.node.target.sv;
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
return "";
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
function getDecoratorArgValue(decorator: any, index: number = 0): string {
|
|
124
|
+
if (!decorator) return "";
|
|
125
|
+
const args = decorator.args || decorator.arguments;
|
|
126
|
+
if (!args) return "";
|
|
127
|
+
const arg = args[index];
|
|
128
|
+
if (arg?.jsValue !== undefined) return String(arg.jsValue);
|
|
129
|
+
if (arg?.value !== undefined) return String(arg.value);
|
|
130
|
+
return "";
|
|
131
|
+
}
|
|
132
|
+
|
|
133
|
+
function getHttpMethod(
|
|
134
|
+
_program: Program,
|
|
135
|
+
operation: Operation,
|
|
136
|
+
): HttpMethod | undefined {
|
|
137
|
+
const decorators = (operation as any).decorators;
|
|
138
|
+
if (!decorators) return undefined;
|
|
139
|
+
|
|
140
|
+
for (const key of Object.keys(decorators)) {
|
|
141
|
+
const decorator = decorators[key];
|
|
142
|
+
const name = getDecoratorName(decorator);
|
|
143
|
+
if (name === "get") return "GET";
|
|
144
|
+
if (name === "post") return "POST";
|
|
145
|
+
if (name === "put") return "PUT";
|
|
146
|
+
if (name === "patch") return "PATCH";
|
|
147
|
+
if (name === "delete") return "DELETE";
|
|
148
|
+
if (name === "head") return "HEAD";
|
|
149
|
+
}
|
|
150
|
+
return undefined;
|
|
151
|
+
}
|
|
152
|
+
|
|
153
|
+
function getRoute(_program: Program, target: Namespace | Operation): string {
|
|
154
|
+
const decorators = (target as any).decorators;
|
|
155
|
+
if (!decorators) return "";
|
|
156
|
+
|
|
157
|
+
for (const key of Object.keys(decorators)) {
|
|
158
|
+
const decorator = decorators[key];
|
|
159
|
+
const name = getDecoratorName(decorator);
|
|
160
|
+
if (name === "route") {
|
|
161
|
+
return getDecoratorArgValue(decorator, 0);
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
return "";
|
|
165
|
+
}
|
|
166
|
+
|
|
167
|
+
function hasAuthDecorator(operation: Operation): boolean {
|
|
168
|
+
const decorators = (operation as any).decorators;
|
|
169
|
+
if (!decorators) return false;
|
|
170
|
+
|
|
171
|
+
for (const key of Object.keys(decorators)) {
|
|
172
|
+
const decorator = decorators[key];
|
|
173
|
+
const name = getDecoratorName(decorator);
|
|
174
|
+
if (name === "useAuth") {
|
|
175
|
+
return true;
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
return false;
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
function getOperationParameters(
|
|
182
|
+
program: Program,
|
|
183
|
+
operation: Operation,
|
|
184
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
185
|
+
): ParameterInfo[] {
|
|
186
|
+
const params: ParameterInfo[] = [];
|
|
187
|
+
const model = operation.parameters;
|
|
188
|
+
|
|
189
|
+
for (const [propName, prop] of model.properties) {
|
|
190
|
+
const decorators = (prop as any).decorators;
|
|
191
|
+
let location: "path" | "query" | "header" | "cookie" = "query";
|
|
192
|
+
let rustName = toRustIdent(propName);
|
|
193
|
+
|
|
194
|
+
if (decorators) {
|
|
195
|
+
if (decorators["$path"]) {
|
|
196
|
+
location = "path";
|
|
197
|
+
} else if (decorators["$query"]) {
|
|
198
|
+
location = "query";
|
|
199
|
+
} else if (decorators["$header"]) {
|
|
200
|
+
location = "header";
|
|
201
|
+
const headerDec = decorators["$header"];
|
|
202
|
+
if (headerDec?.value) {
|
|
203
|
+
rustName = headerDec.value;
|
|
204
|
+
}
|
|
205
|
+
} else if (decorators["$cookie"]) {
|
|
206
|
+
location = "cookie";
|
|
207
|
+
}
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
const { type: rustType } = getRustTypeForProperty(
|
|
211
|
+
prop.type,
|
|
212
|
+
program,
|
|
213
|
+
anonymousEnums,
|
|
214
|
+
);
|
|
215
|
+
|
|
216
|
+
params.push({
|
|
217
|
+
name: propName,
|
|
218
|
+
rustName,
|
|
219
|
+
rustType,
|
|
220
|
+
location,
|
|
221
|
+
required: !prop.optional,
|
|
222
|
+
optional: prop.optional,
|
|
223
|
+
});
|
|
224
|
+
}
|
|
225
|
+
|
|
226
|
+
return params;
|
|
227
|
+
}
|
|
228
|
+
|
|
229
|
+
function getOperationBody(operation: Operation): ModelProperty | undefined {
|
|
230
|
+
for (const [_propName, prop] of operation.parameters.properties) {
|
|
231
|
+
const decorators = (prop as any).decorators;
|
|
232
|
+
if (decorators?.["$body"] || decorators?.["$bodyRoot"]) {
|
|
233
|
+
return prop;
|
|
234
|
+
}
|
|
235
|
+
}
|
|
236
|
+
return undefined;
|
|
237
|
+
}
|
|
238
|
+
|
|
239
|
+
function getOperationResponses(
|
|
240
|
+
program: Program,
|
|
241
|
+
operation: Operation,
|
|
242
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
243
|
+
): ResponseInfo[] {
|
|
244
|
+
const responses: ResponseInfo[] = [];
|
|
245
|
+
const returnType = operation.returnType;
|
|
246
|
+
|
|
247
|
+
if (returnType.kind === "Union") {
|
|
248
|
+
const union = returnType as Union;
|
|
249
|
+
for (const variant of union.variants.values()) {
|
|
250
|
+
const statusCode = getStatusCode(variant);
|
|
251
|
+
const bodyInfo = getBodyFromResponse(variant, program, anonymousEnums);
|
|
252
|
+
responses.push({
|
|
253
|
+
statusCode,
|
|
254
|
+
bodyType: bodyInfo.type,
|
|
255
|
+
bodyDescription: bodyInfo.description,
|
|
256
|
+
});
|
|
257
|
+
}
|
|
258
|
+
} else if (returnType.kind === "Model") {
|
|
259
|
+
const model = returnType as Model;
|
|
260
|
+
for (const [propName, prop] of model.properties) {
|
|
261
|
+
if (propName === "body") {
|
|
262
|
+
const { type: rustType } = getRustTypeForProperty(
|
|
263
|
+
prop.type,
|
|
264
|
+
program,
|
|
265
|
+
anonymousEnums,
|
|
266
|
+
);
|
|
267
|
+
responses.push({
|
|
268
|
+
statusCode: 200,
|
|
269
|
+
bodyType: rustType,
|
|
270
|
+
bodyDescription: getDoc(program, prop),
|
|
271
|
+
});
|
|
272
|
+
}
|
|
273
|
+
}
|
|
274
|
+
}
|
|
275
|
+
|
|
276
|
+
return responses;
|
|
277
|
+
}
|
|
278
|
+
|
|
279
|
+
function getStatusCode(variant: { type: Type }): number {
|
|
280
|
+
if (variant.type.kind === "Model") {
|
|
281
|
+
const model = variant.type as Model;
|
|
282
|
+
for (const [propName, prop] of model.properties) {
|
|
283
|
+
if (propName === "statusCode") {
|
|
284
|
+
const typeAny = prop.type as any;
|
|
285
|
+
if (typeAny.value !== undefined) {
|
|
286
|
+
return typeAny.value as number;
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
}
|
|
290
|
+
}
|
|
291
|
+
return 200;
|
|
292
|
+
}
|
|
293
|
+
|
|
294
|
+
function getBodyFromResponse(
|
|
295
|
+
variant: { type: Type },
|
|
296
|
+
program: Program,
|
|
297
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
298
|
+
): { type: string | undefined; description: string | undefined } {
|
|
299
|
+
if (variant.type.kind === "Model") {
|
|
300
|
+
const model = variant.type as Model;
|
|
301
|
+
for (const [propName, prop] of model.properties) {
|
|
302
|
+
if (propName === "body") {
|
|
303
|
+
const { type: rustType } = getRustTypeForProperty(
|
|
304
|
+
prop.type,
|
|
305
|
+
program,
|
|
306
|
+
anonymousEnums,
|
|
307
|
+
);
|
|
308
|
+
return { type: rustType, description: getDoc(program, prop) };
|
|
309
|
+
}
|
|
310
|
+
}
|
|
311
|
+
}
|
|
312
|
+
return { type: undefined, description: undefined };
|
|
313
|
+
}
|
|
314
|
+
|
|
315
|
+
function getAllOperations(
|
|
316
|
+
program: Program,
|
|
317
|
+
): { namespace: Namespace; operations: Operation[] }[] {
|
|
318
|
+
const seenNamespaces = new Set<string>();
|
|
319
|
+
const namespaceOps = new Map<string, { ns: Namespace; ops: Operation[] }>();
|
|
320
|
+
|
|
321
|
+
navigateProgram(program, {
|
|
322
|
+
namespace(ns: Namespace) {
|
|
323
|
+
const route = getRoute(program, ns);
|
|
324
|
+
if (!route) return;
|
|
325
|
+
if (!seenNamespaces.has(ns.name)) {
|
|
326
|
+
seenNamespaces.add(ns.name);
|
|
327
|
+
namespaceOps.set(ns.name, { ns, ops: [] });
|
|
328
|
+
}
|
|
329
|
+
},
|
|
330
|
+
operation(op: Operation) {
|
|
331
|
+
const method = getHttpMethod(program, op);
|
|
332
|
+
if (!method) return;
|
|
333
|
+
const ns = op.namespace;
|
|
334
|
+
if (!ns) return;
|
|
335
|
+
if (!seenNamespaces.has(ns.name)) {
|
|
336
|
+
seenNamespaces.add(ns.name);
|
|
337
|
+
namespaceOps.set(ns.name, { ns, ops: [] });
|
|
338
|
+
}
|
|
339
|
+
const entry = namespaceOps.get(ns.name);
|
|
340
|
+
if (entry) {
|
|
341
|
+
entry.ops.push(op);
|
|
342
|
+
}
|
|
343
|
+
},
|
|
344
|
+
});
|
|
345
|
+
|
|
346
|
+
const result: { namespace: Namespace; operations: Operation[] }[] = [];
|
|
347
|
+
for (const [, entry] of namespaceOps) {
|
|
348
|
+
if (entry.ops.length > 0) {
|
|
349
|
+
result.push({ namespace: entry.ns, operations: entry.ops });
|
|
350
|
+
}
|
|
351
|
+
}
|
|
352
|
+
return result;
|
|
353
|
+
}
|
|
354
|
+
|
|
355
|
+
function buildFullPath(namespaceRoute: string, operationRoute: string): string {
|
|
356
|
+
let fullPath = namespaceRoute + operationRoute;
|
|
357
|
+
fullPath = fullPath.replace(/\/+/g, "/");
|
|
358
|
+
if (fullPath.length > 1 && fullPath.endsWith("/")) {
|
|
359
|
+
fullPath = fullPath.slice(0, -1);
|
|
360
|
+
}
|
|
361
|
+
return fullPath;
|
|
362
|
+
}
|
|
363
|
+
|
|
364
|
+
function emitOperationInfo(
|
|
365
|
+
program: Program,
|
|
366
|
+
op: Operation,
|
|
367
|
+
nsRoute: string,
|
|
368
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
369
|
+
): OperationInfo | undefined {
|
|
370
|
+
const method = getHttpMethod(program, op);
|
|
371
|
+
if (!method) return undefined;
|
|
372
|
+
|
|
373
|
+
const opRoute = getRoute(program, op);
|
|
374
|
+
const fullPath = buildFullPath(nsRoute, opRoute);
|
|
375
|
+
const tags = getTags(program, op) ?? [];
|
|
376
|
+
const params = getOperationParameters(program, op, anonymousEnums);
|
|
377
|
+
const body = getOperationBody(op);
|
|
378
|
+
const responses = getOperationResponses(program, op, anonymousEnums);
|
|
379
|
+
const doc = getDoc(program, op);
|
|
380
|
+
|
|
381
|
+
const opName = op.name.replace(/[^a-zA-Z0-9_]/g, "_");
|
|
382
|
+
|
|
383
|
+
return {
|
|
384
|
+
name: opName,
|
|
385
|
+
method,
|
|
386
|
+
path: fullPath,
|
|
387
|
+
tags,
|
|
388
|
+
parameters: params,
|
|
389
|
+
body: body,
|
|
390
|
+
responses,
|
|
391
|
+
doc,
|
|
392
|
+
};
|
|
393
|
+
}
|
|
394
|
+
|
|
395
|
+
function getStatusVariantName(statusCode: number): string {
|
|
396
|
+
const statusNames: Record<number, string> = {
|
|
397
|
+
200: "Ok",
|
|
398
|
+
201: "Created",
|
|
399
|
+
202: "Accepted",
|
|
400
|
+
204: "NoContent",
|
|
401
|
+
400: "BadRequest",
|
|
402
|
+
401: "Unauthorized",
|
|
403
|
+
403: "Forbidden",
|
|
404
|
+
404: "NotFound",
|
|
405
|
+
409: "Conflict",
|
|
406
|
+
500: "InternalServerError",
|
|
407
|
+
};
|
|
408
|
+
return statusNames[statusCode] || `Status${statusCode}`;
|
|
409
|
+
}
|
|
410
|
+
|
|
411
|
+
function getHttpStatusCode(statusCode: number): string {
|
|
412
|
+
const statusCodes: Record<number, string> = {
|
|
413
|
+
200: "StatusCode::OK",
|
|
414
|
+
201: "StatusCode::CREATED",
|
|
415
|
+
202: "StatusCode::ACCEPTED",
|
|
416
|
+
204: "StatusCode::NO_CONTENT",
|
|
417
|
+
400: "StatusCode::BAD_REQUEST",
|
|
418
|
+
401: "StatusCode::UNAUTHORIZED",
|
|
419
|
+
403: "StatusCode::FORBIDDEN",
|
|
420
|
+
404: "StatusCode::NOT_FOUND",
|
|
421
|
+
409: "StatusCode::CONFLICT",
|
|
422
|
+
500: "StatusCode::INTERNAL_SERVER_ERROR",
|
|
423
|
+
};
|
|
424
|
+
return statusCodes[statusCode] || `StatusCode::from_u16(${statusCode})`;
|
|
425
|
+
}
|
|
426
|
+
|
|
427
|
+
function generateServerTrait(
|
|
428
|
+
program: Program,
|
|
429
|
+
namespaceGroups: { namespace: Namespace; operations: Operation[] }[],
|
|
430
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
431
|
+
): string {
|
|
432
|
+
const parts: string[] = [];
|
|
433
|
+
|
|
434
|
+
parts.push(`use super::types::*;
|
|
435
|
+
use async_trait::async_trait;
|
|
436
|
+
use axum::{http::StatusCode, Json};
|
|
437
|
+
use eyre::Result;
|
|
438
|
+
|
|
439
|
+
#[async_trait]
|
|
440
|
+
pub trait Server: Send + Sync {
|
|
441
|
+
type Claims: Send + Sync + 'static;
|
|
442
|
+
|
|
443
|
+
`);
|
|
444
|
+
|
|
445
|
+
for (const group of namespaceGroups) {
|
|
446
|
+
const nsName = toPascalCase(
|
|
447
|
+
group.namespace.name.replace(/[^a-zA-Z0-9_]/g, "_"),
|
|
448
|
+
);
|
|
449
|
+
|
|
450
|
+
for (const op of group.operations) {
|
|
451
|
+
const opInfo = emitOperationInfo(program, op, "", anonymousEnums);
|
|
452
|
+
if (!opInfo) continue;
|
|
453
|
+
|
|
454
|
+
if (opInfo.doc) {
|
|
455
|
+
parts.push(` ${formatDoc(opInfo.doc)}`);
|
|
456
|
+
}
|
|
457
|
+
|
|
458
|
+
const requestName = `${nsName}${toPascalCase(opInfo.name)}Request`;
|
|
459
|
+
const responseName = `${nsName}${toPascalCase(opInfo.name)}Response`;
|
|
460
|
+
const fnName = toRustIdent(`${nsName}_${opInfo.name}`);
|
|
461
|
+
const isProtected = hasAuthDecorator(op);
|
|
462
|
+
|
|
463
|
+
if (isProtected) {
|
|
464
|
+
parts.push(
|
|
465
|
+
` async fn ${fnName}(&self, claims: Self::Claims, request: ${requestName}) -> Result<${responseName}>;`,
|
|
466
|
+
);
|
|
467
|
+
} else {
|
|
468
|
+
parts.push(
|
|
469
|
+
` async fn ${fnName}(&self, request: ${requestName}) -> Result<${responseName}>;`,
|
|
470
|
+
);
|
|
471
|
+
}
|
|
472
|
+
}
|
|
473
|
+
}
|
|
474
|
+
|
|
475
|
+
parts.push("}");
|
|
476
|
+
|
|
477
|
+
return parts.join("\n");
|
|
478
|
+
}
|
|
479
|
+
|
|
480
|
+
function generateRequestStructs(
|
|
481
|
+
program: Program,
|
|
482
|
+
namespaceGroups: { namespace: Namespace; operations: Operation[] }[],
|
|
483
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
484
|
+
): string {
|
|
485
|
+
const parts: string[] = [];
|
|
486
|
+
|
|
487
|
+
for (const group of namespaceGroups) {
|
|
488
|
+
const nsName = toPascalCase(
|
|
489
|
+
group.namespace.name.replace(/[^a-zA-Z0-9_]/g, "_"),
|
|
490
|
+
);
|
|
491
|
+
|
|
492
|
+
for (const op of group.operations) {
|
|
493
|
+
const opInfo = emitOperationInfo(program, op, "", anonymousEnums);
|
|
494
|
+
if (!opInfo) continue;
|
|
495
|
+
|
|
496
|
+
const params = opInfo.parameters;
|
|
497
|
+
const requestName = `${nsName}${toPascalCase(opInfo.name)}Request`;
|
|
498
|
+
|
|
499
|
+
const fields: string[] = [];
|
|
500
|
+
|
|
501
|
+
for (const param of params) {
|
|
502
|
+
const rustType = param.optional
|
|
503
|
+
? `Option<${param.rustType}>`
|
|
504
|
+
: param.rustType;
|
|
505
|
+
fields.push(` #[serde(rename = "${param.name}", flatten)]`);
|
|
506
|
+
fields.push(` pub ${param.rustName}: ${rustType},`);
|
|
507
|
+
}
|
|
508
|
+
|
|
509
|
+
if (opInfo.body) {
|
|
510
|
+
const bodyType = getRustTypeForProperty(
|
|
511
|
+
opInfo.body.type,
|
|
512
|
+
program,
|
|
513
|
+
anonymousEnums,
|
|
514
|
+
);
|
|
515
|
+
fields.push(` #[serde(rename = "body")]`);
|
|
516
|
+
fields.push(` pub body: ${bodyType.type},`);
|
|
517
|
+
}
|
|
518
|
+
|
|
519
|
+
parts.push(`#[derive(Debug, Clone, serde::Deserialize)]
|
|
520
|
+
pub struct ${requestName} {
|
|
521
|
+
${fields.join("\n")}
|
|
522
|
+
}
|
|
523
|
+
`);
|
|
524
|
+
}
|
|
525
|
+
}
|
|
526
|
+
|
|
527
|
+
return parts.join("\n");
|
|
528
|
+
}
|
|
529
|
+
|
|
530
|
+
function generateResponseEnums(
|
|
531
|
+
program: Program,
|
|
532
|
+
namespaceGroups: { namespace: Namespace; operations: Operation[] }[],
|
|
533
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
534
|
+
): string {
|
|
535
|
+
const parts: string[] = [];
|
|
536
|
+
|
|
537
|
+
for (const group of namespaceGroups) {
|
|
538
|
+
const nsName = toPascalCase(
|
|
539
|
+
group.namespace.name.replace(/[^a-zA-Z0-9_]/g, "_"),
|
|
540
|
+
);
|
|
541
|
+
|
|
542
|
+
for (const op of group.operations) {
|
|
543
|
+
const opInfo = emitOperationInfo(program, op, "", anonymousEnums);
|
|
544
|
+
if (!opInfo) continue;
|
|
545
|
+
|
|
546
|
+
if (opInfo.responses.length === 0) continue;
|
|
547
|
+
|
|
548
|
+
const responseName = `${nsName}${toPascalCase(opInfo.name)}Response`;
|
|
549
|
+
|
|
550
|
+
const variants: string[] = [];
|
|
551
|
+
for (const resp of opInfo.responses) {
|
|
552
|
+
const variantName = getStatusVariantName(resp.statusCode);
|
|
553
|
+
if (!resp.bodyType) {
|
|
554
|
+
variants.push(` ${variantName},`);
|
|
555
|
+
} else {
|
|
556
|
+
variants.push(` ${variantName}(Json<${resp.bodyType}>),`);
|
|
557
|
+
}
|
|
558
|
+
}
|
|
559
|
+
parts.push(`pub enum ${responseName} {
|
|
560
|
+
${variants.join("\n")}
|
|
561
|
+
}
|
|
562
|
+
`);
|
|
563
|
+
|
|
564
|
+
parts.push(`impl IntoResponse for ${responseName} {
|
|
565
|
+
fn into_response(self) -> axum::response::Response {
|
|
566
|
+
match self {
|
|
567
|
+
`);
|
|
568
|
+
for (const resp of opInfo.responses) {
|
|
569
|
+
const variantName = getStatusVariantName(resp.statusCode);
|
|
570
|
+
const statusCodeStr = getHttpStatusCode(resp.statusCode);
|
|
571
|
+
if (!resp.bodyType) {
|
|
572
|
+
parts.push(
|
|
573
|
+
` ${responseName}::${variantName} => ${statusCodeStr}.into_response(),`,
|
|
574
|
+
);
|
|
575
|
+
} else {
|
|
576
|
+
parts.push(
|
|
577
|
+
` ${responseName}::${variantName}(body) => (${statusCodeStr}, body).into_response(),`,
|
|
578
|
+
);
|
|
579
|
+
}
|
|
580
|
+
}
|
|
581
|
+
parts.push(` }
|
|
582
|
+
}
|
|
583
|
+
}
|
|
584
|
+
`);
|
|
585
|
+
}
|
|
586
|
+
}
|
|
587
|
+
|
|
588
|
+
return parts.join("\n");
|
|
589
|
+
}
|
|
590
|
+
|
|
591
|
+
function generateRouter(
|
|
592
|
+
program: Program,
|
|
593
|
+
namespaceGroups: { namespace: Namespace; operations: Operation[] }[],
|
|
594
|
+
anonymousEnums: Map<string, AnonymousStringLiteralUnion>,
|
|
595
|
+
): string {
|
|
596
|
+
const handlers: string[] = [];
|
|
597
|
+
const publicRoutes: string[] = [];
|
|
598
|
+
const protectedRoutes: string[] = [];
|
|
599
|
+
const usedMethods = new Set<string>();
|
|
600
|
+
|
|
601
|
+
for (const group of namespaceGroups) {
|
|
602
|
+
const nsRoute = getRoute(program, group.namespace);
|
|
603
|
+
if (!nsRoute) continue;
|
|
604
|
+
|
|
605
|
+
const nsName = toPascalCase(
|
|
606
|
+
group.namespace.name.replace(/[^a-zA-Z0-9_]/g, "_"),
|
|
607
|
+
);
|
|
608
|
+
|
|
609
|
+
for (const op of group.operations) {
|
|
610
|
+
const opInfo = emitOperationInfo(program, op, nsRoute, anonymousEnums);
|
|
611
|
+
if (!opInfo) continue;
|
|
612
|
+
|
|
613
|
+
const method = opInfo.method.toLowerCase();
|
|
614
|
+
usedMethods.add(method);
|
|
615
|
+
|
|
616
|
+
const handlerFnName = toRustIdent(`${nsName}_${opInfo.name}`);
|
|
617
|
+
const traitFnName = handlerFnName;
|
|
618
|
+
const requestName = `${nsName}${toPascalCase(opInfo.name)}Request`;
|
|
619
|
+
const isProtected = hasAuthDecorator(op);
|
|
620
|
+
|
|
621
|
+
const pathParams = opInfo.parameters.filter((p) => p.location === "path");
|
|
622
|
+
const queryParams = opInfo.parameters.filter(
|
|
623
|
+
(p) => p.location === "query",
|
|
624
|
+
);
|
|
625
|
+
const hasPathParams = pathParams.length > 0;
|
|
626
|
+
const hasQueryParams = queryParams.length > 0;
|
|
627
|
+
const hasBody = !!opInfo.body;
|
|
628
|
+
|
|
629
|
+
// Build extractor lines and request construction expression
|
|
630
|
+
const extractorLines: string[] = [];
|
|
631
|
+
let requestExpr = "";
|
|
632
|
+
|
|
633
|
+
if (hasPathParams) {
|
|
634
|
+
const pathTypes = pathParams.map((p) => p.rustType).join(", ");
|
|
635
|
+
const pathFields = pathParams.map((p) => p.rustName).join(", ");
|
|
636
|
+
if (pathParams.length === 1) {
|
|
637
|
+
extractorLines.push(
|
|
638
|
+
` axum::extract::Path(${pathFields}): axum::extract::Path<${pathTypes}>,`,
|
|
639
|
+
);
|
|
640
|
+
} else {
|
|
641
|
+
extractorLines.push(
|
|
642
|
+
` axum::extract::Path((${pathFields})): axum::extract::Path<(${pathTypes})>,`,
|
|
643
|
+
);
|
|
644
|
+
}
|
|
645
|
+
}
|
|
646
|
+
|
|
647
|
+
if (hasQueryParams) {
|
|
648
|
+
extractorLines.push(
|
|
649
|
+
` axum::extract::Query(query): axum::extract::Query<${requestName}>,`,
|
|
650
|
+
);
|
|
651
|
+
} else if (hasBody) {
|
|
652
|
+
extractorLines.push(
|
|
653
|
+
` axum::Json(body): axum::Json<${requestName}Body>,`,
|
|
654
|
+
);
|
|
655
|
+
}
|
|
656
|
+
|
|
657
|
+
if (isProtected) {
|
|
658
|
+
extractorLines.push(
|
|
659
|
+
` axum::Extension(claims): axum::Extension<S::Claims>,`,
|
|
660
|
+
);
|
|
661
|
+
}
|
|
662
|
+
|
|
663
|
+
// Build request struct expression
|
|
664
|
+
if (hasOnlyPathParams(hasPathParams, hasQueryParams, hasBody)) {
|
|
665
|
+
const pathAssignments = pathParams
|
|
666
|
+
.map((p) => `${p.rustName},`)
|
|
667
|
+
.join(" ");
|
|
668
|
+
requestExpr = `${requestName} { ${pathAssignments} }`;
|
|
669
|
+
} else if (hasQueryParams) {
|
|
670
|
+
if (hasPathParams) {
|
|
671
|
+
const pathAssignments = pathParams
|
|
672
|
+
.map((p) => `${p.rustName},`)
|
|
673
|
+
.join(" ");
|
|
674
|
+
requestExpr = `${requestName} { ${pathAssignments} ..query }`;
|
|
675
|
+
} else {
|
|
676
|
+
requestExpr = "query";
|
|
677
|
+
}
|
|
678
|
+
} else if (hasBody) {
|
|
679
|
+
if (hasPathParams) {
|
|
680
|
+
const pathAssignments = pathParams
|
|
681
|
+
.map((p) => `${p.rustName},`)
|
|
682
|
+
.join(" ");
|
|
683
|
+
requestExpr = `${requestName} { ${pathAssignments} body }`;
|
|
684
|
+
} else {
|
|
685
|
+
requestExpr = `${requestName} { body }`;
|
|
686
|
+
}
|
|
687
|
+
} else {
|
|
688
|
+
requestExpr = `${requestName} {}`;
|
|
689
|
+
}
|
|
690
|
+
|
|
691
|
+
// Server method call
|
|
692
|
+
const serverCall = isProtected
|
|
693
|
+
? `service.${traitFnName}(claims, ${requestExpr}).await`
|
|
694
|
+
: `service.${traitFnName}(${requestExpr}).await`;
|
|
695
|
+
|
|
696
|
+
// All handlers use <S> generics, Claims is now an associated type
|
|
697
|
+
let handlerCode = `pub async fn ${handlerFnName}_handler<S>(
|
|
698
|
+
axum::extract::State(service): axum::extract::State<S>,
|
|
699
|
+
${extractorLines.join("\n")}
|
|
700
|
+
) -> impl axum::response::IntoResponse
|
|
701
|
+
where
|
|
702
|
+
S: Server + Clone + Send + Sync + 'static,
|
|
703
|
+
S::Claims: Send + Sync + Clone + 'static,
|
|
704
|
+
{
|
|
705
|
+
let result = ${serverCall};
|
|
706
|
+
match result {
|
|
707
|
+
Ok(response) => response.into_response(),
|
|
708
|
+
Err(e) => (
|
|
709
|
+
axum::http::StatusCode::INTERNAL_SERVER_ERROR,
|
|
710
|
+
format!("Internal error: {e}"),
|
|
711
|
+
)
|
|
712
|
+
.into_response(),
|
|
713
|
+
}
|
|
714
|
+
}`;
|
|
715
|
+
|
|
716
|
+
handlers.push(handlerCode);
|
|
717
|
+
|
|
718
|
+
let routePath = `"${opInfo.path}"`;
|
|
719
|
+
let routeStmt = "";
|
|
720
|
+
if (isProtected) {
|
|
721
|
+
routeStmt = `.route(${routePath}, ${method}(${handlerFnName}_handler::<S>))`;
|
|
722
|
+
protectedRoutes.push(routeStmt);
|
|
723
|
+
} else {
|
|
724
|
+
routeStmt = `.route(${routePath}, ${method}(${handlerFnName}_handler::<S>))`;
|
|
725
|
+
publicRoutes.push(routeStmt);
|
|
726
|
+
}
|
|
727
|
+
}
|
|
728
|
+
}
|
|
729
|
+
|
|
730
|
+
const methodImports = Array.from(usedMethods).sort().join(", ");
|
|
731
|
+
|
|
732
|
+
const routerBody = buildRouterBody(publicRoutes, protectedRoutes);
|
|
733
|
+
|
|
734
|
+
const parts: string[] = [];
|
|
735
|
+
parts.push(`use axum::response::IntoResponse;
|
|
736
|
+
use axum::routing::{${methodImports}};
|
|
737
|
+
use axum::Router;
|
|
738
|
+
|
|
739
|
+
`);
|
|
740
|
+
parts.push(handlers.join("\n\n"));
|
|
741
|
+
parts.push(`
|
|
742
|
+
pub fn create_router<S, M>(service: S, middleware: M) -> Router
|
|
743
|
+
where
|
|
744
|
+
S: Server + Clone + Send + Sync + 'static,
|
|
745
|
+
S::Claims: Send + Sync + Clone + 'static,
|
|
746
|
+
M: FnOnce(Router<S>) -> Router<S> + Clone + Send + Sync + 'static,
|
|
747
|
+
{
|
|
748
|
+
${routerBody}
|
|
749
|
+
}`);
|
|
750
|
+
|
|
751
|
+
return parts.join("\n");
|
|
752
|
+
}
|
|
753
|
+
|
|
754
|
+
function hasOnlyPathParams(
|
|
755
|
+
hasPath: boolean,
|
|
756
|
+
hasQuery: boolean,
|
|
757
|
+
hasBody: boolean,
|
|
758
|
+
): boolean {
|
|
759
|
+
return hasPath && !hasQuery && !hasBody;
|
|
760
|
+
}
|
|
761
|
+
|
|
762
|
+
function buildRouterBody(
|
|
763
|
+
publicRoutes: string[],
|
|
764
|
+
protectedRoutes: string[],
|
|
765
|
+
): string {
|
|
766
|
+
const lines: string[] = [];
|
|
767
|
+
lines.push(" let mut router = Router::new();");
|
|
768
|
+
if (publicRoutes.length > 0) {
|
|
769
|
+
lines.push(" let public = Router::new()");
|
|
770
|
+
for (const r of publicRoutes) {
|
|
771
|
+
lines.push(` ${r.trim()}`);
|
|
772
|
+
}
|
|
773
|
+
lines.push(" ;");
|
|
774
|
+
lines.push(" router = router.merge(public);");
|
|
775
|
+
}
|
|
776
|
+
if (protectedRoutes.length > 0) {
|
|
777
|
+
lines.push(" let protected = Router::new()");
|
|
778
|
+
for (const r of protectedRoutes) {
|
|
779
|
+
lines.push(` ${r.trim()}`);
|
|
780
|
+
}
|
|
781
|
+
lines.push(" ;");
|
|
782
|
+
lines.push(" router = router.merge(middleware(protected));");
|
|
783
|
+
}
|
|
784
|
+
lines.push(" router.with_state(service)");
|
|
785
|
+
return lines.join("\n");
|
|
786
|
+
}
|
|
787
|
+
|
|
76
788
|
const scalarToRust: Record<string, string> = {
|
|
77
789
|
string: "String",
|
|
78
790
|
int8: "i8",
|
|
@@ -403,10 +1115,8 @@ function emitModel(
|
|
|
403
1115
|
parts.push('#[error("{code}: {message}")]');
|
|
404
1116
|
}
|
|
405
1117
|
|
|
406
|
-
parts.push(`pub struct ${name}`);
|
|
407
|
-
|
|
408
1118
|
if (allProps.size > 0) {
|
|
409
|
-
|
|
1119
|
+
const fields: string[] = [];
|
|
410
1120
|
for (const [propName, prop] of allProps) {
|
|
411
1121
|
const doc = getDoc(program, prop);
|
|
412
1122
|
const { type: rustType } = getRustTypeForProperty(
|
|
@@ -420,20 +1130,23 @@ function emitModel(
|
|
|
420
1130
|
propName !== fieldName ? `#[serde(rename = "${propName}")]` : "";
|
|
421
1131
|
|
|
422
1132
|
if (doc) {
|
|
423
|
-
|
|
1133
|
+
fields.push(` ${formatDoc(doc)}`);
|
|
424
1134
|
}
|
|
425
1135
|
if (serdeRename) {
|
|
426
|
-
|
|
1136
|
+
fields.push(` ${serdeRename}`);
|
|
427
1137
|
}
|
|
428
1138
|
if (optional) {
|
|
429
|
-
|
|
1139
|
+
fields.push(` #[serde(skip_serializing_if = "Option::is_none")]`);
|
|
430
1140
|
}
|
|
431
|
-
|
|
1141
|
+
fields.push(
|
|
432
1142
|
` pub ${fieldName}: ${optional ? `Option<${rustType}>` : rustType},`,
|
|
433
1143
|
);
|
|
434
1144
|
}
|
|
435
|
-
parts.push(
|
|
1145
|
+
parts.push(`pub struct ${name} {
|
|
1146
|
+
${fields.join("\n")}
|
|
1147
|
+
}`);
|
|
436
1148
|
} else {
|
|
1149
|
+
parts.push(`pub struct ${name}`);
|
|
437
1150
|
parts.push("(());");
|
|
438
1151
|
}
|
|
439
1152
|
|
|
@@ -676,10 +1389,47 @@ export async function $onEmit(
|
|
|
676
1389
|
content.push("");
|
|
677
1390
|
}
|
|
678
1391
|
|
|
1392
|
+
const namespaceGroups = getAllOperations(context.program);
|
|
1393
|
+
|
|
679
1394
|
const outputContent = content.join("\n");
|
|
680
1395
|
|
|
681
1396
|
await emitFile(context.program, {
|
|
682
1397
|
path: resolvePath(context.emitterOutputDir, "types.rs"),
|
|
683
1398
|
content: outputContent,
|
|
684
1399
|
});
|
|
1400
|
+
|
|
1401
|
+
if (namespaceGroups.length > 0) {
|
|
1402
|
+
const serverTrait = generateServerTrait(
|
|
1403
|
+
context.program,
|
|
1404
|
+
namespaceGroups,
|
|
1405
|
+
anonymousEnums,
|
|
1406
|
+
);
|
|
1407
|
+
const requestStructs = generateRequestStructs(
|
|
1408
|
+
context.program,
|
|
1409
|
+
namespaceGroups,
|
|
1410
|
+
anonymousEnums,
|
|
1411
|
+
);
|
|
1412
|
+
const responseEnums = generateResponseEnums(
|
|
1413
|
+
context.program,
|
|
1414
|
+
namespaceGroups,
|
|
1415
|
+
anonymousEnums,
|
|
1416
|
+
);
|
|
1417
|
+
const router = generateRouter(
|
|
1418
|
+
context.program,
|
|
1419
|
+
namespaceGroups,
|
|
1420
|
+
anonymousEnums,
|
|
1421
|
+
);
|
|
1422
|
+
|
|
1423
|
+
const serverContent = [
|
|
1424
|
+
serverTrait,
|
|
1425
|
+
requestStructs,
|
|
1426
|
+
responseEnums,
|
|
1427
|
+
router,
|
|
1428
|
+
].join("\n");
|
|
1429
|
+
|
|
1430
|
+
await emitFile(context.program, {
|
|
1431
|
+
path: resolvePath(context.emitterOutputDir, "server.rs"),
|
|
1432
|
+
content: serverContent,
|
|
1433
|
+
});
|
|
1434
|
+
}
|
|
685
1435
|
}
|