!636
!quickmain
!auto-import
!standard functions

interface Learner {
  void processInOut(String in, String out);
  String processIn(String in);
}

// only "learns" the "id" function
class LId implements Learner {
  public void processInOut(String in, String out) {
  }
  
  public String processIn(String in) {
    return in;
  }
}

main {
  static List<String[]> fullExamples = new ArrayList<String[]>();
  static List<String> halfExamples = new ArrayList<String>();
  static List<String[]> examples1, examples2;
  
  psvm {
    parse(args.length != 0 ? args[0] : null);
    calculate();
  }
  
  static void parse(String arg) tex {
    String text;
    if (arg != null)
      text = loadSnippet(arg);
    else {
      text = loadTextFile("input/input.txt", null);
      if (text == null) text = loadSnippet("#2000377");
    }
      
    System.out.println(text);
    String in = null, out = null;
    
    for (String line : toLines(text)) {
      if (line.startsWith("I")) { // "In: " or "I: "
        if (in != null)
          halfExamples.add(in);
        in = unquote(line.substring(line.indexOf(':')+1).trim());
        out = null;
      } else if (line.startsWith("O")) { // "Out: " or "O: "
        out = unquote(line.substring(line.indexOf(':')+1).trim());
        System.out.println(quote(in) + " => " + quote(out));
        fullExamples.add(new String[] {in, out});
        in = out = null;
      }
    }
    
    if (in != null)
      halfExamples.add(in);
  }
  
  static void calculate() tex {
    if (fullExamples.size() < 2)
      throw new RuntimeException("Too few examples (" + fullExamples.size() + ")");
    int splitPoint = fullExamples.size()-1;
    System.out.println("Full examples: " + fullExamples.size() + ", splitPoint: " + splitPoint);
    examples1 = fullExamples.subList(0, splitPoint);
    examples2 = fullExamples.subList(splitPoint, fullExamples.size());
    
    Learner learner = findOKLearner();
    if (learner == null)
      print "\nProblem not solved"
    else {
      print "\nSolved!\n"
      for (String in : halfExamples) {
        String out = learner.processIn(in);
        System.out.println(quote(in) + " =>! " + quote(out));
      }
    }
  }
  
  static Learner findOKLearner() {
    for (Learner learner : makeLearners()) try {
      if (learnerOK(learner))
        return learner;
    } catch (Throwable e) {
      e.printStackTrace();
    }
    return null;
  }
  
  static boolean learnerOK(Learner learner) {
    for (String[] e : examples1) {
      learner.processInOut(e[0], e[1]);
    }
    for (String[] e : fullExamples) {
      String out = learner.processIn(e[0]);
      if (!e[1].equals(out)) {
        System.out.println("[fail] " + learner + " on " + quote(e[0]) + " - " + quote(out) + " vs " + quote(e[1]));
        return false;
      }
    }
    return true; // all test examples passed
  }
  
  static Iterable<Learner> makeLearners() {
    List<Learner> list = new ArrayList<Learner>();
    list.add(new LId());
    list.add(new LPrefixSuffix());
    return list;
  }
  
  public static String unquote(String s) {
    if (s.startsWith("\"") && s.endsWith("\"") && s.length() > 1)
      return s.substring(1, s.length()-1).replace("\\\"", "\"").replace("\\\\", "\\"); // SHOULD work...
    else
      return s; // Return SOMETHING
  }
  
  public static String quote(String s) {
    if (s == null) return "null";
    return "\"" + s.replace("\\", "\\\\").replace("\"", "\\\"") + "\"";
  }
  
  // learns to exchange common prefixes and suffixes
  static class LPrefixSuffix implements Learner {
    String prefixIn, suffixIn, prefixOut, suffixOut;
    
    public void processInOut(String in, String out) {
      updateIn(in);
      prefixOut = prefixOut == null ? out : commonPrefix(prefixOut, out);
      suffixOut = suffixOut == null ? out : commonSuffix(suffixOut, out);
    }
    
    void updateIn(String in) {
      prefixIn = prefixIn == null ? in : commonPrefix(prefixIn, in);
      suffixIn = suffixIn == null ? in : commonSuffix(suffixIn, in);
    }
    
    public String processIn(String in) {
      //System.out.println("[before last info] " + quote(prefixIn) + " " + quote(suffixIn) + " " + quote(prefixOut) + " " + quote(suffixOut));
      //System.out.println("[last info] " + quote(in));
      
      // use latest information
      String p = prefixIn, s = suffixIn;
      updateIn(in);
      prefixOut = prefixOut.substring(0, prefixOut.length()-(p.length()-prefixIn.length()));
      suffixOut = suffixOut.substring(s.length()-suffixIn.length());
      
      //System.out.println("[after last info] " + quote(prefixIn) + " " + quote(suffixIn) + " " + quote(prefixOut) + " " + quote(suffixOut));
      String core = in.substring(prefixIn.length(), in.length()-suffixIn.length());
      return prefixOut + core + suffixOut;
    }
  }

}