!7

sS background = #1009931;
static JDesktopPane desktop;
static ReliableSingleThread rst = new(f updateModules);
static volatile long updateCycles;

p-noconsole {
  autoRestart();
  db();

  desktop = jDesktopPaneWithSkyPicture(background, Color.black);
  autoFixDesktopPane(desktop);
  cleanExitOnFrameClose(frameIcon(#1101272, showMaximizedFrame(desktop)));
  titlePopupMenu_top(desktop, voidfunc(JPopupMenu menu) {
    addMenuItems(menu,
      "Update One Cycle", rst,
      "New Session", rThreadPcallMessageBox(r deleteAllModules),
    );
  });
    
  initAfterDBLoad();
}

svoid initAfterDBLoad {
  initialModules();
  
  UpdateCycles uc = conceptWhere(UpdateCycles);
  if (uc != null) updateCycles = uc.value;
  
  for (Module m : onModules()) startModule(m);
  
  addConceptIndex(simpleConceptIndex(rst));
  rst.trigger();
}

svoid initialModules {
  if (!hasConcept(ModuleClasses))
    showModule(uniq(ModuleClasses));
}

svoid updateModules {
  ++updateCycles;
  for (Module m : onModules())
    pcall { m.update(); }
}

svoid cleanMeUp {
  for (Module m) if (m.vis != null) pcall {
    m.unvisualize();
  }
  
  for (Module m) cleanUp(m);
}

static L<Module> onModules() { ret conceptsWhere(Module, on := true); }

sbool hasModuleWithFields(Class<? extends Module> c, O... params) {
  ret hasConcept(c, concatArrays(new O[] {on := true}, params));
}

svoid startModule(Module m) {
  //addIfNotThere(modules, m);
  lock m.lock;
  if (m.started) ret;
  m.started = true;
  print("Starting module " + m);
  m.start();
  if (m.visible) showModule(m);
}

svoid showModule(final Module m) {
  if (m == null) ret;
  startModule(m);
  lock m.lock;
  if (m.vis != null) ret;
  cset(m, visible := true);
  visualizeModule(m);
  if (m.vis != null) swing {
    Rect r = m.frameRect;
    if (r == null) r = randomRect(desktop.getWidth(), desktop.getHeight(), 10, 150, 100);
    if (r == null) r = Rect(10, 10, 200, 100);
    print("Showing frame at " + r);
    S frameTitle = humanizeFormLabel(shortClassName(m));
    final JInternalFrame f = showInternalFrame(desktop, frameTitle, r.x, r.y, r.w, r.h, m.vis);
    
    internalFrameTitlePopupMenuItem(f, "Module source", r-thread { pcall-messagebox {
      S src = m.sourceCode();
      if (src != null) showText(internalFrameTitle(f) + " [Source]", src); else infoBox("No source code found");
    }});
    onInternalFrameIconified(f, r { hideModule(m) });
    onInternalFrameClosing(f, r { deleteModule(m) /* really? */ });
    internalFrameIcon(f, m.iconID);
    m.enhanceFrame(f);
  }
}

svoid deleteModule(Module m) { removeConcept(m); }

svoid visualizeModule(Module m) {
  pcall-messagebox {
    if (m.vis == null) m.vis = m.visualize();
    if (m.vis == null) m.vis = defaultVisualize(m);
  }
}

svoid hideModule(final Module m) {
  if (m == null) ret;
  lock m.lock;
  cset(m, visible := false);
  pcall-messagebox { m.unvisualize(); }
}

svoid revisualizeModule(Module m) {
  pcall-messagebox {
    if (m == null) ret;
    lock m.lock;
    JInternalFrame frame = getInternalFrame(m.vis);
    m.unvisualize1b();
    m.unvisualize2();
    visualizeModule(m);
    setInternalFrameContents(frame, m.vis);
  }
}



abstract concept Module {
  transient Component vis;
  transient bool started;
  transient Lock lock = lock();
  
  bool on = true, visible;
  Rect frameRect;
  S iconID;
  
  JComponent visualize() { null; }
  void unvisualize() { unvisualize1(); unvisualize2(); }
  void enhanceFrame(JInternalFrame frame) {}
  void start() {}
  void unvisualize1() {
    disposeInternalFrame(getInternalFrame(vis));
    unvisualize1b();
  }
  
  void unvisualize1b() {
    grabFrameRect();
    vis = null;
  }
  void unvisualize2() {}
  void update() {}
  
  void grabFrameRect() {
    JInternalFrame f = getInternalFrame(vis);
    if (f != null)
      cset(this, frameRect := toRect(getBounds(f)));
  }
  
  void cleanMeUp_started() { started = false; }

  void delete() {
    unvisualize();
    cleanUp(this);
    super.delete();
  }

  S sourceCode() {  
    ret javaxSourceOfMyClass1(shortClassName(this));
  }
  
  void triggerUpdate { rst.trigger(); }
  
  void setModuleIcon(S iconID) {
    if (eq(iconID, this.iconID)) ret;
    this.iconID = iconID;
    internalFrameIcon(vis, iconID);
  }
  
} // END CONCEPT MODULE

static JComponent defaultVisualize(Module m) {
  ret jCenteredMultiLineLabel(renderConcept(m)); 
}

static <A extends Module> A findModule(Class<A> c) {
  ret findConcept(c, on := true);
}

static O findDynModuleOfType(S type) {
  DynamicModule m = findConcept(DynamicModule, on := true, _className := "main$" + type);
  ret m == null ? null : m.o;
}

sS getInterestingString {
  InterestingString m = findModule(InterestingString);
  ret m == null ? null : m.theString;
}

sS modulesSessionGrab() {
  grabFrameRects();
  ret struct(ll(programID(), localDateWithMilliseconds())) + "\n"
    + mainConcepts.xfullgrab();
}

svoid autoSaveModulesSession() {
  infoBox("Auto-saving session.");
  S grab;
  logQuoted(javaxBackupDir(fsI(programID()) + "/auto-saved-sessions.txt"), grab = modulesSessionGrab());
  infoBox("Auto-save done (" + l(grab) + " chars)");
}

svoid deleteAllModules() {
  autoSaveModulesSession();
  deleteConcepts(Module);
  initialModules;
}

svoid restoreModulesSession(S text) {
  autoSaveModulesSession();
  infoBox("Releasing session");
  cleanMeUp();
  cleanUp(mainConcepts);
  mainConcepts = null;
  //sleepSeconds(1);
  infoBox("Loading session");
  mainConcepts = new Concepts().load(dropFirstLine(text));
  initAfterDBLoad();
  infoBox("Session restore done");
}

svoid grabFrameRects {
  for (Module m : onModules()) m.grabFrameRect();
}

sclass DynamicModule extends Module {
  S moduleID, _className; // "className" is taken by DynamicObject; _className == null for old-style dyn module
  S oStruct; // serialized dynamic object
  
  transient Class c;
  transient O o;

  *() {}
  *(S *moduleID, S *_className) {}
  *(S *moduleID, S *_className, Class *c) {}

  JComponent visualize() { ret (JComponent) callOpt(o, 'visualize); }
  void enhanceFrame(final JInternalFrame f) {
    internalFrameTitlePopupMenuItem(f, "Reload", rThread(r reload));
    thread { internalFrameTitle(f, dropSuffixICTrim("[Dyn Module]", snippetTitle(moduleID))); }
  }
  void start() {
    if (moduleID == null) ret;
    if (c == null) c = hotwireDependent(moduleID);
    if (oStruct != null) pcall {
      o = unstructureInRealm(oStruct, c);
    }
    if (o == null)
      if (_className == null)
        o = c;
      else
        o = nu(_getClass(c, _className));
    setOpt(o, _host := this);
    if (o instanceof Class)
      callMain(o);
    else
      callOpt(o, 'start);
  }
  void unvisualize2() { callOpt(o, 'unvisualize2); }
  void update() { callOpt(o, 'update); }
  void cleanMeUp() {
    oStruct = null; pcall { if (o != null && !o instanceof Class) oStruct = struct(o); }
    cleanUp(o);
    if (!o instanceof Class)
      cleanUp(_getClass(o, "main"));
    o = null;
    c = null;
  }
  
  void reload() {
    JInternalFrame frame = getInternalFrame(vis);
    unvisualize1b();
    unvisualize2();
    cleanUp(this); // also sets started to false
    if (frame != null)
      setInternalFrameContents(frame, jcenteredlabel("Reloading..."));
    visible = false;
    startModule(this);
    if (frame != null) {
      cset(this, visible := true);
      visualizeModule(this);
      print("New content: " + vis);
      setInternalFrameContents(frame, vis);
    }
    rst.trigger();
  }
  
  S sourceCode() {  
    ret loadSnippet(moduleID);
  }
}

!include once #1015885 // standard modules
!include once #1015842 // SavedSessions