!636
!sf
!multi-line strings
!1000635 // stringfunc {
!723 // S, O etc.
!1000636 // compare strings with ==
!1000482 // quick cast
!JavaTok
!class _javax 19

interface StringFunc {
  S get(S s);
}

main {
  static O in;
  
  psvm {
    if (in == null) // may have been set by hotwire client
      in = "dump for table `mytable` and table `anothertable`";
    
    if (args.length != 0) in = loadSnippet(args[0]);
    
    S s = [[
      example output: "mytable, anothertable"
      javatok input // rather, mysqltok... ^^
      find "table" "`" *
      fix duplicates
      fix output
    ]];
    L<S> ll = toLinesFullTrim(s);
    //System.out.println(ll);
    
    ll = map(ll, stringfunc { s.replaceAll("//.*$", "").trim() });
    //System.out.println(ll);
    O expectedOut = null;

    for (S c : ll) {
      if (c.startsWith("example output:"))
        expectedOut = unquote(c.substring("example output:".length()).trim());
      else if (c == "javatok input")
        in = JavaTok.split((String) in);
      else if (c.startsWith("find")) {
        L<S> tok = JavaTok.split(c);
        tok.remove(0); tok.remove(0);
        tok = map(tok, stringfunc { unquote(s) });
        L<S> inp = cast in;
        //System.out.println("inp: " + inp);
        //System.out.println("pat: " + tok);
        in = matchTokensList(inp, tok);
        //System.out.println("Result: " + in);
      } else if (c == "fix output") {
        if (in instanceof List && expectedOut instanceof String)
          in = join(", ", in);
      } else if (c == "fix duplicates") {
        new L<S> list;
        L<S> l = cast in;
        for (int i = 0; i < l.size(); i++)
          if (i == 0 || !l.get(i).equals(l.get(i-1)))
            list.add(l.get(i));
        in = list;
      }
    }
    
    if (expectedOut != null)
      System.out.println("Expected: " + expectedOut);
    System.out.println("Got: " + in);
    if (expectedOut != null && expectedOut.equals(in))
      print "OK!"
  }
  
  static S join(S glue, O o) {
    return join(glue, (L<S>) o);
  }
  
  !include #2000507 // original join function
  
  // returns a list of the matched tokens
  static L<S> matchTokensList(L<S> inp, L<S> pat) {
    new L<S> list;
    for (int i = 1; i+pat.size()-3 < inp.size(); i += 2) {
      S result = matchTokens_step(inp, pat, i);
      if (result != null)
        list.add(result);
    }
    return list;
  }
  
  // returns
  // -token matched for *
  // -"" on match without *
  // -null on non-match
  static String matchTokens(L<S> inp, L<S> pat) {
    for (int i = 1; i+pat.size()-3 < inp.size(); i += 2) {
      S result = matchTokens_step(inp, pat, i);
      if (result != null)
        return result;
    }
    return null;
  }
  
  static String matchTokens_step(L<S> inp, L<S> pat, int i) {
    S result = "";
    for (int j = 1; j < pat.size(); j += 2)
      if (pat.get(j).equals("*"))
        result = inp.get(i+j-1);
      else {
        String found = inp.get(i + j - 1);
        String expected = pat.get(j);
        if (!found.equals(expected))
          return null;
      }
    return result;
  }
  
  static L<S> map(L<S> l, StringFunc f) {
    new L<S> l2;
    for (S s : l)
      l2.add(f.get(s));
    return l2;
  }
}