v3/src/ts-protoc-gen/util.ts

import { parse } from 'querystring';
import { FileDescriptorProto } from 'google-protobuf/google/protobuf/descriptor_pb';
import { ExportEnumEntry, ExportMessageEntry } from './ExportMap';
import { ServiceParameter, ModeParameter } from './parameters';
export function filePathToPseudoNamespace(filePath: string): string {
    return (
        filePath
            .replace('.proto', '')
            .replace(/\//g, '_')
            .replace(/\./g, '_')
            .replace(/\-/g, '_') + '_pb'
    );
}

export function stripPrefix(str: string, prefix: string) {
    if (str.substr(0, prefix.length) === prefix) {
        return str.substr(prefix.length);
    }
    return str;
}

export function snakeToCamel(str: string): string {
    return str.replace(/(\_\w)/g, function (m) {
        return m[1].toUpperCase();
    });
}

export function uppercaseFirst(str: string): string {
    return str.charAt(0).toUpperCase() + str.slice(1);
}

const PROTO2_SYNTAX = 'proto2';
export function isProto2(fileDescriptor: FileDescriptorProto): boolean {
    // Empty syntax defaults to proto2
    return (
        fileDescriptor.getSyntax() === '' || fileDescriptor.getSyntax() === PROTO2_SYNTAX
    );
}

export function oneOfName(name: string) {
    return uppercaseFirst(snakeToCamel(name.toLowerCase()));
}

export function generateIndent(indentLevel: number): string {
    let indent = '';
    for (let i = 0; i < indentLevel; i++) {
        indent += '  ';
    }
    return indent;
}

export function getPathToRoot(fileName: string) {
    const depth = fileName.split('/').length;
    return depth === 1 ? './' : new Array(depth).join('../');
}

export function withinNamespaceFromExportEntry(
    name: string,
    exportEntry: ExportMessageEntry | ExportEnumEntry
) {
    return exportEntry.pkg ? name.substring(exportEntry.pkg.length + 1) : name;
}

export function replaceProtoSuffix(protoFilePath: string): string {
    const suffix = '.proto';
    const hasProtoSuffix =
        protoFilePath.slice(protoFilePath.length - suffix.length) === suffix;
    return hasProtoSuffix
        ? protoFilePath.slice(0, -suffix.length) + '_pb'
        : protoFilePath;
}

export function withAllStdIn(callback: (buffer: Buffer) => void): void {
    const ret: Buffer[] = [];
    let len = 0;

    const stdin = process.stdin;
    stdin.on('readable', function () {
        let chunk = stdin.read();

        while (chunk) {
            if (!(chunk instanceof Buffer)) throw new Error('Did not receive buffer');
            ret.push(chunk);
            len += chunk.length;
            chunk = stdin.read();
        }
    });

    stdin.on('end', function () {
        callback(Buffer.concat(ret, len));
    });
}

// normaliseFieldObjectName modifies the field name that appears in the `asObject` representation
// to match the logic found in `protobuf/compiler/js/js_generator.cc`. See: https://goo.gl/tX1dPQ
export function normaliseFieldObjectName(name: string): string {
    switch (name) {
        case 'abstract':
        case 'boolean':
        case 'break':
        case 'byte':
        case 'case':
        case 'catch':
        case 'char':
        case 'class':
        case 'const':
        case 'continue':
        case 'debugger':
        case 'default':
        case 'delete':
        case 'do':
        case 'double':
        case 'else':
        case 'enum':
        case 'export':
        case 'extends':
        case 'false':
        case 'final':
        case 'finally':
        case 'float':
        case 'for':
        case 'function':
        case 'goto':
        case 'if':
        case 'implements':
        case 'import':
        case 'in':
        case 'instanceof':
        case 'int':
        case 'interface':
        case 'long':
        case 'native':
        case 'new':
        case 'null':
        case 'package':
        case 'private':
        case 'protected':
        case 'public':
        case 'return':
        case 'short':
        case 'static':
        case 'super':
        case 'switch':
        case 'synchronized':
        case 'this':
        case 'throw':
        case 'throws':
        case 'transient':
        case 'try':
        case 'typeof':
        case 'var':
        case 'void':
        case 'volatile':
        case 'while':
        case 'with':
            return `pb_${name}`;
    }
    return name;
}

export function getServiceParameter(service?: string): ServiceParameter {
    switch (service) {
        case 'true':
            console.warn(
                'protoc-gen-ts warning: The service=true parameter has been deprecated. Use service=grpc-web instead.'
            );
            return ServiceParameter.GrpcWeb;
        case 'grpc-web':
            return ServiceParameter.GrpcWeb;
        case 'grpc-node':
            return ServiceParameter.GrpcNode;
        case undefined:
            return ServiceParameter.None;
        default:
            throw new Error(`Unrecognised service parameter: ${service}`);
    }
}

export function getModeParameter(mode?: string): ModeParameter {
    switch (mode) {
        case 'grpc-js':
            return ModeParameter.GrpcJs;
        case undefined:
            return ModeParameter.None;
        default:
            throw new Error(`Unrecognised mode parameter: ${mode}`);
    }
}

export function getParameterEnums(
    parameter: string
): {
    service: ServiceParameter;
    mode: ModeParameter;
} {
    const { service, mode } = parse(parameter, ',');
    return {
        service: getServiceParameter(service as string),
        mode: getModeParameter(mode as string),
    };
}