package state.ui.main;

import state.ui.UI;
import state.ui.UIMenu;
import state.ui.UIMenuBuilder;

class Control {
  final State EXITED;
  final State EXIT;
  private State state;
  Control(UI ui) {
    EXITED = new ExitedState();
    EXIT = new ExitState(this,ui);
    state = EXIT;
  }

  void run() {
    while (state != EXITED) {
      state = state.run();
    }
  }
}

interface State {
  public State run();
}

final class ExitedState implements State {
  public State run() {
    return this;
  }
}

final class ExitState implements State {
  Control control;
  UI ui;
  UIMenu m;
  ExitState(Control control, UI ui) {
    this.control = control;
    this.ui = ui;

    UIMenuBuilder mb;
    mb = new UIMenuBuilder();

    //mb.add("Default", new UIMenuAction() { public Object run() {return this;} });
    mb.add("Default", () -> this);
    mb.add("Yes", () -> control.EXITED);
    mb.add("No", () -> control.EXIT);
    this.m = mb.toUIMenu("Are you sure you want to exit?");
  }
  public State run() {
    return (State) ui.processMenu(m);
  }
}