diff --git a/src/grpc/server.ts b/src/grpc/server.ts index 67b4c2d..9a7c1e8 100644 --- a/src/grpc/server.ts +++ b/src/grpc/server.ts @@ -5,6 +5,10 @@ import winston from 'winston'; import { DiscoveryServer } from './discovery.js'; import { PluginServer } from './plugin.js'; +export enum Network { + TCP = 'tcp', +} + export const getServer = () => { const server = new grpc.Server(); server.addService(pluginV3.cloudquery.plugin.v3.UnimplementedPluginService.definition, new PluginServer()); diff --git a/src/serve/serve.ts b/src/serve/serve.ts index 9f4dfc3..9cbebee 100644 --- a/src/serve/serve.ts +++ b/src/serve/serve.ts @@ -1,16 +1,14 @@ import yargs from 'yargs'; import { hideBin } from 'yargs/helpers'; -import { startServer } from '../grpc/server.js'; +import { startServer, Network } from '../grpc/server.js'; import { LogFormat, LogLevel, createLogger } from '../logger/logger.js'; -const NETWORK_CHOICES = ['tcp', 'tcp4', 'tcp6', 'unix', 'unixpacket'] as const; - const TELEMETRY_LEVEL_CHOICES = ['none', 'errors', 'stats', 'all'] as const; export type ServeArguments = { address: string; - network: (typeof NETWORK_CHOICES)[number]; + network: Network; logLevel: LogLevel; logFormat: LogFormat; sentry: boolean; @@ -40,7 +38,7 @@ export const serve = yargs(hideBin(process.argv)) network: { alias: 'n', type: 'string', - choices: NETWORK_CHOICES, + choices: Object.values(Network), description: 'network to bind to', default: 'tcp', },