Myujth

This is actually quite difficult. First of all, your if "onPause" in sourcecode approach currently doesn't distinguish between defining onPause() and calling it. And second of all, finding the correct closing } isn't trivial. Naively, you might just count opening and closing curlies ( increments the blocklevel, decrements it), and assume that the } that makes the blocklevel zero is the closing curly of the method. However, this might be wrong! Because the method might contain some string literal containing (possibly unbalanced) curlies. Or comments with curlies. This would mess up the blocklevel count.

To do this properly, you would have to build an actual Java parser. That's a lot of work, even when using libraries such as tatsu.

If you're fine with a rather volatile kludge, you can try and use the blocklevel count mentioned above together with the indentation as a clue (assuming your source code is decently indented). Here's something I've hacked up as a starting point:

def augment_function(sourcecode, function, line_to_insert):
 in_function = False
 blocklevel = 0
 insert_before = None
 source = sourcecode.split("n")
 for line_no, line in enumerate(source):
 if in_function:
 if "" in line:
 blocklevel += 1
 if "" in line:
 blocklevel -= 1
 if blocklevel == 0:
 insert_before = line_no
 indent = len(line) - len(line.lstrip(" ")) + 4 #4=your indent level
 break
 elif function in line and "public " in line:
 in_function = True
 if "" in line:
 blocklevel += 1
 if insert_before:
 source.insert(insert_before, " "*indent + line_to_insert)
 return "n".join(source)

# test code:
java_code = """class Foo 
 private int foo;
 public void main(String args) 
 foo = 1;
 
 public void setFoo(int f)
 
 foo = f;
 
 public int getFoo(int f) 
 return foo;
 

"""
print(augment_function(java_code, "setFoo", "log.debug("setFoo")"))

Note that this is vulnerable to all sorts of edge cases (such as in a string or in a comment, or tab indent instead of space, or possibly a thousand other things). This is just a starting point for you.