用java代码编程实现为java文件自动加入合适的包名

在学习java编程思想这本书的时候，发现配套提供的java代码的包名很多都不对，每打开一个文件都需要手动去点一下eclipse的建议修改，然后再点“add package”，比较麻烦。

在学习了第18章java IO系统后，活学活用，编写了一段代码，为所有需要加入包名的文件自己加入合适的包名。

以下是代码：

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class Test {

	public static void main(String args[]) throws IOException {

		TreeInfo treeInfo = recurseDirs(new File("./src"), ".*\\.java");
		System.out.println("共获取到" + treeInfo.files.size() + "个文件！");
		int count = 0;
		for (File file : treeInfo) {
			if (file.getName().startsWith("Test.java")) {
				continue;
			}
			// String name = file.getName();
			// System.out.println(name);
			if (file.getPath().indexOf("src") == -1) {
				continue;
			}
			addPackage(file);
			count++;
		}
		System.out.println("操作完成！共操作 " + count + "个文件，简直棒棒哒！");
	}

	public static void addPackage(File file) {
		String packageName = getPackage(file.getPath());
		if (packageName == null) {
			return;
		}
		try {
			BufferedReader in = new BufferedReader(new FileReader(file));
			String s;
			StringBuilder sb = new StringBuilder("package " + packageName + ";\n");
			while ((s = in.readLine()) != null) {
				if (s.indexOf("package") == 0) {
					return;
				}
				sb.append(s + "\n");
			}
			in.close();
			// rewrite the file
			PrintWriter out = new PrintWriter(new FileWriter(file));
			out.print(sb.toString());
			out.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	static TreeInfo recurseDirs(File startDir, String regex) {
		TreeInfo result = new TreeInfo();
		for (File item : startDir.listFiles()) {
			if (item.isDirectory()) {
				result.dirs.add(item);
				result.addAll(recurseDirs(item, regex));
			} else {
				if (item.getName().matches(regex)) {
					result.files.add(item);
				}
			}
		}
		return result;
	}

	public static String getPackage(String pathWithName) {
		StringBuffer pACKAGE = new StringBuffer();
		int p = pathWithName.indexOf("src");
		if (p == -1) {
			return null;
		}
		pathWithName = pathWithName.substring(p + 4);
		while ((p = pathWithName.indexOf("\\")) != -1) {
			pACKAGE.append(pathWithName.substring(0, p));
			pathWithName = pathWithName.substring(p + 1);
			if ((p = pathWithName.indexOf("\\")) != -1) {
				pACKAGE.append('.');
			}
		}

		return pACKAGE.toString();
	}
}

class TreeInfo implements Iterable<File> {
	public List<File> files = new ArrayList<File>();
	public List<File> dirs = new ArrayList<File>();

	// The default iterable element is the file list:
	public Iterator<File> iterator() {
		return files.iterator();
	}

	void addAll(TreeInfo other) {
		files.addAll(other.files);
		dirs.addAll(other.dirs);
	}
}