!7

concept RewriteExample {
  S in, out;
  O trail;
}

concept Result {
  S in, out;
  O trail;
  RewriteExample existing;
}

cmodule RewriteDatabase {
  transient CRUD<RewriteExample> rewriteExamplesCRUD;
  transient CRUD<Result> resultsCRUD;
  transient SingleComponentPanel scpSuggestions;
  transient ReliableSingleThread rstMakeSuggestions = dm_rst(me(), r makeSuggestions).cancelBeforeTrigger();
  transient ReliableSingleThread rstProcessInput = dm_rst(me(), r processInput).cancelBeforeTrigger();
  S input;
  int selectedTab;
  switchable int maxResults = 1000;
  
  start {
    dbIndexingCI(RewriteExample, 'in);
    dbIndexingCI(Result, 'out);
    rewriteExamplesCRUD = CRUD(RewriteExample);
    resultsCRUD = CRUD(Result);
  }
  
  visualize {
    JComponent c = dm_rememberSelectedTab selectedTab(jtabs(
      "Rewrite Examples",
        centerAndSouthWithMargin(
          rewriteExamplesCRUD.visualize(),
          jCenteredSection("Suggestions", jMinHeight(100, scpSuggestions = singleComponentPanel()))),
      "Input",
        northAndCenterWithMargins(
          withRightAlignedButtons(dm_centeredTextFieldAsSection input(),
            "Make rule...", rThreadEnter makeRuleDialog),
          jCenteredSection("Results", resultsCRUD.visualize()))
    ));
    resultsCRUD.addSelectionDependentButton("Accept", r { acceptResult(resultsCRUD.selected()) });
    resultsCRUD.addSelectionDependentButton("Use as new input", r { setField(input := resultsCRUD.selected().out) });
    onTableSelectionChanged(rewriteExamplesCRUD.table(), rstMakeSuggestions);
    onConceptChangeByClass_notOnAllChanged RewriteExample(r { rstMakeSuggestions.trigger(); rstProcessInput.trigger(); });
    dm_watchFieldAndNow input(rstProcessInput);
    ret c;
  }

  void makeSuggestions {
    RewriteExample e = rewriteExamplesCRUD.selected(), ret if null;
    new L<JComponent> buttons;
    
    S transferred = transferCurlyBracketPhrases(e.out, e.in);
    if (neq(transferred, e.in))
      buttons.add(jbutton("Change LHS to: " + transferred, r { cset(e, in := transferred) }));

    Set<S> sharedPhrases = sharedCurlyBracketPhrases(e.in, e.out);
    if (nempty(sharedPhrases)) {
      S var = firstUnusedCountingDollarVar(e.in);
      for (S phrase : sharedPhrases) {
        S in2 = jreplaceIC_literal(e.in, curly(phrase), var);
        S out2 = jreplaceIC_literal(e.out, curly(phrase), var);
        if (!hasConceptIC RewriteExample(in := in2, out := out2))
          buttons.add(jbutton("Replace " + curly(phrase) + " with " + var, r {
            addExample(in2, out2, ll('sharedCurlyBracketPhrases, e))
          }));
      }
    }

    setComponent(scpSuggestions, scrollableStackWithSpacing(buttons));
  }

  void processInput {
    input = simplify(this.input);
    print("Processing input: " + input);
    deleteConcepts Result();

    // flexMatch input with all examples
    for ping (RewriteExample e) {
      if (containsDollarVars(e.in))
        for (SS map : flexMatchDollarVarsIC_all(e.in, input))
          addResult(cnew Result(in := input, out := replaceVars(e.out, map), trail := ll('flexMatch, e)));
    }

    // Offer grouping last tokens
    LS tok = javaTokWithAllBrackets_cached(input);
    for (S s : tok_groupAfterNTokens_all(tok))
      addResult(cnew Result(in := input, out := s));

    // Offer grouping first tokens
    for (S s : tok_groupFirstNTokens_all(tok))
      addResult(cnew Result(in := input, out := s));      
  }

  void acceptResult(Result r) {
    if (r == null || r.existing != null) ret;
    RewriteExample e = addExample(r.in, r.out, r.trail);
    if (e == null) ret;
    cset(r, existing := e);
  }

  void addResult(Result r) {
    cset(r, existing := conceptWhereCI RewriteExample(in := r.in, out := r.out));
    if (countConcepts Result() >= maxResults) quickFail("Maximum number of results reached");
  }

  void makeRuleDialog {
    JTextField tfIn = jTextField(input);
    JTextField tfOut = jTextField(input);
    showFormTitled2("Make rule",
      "In",  tfIn,
      "Out", tfOut,
      r {
        RewriteExample e = addExample(getText(tfIn), getText(tfOut), 'userDialog);
        infoBox(e == null ? "Rule exists" : "Rule made: " + e.id);
      });
  }

  S simplify(S s) {
    ret simpleSpacesAndTrim(s);
  }

  RewriteExample addExample(S in, S out, O trail) {
    ret csetAndReturn(uniqIC_returnIfNew RewriteExample(in := simplify(in), out := simplify(out)), +trail);
  }
}