// See #593 for an example!
// Needs a translator (e.g. #595)

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

public class main {
  static String[] standardClasses = {
    "java.util.*",
    "javax.swing.*",
    "java.io.*"
  };
  
  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;
  }

  static String loadMainJava() throws IOException {
    return loadTextFile("input/main.java", "");
  }

  static void saveMainJava(String s) throws IOException {
    saveTextFile("output/main.java", s);
  }

 public static List<String> toLines(String s) {
    List<String> lines = new ArrayList<String>();
    int start = 0;
    while (true) {
      int i = toLines_nextLineBreak(s, start);
      if (i < 0) {
        if (s.length() > start) lines.add(s.substring(start));
        break;
      }

      lines.add(s.substring(start, i));
      if (s.charAt(i) == '\r' && i+1 < s.length() && s.charAt(i+1) == '\n')
        i += 2;
      else
        ++i;

      start = i;
    }
    return lines;
  }

  private static int toLines_nextLineBreak(String s, int start) {
    for (int i = start; i < s.length(); i++) {
      char c = s.charAt(i);
      if (c == '\r' || c == '\n')
        return i;
    }
    return -1;
  }

  /** writes safely (to temp file, then rename) */
  public static void saveTextFile(String fileName, String contents) throws IOException {
    File file = new File(fileName);
    File parentFile = file.getParentFile();
    if (parentFile != null)
      parentFile.mkdirs();
    String tempFileName = fileName + "_temp";
    FileOutputStream fileOutputStream = new FileOutputStream(tempFileName);
    OutputStreamWriter outputStreamWriter = new OutputStreamWriter(fileOutputStream, "UTF-8");
    PrintWriter printWriter = new PrintWriter(outputStreamWriter);
    printWriter.print(contents);
    printWriter.close();
    if (file.exists() && !file.delete())
      throw new IOException("Can't delete " + fileName);

    if (!new File(tempFileName).renameTo(file))
      throw new IOException("Can't rename " + tempFileName + " to " + fileName);
  }

  public static String loadTextFile(String fileName, String defaultContents) throws IOException {
    if (!new File(fileName).exists())
      return defaultContents;

    FileInputStream fileInputStream = new FileInputStream(fileName);
    InputStreamReader inputStreamReader = new InputStreamReader(fileInputStream, "UTF-8");
    return loadTextFile(inputStreamReader);
  }

  public static String loadTextFile(Reader reader) throws IOException {
    StringBuilder builder = new StringBuilder();
    try {
      BufferedReader bufferedReader = new BufferedReader(reader);
      String line;
      while ((line = bufferedReader.readLine()) != null)
        builder.append(line).append('\n');
    } finally {
      reader.close();
    }
    return builder.length() == 0 ? "" : builder.substring(0, builder.length()-1);
  }
}