!7

sinterface StringFunc {
  S get(S s);
}

static O in;
static L<S> markedStrings;

p {
  if (in == null) // may have been set by hotwire client
    in = "create table `bla`(hello int);";
  
  if (args.length != 0) in = loadSnippet(args[0]);
  
  S s = [[
    mysqltok input
    find "insert" ... ";"
  ]];
  L<S> ll = toLinesFullTrim(s);
  //System.out.println(ll);
  
  // remove comments
  ll = map(ll, func(S s) -> S { s.replaceAll("//.*$", "").trim() });
  
  L<S> tok = null;
  
  for (S c : ll) {
    if (eq(c, "mysqltok input"))
      in = tok = mysqlTok((String) in);
    else if (c.startsWith("find")) {
      L<S> pat = javaTok(c);
      pat.remove(0); pat.remove(0);
      L<S> inp = cast in;
      //System.out.println("inp: " + inp);
      //System.out.println("pat: " + pat);
      in = matchTokensList(inp, pat);
      //System.out.println("Result: " + structure(in));
    }
  }
  
  if (in instanceof List) {
    List<int[]> list = cast in;
    markedStrings = getMarkedStrings(tok, list);
    markTokens(tok, list);
    in = join(tok);
  
    System.out.println("Markings found: " + size(list));
    for (int[] ij : list) {
      S bla = join(tok.subList(ij[0], ij[1]+1));
      System.out.println("Got: " + bla);
    }
  }
}

static List<String> getMarkedStrings(L<S> tok, L<int[]> markings) {
  new L<S> result;
  for (int i = 0; i < markings.size(); i++)
    result.add(join(tok.subList(markings.get(i)[0], markings.get(i)[1])));
  return result;
}

static int size(Object o) {
  return ((List) o).size();
}

static void markTokens(L<S> tok, L<int[]> marks) {
  System.out.println(tok.size());
  for (int i = 0; i < tok.size(); i++) {
    String t = tok.get(i);
    if (isStartMark(i, marks))
      t = "[[" + t;
    if (isEndMark(i, marks))
      t += "]]";
    tok.set(i, t);
  }
}

static boolean isStartMark(int i, L<int[]> marks) {
  for (int[] ij : marks)
    if (ij[0] == i)
      return true;
  return false;
}

static boolean isEndMark(int i, L<int[]> marks) {
  for (int[] ij : marks)
    if (ij[1] == i+1)
      return true;
  return false;
}