获得java项目，并且在某个方法中添加内容

获得工作区的第一个项目，找到类“x.y.z.Test”，并且在方法中添加一段代码

eg：

public static void doS() {

IProject[] projects = ResourcesPlugin.getWorkspace().getRoot().getProjects();

System.out.println(projects.length);

if (projects.length == 0) {

return;

}

IProject project = projects[0];

System.out.println();

IJavaProject jp = JavaCore.create(project);

try {

IType tp = jp.findType("x.y.z.Test");

ICompilationUnit cu = tp.getCompilationUnit();

File file = cu.getResource().getLocation().toFile();

Document doc = new Document(getFileAsString(file));

ASTParser parsert = ASTParser.newParser(AST.JLS3);

parsert.setSource(doc.get().toCharArray());

System.out.println("!!!!!!!!!!---1---" + doc.getLength());

CompilationUnit result = (CompilationUnit) parsert.createAST(null);

result.recordModifications();

AST ast = result.getAST();

List types = result.types();

TypeDeclaration typeDec = (TypeDeclaration) types.get(0);

List importList = result.imports();

PackageDeclaration packetDec = result.getPackage();

String className = typeDec.getName().toString();

MethodDeclaration methodDec[] = typeDec.getMethods();

FieldDeclaration fieldDec[] = typeDec.getFields();

System.out.println("包:");

System.out.println(packetDec.getName());

System.out.println("引用import:");

for (Object obj : importList) {

ImportDeclaration importDec = (ImportDeclaration) obj;

System.out.println(importDec.getName());

}

System.out.println("类:");

System.out.println(className);

System.out.println("========================");

System.out.println("函数:");

for (MethodDeclaration method : methodDec) {

System.out.println("=============");

System.out.println(method);

List lst = method.getBody().statements();

MethodInvocation mi = ast.newMethodInvocation();

mi.setName(ast.newSimpleName("setLayout"));

mi.arguments().add(ast.newSimpleName("gridLayout"));

lst.add(ast.newExpressionStatement(mi));

Assignment assign = ast.newAssignment();

assign.setLeftHandSide(ast.newSimpleName("i"));

assign.setOperator(Assignment.Operator.ASSIGN);

assign.setRightHandSide(ast.newNumberLiteral("5"));

ExpressionStatement statement = ast.newExpressionStatement(assign);

lst.add(statement); // 可以追加到指定位置

Map<String, String> options = new HashMap<String, String>();

options.put(

DefaultCodeFormatterConstants.FORMATTER_INDENT_SWITCHSTATEMENTS_COMPARE_TO_SWITCH,

"true");

TextEdit edits = result.rewrite(doc, options);

edits.apply(doc);

string2File(doc.get(), file);

cu.getResource().refreshLocal(0, null);

break;

}

System.out.println("变量:");

for (FieldDeclaration fieldDecEle : fieldDec) {

//public static

for (Object modifiObj : fieldDecEle.modifiers()) {

Modifier modify = (Modifier) modifiObj;

System.out.print(modify + "-");

}

System.out.println(fieldDecEle.getType());

for (Object obj : fieldDecEle.fragments()) {

VariableDeclarationFragment frag = (VariableDeclarationFragment) obj;

System.out.println("[FIELD_NAME:]" + frag.getName());

}

}

} catch (Exception e) {

e.printStackTrace();

}

}

public static String getFileAsString(File file) throws IOException {

BufferedReader in = new BufferedReader(new FileReader(file));

StringBuffer buffer = new StringBuffer();

String line = null;

// iterator read line

while (null != (line = in.readLine())) {

buffer.append(line);

buffer.append("\n");

}

return buffer.toString();

}

public static void string2File(String content, File file) throws IOException {

// create file if it necessary

if (!file.exists()) {

file.createNewFile();

}

// write to file

FileWriter fw = new FileWriter(file);

fw.write(content);

fw.flush();

}