import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.io.FileWriter;
import java.io.RandomAccessFile;
import javax.swing.BorderFactory;
import javax.swing.JButton;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
/**
* (1)建立一个窗口,用户可以设置一个目录、编写头注释信息;
* (2)查找目录下所有子文件,如果是Java文件则处理之,如果是目录则递归处理;
* (3)处理Java文件时,打开后,找到package语句或者第一个import语句,作为注释的插入点,插入注释;
* (4)将增加了头注释的文件内容写回文件。
*
* @author Chennzhi
*
*/
public class HeaderCommentsGenerator {
private static int count = 0;
public static void main(String[] args) {
final JFrame frame = new JFrame("批量修改头注释");
JPanel contentPane = (JPanel) frame.getContentPane();
JPanel centerPane = new JPanel(new BorderLayout(10, 10));
centerPane.setBorder(BorderFactory.createEmptyBorder(10, 10, 0, 10));
JPanel pathPane = new JPanel(new BorderLayout());
final JTextField txtPath = new JTextField();
txtPath.setText("Please select your file or path.");
pathPane.add(txtPath, BorderLayout.CENTER);
JButton btnSelectPath = new JButton("Browser");
btnSelectPath.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
JFileChooser chooser = new JFileChooser();
chooser.setFileSelectionMode(JFileChooser.FILES_AND_DIRECTORIES);
int returnVal = chooser.showOpenDialog(frame);
if (returnVal == JFileChooser.APPROVE_OPTION) {
txtPath.setText(chooser.getSelectedFile().getAbsolutePath());
}
}
});
btnSelectPath.setMnemonic('B');
pathPane.add(btnSelectPath, BorderLayout.EAST);
centerPane.add(pathPane, BorderLayout.NORTH);
final JTextArea txtComments = new JTextArea();
txtComments
.setText("/*\n"
+ " * Copyright 1992-2013 chennzhi Software, Inc. All rights reserved.\n"
+ " */\n\n");
centerPane.add(new JScrollPane(txtComments), BorderLayout.CENTER);
contentPane.add(centerPane, BorderLayout.CENTER);
JPanel buttonPane = new JPanel(new FlowLayout(FlowLayout.RIGHT, 10, 10));
JButton btnOK = new JButton("Generate!");
btnOK.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
String path = txtPath.getText();
File file = new File(path);
if (!file.exists()) {
JOptionPane.showMessageDialog(frame, "Path '" + path
+ "' not exist.", "Error",
JOptionPane.ERROR_MESSAGE);
} else {
commentFile(file, txtComments.getText());
JOptionPane.showMessageDialog(frame, "Finish, total "
+ count + " files are processed.", "Information",
JOptionPane.INFORMATION_MESSAGE);
}
}
});
btnOK.setMnemonic('G');
JButton btnClose = new JButton("Close");
btnClose.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
System.exit(0);
}
});
btnClose.setMnemonic('C');
buttonPane.add(btnOK);
buttonPane.add(btnClose);
contentPane.add(buttonPane, BorderLayout.SOUTH);
frame.setSize(500, 300);
frame.show();
}
private static void commentFile(File file, String comments) {
if (file != null && file.exists()) {
if (file.isDirectory()) {
String[] children = file.list();
for (int i = 0; i < children.length; i++) {
File child = new File(file.getPath()
+ System.getProperty("file.separator")
+ children[i]);
commentFile(child, comments);
}
} else {
if (file.getName().toLowerCase().endsWith(".java")) {
System.out.println(file.getName());
count++;
try {
RandomAccessFile raFile = new RandomAccessFile(file,
"rw");
byte[] content = new byte[(int) raFile.length()];
raFile.readFully(content);
String all = new String(content);
all = all.trim();
while (all.startsWith("\n")) {
all = all.substring(1);
}
if (all.indexOf("package") != -1) {
all = all.substring(all.indexOf("package"));
}
if (all.indexOf("import") != -1) {
all = all.substring(all.indexOf("package"));
}
all = comments + "\n" + all;
raFile.close();
FileWriter writer = new FileWriter(file);
writer.write(all);
writer.close();
} catch (Exception ex) {
ex.printStackTrace();
}
}
}
}
}
}