01: package net.sf.mockcreator.plugin.popup.actions;
02:
03: import java.io.BufferedReader;
04: import java.io.FileNotFoundException;
05: import java.io.FileReader;
06: import java.io.FileWriter;
07: import java.util.ArrayList;
08: import java.util.Iterator;
09: import java.util.List;
10:
11: public class FileUpdater {
12:
13: public void updateFile(String fileName, String className) {
14: className = className.trim();
15:
16: List lst = new ArrayList();
17:
18: BufferedReader br = null;
19: try {
20: br = new BufferedReader(new FileReader(fileName));
21: } catch (FileNotFoundException ex) {
22: // that's ok, no file yet
23: }
24:
25: try {
26: if (br != null) {
27: while (true) {
28: String line = br.readLine();
29: if (line == null)
30: break;
31:
32: line = line.trim();
33: if (className.equals(line))
34: return;
35: lst.add(line);
36: }
37: br.close();
38: }
39:
40: lst.add(className);
41:
42: FileWriter fw = new FileWriter(fileName);
43: for (Iterator it = lst.iterator(); it.hasNext();) {
44: String cn = (String) it.next();
45: fw.write(cn);
46: fw.write('\n');
47: }
48: fw.close();
49: } catch (Exception ex) {
50: throw new RuntimeException(ex);
51: }
52: }
53: }
|