Popular
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607import * as ast from "../ast/index.ts";import { Token } from "../parser/recursive-descent-parser.ts";
export interface Visitor { visitProto: VisitFn<ast.Proto>; visitTopLevelStatement: VisitFn<ast.TopLevelStatement>; visitSyntax: VisitFn<ast.Syntax>; visitImport: VisitFn<ast.Import>; visitPackage: VisitFn<ast.Package>; visitOption: VisitFn<ast.Option>; visitOptionName: VisitFn<ast.OptionName>; visitOptionNameSegment: VisitFn<ast.OptionNameSegment>; visitTopLevelDef: VisitFn<ast.TopLevelDef>; visitMessage: VisitFn<ast.Message>; visitMessageBody: VisitFn<ast.MessageBody>; visitMessageBodyStatement: VisitFn<ast.MessageBodyStatement>; visitEnum: VisitFn<ast.Enum>; visitEnumBody: VisitFn<ast.EnumBody>; visitEnumBodyStatement: VisitFn<ast.EnumBodyStatement>; visitEnumField: VisitFn<ast.EnumField>; visitExtend: VisitFn<ast.Extend>; visitExtendBody: VisitFn<ast.ExtendBody>; visitExtendBodyStatement: VisitFn<ast.ExtendBodyStatement>; visitService: VisitFn<ast.Service>; visitServiceBody: VisitFn<ast.ServiceBody>; visitServiceBodyStatement: VisitFn<ast.ServiceBodyStatement>; visitRpc: VisitFn<ast.Rpc>; visitRpcBody: VisitFn<ast.RpcBody>; visitRpcBodyStatement: VisitFn<ast.RpcBodyStatement>; visitRpcType: VisitFn<ast.RpcType>; visitEmpty: VisitFn<ast.Empty>; visitField: VisitFn<ast.Field>; visitFieldOptions: VisitFn<ast.FieldOptions>; visitFieldOption: VisitFn<ast.FieldOption>; visitGroup: VisitFn<ast.Group>; visitOneof: VisitFn<ast.Oneof>; visitOneofBody: VisitFn<ast.OneofBody>; visitOneofBodyStatement: VisitFn<ast.OneofBodyStatement>; visitOneofField: VisitFn<ast.OneofField>; visitOneofGroup: VisitFn<ast.OneofGroup>; visitMapField: VisitFn<ast.MapField>; visitExtensions: VisitFn<ast.Extensions>; visitRanges: VisitFn<ast.Ranges>; visitRange: VisitFn<ast.Range>; visitMax: VisitFn<ast.Max>; visitReserved: VisitFn<ast.Reserved>; visitFieldNames: VisitFn<ast.FieldNames>; visitConstant: VisitFn<ast.Constant>; visitCommentGroup: VisitFn<ast.CommentGroup>; visitComment: VisitFn<ast.Comment>; visitSinglelineComment: VisitFn<ast.SinglelineComment>; visitMultilineComment: VisitFn<ast.MultilineComment>; visitKeyword: VisitFn<ast.Keyword>; visitType: VisitFn<ast.Type>; visitFullIdent: VisitFn<ast.FullIdent>; visitIntLit: VisitFn<ast.IntLit>; visitSignedIntLit: VisitFn<ast.SignedIntLit>; visitFloatLit: VisitFn<ast.FloatLit>; visitSignedFloatLit: VisitFn<ast.SignedFloatLit>; visitStrLit: VisitFn<ast.StrLit>; visitBoolLit: VisitFn<ast.BoolLit>; visitAggregate: VisitFn<ast.Aggregate>; visitIdent: VisitFn<ast.Ident>; visitDot: VisitFn<ast.Dot>; visitComma: VisitFn<ast.Comma>; visitSemi: VisitFn<ast.Semi>; visitToken: VisitFn<Token>;}
export interface VisitFn<T> { (visitor: Visitor, node: T): void;}
export const visitor: Visitor = { visitProto(visitor, node) { for (const statement of node.statements) { visitor.visitTopLevelStatement(visitor, statement); } }, visitTopLevelStatement(visitor, node) { switch (node.type) { case "syntax": return visitor.visitSyntax(visitor, node); case "import": return visitor.visitImport(visitor, node); case "package": return visitor.visitPackage(visitor, node); case "option": return visitor.visitOption(visitor, node); case "message": case "enum": case "extend": case "service": return visitor.visitTopLevelDef(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitSyntax(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.eq); visitor.visitToken(visitor, node.quoteOpen); visitor.visitToken(visitor, node.syntax); visitor.visitToken(visitor, node.quoteClose); visitor.visitSemi(visitor, node.semi); }); }, visitImport(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); node.weakOrPublic && visitor.visitToken(visitor, node.weakOrPublic); visitor.visitStrLit(visitor, node.strLit); visitor.visitSemi(visitor, node.semi); }); }, visitPackage(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitFullIdent(visitor, node.fullIdent); visitor.visitSemi(visitor, node.semi); }); }, visitOption(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitOptionName(visitor, node.optionName); visitor.visitToken(visitor, node.eq); visitor.visitConstant(visitor, node.constant); visitor.visitSemi(visitor, node.semi); }); }, visitOptionName(visitor, node) { for (const optionNameSegmentOrDot of node.optionNameSegmentOrDots) { switch (optionNameSegmentOrDot.type) { case "option-name-segment": visitor.visitOptionNameSegment(visitor, optionNameSegmentOrDot); continue; case "dot": visitor.visitDot(visitor, optionNameSegmentOrDot); continue; } } }, visitOptionNameSegment(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); visitor.visitFullIdent(visitor, node.name); node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitTopLevelDef(visitor, node) { switch (node.type) { case "message": return visitor.visitMessage(visitor, node); case "enum": return visitor.visitEnum(visitor, node); case "extend": return visitor.visitExtend(visitor, node); case "service": return visitor.visitService(visitor, node); } }, visitMessage(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.messageName); visitor.visitMessageBody(visitor, node.messageBody); }); }, visitMessageBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitMessageBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitMessageBodyStatement(visitor, node) { switch (node.type) { case "field": return visitor.visitField(visitor, node); case "enum": return visitor.visitEnum(visitor, node); case "message": return visitor.visitMessage(visitor, node); case "extend": return visitor.visitExtend(visitor, node); case "extensions": return visitor.visitExtensions(visitor, node); case "group": return visitor.visitGroup(visitor, node); case "option": return visitor.visitOption(visitor, node); case "oneof": return visitor.visitOneof(visitor, node); case "map-field": return visitor.visitMapField(visitor, node); case "reserved": return visitor.visitReserved(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitEnum(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.enumName); visitor.visitEnumBody(visitor, node.enumBody); }); }, visitEnumBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitEnumBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitEnumBodyStatement(visitor, node) { switch (node.type) { case "option": return visitor.visitOption(visitor, node); case "reserved": return visitor.visitReserved(visitor, node); case "enum-field": return visitor.visitEnumField(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitEnumField(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitToken(visitor, node.fieldName); visitor.visitToken(visitor, node.eq); visitor.visitSignedIntLit(visitor, node.fieldNumber); node.fieldOptions && visitor.visitFieldOptions(visitor, node.fieldOptions); visitor.visitSemi(visitor, node.semi); }); }, visitExtend(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitType(visitor, node.messageType); visitor.visitExtendBody(visitor, node.extendBody); }); }, visitExtendBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitExtendBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitExtendBodyStatement(visitor, node) { switch (node.type) { case "field": return visitor.visitField(visitor, node); case "group": return visitor.visitGroup(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitService(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.serviceName); visitor.visitServiceBody(visitor, node.serviceBody); }); }, visitServiceBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitServiceBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitServiceBodyStatement(visitor, node) { switch (node.type) { case "option": return visitor.visitOption(visitor, node); case "rpc": return visitor.visitRpc(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitRpc(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.rpcName); visitor.visitRpcType(visitor, node.reqType); visitor.visitToken(visitor, node.returns); visitor.visitRpcType(visitor, node.resType); if (node.semiOrRpcBody.type === "semi") { visitor.visitSemi(visitor, node.semiOrRpcBody); } else { visitor.visitRpcBody(visitor, node.semiOrRpcBody); } }); }, visitRpcBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitRpcBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitRpcBodyStatement(visitor, node) { switch (node.type) { case "option": return visitor.visitOption(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitRpcType(visitor, node) { visitor.visitToken(visitor, node.bracketOpen); node.stream && visitor.visitKeyword(visitor, node.stream); visitor.visitType(visitor, node.messageType); visitor.visitToken(visitor, node.bracketClose); }, visitEmpty(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitSemi(visitor, node.semi); }); }, visitField(visitor, node) { visitStatementBase(visitor, node, () => { node.fieldLabel && visitor.visitKeyword(visitor, node.fieldLabel); visitor.visitType(visitor, node.fieldType); visitor.visitToken(visitor, node.fieldName); visitor.visitToken(visitor, node.eq); visitor.visitIntLit(visitor, node.fieldNumber); node.fieldOptions && visitor.visitFieldOptions(visitor, node.fieldOptions); visitor.visitSemi(visitor, node.semi); }); }, visitFieldOptions(visitor, node) { visitor.visitToken(visitor, node.bracketOpen); for (const fieldOptionOrComma of node.fieldOptionOrCommas) { switch (fieldOptionOrComma.type) { case "field-option": visitor.visitFieldOption(visitor, fieldOptionOrComma); continue; case "comma": visitor.visitComma(visitor, fieldOptionOrComma); continue; } } visitor.visitToken(visitor, node.bracketClose); }, visitFieldOption(visitor, node) { visitor.visitOptionName(visitor, node.optionName); visitor.visitToken(visitor, node.eq); visitor.visitConstant(visitor, node.constant); }, visitGroup(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.groupLabel); visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.groupName); visitor.visitToken(visitor, node.eq); visitor.visitIntLit(visitor, node.fieldNumber); node.fieldOptions && visitor.visitFieldOptions(visitor, node.fieldOptions); visitor.visitMessageBody(visitor, node.messageBody); }); }, visitOneof(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.oneofName); visitor.visitOneofBody(visitor, node.oneofBody); }); }, visitOneofBody(visitor, node) { node.bracketOpen && visitor.visitToken(visitor, node.bracketOpen); for (const statement of node.statements) { visitor.visitOneofBodyStatement(visitor, statement); } node.bracketClose && visitor.visitToken(visitor, node.bracketClose); }, visitOneofBodyStatement(visitor, node) { switch (node.type) { case "option": return visitor.visitOption(visitor, node); case "oneof-field": return visitor.visitOneofField(visitor, node); case "empty": return visitor.visitEmpty(visitor, node); } }, visitOneofField(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitType(visitor, node.fieldType); visitor.visitToken(visitor, node.fieldName); visitor.visitToken(visitor, node.eq); visitor.visitIntLit(visitor, node.fieldNumber); node.fieldOptions && visitor.visitFieldOptions(visitor, node.fieldOptions); visitor.visitSemi(visitor, node.semi); }); }, visitOneofGroup(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.groupName); visitor.visitToken(visitor, node.eq); visitor.visitIntLit(visitor, node.fieldNumber); visitor.visitMessageBody(visitor, node.messageBody); }); }, visitMapField(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitToken(visitor, node.typeBracketOpen); visitor.visitType(visitor, node.keyType); visitor.visitToken(visitor, node.typeSep); visitor.visitType(visitor, node.valueType); visitor.visitToken(visitor, node.typeBracketClose); visitor.visitToken(visitor, node.mapName); visitor.visitToken(visitor, node.eq); visitor.visitIntLit(visitor, node.fieldNumber); node.fieldOptions && visitor.visitFieldOptions(visitor, node.fieldOptions); visitor.visitSemi(visitor, node.semi); }); }, visitExtensions(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); visitor.visitRanges(visitor, node.ranges); visitor.visitSemi(visitor, node.semi); }); }, visitRanges(visitor, node) { for (const rangeOrComma of node.rangeOrCommas) { switch (rangeOrComma.type) { case "range": visitor.visitRange(visitor, rangeOrComma); continue; case "comma": visitor.visitComma(visitor, rangeOrComma); continue; } } }, visitRange(visitor, node) { visitor.visitIntLit(visitor, node.rangeStart); node.to && visitor.visitKeyword(visitor, node.to); if (node.rangeEnd) { switch (node.rangeEnd.type) { case "int-lit": visitor.visitIntLit(visitor, node.rangeEnd); break; case "max": visitor.visitMax(visitor, node.rangeEnd); break; } } }, visitMax(visitor, node) { visitor.visitToken(visitor, node); }, visitReserved(visitor, node) { visitStatementBase(visitor, node, () => { visitor.visitKeyword(visitor, node.keyword); switch (node.reserved.type) { case "ranges": visitor.visitRanges(visitor, node.reserved); break; case "field-names": visitor.visitFieldNames(visitor, node.reserved); break; } visitor.visitSemi(visitor, node.semi); }); }, visitFieldNames(visitor, node) { for (const strLitOrComma of node.strLitOrCommas) { switch (strLitOrComma.type) { case "str-lit": visitor.visitStrLit(visitor, strLitOrComma); continue; case "comma": visitor.visitComma(visitor, strLitOrComma); continue; } } }, visitConstant(visitor, node) { switch (node.type) { case "full-ident": return visitor.visitFullIdent(visitor, node); case "signed-int-lit": return visitor.visitSignedIntLit(visitor, node); case "signed-float-lit": return visitor.visitSignedFloatLit(visitor, node); case "str-lit": return visitor.visitStrLit(visitor, node); case "bool-lit": return visitor.visitBoolLit(visitor, node); case "aggregate": return visitor.visitAggregate(visitor, node); } }, visitCommentGroup(visitor, node) { for (const comment of node.comments) { visitor.visitComment(visitor, comment); } }, visitComment(visitor, node) { switch (node.type) { case "singleline-comment": return visitor.visitSinglelineComment(visitor, node); case "multiline-comment": return visitor.visitMultilineComment(visitor, node); } }, visitSinglelineComment(visitor, node) { visitor.visitToken(visitor, node); }, visitMultilineComment(visitor, node) { visitor.visitToken(visitor, node); }, visitKeyword(visitor, node) { visitor.visitToken(visitor, node); }, visitType(visitor, node) { for (const identOrDot of node.identOrDots) { switch (identOrDot.type) { case "ident": visitor.visitIdent(visitor, identOrDot); continue; case "dot": visitor.visitDot(visitor, identOrDot); continue; } } }, visitFullIdent(visitor, node) { for (const identOrDot of node.identOrDots) { switch (identOrDot.type) { case "ident": visitor.visitIdent(visitor, identOrDot); continue; case "dot": visitor.visitDot(visitor, identOrDot); continue; } } }, visitIntLit(visitor, node) { visitor.visitToken(visitor, node); }, visitSignedIntLit(visitor, node) { node.sign && visitor.visitToken(visitor, node.sign); visitor.visitIntLit(visitor, node.value); }, visitFloatLit(visitor, node) { visitor.visitToken(visitor, node); }, visitSignedFloatLit(visitor, node) { node.sign && visitor.visitToken(visitor, node.sign); visitor.visitFloatLit(visitor, node.value); }, visitStrLit(visitor, node) { for (const token of node.tokens) { visitor.visitToken(visitor, token); } }, visitBoolLit(visitor, node) { visitor.visitToken(visitor, node); }, visitAggregate() {}, visitIdent(visitor, node) { visitor.visitToken(visitor, node); }, visitDot(visitor, node) { visitor.visitToken(visitor, node); }, visitComma(visitor, node) { visitor.visitToken(visitor, node); }, visitSemi(visitor, node) { visitor.visitToken(visitor, node); }, visitToken() {},};
export function visitStatementBase<T extends ast.StatementBase>( visitor: Visitor, node: T, visit: () => void,): void { for (const commentGroup of node.leadingDetachedComments) { visitor.visitCommentGroup(visitor, commentGroup); } for (const commentGroup of node.leadingComments) { visitor.visitCommentGroup(visitor, commentGroup); } visit(); for (const commentGroup of node.trailingComments) { visitor.visitCommentGroup(visitor, commentGroup); }}