vox_codegen/targets/typescript/
server.rs

1//! TypeScript server/handler generation.
2//!
3//! Generates the handler interface and a Dispatcher class that routes calls
4//! to handler methods. All encode/decode is handled by the runtime via the
5//! service descriptor — no serialization code in generated output.
6
7use heck::{ToLowerCamelCase, ToUpperCamelCase};
8use vox_types::{ServiceDescriptor, ShapeKind, classify_shape};
9
10use super::types::{ts_type_server_arg, ts_type_server_return};
11
12/// Generate handler interface (for implementing the service).
13///
14/// r[impl rpc.channel.binding] - Handler binds channels in args.
15pub fn generate_handler_interface(service: &ServiceDescriptor) -> String {
16    let mut out = String::new();
17    let service_name = service.service_name.to_upper_camel_case();
18
19    out.push_str(&format!("// Handler interface for {service_name}\n"));
20    out.push_str(&format!("export interface {service_name}Handler {{\n"));
21
22    for method in service.methods {
23        let method_name = method.method_name.to_lower_camel_case();
24        let args = method
25            .args
26            .iter()
27            .map(|a| {
28                format!(
29                    "{}: {}",
30                    a.name.to_lower_camel_case(),
31                    ts_type_server_arg(a.shape)
32                )
33            })
34            .collect::<Vec<_>>()
35            .join(", ");
36        let ret_ty = ts_type_server_return(method.return_shape);
37
38        out.push_str(&format!(
39            "  {method_name}({args}): Promise<{ret_ty}> | {ret_ty};\n"
40        ));
41    }
42
43    out.push_str("}\n\n");
44    out
45}
46
47/// Generate the Dispatcher class.
48///
49/// Implements `Dispatcher` from vox-core:
50/// - `getDescriptor()` returns the service descriptor
51/// - `dispatch(context, method, args, call)` routes by method ID and calls handler methods
52///
53/// The runtime handles all arg decoding (using method.args tuple schema) and
54/// response encoding (using method.result schema via call.reply/replyErr).
55/// Generated dispatch code only does type casts and handler invocation.
56pub fn generate_dispatcher_class(service: &ServiceDescriptor) -> String {
57    use crate::render::hex_u64;
58
59    let mut out = String::new();
60    let service_name = service.service_name.to_upper_camel_case();
61    let service_name_lower = service.service_name.to_lower_camel_case();
62
63    out.push_str(&format!("// Dispatcher for {service_name}\n"));
64    out.push_str(&format!(
65        "export class {service_name}Dispatcher implements Dispatcher {{\n"
66    ));
67    out.push_str(&format!(
68        "  private readonly handler: {service_name}Handler;\n\n"
69    ));
70    out.push_str(&format!(
71        "  constructor(handler: {service_name}Handler) {{\n"
72    ));
73    out.push_str("    this.handler = handler;\n");
74    out.push_str("  }\n\n");
75
76    // getDescriptor()
77    out.push_str("  getDescriptor(): ServiceDescriptor {\n");
78    out.push_str(&format!("    return {service_name_lower}_descriptor;\n"));
79    out.push_str("  }\n\n");
80
81    // dispatch()
82    out.push_str(
83        "  async dispatch(_context: RequestContext, method: MethodDescriptor, args: unknown[], call: VoxCall): Promise<void> {\n",
84    );
85
86    let mut first = true;
87    for method in service.methods {
88        let method_name = method.method_name.to_lower_camel_case();
89        // r[impl rpc.method-id]
90        let id = crate::method_id(method);
91        let is_fallible = matches!(
92            classify_shape(method.return_shape),
93            ShapeKind::Result { .. }
94        );
95
96        // Build typed arg list from args array
97        let typed_args: Vec<_> = method
98            .args
99            .iter()
100            .enumerate()
101            .map(|(i, a)| format!("args[{i}] as {}", ts_type_server_arg(a.shape)))
102            .collect();
103
104        let keyword = if first { "if" } else { "} else if" };
105        first = false;
106
107        out.push_str(&format!(
108            "    {keyword} (method.id === {}n) {{\n",
109            hex_u64(id)
110        ));
111        out.push_str("      try {\n");
112        out.push_str(&format!(
113            "        const result = await this.handler.{method_name}({});\n",
114            typed_args.join(", ")
115        ));
116
117        if is_fallible {
118            out.push_str("        if (result.ok) call.reply(result.value); else call.replyErr(result.error);\n");
119        } else {
120            out.push_str("        call.reply(result);\n");
121        }
122
123        out.push_str("      } catch (error) {\n");
124        out.push_str(
125            "        call.replyInternalError(error instanceof Error ? error.message : String(error));\n",
126        );
127        out.push_str("      }\n");
128    }
129
130    if !first {
131        out.push_str("    }\n");
132    }
133
134    out.push_str("  }\n");
135    out.push_str("}\n\n");
136    out
137}
138
139/// Generate complete server code (handler interface + dispatcher class).
140pub fn generate_server(service: &ServiceDescriptor) -> String {
141    let mut out = String::new();
142    out.push_str(&generate_handler_interface(service));
143    out.push_str(&generate_dispatcher_class(service));
144    out
145}
vox_codegen/targets/typescript/server.rs

vox_codegen/targets/typescript/
server.rs
