!595

import java.util.*;
import java.io.*;
import java.util.regex.*;

public class main {
  static String[] standardClasses = {
    "java.util.*",
    "java.util.zip.*",
    "java.util.List",
    "java.util.regex.*",
    "java.util.concurrent.*",
    "java.util.concurrent.atomic.*",
    "javax.swing.*",
    "javax.swing.event.*",
    "javax.swing.text.*",
    "javax.swing.table.*",
    "java.io.*",
    "java.net.*",
    "java.lang.reflect.*",
    "java.lang.ref.*",
    "java.lang.management.*",
    "java.security.*",
    "java.security.spec.*",
    "java.awt.*",
    "java.awt.event.*",
    "java.awt.image.*",
    "javax.imageio.*",
    "java.math.*"
  };
  
  public static void main(String[] args) throws IOException {
    String s = loadMainJava();
    List<String> imports = findImports(s);
    for (String c : standardClasses)
      if (!(imports.contains(c)))
        s = "import " + c + ";\n" + s;
    saveMainJava(s);
  }
  
  static List<String> findImports(String src) {
    List<String> imports = new ArrayList<String>();
    for (String line : toLines(src)) {
      Matcher matcher = Pattern.compile("^\\s*import\\s+(.*?)\\s*;").matcher(line);
      if (matcher.find())
        imports.add(matcher.group(1));
    }
    return imports;
  }
}