/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use glib::IsA;
use gtk::{prelude::*, Box, Button, Widget};

pub trait ClosableTabImplementedNotebook {
    fn prepend_closable_page<T: IsA<Widget>, U: IsA<Widget>>(
        &self,
        child: &T,
        tab_label: Option<&U>,
    ) -> u32;
    fn append_closable_page<T: IsA<Widget>, U: IsA<Widget>>(
        &self,
        child: &T,
        tab_label: Option<&U>,
    ) -> u32;

    fn create_closable_tab<U: IsA<Widget>>(tab_label: Option<&U>) -> (Box, Button);
}

impl ClosableTabImplementedNotebook for gtk::Notebook {
    fn create_closable_tab<U: IsA<Widget>>(tab_label: Option<&U>) -> (Box, Button) {
        let tab = Box::new(gtk::Orientation::Horizontal, 5);
        if tab_label.is_some() {
            tab.append(tab_label.unwrap());
        }

        let button = gtk::Button::new();

        button.set_icon_name("window-close-symbolic");
        button.set_has_frame(false);

        tab.append(&button);

        (tab, button)
    }

    fn prepend_closable_page<T: IsA<Widget>, U: IsA<Widget>>(
        &self,
        child: &T,
        tab_label: Option<&U>,
    ) -> u32 {
        let (tab, button) = &Self::create_closable_tab(tab_label);
        let page = self.prepend_page(child, Some(tab));

        button.connect_clicked(glib::clone!(@weak self as notebook =>
            move |_| {
            notebook.remove_page(Some(page));
        }));

        page
    }

    fn append_closable_page<T: IsA<Widget>, U: IsA<Widget>>(
        &self,
        child: &T,
        tab_label: Option<&U>,
    ) -> u32 {
        let (tab, button) = &Self::create_closable_tab(tab_label);
        let page = self.append_page(child, Some(tab));

        button.connect_clicked(glib::clone!(@weak self as notebook =>
            move |_| {
            notebook.remove_page(Some(page));
        }));

        page
    }
}