!7

// only from machines that run #1020927 with postToServer=true

cmodule ServerLoadsTrayIcon > DynPrintLog {

  int size = goodSystemTrayIconSize().x; // icon width+height

  transient double expiryTime = 10.0;

  transient TrayIcon trayIcon; 

  transient Map<S, Load> loads = expiringMap(expiryTime);

  transient LS colors = ll("FF0000", "00FF00", "0000FF");

  transient S status;

  srecord Load(int cores, double percent) {}

  start {

    ownResource(tempTrayIcon(trayIcon = installTrayIcon(whiteImage(size, size), "Server CPUs", rThread dm_activateOSAndModule)));

    doEvery(1.0, r updateIcon);

    dm_onRemoteSystemStatusPosted((computerID, msg) -> {

      LS l = regexpFirstGroupsIC([[System CPU: (\d+)% \((\d+) core]], msg);

      if (l == null) ret;

      loads.put(computerID, new Load(parseInt(second(l)), parseInt(first(l))));

    });

  }

  void updateIcon enter {

    Map<S, Load> map = cloneMap(loads);

    setField(status := joinMap(", ", map, (server, load)

      -> server + ": " + iround(load.percent/load.cores) + "%"));

    int cores = 0;

    for (Load l : values(map)) cores += l.cores;

    BufferedImage img;

    double totalLoad = 0;

    if (cores == 0) img = whiteImage(1, 1);

    else {

      img = whiteImage(cores, size);

      int x = 0, i = 0;

      // go alphabetical in computer IDs

      for (Load l : valuesSortedByKeys(map)) {

        totalLoad += l.percent;

        Color color = colorFromHex(colors.get((i++) % l(colors)));

        int h = iround(l.percent/l.cores/100*size);

        fillRect(img, x, size-h, l.cores, h, color);

        x += l.cores;

      }

    }

    setTrayIconImage(trayIcon, resizeImage(img, size, size));

    trayIcon.setToolTip(nServers(map) + ", " + nCores(cores) + ", " + "total load: " + iround(totalLoad/cores) + "%");

  }

  visual centerAndSouthWithMargins(super, jCenteredSection("Server loads", dm_fieldLabel status()));

}