#include <functional>        // for function
#include <initializer_list>  // for initializer_list
#include <string>            // for wstring, allocator
#include <vector>            // for vector

#include "ftxui/component/component.hpp"           // for Component
#include "ftxui/component/container.hpp"           // for Container
#include "ftxui/component/menu.hpp"                // for Menu
#include "ftxui/component/screen_interactive.hpp"  // for ScreenInteractive
#include "ftxui/dom/elements.hpp"                  // for operator|, Element
#include "ftxui/screen/box.hpp"                    // for ftxui
#include "ftxui/screen/color.hpp"                  // for Color, Color::Blue

using namespace ftxui;

class MyComponent : public Component {
 public:
  MyComponent() {
    Add(&container);

    for (Menu* menu : {
             &menu_1,
             &menu_2,
             &menu_3,
             &menu_4,
             &menu_5,
             &menu_6,
         }) {
      container.Add(menu);
      menu->entries = {
          L"Monkey", L"Dog", L"Cat", L"Bird", L"Elephant",
      };
      menu->on_enter = [this]() { on_enter(); };
    }

    menu_2.focused_style = bold | color(Color::Blue);
    menu_2.selected_style = color(Color::Blue);
    menu_2.selected_focused_style = bold | color(Color::Blue);

    menu_3.selected_style = color(Color::Blue);
    menu_3.focused_style = bgcolor(Color::Blue);
    menu_3.selected_focused_style = bgcolor(Color::Blue);

    menu_4.selected_style = bgcolor(Color::Blue);
    menu_4.focused_style = bgcolor(Color::BlueLight);
    menu_4.selected_focused_style = bgcolor(Color::BlueLight);

    menu_5.normal_style = bgcolor(Color::Blue);
    menu_5.selected_style = bgcolor(Color::Yellow);
    menu_5.focused_style = bgcolor(Color::Red);
    menu_5.selected_focused_style = bgcolor(Color::Red);

    menu_6.normal_style = dim | color(Color::Blue);
    menu_6.selected_style = color(Color::Blue);
    menu_6.focused_style = bold | color(Color::Blue);
    menu_6.selected_focused_style = bold | color(Color::Blue);
  }

  std::function<void()> on_enter = []() {};

 private:
  Container container = Container::Horizontal();
  Menu menu_1;
  Menu menu_2;
  Menu menu_3;
  Menu menu_4;
  Menu menu_5;
  Menu menu_6;

  // clang-format off
   Element Render() override {
     return
        hbox({
          menu_1.Render() | flex, separator(),
          menu_2.Render() | flex, separator(),
          menu_3.Render() | flex, separator(),
          menu_4.Render() | flex, separator(),
          menu_5.Render() | flex, separator(),
          menu_6.Render() | flex,
        }) | border;
   }
  // clang-format on
};

int main(int argc, const char* argv[]) {
  auto screen = ScreenInteractive::TerminalOutput();
  MyComponent component;
  component.on_enter = screen.ExitLoopClosure();
  screen.Loop(&component);
}

// Copyright 2020 Arthur Sonzogni. All rights reserved.
// Use of this source code is governed by the MIT license that can be found in
// the LICENSE file.