import { EnumDescriptorProto } from 'google-protobuf/google/protobuf/descriptor_pb';
import { Printer } from '../Printer';

export function printEnum(enumDescriptor: EnumDescriptorProto, indentLevel: number) {
    const printer = new Printer(indentLevel);
    const enumInterfaceName = `${enumDescriptor.getName()}Map`;
    printer.printEmptyLn();
    printer.printLn(`export interface ${enumInterfaceName} {`);
    enumDescriptor.getValueList().forEach((value) => {
        printer.printIndentedLn(
            `${(value.getName() as string).toUpperCase()}: ${value.getNumber()};`
        );
    });
    printer.printLn(`}`);
    printer.printEmptyLn();
    printer.printLn(`export const ${enumDescriptor.getName()}: ${enumInterfaceName};`);
    return printer.getOutput();
}