#include <GL/sgl.hpp>

class ColorBoxWindow: public sgl::Window {
protected:
    //  Access to this custom popup menu class is restricted to
    //  code within this ColorBoxWindow class.
    class ColorSelectorPopup : public sgl::PopupMenu {
        ColorBoxWindow *win;
    public:
        //  The constructor associates a menu item with an action
        //  to perform.  Each action is a method within this class.
        ColorSelectorPopup(ColorBoxWindow *parent) : win(parent) {
            add_menu_item("Red", [this]() { win->set_color_red(); });
            add_menu_item("Green", [this]() { win->set_color_green(); });
            add_menu_item("Blue", [this]() { win->set_color_blue(); });
            add_menu_item("Black", [this]() { win->set_color_black(); });
            add_menu_item("Quit", [this]() { win->quit(); });
        }
    };
    sgl::Color color;
    ColorSelectorPopup menu;
public:
    ColorBoxWindow(): sgl::Window("Color Box", 100, 100, 400, 300,
        0, 100, 0, 100), color(sgl::RED), menu(this){}

    void paint() override {
        sgl::set_color(color);
        sgl::fill_rectangle(25, 25, 50, 50);
    }

    virtual void set_box_color(const sgl::Color& c) {
        color = c;
        repaint();
    }

    virtual void set_color_green() {
        color = sgl::GREEN;
        repaint();
    }

    virtual void set_color_blue() {
        color = sgl::BLUE;
        repaint();
    }

    virtual void set_color_red() {
        color = sgl::RED;
        repaint();
    }

    virtual void set_color_black() {
        color = sgl::BLACK;
        repaint();
    }

    virtual void quit() {
        exit(0);
    }

    //  Pressing the left mouse button lightens the box
    //  Pressing the left mouse button and the shift key
    //  darkens the box
    void mouse_pressed(double, double, sgl::MouseButton) override {
        double inc = (get_key_modifiers() == sgl::KeyModifier::SHIFT_KEY)
                   ? -0.05 : 0.05;
        color.red += inc;
        color.green += inc;
        color.blue += inc;
        repaint();
    }
};

int main() {
    sgl::run<ColorBoxWindow>();
}