mirror of
https://github.com/cline/cline.git
synced 2026-09-19 02:05:44 +08:00
Sarah Fortune
600a6e33ed
* Move duplicated code for laoding the protobuf descriptor set into a shared util file. * Formatting * Update comment
51 lines
1.5 KiB
JavaScript
Executable File
51 lines
1.5 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
|
|
import * as fs from "fs/promises"
|
|
import * as path from "path"
|
|
import * as grpc from "@grpc/grpc-js"
|
|
import * as protoLoader from "@grpc/proto-loader"
|
|
|
|
const DESCRIPTOR_SET = path.resolve("dist-standalone/proto/descriptor_set.pb")
|
|
|
|
const typeNameToFQN = new Map()
|
|
|
|
function addTypeNameToFqn(name, fqn) {
|
|
if (typeNameToFQN.has(name)) {
|
|
throw new Error(`Proto type ${name} redefined (${fqn}).`)
|
|
}
|
|
typeNameToFQN.set(name, fqn)
|
|
}
|
|
// Get the fully qualified name for a proto type, e.g. getFqn('StringRequest') returns 'cline.StringRequest'
|
|
export function getFqn(name) {
|
|
if (!typeNameToFQN.has(name)) {
|
|
throw Error(`No FQN for ${name}`)
|
|
}
|
|
return typeNameToFQN.get(name)
|
|
}
|
|
|
|
export async function loadServicesFromProtoDescriptor() {
|
|
// Load service definitions from descriptor set
|
|
const descriptorBuffer = await fs.readFile(DESCRIPTOR_SET)
|
|
const packageDefinition = protoLoader.loadFileDescriptorSetFromBuffer(descriptorBuffer)
|
|
const proto = grpc.loadPackageDefinition(packageDefinition)
|
|
|
|
// Extract host services and proto messages from the proto definition
|
|
const hostServices = {}
|
|
for (const [name, def] of Object.entries(proto.host)) {
|
|
if (def && "service" in def) {
|
|
hostServices[name] = def
|
|
} else {
|
|
addTypeNameToFqn(name, `proto.host.${name}`)
|
|
}
|
|
}
|
|
const protobusServices = {}
|
|
for (const [name, def] of Object.entries(proto.cline)) {
|
|
if (def && "service" in def) {
|
|
protobusServices[name] = def
|
|
} else {
|
|
addTypeNameToFqn(name, `proto.cline.${name}`)
|
|
}
|
|
}
|
|
return { protobusServices, hostServices }
|
|
}
|