/*
 * Copyright 2023 jacqueline <me@jacqueline.id.au>
 *
 * SPDX-License-Identifier: GPL-3.0-only
 */

#pragma once

#include <memory>
#include <optional>
#include <vector>

#include "core/lv_group.h"
#include "core/lv_obj.h"
#include "core/lv_obj_tree.h"
#include "lvgl.h"

namespace ui {

/*
 * Base class for ever discrete screen in the app. Provides a consistent
 * interface that can be used for transitioning between screens, adding them to
 * back stacks, etc.
 */
class Screen {
 public:
  Screen();
  virtual ~Screen();

  auto root() -> lv_obj_t* { return root_; }
  auto content() -> lv_obj_t* { return content_; }
  auto alert() -> lv_obj_t* { return alert_; }

  auto modal_content() -> lv_obj_t* { return modal_content_; }
  auto modal_group(lv_group_t* g) -> void { modal_group_ = g; }
  auto group() -> lv_group_t* {
    if (modal_group_) {
      return modal_group_;
    }
    return group_;
  }

 protected:
  lv_obj_t* const root_;
  lv_obj_t* content_;
  lv_obj_t* modal_content_;
  lv_obj_t* alert_;

  lv_group_t* const group_;
  lv_group_t* modal_group_;
};

}  // namespace ui