1. import java.io.IOException;
  2. import java.util.ArrayList;
  3. import java.util.List;
  4. import java.util.regex.Matcher;
  5. import java.util.regex.Pattern;
  7. public class CodeInstrumenter {
  9. private static final String INSTRUMENTATION_START = "/*--- START INSTRUMENTATION ---*/";
  10. private static final String INSTRUMENTATION_END = "/*--- END INSTRUMENTATION ---*/";
  11. private static final String LOG_START = "System.out.println(\"DEBUG: \");";
  12. private static final String LOG_END = "System.out.println(\"DEBUG: \");";
  15. public static String instrumentCode(String code) {
  16. List<String> lines = new ArrayList<>();
  17. for (String line : code.split("\\r?\\n")) {
  18. lines.add(line);
  19. }
  21. StringBuilder instrumentedCode = new StringBuilder();
  22. boolean inInstrumentationBlock = false;
  24. for (int i = 0; i < lines.size(); i++) {
  25. String line = lines.get(i);
  27. if (line.contains(INSTRUMENTATION_START)) {
  28. instrumentedCode.append(line).append("\n");
  29. inInstrumentationBlock = true;
  30. } else if (line.contains(INSTRUMENTATION_END)) {
  31. instrumentedCode.append(line).append("\n");
  32. inInstrumentationBlock = false;
  33. } else if (inInstrumentationBlock) {
  34. instrumentedCode.append(LOG_START).append("\n");
  35. instrumentedCode.append(line).append("\n");
  36. instrumentedCode.append(LOG_END).append("\n");
  37. } else {
  38. instrumentedCode.append(line).append("\n");
  39. }
  40. }
  42. return instrumentedCode.toString();
  43. }
  47. public static void main(String[] args) {
  48. String code = """
  49. public class MyClass {
  50. public void myMethod() {
  51. int x = 10;
  52. int y = 20;
  53. int sum = x + y;
  54. System.out.println("Sum: " + sum);
  55. }
  56. }
  57. """;
  59. String instrumentedCode = instrumentCode(code);
  60. System.out.println(instrumentedCode);
  61. }
  62. }

Add your comment