public override object VisitClass_declaration([NotNull] CSharpParser.Class_declarationContext context) { bool matchNames = context.identifier().GetText() == ServiceClassName; bool isTypeParamsMatch = true; var typeParams = context?.type_parameter_list()?.type_parameter(); if (!(typeParams is null && (TypeParameters is null || TypeParameters.Count == 0))) { if ((typeParams?.Length ?? 0) != (TypeParameters?.Count ?? 0)) { isTypeParamsMatch = false; } else { for (int i = 0; i < typeParams.Length; ++i) { if (typeParams[i].identifier().GetText() != TypeParameters[i].TypeParam) { isTypeParamsMatch = false; break; } } } } if (matchNames && isTypeParamsMatch) { HasServiceClass = true; Results.ServiceNamespace = string.Join(".", _currentNamespace.ToArray().Reverse()); var classDeclaration = new ClassInterfaceDeclaration() { IsInterface = false, Attributes = _cSharpParserService.GetTextWithWhitespace(Tokens, context?.attributes()), Identifier = ServiceClassName }; //Results.ClassInterfaceDeclarations.Add(className, classDeclaration); Results.ServiceDeclaration = classDeclaration; if (context?.PARTIAL()?.GetText() != null) { classDeclaration.Modifiers.Add(context.PARTIAL().GetText()); } if (context?.class_modifier() != null) { foreach (var modifier in context.class_modifier()) { classDeclaration.Modifiers.Add(modifier.GetText()); } } var typeParameters = context?.type_parameter_list()?.type_parameter(); var constraintsClauses = context?.type_parameter_constraints_clauses()?.type_parameter_constraints_clause(); classDeclaration.TypeParameters = _cSharpParserService.ParseTypeParameterList(Tokens, typeParameters, constraintsClauses); var baseClassType = _cSharpParserService.GetTextWithWhitespaceMinifiedLite( Tokens, context?.class_base()?.class_type()); var interfaceTypes = context?.class_base()?.interface_type_list()?.interface_type(); if (interfaceTypes != null || baseClassType != null) { classDeclaration.Base = new ClassInterfaceBase(); if (baseClassType != null) { classDeclaration.Base.ClassType = baseClassType; } if (interfaceTypes != null) { foreach (var interfaceType in interfaceTypes) { classDeclaration.Base.InterfaceTypeList.Add( _cSharpParserService.GetTextWithWhitespaceMinifiedLite(Tokens, interfaceType)); } } } var memberDeclarations = context?.class_body()?.class_member_declarations()?.class_member_declaration(); if (memberDeclarations != null) { foreach (var memberDeclaration in memberDeclarations) { bool isPublic = false; var classMethodDeclaration = memberDeclaration?.method_declaration(); var classPropertyDeclaration = memberDeclaration?.property_declaration(); if (classMethodDeclaration != null) { var methodDeclaration = new MethodDeclaration() { Attributes = _cSharpParserService.GetTextWithWhitespace( Tokens, classMethodDeclaration?.method_header()?.attributes()), ReturnType = _cSharpParserService.GetTextWithWhitespaceMinifiedLite( Tokens, classMethodDeclaration.method_header().return_type()), Identifier = classMethodDeclaration.method_header().member_name().GetText(), Body = _cSharpParserService.GetTextWithWhitespaceUntab( Tokens, classMethodDeclaration.method_body()) }; var methodModifiers = classMethodDeclaration?.method_header()?.method_modifier(); if (methodModifiers != null) { foreach (var methodMod in methodModifiers) { if (methodMod.GetText() == Keywords.Public) { isPublic = true; } methodDeclaration.Modifiers.Add(methodMod.GetText()); } } if (!isPublic) { continue; } var formalParameterList = classMethodDeclaration?.method_header()?.formal_parameter_list(); methodDeclaration.FormalParameterList = _cSharpParserService.ParseFormalParameterList(Tokens, formalParameterList); var methodTypeParameters = classMethodDeclaration?.method_header()?.type_parameter_list()?.type_parameter(); var methodConstraintsClauses = classMethodDeclaration ?.method_header() ?.type_parameter_constraints_clauses() ?.type_parameter_constraints_clause(); methodDeclaration.TypeParameters = _cSharpParserService .ParseTypeParameterList(Tokens, methodTypeParameters, methodConstraintsClauses); classDeclaration.Body.MethodDeclarations.Add(methodDeclaration); } else if (classPropertyDeclaration != null) { var propertyDeclaration = new PropertyDeclaration() { Attributes = _cSharpParserService.GetTextWithWhitespace( Tokens, classPropertyDeclaration?.attributes()), Type = _cSharpParserService.GetTextWithWhitespaceMinifiedLite( Tokens, classPropertyDeclaration.type_()), Identifier = classPropertyDeclaration.member_name().GetText(), Body = new PropertyBody() { Text = _cSharpParserService.GetTextWithWhitespaceUntab( Tokens, classPropertyDeclaration.property_body()), HasGetAccessor = classPropertyDeclaration.property_body().expression() != null | classPropertyDeclaration.property_body() ?.accessor_declarations() ?.get_accessor_declaration() != null, HasSetAccessor = classPropertyDeclaration.property_body() ?.accessor_declarations() ?.set_accessor_declaration() != null } }; var propertyModifiers = classPropertyDeclaration?.property_modifier(); if (propertyModifiers != null) { foreach (var propModifier in propertyModifiers) { if (propModifier.GetText() == Keywords.Public) { isPublic = true; } propertyDeclaration.Modifiers.Add(propModifier.GetText()); } } if (!isPublic) { continue; } classDeclaration.Body.PropertyDeclarations.Add(propertyDeclaration); } } } } VisitChildren(context); return(null); }
public override object VisitClass_declaration([NotNull] CSharpParser.Class_declarationContext context) { _currentClass.Push(context.identifier().GetText()); var typeParameters = _serviceFile.ServiceDeclaration.TypeParameters; var currentNamespace = GetCurrentNamespace(); bool isTypeParamsMatch = true; var typeParams = context?.type_parameter_list()?.type_parameter(); if (!(typeParams is null && (typeParameters is null || typeParameters.Count == 0))) { if ((typeParams?.Length ?? 0) != (typeParameters?.Count ?? 0)) { isTypeParamsMatch = false; } else { for (int i = 0; i < typeParams.Length; ++i) { if (typeParams[i].identifier().GetText() != typeParameters[i].TypeParam) { isTypeParamsMatch = false; break; } } } } if (GetCurrentClass() == _serviceClassInterfaceName && currentNamespace == _serviceFile.ServiceNamespace && isTypeParamsMatch) { _isCorrectClass.Push(true); _hasServiceClass = true; } else { _isCorrectClass.Push(false); } VisitChildren(context); if (_isCorrectClass.Peek()) { _hasServiceClass = true; var preclassWhitespace = Tokens.GetHiddenTokensToLeft(context.Start.TokenIndex, Lexer.Hidden); int tabLevels = 1 + ((preclassWhitespace?.Count ?? 0) > 0 ? _stringUtilService.CalculateTabLevels(preclassWhitespace[0]?.Text ?? string.Empty, _tabString) : 0); int?finalProperty = null; int?finalMethod = null; int?finalField = null; int?finalConstantOrField = null; int?finalConstructorOrDestructor = null; var members = context?.class_body()?.class_member_declarations()?.class_member_declaration(); if (members != null) { foreach (var member in members) { if (member.method_declaration() != null) { finalMethod = member.method_declaration().Stop.TokenIndex; } else if (member.property_declaration() != null) { finalProperty = member.property_declaration().Stop.TokenIndex; } else if (member.constant_declaration() != null) { finalConstantOrField = member.constant_declaration().Stop.TokenIndex; } else if (member.field_declaration() != null) { finalConstantOrField = member.field_declaration().Stop.TokenIndex; finalField = member.field_declaration().Stop.TokenIndex; } else if (member.constructor_declaration() != null) { finalConstructorOrDestructor = member.constructor_declaration().Stop.TokenIndex; } else if (member.static_constructor_declaration() != null) { finalConstructorOrDestructor = member.static_constructor_declaration().Stop.TokenIndex; } else if (member.destructor_declaration() != null) { finalConstructorOrDestructor = member.destructor_declaration().Stop.TokenIndex; } } } int fieldStopIndex = finalField ?? finalConstantOrField ?? context.class_body().OPEN_BRACE().Symbol.TokenIndex; int?constructorStopIndex = null; int?propertyStopIndex = null; int?methodStopIndex = null; var fieldStringBuilder = new StringBuilder(); StringBuilder constructorStringBuilder = null; StringBuilder propertyStringBuilder = null; StringBuilder methodStringBuilder = null; if (_fieldDict.Keys.Count > 0) { // add fields foreach (var field in _fieldDict.Values) { fieldStringBuilder.Append(_cSharpParserService.GenerateFieldDeclaration( field, tabLevels, _tabString)); } } if (!_hasServiceConstructor) { // add ctor constructorStopIndex = finalProperty ?? finalConstantOrField ?? fieldStopIndex; constructorStringBuilder = constructorStopIndex == fieldStopIndex ? fieldStringBuilder : new StringBuilder(); constructorStringBuilder.Append(_cSharpParserService.GenerateConstructorDeclaration( _serviceFile.ServiceDeclaration.Body.ConstructorDeclaration, tabLevels, _tabString)); } if (_serviceFile.ServiceDeclaration.Body?.PropertyDeclarations != null && _serviceFile.ServiceDeclaration.Body.PropertyDeclarations.Count > 0) { // add properties propertyStopIndex = finalProperty ?? finalConstructorOrDestructor ?? constructorStopIndex ?? fieldStopIndex; propertyStringBuilder = propertyStopIndex == fieldStopIndex ? fieldStringBuilder : (propertyStopIndex == constructorStopIndex ? constructorStringBuilder : new StringBuilder()); propertyStringBuilder.Append( _cSharpParserService.GeneratePropertyDeclarations( _serviceFile.ServiceDeclaration.Body.PropertyDeclarations, tabLevels, _tabString)); } if (_serviceFile.ServiceDeclaration.Body?.MethodDeclarations != null && _serviceFile.ServiceDeclaration.Body.MethodDeclarations.Count > 0) { // add methods methodStopIndex = finalMethod ?? finalProperty ?? finalConstructorOrDestructor ?? constructorStopIndex ?? fieldStopIndex; methodStringBuilder = methodStopIndex == fieldStopIndex ? fieldStringBuilder : (methodStopIndex == constructorStopIndex ? constructorStringBuilder : (methodStopIndex == propertyStopIndex ? propertyStringBuilder : new StringBuilder())); methodStringBuilder.Append( _cSharpParserService.GenerateMethodDeclarations( _serviceFile.ServiceDeclaration.Body.MethodDeclarations, tabLevels, _tabString)); } var fieldString = fieldStringBuilder.ToString(); if (fieldString.Length > 0) { IsModified = true; Rewriter.InsertAfter(fieldStopIndex, fieldString); } if (constructorStringBuilder != null && constructorStopIndex != fieldStopIndex) { var constructorString = constructorStringBuilder.ToString(); if (constructorString.Length > 0) { IsModified = true; Rewriter.InsertAfter(constructorStopIndex ?? -1, constructorString); } } if (propertyStringBuilder != null && propertyStopIndex != constructorStopIndex && propertyStopIndex != fieldStopIndex) { var propertyString = propertyStringBuilder.ToString(); if (propertyString.Length > 0) { IsModified = true; Rewriter.InsertAfter(propertyStopIndex ?? -1, propertyString); } } if (methodStringBuilder != null && methodStopIndex != constructorStopIndex && methodStopIndex != fieldStopIndex && methodStopIndex != propertyStopIndex) { var methodString = methodStringBuilder.ToString(); if (methodString.Length > 0) { IsModified = true; Rewriter.InsertAfter(methodStopIndex ?? -1, methodString); } } } _ = _isCorrectClass.Pop(); _ = _currentClass.Pop(); return(null); }