def enterMethodDeclaration(self, ctx: JavaParserLabeled.MethodDeclarationContext): if self.is_source_class or self.is_target_class: if ctx.formalParameters().formalParameterList(): method_parameters = [ctx.formalParameters().formalParameterList().children[i] for i in range(len(ctx.formalParameters().formalParameterList().children)) if i % 2 == 0] else: method_parameters = [] method_text = '' for modifier in ctx.parentCtx.parentCtx.modifier(): method_text += modifier.getText() + ' ' type_text = ctx.typeTypeOrVoid().getText() if type_text == self.source_class: type_text = self.target_class if self.is_target_class: self.token_stream_rewriter.replace( program_name=self.token_stream_rewriter.DEFAULT_PROGRAM_NAME, from_idx=ctx.typeTypeOrVoid().start.tokenIndex, to_idx=ctx.typeTypeOrVoid().stop.tokenIndex, text=type_text ) method_text += type_text + ' ' + ctx.IDENTIFIER().getText() method_text += ' ( ' for parameter in method_parameters: method_text += parameter.typeType().getText() + ' ' method_text += parameter.variableDeclaratorId().getText() + ', ' if method_parameters: method_text = method_text[:len(method_text) - 2] method_text += ')\n\t{' method_text += self.token_stream_rewriter.getText( program_name=self.token_stream_rewriter.DEFAULT_PROGRAM_NAME, start=ctx.methodBody().start.tokenIndex + 1, stop=ctx.methodBody().stop.tokenIndex - 1 ) method_text += '}\n' if self.is_source_class: self.source_class_data['methods'].append(ConstructorOrMethod( name=ctx.IDENTIFIER().getText(), parameters=[Parameter( parameter_type=p.typeType().getText(), name=p.variableDeclaratorId().IDENTIFIER().getText()) for p in method_parameters], text=method_text)) else: self.target_class_data['methods'].append(ConstructorOrMethod( name=ctx.IDENTIFIER().getText(), parameters=[Parameter( parameter_type=p.typeType().getText(), name=p.variableDeclaratorId().IDENTIFIER().getText()) for p in method_parameters], text=method_text))
def exitMethodDeclaration(self, ctx: JavaParserLabeled.MethodDeclarationContext): if not self.is_source_class: return None grand_parent_ctx = ctx.parentCtx.parentCtx method_identifier = ctx.IDENTIFIER().getText() if self.method_name in method_identifier: if grand_parent_ctx.modifier() == []: self.token_stream_rewriter.replaceRange( from_idx=ctx.typeTypeOrVoid().start.tokenIndex, to_idx=ctx.typeTypeOrVoid().stop.tokenIndex, text='public ' + ctx.typeTypeOrVoid().getText() ) elif grand_parent_ctx.modifier(0).getText() == 'private': self.token_stream_rewriter.replaceRange( from_idx=grand_parent_ctx.modifier(0).start.tokenIndex, to_idx=grand_parent_ctx.modifier(0).stop.tokenIndex, text='public') elif grand_parent_ctx.modifier(0).getText() != 'public': self.token_stream_rewriter.replaceRange( from_idx=grand_parent_ctx.modifier(0).start.tokenIndex, to_idx=grand_parent_ctx.modifier(0).stop.tokenIndex, text='public ' + grand_parent_ctx.modifier(0).getText())
def exitMethodDeclaration(self, ctx: JavaParserLabeled.MethodDeclarationContext): if not self.is_source_class: return None grand_parent_ctx = ctx.parentCtx.parentCtx method_identifier = ctx.IDENTIFIER().getText() if self.method_name in method_identifier: if grand_parent_ctx.modifier() == []: self.token_stream_rewriter.replaceRange( from_idx=ctx.typeTypeOrVoid().start.tokenIndex, to_idx=ctx.typeTypeOrVoid().stop.tokenIndex, text='final ' + ctx.typeTypeOrVoid().getText()) else: for i in range(0, len(grand_parent_ctx.modifier())): if grand_parent_ctx.modifier(i).getText() == "final": self.is_static = True break if not self.is_static: self.token_stream_rewriter.replaceRange( from_idx=grand_parent_ctx.modifier(0).start.tokenIndex, to_idx=grand_parent_ctx.modifier(0).stop.tokenIndex, text=grand_parent_ctx.modifier(0).getText() + ' final')
def enterMethodDeclaration( self, ctx: JavaParserLabeled.MethodDeclarationContext): if len(self.enter_method_body_stack) > 0: return if hasattr(ctx.parentCtx.parentCtx, 'modifier'): modifiers = ctx.parentCtx.parentCtx.modifier() else: modifiers = ctx.parentCtx.parentCtx.parentCtx.modifier() do_extract = True for modifier in modifiers: if modifier.classOrInterfaceModifier() is not None: # print('modifier.classOrInterfaceModifier().getText()', modifier.classOrInterfaceModifier().getText()) # Todo: Requires better handling if "private" in modifier.classOrInterfaceModifier().getText() or \ "static" in modifier.classOrInterfaceModifier().getText() or \ '?' in modifier.classOrInterfaceModifier().getText(): do_extract = False break # Todo: Requires better handling if '?' in ctx.getChild(0).getText(): do_extract = False if do_extract: method = { 'name': ctx.IDENTIFIER().getText(), 'return_type': ctx.typeTypeOrVoid().getText(), 'formal_parameters': [] } if ctx.formalParameters().formalParameterList() is not None: if hasattr(ctx.formalParameters().formalParameterList(), 'formalParameter'): for f in ctx.formalParameters().formalParameterList( ).formalParameter(): _type = f.typeType().getText() identifier = f.variableDeclaratorId().getText() method['formal_parameters'].append([_type, identifier]) self.interface_info['methods'].append(method)
def enterMethodDeclaration( self, ctx: JavaParserLabeled.MethodDeclarationContext): methods.append(f"{ctx.IDENTIFIER()} ") returntype.append(ctx.typeTypeOrVoid().getText()) if ctx.IDENTIFIER().getText() == self.method_name: found_func.append(ctx.IDENTIFIER())