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

  }

}