给java代码加一行PACKAGE(在文件头添加一行)

给java代码加一行PACKAGE(在文件头添加一行)

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
public class Tet {
    private static String initDir;
    
    private void getFiles(String dir) throws IOException{
	File file=new File(dir);
	File[] files=file.listFiles();
	for (File file2 : files) {
	    if(file2.isDirectory()){
		getFiles(file2.getPath());
	    }else if(file2.isFile()&&file2.getName().endsWith("java")){
		modify(file2.getPath(), dir);
	    }
	}
    }
    
    private void modify(String filePath,String dir) throws IOException{
	System.out.println(filePath);
	File file=new File(filePath);
	    BufferedReader br = new BufferedReader(new InputStreamReader(
		    new FileInputStream(file)));
	    List list = new ArrayList();
	    String f2=file.getParent();
	    System.out.println(f2+"\n"+f2.replace(initDir, ""));
	    String packageString="package "+f2.replace(initDir, "").replace("\\", ".").replaceFirst(".", "")+";";
	    list.add(packageString);
	    while (true) {
		String str = br.readLine();
		if (str == null)
		    break;
		int index = str.indexOf(packageString);
		if (index != -1) {
//		    str.replace(packageString, "");
		    continue;
		}
		list.add(str);

	    }
	    br.close();

	    PrintWriter pw = new PrintWriter(file);
	    for (int i = 0; i < list.size(); i++) {
		String str = (String) list.get(i);
		pw.println(str);
	    }
	    pw.close();
    }
    
    public static void main(String[] args) {
	initDir="D:\\Work\\Workspace64\\Think in Java 4 code\\src";
	try {
	    new Tet().getFiles(initDir);
	} catch (IOException e) {
	    e.printStackTrace();
	}
    }
}

猜你喜欢

转载自crabdave.iteye.com/blog/2233241