!7

!include once #1017126 // transpileRaw

sclass DevelopFromExamples > DynTextArea {
  S whatToTry;
  transient ReliableSingleThread rst = new(r think);
  transient volatile PlanInMotion<Runnable> currentPlan;
  
  transient int delay = 0; //2000; // to see stuff

  JComponent visualize() {
    final JLabel lblStatus = jlabel();
    awtEvery(lblStatus, 100, r { main.setText(lblStatus, planStatus()) });
    ret withCenteredButtons(centerAndSouthWithMargins(
      jhsplit(super.visualize(), dm_printLogComponent()),
      vstackWithSpacing(
        jsection("Stuff to try", onEnter(dm_fieldTextField('whatToTry), rst)),
        lblStatus
      )), "Fresh", rThread fresh, "Think", rst, "[disabled] Cancel", r cancel);
  }
  
  S planStatus() {
    PlanInMotion plan = currentPlan;
    if (plan == null) ret " ";
    ret "Steps done: " + l(plan.doneSteps) + ". Steps planned: " + l(plan.plannedSteps);
  }
  
  void cancel {
    PlanInMotion plan = currentPlan;
    if (plan != null) plan.cancelled = true;
  }
  
  void fresh {
    dm_refreshTranspiler();
    forgetTranspiledStandardFunctions(words(whatToTry));
    loadFunctions_clearCache();
    rst.trigger();
  }
  
  void think {
    if (currentPlan != null) ret;
    new PlanInMotion<Runnable> plan;
    currentPlan = plan;
    temp tempAfterwards(r { currentPlan = null; });
    temp tempEnableButton(findButton(dm_vis(), "Cancel"));
    
    clearPrintLog();
    loadFunctions_preferDiskCache();
    
    S text = rtrimAll(getText());
    Pair<File, Bool> examplesFile = ai_makeExamplesFile(text);
    if (examplesFile.b) print("Wrote: " + f2s(examplesFile.a));
    final new L<Pair<S>> examples;
    bool eqic = false;
    
    L<S> lines = tlftj(text);
    if (eq(first(lines), "splitAtEmptyLines"))
      lines = splitAtEmptyLines(text);
      
    for (S line : lines) pcall {
      if (eqic(line, "eqic")) eqic = true;
      L<S> l = splitAtDoubleArrow_bothDirections(line);
      if (l(l) == 2) addPair(examples,
        tok_multiLineQuotesToStandardQuotes(first(l)),
        safeCanonicalizeStructure(tok_multiLineQuotesToStandardQuotes(second(l))));
    }
    
    final File logFile = replaceExtension(examplesFile.a, ".log");
    final LineBuffer lineBuffer = new(voidfunc(S line) { appendToFile(logFile, line + "\n") });
    temp tempInterceptPrint(func(S s) -> Bool { lineBuffer.append(s); true; });
    
    print(localDateWithMilliseconds());
    print("Thinking about " + n2(examples, "example"));
    
    final new LinkedHashSet<S> thingsThatWork;
    final F2<O, O, Bool> equalsFunction = eqic
        ? func(O a, O b) -> Bool { eqic((S) a, (S) b) }
        : func(O a, O b) -> Bool { eq(a, b) };

    // Make plan
    for (fS s : words(whatToTry)) if (isStandardFunction(s)) plan.add(r {
      sleep(delay);
      F1<S, S> f = func(S in) -> S { struct(callAndMake_orDirect(s, unstruct(in))) };
      T3<S, S, O> counterExample = worksOnAllExamples_returnCounterExample(f, examples, equalsFunction);
      if (counterExample == null)
        print("WORKS: " + addAndReturn(thingsThatWork, s));
      else {
        print("Nope: " + s);
        bool exc = counterExample.c instanceof Throwable;
        print((exc ? "Exception on:    "
                   : "Wrong result on: ") + sfu(counterExample.a));
        print(       "Expected:        " + sfu(counterExample.b));
        if (exc)
          printStackTrace((Throwable) counterExample.c);
        else
          print(     "         Result: " + sfu(counterExample.c));
      }
    });
    
    for (Runnable r : plan.master())
      callF(r);
      
    print("\n" + (plan.cancelled ? "Cancelled. " : "Done. ") + (empty(thingsThatWork) ? (plan.cancelled ? "Nothing worked so far." : "Nothing works.") : (plan.cancelled ? "Things that worked so far: " : "Things that work: ") + joinWithComma(thingsThatWork)));
    print();
  }
}