!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;

  }

}