!7

cmodule DiscordFan > DynPrintLogAndEnabled {

  transient Map<S, BWImage> protos = syncCIMap();

  ISegmenter segmenter;

  transient new ScreenShooter shooter;

  transient ReliableSingleThread rstSegment = dm_rst(module(), r segment);

  transient ReliableSingleThread rstAOA = dm_rst(module(), r aoa);

  transient ImageInfo info;

  transient ImageInfo latestWithAOA;

  transient ImageSurface isThings;

  transient double minScore = 95;

  transient new Thinker thinker;

  switchable double waitBeforeClick = 1.0;

  switchable double patience = 1.0;

  sclass Segment {

    Rect rect;

    BWImage bw;

    *() {}

    *(Rect *rect) {}

  }

  sclass ImageInfo {

    BufferedImage image;

    L<Segment> segments;

    new AOA_CompareImagesToProtos aoa;

    Map<S, Segment> foundProtos;

  }

  visual jvsplit(

    jCenteredSection("Things found on screen", jscroll_center(isThings = jDoubleBufferedImageSurface())),

    super);

  start-thread {

    dm_watchFieldAndNow enabled(r { set(shooter, +enabled) });

    if (empty(protos))

      putAll(protos, mapValues loadBWImage(imageSnippetIDsForNames(llQuoted("Open", "Open discord"))));

    print("Have " + n2(protos, "proto"));

    if (segmenter == null)

      setField(segmenter := segmenterFromAGIBlue("Find things on discordapp.com #1"));

    print("Have segmenter: " + segmenter);

    shooter.onNewImage.add(rstSegment);

    shooter.start(enter := f<AutoCloseable> enter);

    doEvery(1.0, rEnter { thinker.think() });

  }

  void segment {

    BufferedImage img = shooter.latestImage;

    if (!enabled || img == null || segmenter == null) ret; // man weiß ja nie...

    // Have picture!!

    long time = sysNow();

    new ImageInfo info;

    info.image = img;

    info.segments = map(segmenter.get(img), r -> new Segment(r));

    this.info = info;

    done2_always(time, "Make " + nSegments(info.segments));

    rstAOA.trigger();

  }

  void aoa {

    ImageInfo info = this.info;

    if (info == null) ret; // man weiß ja nie...

    long time = sysNow();

    AOA_CompareImagesToProtos aoa = info.aoa;

    aoa.addProtos(protos);

    for (Segment s : info.segments)

      s.bw = toBWImage(bufferedImageClip(info.image, s.rect));

    aoa.addImages(iotaKeys_str(collect bw(info.segments)));

    stepAll(aoa);

    Map<S, Scored<S>> bestImageForProto = aoa.bestImageForProto;

    Map<S, Scored<S>> foundProtos1 = filterMapByValuePredicate(bestImageForProto, v -> score(v) >= minScore/100);

    //print(+foundProtos1);

    Map<S, Segment> foundProtos = ciMap();

    for (S proto, Scored<S> scoredImage : foundProtos1)

      mapPut(foundProtos, proto, getStartingWith1(info.segments, parseInt(scoredImage!)));

    info.foundProtos = foundProtos;

    //print('foundProtos, info.foundProtos);

    if (isThings != null) isThings.setImage(

      mergeBWImagesHorizontally(map bwInverted(collect bw(values(info.foundProtos)))));

    print(mapValues doubleScoreAsIntPercent(bestImageForProto));

    latestWithAOA = info;

    done2_always("AOA", time);

  }

  class Thinker {

    WaitForStableValue<Rect> wait = new(patience);

    void think {

      //print("think " + enabled);

      if (!enabled) ret;

      ImageInfo info = latestWithAOA;

      if (info == null) ret; // man weiß ja nie...

      //print("Keys", sfu(keys(info.foundProtos)));

      Segment s = mapGet(info.foundProtos, quote("Open discord"));

      Rect r = s == null ? null : s.rect;

      print(+r);

      if (wait.set(r) != null) {

        print("Moving mouse to Discord button!!!!");

        O mover = vmBus_query mouseMover();

        if (mover == null) ret with print("No mover");

        callWithImportExport(mover, 'moveMouse, centerOfRect(wait!));

        print("MOVED");

        long startTime = sysNow();

        while (licensed() && sysNow() < startTime+toMS(waitBeforeClick)) {

          if (!enabled) ret;

          if (isTrue_rcall blocked(mover))

            ret with print("INTERFERENCE, EXITING");

          sleepSeconds(0.05);

        }

        // TODO: check mouse position

        print("CLICKING!");

        rcall click(mover);

        print("CLICKED!");

        wait.clear();

      }

    }

  }

  void cleanMeUp {

    dispose shooter;

  }

}