@@ -18,12 +18,16 @@ import {
1818 factory ,
1919} from "typescript" ;
2020
21+ import { create , fromBinary , toBinary } from "@bufbuild/protobuf" ;
2122import {
2223 GenerateRequest ,
24+ GenerateRequestSchema ,
2325 GenerateResponse ,
26+ GenerateResponseSchema ,
2427 Parameter ,
2528 Column ,
2629 File ,
30+ FileSchema ,
2731 Query ,
2832} from "./gen/plugin/codegen_pb" ;
2933
@@ -200,7 +204,7 @@ ${query.text}`
200204 }
201205 if ( nodes ) {
202206 files . push (
203- new File ( {
207+ create ( FileSchema , {
204208 name : `${ filename . replace ( "." , "_" ) } .ts` ,
205209 contents : new TextEncoder ( ) . encode ( printNode ( nodes ) ) ,
206210 } )
@@ -209,15 +213,15 @@ ${query.text}`
209213 }
210214 }
211215
212- return new GenerateResponse ( {
216+ return create ( GenerateResponseSchema , {
213217 files : files ,
214218 } ) ;
215219}
216220
217221// Read input from stdin
218222function readInput ( ) : GenerateRequest {
219223 const buffer = readFileSync ( STDIO . Stdin ) ;
220- return GenerateRequest . fromBinary ( buffer ) ;
224+ return fromBinary ( GenerateRequestSchema , buffer ) ;
221225}
222226
223227function queryDecl ( name : string , sql : string ) {
@@ -299,7 +303,6 @@ function printNode(nodes: Node[]): string {
299303
300304// Write output to stdout
301305function writeOutput ( output : GenerateResponse ) {
302- const encodedOutput = output . toBinary ( ) ;
303- const buffer = new Uint8Array ( encodedOutput ) ;
306+ const buffer = toBinary ( GenerateResponseSchema , output ) ;
304307 writeFileSync ( STDIO . Stdout , buffer ) ;
305308}
0 commit comments