#include "GUI_Utils.hpp"
#include "GUI_App.hpp"
#include <wx/panel.h>
#include <wx/bitmap.h>
#include <wx/image.h>
#include <wx/sizer.h>
#include <wx/gbsizer.h>
#include <wx/webrequest.h>
#include <wx/control.h>
#include <wx/dcclient.h>
#include <wx/display.h>
#include <wx/mstream.h>
#include <wx/sstream.h>
#include <wx/zstream.h>
#include <wx/window.h>
#include <wx/dcgraph.h>
#include <wx/glcanvas.h>
#include <wx/utils.h>
#include <boost/format.hpp>

#include "Widgets/CheckBox.hpp"
#include "Widgets/Label.hpp"

#include "MsgDialog.hpp"
#include "Printer/PrinterFileSystem.h"
#include "PartSkipDialog.hpp"
#include "SkipPartCanvas.hpp"
#include "MediaPlayCtrl.h"


namespace Slic3r { namespace GUI {

extern wxString hide_passwd(wxString url, std::vector<wxString> const &passwords);
extern void     refresh_agora_url(char const *device, char const *dev_ver, char const *channel, void *context, void (*callback)(void *context, char const *url));


PartSkipDialog::PartSkipDialog(wxWindow* parent): DPIDialog(parent, wxID_ANY, _L("Skip Objects"), wxDefaultPosition, wxDefaultSize, wxCAPTION | wxCLOSE_BOX)
{
    std::time_t t = std::time(0);
    std::stringstream buf;
    buf << put_time(std::localtime(&t), "%a_%b_%d_%H_%M_%S/");
    m_timestamp = buf.str();
    
    std::string icon_path = (boost::format("%1%/images/BambuStudioTitle.ico") % Slic3r::resources_dir()).str();
    SetIcon(wxIcon(Slic3r::encode_path(icon_path.c_str()), wxBITMAP_TYPE_ICO));
    SetBackgroundColour(*wxWHITE);
	SetSizeHints( wxDefaultSize, wxDefaultSize );
	
	m_sizer = new wxBoxSizer( wxVERTICAL );
    auto m_line_top = new wxPanel(this, wxID_ANY, wxDefaultPosition, wxSize(-1, 1), wxTAB_TRAVERSAL);
    m_line_top->SetBackgroundColour(wxColour(0xA6, 0xa9, 0xAA));
    m_sizer->Add(m_line_top, 0, wxEXPAND, FromDIP(5));
    m_sizer->Add(0, 0, 0, wxTOP, FromDIP(5));
    

	m_simplebook       = new wxSimplebook(this, wxID_ANY, wxDefaultPosition, wxDefaultSize, 0);
	m_book_first_panel = new wxPanel( m_simplebook, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL );
    m_book_third_panel = new wxPanel(m_simplebook, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL);
    m_book_third_panel->SetBackgroundColour(*wxWHITE);

	m_dlg_sizer = new wxBoxSizer( wxVERTICAL );
	m_dlg_content_sizer = new wxBoxSizer( wxHORIZONTAL );
	m_canvas_sizer = new wxBoxSizer( wxVERTICAL );

    // page 3
	wxGLAttributes canvasAttrs;
    canvasAttrs.PlatformDefaults().Defaults().Stencil(8).EndList();
    m_canvas = new SkipPartCanvas(m_book_third_panel, canvasAttrs);
    m_canvas->SetPosition(wxPoint(FromDIP(10), FromDIP(10)));
	m_canvas->SetMinSize( wxSize( FromDIP(400),FromDIP(400) ) );

	m_canvas_sizer->Add( m_canvas, 0, wxALL, FromDIP(10) );

	m_canvas_btn_sizer = new wxBoxSizer( wxHORIZONTAL );
	m_canvas_btn_sizer->SetMinSize( wxSize( FromDIP(214),FromDIP(28) ) );
    
    StateColor zoom_bg(std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Disabled), std::pair<wxColour, int>(wxColour(206, 206, 206), StateColor::Pressed),
                     std::pair<wxColour, int>(wxColour(238, 238, 238), StateColor::Hovered), std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Enabled),
                       std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Normal));
    zoom_bg.setTakeFocusedAsHovered(false);
    StateColor zoom_bd(std::pair<wxColour, int>(wxColour(144, 144, 144), StateColor::Disabled), std::pair<wxColour, int>(wxColour(38, 46, 48), StateColor::Enabled));
    StateColor zoom_text(std::pair<wxColour, int>(wxColour(144, 144, 144), StateColor::Disabled), std::pair<wxColour, int>(wxColour(38, 46, 48), StateColor::Enabled));

    m_zoom_out_btn = new Button(m_book_third_panel, _L(""));
    m_zoom_out_btn->SetIcon("canvas_zoom_out");
    m_zoom_out_btn->SetToolTip(_L("Zoom Out"));
    m_zoom_out_btn->SetBackgroundColor(zoom_bg);
    m_zoom_out_btn->SetBorderColor(wxColour(238, 238, 238));
    m_zoom_out_btn->SetCornerRadius(0);
    m_zoom_out_btn->SetMinSize(wxSize(FromDIP(54), FromDIP(26)));  

	m_canvas_btn_sizer->Add(m_zoom_out_btn, 0, wxEXPAND, 0);

    StateColor percent_bg(std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Disabled), std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Pressed),
                          std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Hovered), std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Enabled),
                       std::pair<wxColour, int>(wxColour(255, 255, 255), StateColor::Normal));
    m_percent_label = new Button(m_book_third_panel, _L("100 %"));
    m_percent_label->SetBackgroundColor(percent_bg);
    m_percent_label->SetBorderColor(wxColour(238, 238, 238));
    m_percent_label->SetMinSize(wxSize(FromDIP(54), FromDIP(26)));
    m_percent_label->SetCornerRadius(0);
	m_canvas_btn_sizer->Add( m_percent_label, 0, wxEXPAND, 0 );

    m_zoom_in_btn = new Button(m_book_third_panel, _L(""));
    m_zoom_in_btn->SetIcon("canvas_zoom_in");
    m_zoom_in_btn->SetToolTip(_L("Zoom In"));
    m_zoom_in_btn->SetBackgroundColor(zoom_bg);
    m_zoom_in_btn->SetBorderColor(wxColour(238, 238, 238));
    m_zoom_in_btn->SetMinSize(wxSize(FromDIP(54), FromDIP(26)));
    m_zoom_in_btn->SetCornerRadius(0);
	m_canvas_btn_sizer->Add(m_zoom_in_btn, 0, wxEXPAND, 0);

	m_switch_drag_btn = new Button(m_book_third_panel, _L(""));
    m_switch_drag_btn->SetIcon("canvas_drag");
    m_switch_drag_btn->SetToolTip(_L("Drag"));
    m_switch_drag_btn->SetBackgroundColor(*wxWHITE);
    m_switch_drag_btn->SetBorderColor(wxColour(238, 238, 238));
    m_switch_drag_btn->SetMinSize(wxSize(FromDIP(54),FromDIP(26)));
    m_switch_drag_btn->SetCornerRadius(0);
	m_canvas_btn_sizer->Add( m_switch_drag_btn, 0, wxEXPAND, FromDIP(5) );

	m_canvas_sizer->Add(m_canvas_btn_sizer, 0, wxALIGN_CENTER_HORIZONTAL, FromDIP(10));

	m_dlg_content_sizer->Add( m_canvas_sizer, 0, wxALL, FromDIP(10) );

	m_list_sizer = new wxBoxSizer( wxVERTICAL );
	m_list_sizer->SetMinSize( wxSize( FromDIP(267),FromDIP(422) ) );

	auto all_checkbox_sizer = new wxBoxSizer(wxHORIZONTAL);
    m_all_checkbox          = new CheckBox(m_book_third_panel, wxID_ANY);
    m_all_checkbox->SetValue(false);
    m_all_checkbox->SetBackgroundColour(*wxWHITE);
    m_all_label = new Label(m_book_third_panel, _L("Select All"));
    m_all_label->Wrap(-1);
    m_all_label->SetBackgroundColour(*wxWHITE);
    
    m_all_label->SetMinSize(wxSize(267, -1));
    m_all_label->SetMaxSize(wxSize(267, -1));
    all_checkbox_sizer->Add(m_all_checkbox, 0, wxALL | wxALIGN_CENTER_VERTICAL, 0);
    all_checkbox_sizer->Add(m_all_label, 0, wxALL | wxALIGN_CENTER_VERTICAL, 0);

	m_list_sizer->Add(all_checkbox_sizer, 0, wxALL, FromDIP(5));

	m_line = new wxPanel(m_book_third_panel, wxID_ANY, wxDefaultPosition, wxSize(FromDIP(267), FromDIP(1)), wxTAB_TRAVERSAL);
	m_line->SetBackgroundColour( wxColor(238, 238, 238) );

	m_list_sizer->Add(m_line, 0, wxEXPAND, 0);

	m_list_view = new wxScrolledWindow( m_book_third_panel, wxID_ANY, wxDefaultPosition, wxSize( 267,-1 ), wxHSCROLL|wxVSCROLL );
	m_list_view->SetScrollRate( 5, 5 );
	m_list_view->SetMinSize( wxSize( FromDIP(267),FromDIP(410) ) );
    m_list_view->SetMaxSize(wxSize(FromDIP(267), FromDIP(410)));
    m_list_view->SetBackgroundColour(*wxWHITE);

    m_scroll_sizer = new wxBoxSizer(wxVERTICAL);
	m_list_view->SetSizer(m_scroll_sizer);
	m_list_view->Layout();

    m_list_sizer->Add(m_list_view, 0, wxEXPAND, 0);

	m_dlg_content_sizer->Add( m_list_sizer, 0, wxALL, FromDIP(15) );


	m_dlg_sizer->Add( m_dlg_content_sizer, 0, wxEXPAND, FromDIP(5) );

	m_dlg_btn_sizer = new wxBoxSizer( wxHORIZONTAL );
    m_dlg_btn_sizer->SetMinSize(wxSize(FromDIP(720), FromDIP(54)));

	m_cnt_label = new Label(m_book_third_panel, _L("cnt 1"));
	m_cnt_label->Wrap( -1 );
    m_cnt_label->SetBackgroundColour(*wxWHITE);
    m_cnt_label->SetForegroundColour(wxColour(0, 174, 66));
     
    m_cnt_label->SetFont(Label::Head_15);
    m_dlg_btn_sizer->Add(0, 0, 0, wxLEFT | wxALIGN_CENTER_VERTICAL, FromDIP(5)); 
	m_dlg_btn_sizer->Add( m_cnt_label, 0, wxALL | wxALIGN_CENTER_VERTICAL, FromDIP(0) );

	m_tot_label = new Label(m_book_third_panel, _L("tot 15"));
	m_tot_label->Wrap( -1 );
    m_tot_label->SetBackgroundColour(*wxWHITE);
    
    m_dlg_btn_sizer->Add(m_tot_label, 0, wxALL | wxALIGN_CENTER_VERTICAL, FromDIP(5));
    m_dlg_btn_sizer->Add(0, 0, 1, wxEXPAND | wxALIGN_CENTER_VERTICAL, FromDIP(5));

    StateColor btn_bg_white(std::pair<wxColour, int>(wxColour(206, 206, 206), StateColor::Pressed), std::pair<wxColour, int>(wxColour(238, 238, 238), StateColor::Hovered),
                            std::pair<wxColour, int>(*wxWHITE, StateColor::Normal));


    StateColor btn_bg_green(std::pair<wxColour, int>(wxColour(27, 136, 68), StateColor::Pressed), std::pair<wxColour, int>(wxColour(61, 203, 115), StateColor::Hovered),
                            std::pair<wxColour, int>(wxColour(0, 177, 66), StateColor::Normal));
             
    m_apply_btn = new Button(m_book_third_panel, wxString(_L("Skip"))); 
    m_apply_btn->SetBackgroundColor(btn_bg_green);
    m_apply_btn->SetTextColor(*wxWHITE);
    // m_apply_btn->SetBorderColor(wxColour(38, 46, 48));
    m_apply_btn->SetFont(Label::Body_14);
    m_apply_btn->SetSize(wxSize(-1, FromDIP(32))); 
    m_apply_btn->SetMinSize(wxSize(FromDIP(80), FromDIP(32)));
    m_apply_btn->SetCornerRadius(FromDIP(16));
    m_dlg_btn_sizer->Add(m_apply_btn, 0, wxALL | wxALIGN_CENTER_VERTICAL, FromDIP(15));

	m_dlg_sizer->Add( m_dlg_btn_sizer, 0, wxEXPAND, FromDIP(5) );

	m_book_third_panel->SetSizer( m_dlg_sizer );
	m_book_third_panel->Layout();
	m_dlg_sizer->Fit( m_book_third_panel );
	
    // page 2
    m_book_second_panel = new wxPanel( m_simplebook, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL );
    m_book_second_panel->SetBackgroundColour(*wxWHITE);
	m_book_second_sizer = new wxBoxSizer( wxVERTICAL );
    m_book_second_btn_sizer = new wxBoxSizer( wxHORIZONTAL );

	m_retry_label = new Label( m_book_second_panel, _L("Load Skipping Objects Information Failed. \nPlease try again."));
	m_retry_label->Wrap( -1 );
    m_retry_label->SetBackgroundColour(*wxWHITE);
    m_book_second_sizer->Add(0, 0, 1, wxEXPAND, 0);
    m_book_second_sizer->Add(m_retry_label, 0, wxALIGN_CENTER_VERTICAL | wxALIGN_CENTER_HORIZONTAL, 0);
    m_book_second_sizer->Add(0, 0, 1, wxEXPAND, 0);

    m_second_retry_btn = new Button(m_book_second_panel, _L("Retry"));
    m_second_retry_btn->SetBackgroundColor(btn_bg_green);
    // m_second_retry_btn->SetBorderColor(wxColour(38, 46, 48));
    m_second_retry_btn->SetTextColor(*wxWHITE);
    m_second_retry_btn->SetFont(Label::Body_14);
    m_second_retry_btn->SetSize(wxSize(FromDIP(80), FromDIP(32)));
    m_second_retry_btn->SetMinSize(wxSize(FromDIP(80), FromDIP(32)));
    m_second_retry_btn->SetCornerRadius(FromDIP(16));
    m_second_retry_btn->Bind(wxEVT_BUTTON, &PartSkipDialog::OnRetryButton, this);
    m_book_second_btn_sizer->Add(m_second_retry_btn, 0, wxALL, FromDIP(24));

    m_book_second_sizer->Add(m_book_second_btn_sizer, 0, wxALIGN_RIGHT, 0);
	m_book_second_panel->SetSizer( m_book_second_sizer );
	m_book_second_panel->Layout();
	m_book_second_sizer->Fit( m_book_second_panel );

    // page 1
	m_book_first_panel = new wxPanel( m_simplebook, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL );
	m_book_first_sizer = new wxBoxSizer( wxVERTICAL );
    m_book_first_sizer->SetMinSize(wxSize(FromDIP(720), FromDIP(500)));

    // auto m_loading_sizer = new wxBoxSizer(wxHORIZONTAL);
    // std::vector<std::string> list{"ams_rfid_1", "ams_rfid_2", "ams_rfid_3", "ams_rfid_4"};
    // m_loading_icon = new AnimaIcon(m_book_first_panel, wxID_ANY, list, "refresh_printer", 100);

	m_loading_label = new Label( m_book_first_panel, wxString(_L("Skipping Objects Information Loading ...")));
	m_loading_label->Wrap( -1 );
    m_loading_label->SetBackgroundColour(*wxWHITE);

    // m_loading_sizer->Add(0, 0, 1, wxEXPAND, 0);
    // m_loading_sizer->Add(m_loading_icon, 0, wxALL, FromDIP(5));
    // m_loading_sizer->Add(m_loading_label, 0, wxALL, FromDIP(5));
    // m_loading_sizer->Add(0, 0, 1, wxEXPAND, 0);

    m_book_first_sizer->Add(0, 0, 1, wxEXPAND, 0);
	m_book_first_sizer->Add( m_loading_label, 0, wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL, 0);
    m_book_first_sizer->Add(0, 0, 1, wxEXPAND, 0);

	m_book_first_panel->SetSizer( m_book_first_sizer );
	m_book_first_panel->Layout();
	m_book_first_sizer->Fit( m_book_first_panel );
	
    m_simplebook->AddPage( m_book_first_panel, _("loading page"), false );
    m_simplebook->AddPage( m_book_second_panel, _("retry page"), false );
    m_simplebook->AddPage( m_book_third_panel, _("dialog page"), false );
	m_sizer->Add( m_simplebook, 1, wxEXPAND | wxALL, 5 );

    
	SetSizer( m_sizer );
    m_zoom_in_btn->Bind(wxEVT_BUTTON, &PartSkipDialog::OnZoomIn, this);
    m_zoom_out_btn->Bind(wxEVT_BUTTON, &PartSkipDialog::OnZoomOut, this);
    m_switch_drag_btn->Bind(wxEVT_BUTTON, &PartSkipDialog::OnSwitchDrag, this);
    m_canvas->Bind(EVT_ZOOM_PERCENT, &PartSkipDialog::OnZoomPercent, this);
    m_canvas->Bind(EVT_CANVAS_PART, &PartSkipDialog::UpdatePartsStateFromCanvas, this);

    m_apply_btn->Bind(wxEVT_BUTTON, &PartSkipDialog::OnApplyDialog, this);
    m_all_checkbox->Bind(wxEVT_TOGGLEBUTTON, &PartSkipDialog::OnAllCheckbox, this);
    
    Fit();
	Layout();
    CentreOnParent();
}

PartSkipDialog::~PartSkipDialog() {}

void PartSkipDialog::on_dpi_changed(const wxRect& suggested_rect) { Fit(); }


std::string PartSkipDialog::create_tmp_path(){
    boost::filesystem::path parent_path(temporary_dir());

    std::stringstream buf;
    buf << "/bamboo_task/";
    buf << m_timestamp;
    if (m_obj) {
        buf << m_obj->dev_id << "_";
        buf << m_obj->job_id_ << "/";
    } else {
        buf << 1 << "_" << 1 << "/";
    }
    std::string tmp_path = (parent_path / buf.str() ).string();

    if (!std::filesystem::exists(tmp_path + "Metadata/")  && !fs::create_directories(tmp_path + "Metadata/")) {
        wxMessageBox("create file failed.");
    }
    return tmp_path;
}


bool PartSkipDialog::is_local_file_existed(const std::vector<string> &local_paths) 
{ 
    for (auto path : local_paths) { 
        if (!std::filesystem::exists(path)) { 
            return false;
        }
    }
    return true;
}

void PartSkipDialog::DownloadPartsFile()
{
    m_tmp_path = create_tmp_path(); //wxGetApp().app_config->get("download_path");

    m_local_paths.clear();
    m_target_paths.clear();
    
    int plate_idx = m_obj ? m_obj->m_plate_index : 1;

    m_local_paths.push_back(m_tmp_path + "Metadata/pick_" + std::to_string(plate_idx) + ".png");
    m_local_paths.push_back(m_tmp_path + "Metadata/model_settings.config");
    m_local_paths.push_back(m_tmp_path + "Metadata/slice_info.config");

    m_target_paths.push_back("Metadata/pick_" + std::to_string(plate_idx) + ".png");
    m_target_paths.push_back("Metadata/model_settings.config");
    m_target_paths.push_back("Metadata/slice_info.config");
    
    if (!is_local_file_existed(m_local_paths)) {
        m_file_sys = boost::make_shared<PrinterFileSystem>();
        m_file_sys->Attached();
        m_file_sys->Bind(EVT_STATUS_CHANGED, &PartSkipDialog::OnFileSystemEvent, this);
        m_file_sys->Bind(EVT_RAMDOWNLOAD, &PartSkipDialog::OnFileSystemResult, this);
        m_file_sys->Start();
    } else {
        m_file_sys->SendExistedFile();
    }
}
void PartSkipDialog::fetchUrl(boost::weak_ptr<PrinterFileSystem> wfs)
{
    boost::shared_ptr fs(wfs.lock());
    if (!fs) return;

    DeviceManager *dm  = GUI::wxGetApp().getDeviceManager();
    MachineObject *obj = dm->get_selected_machine();

    if (obj == nullptr) {
        fs->SetUrl("0");
        return;
    }
    std::string dev_ver      = obj->get_ota_version();
    std::string dev_id       = obj->dev_id;
    int         remote_proto = obj->get_file_remote();

    NetworkAgent *agent         = wxGetApp().getAgent();
    std::string   agent_version = agent ? agent->get_version() : "";

    if (agent) {
        switch (m_url_state) {
            case URL_TCP: {
                    std::string devIP      = obj->dev_ip;
                    std::string accessCode = obj->get_access_code();
                    std::string tcp_url    = "bambu:///local/" + devIP + "?port=6000&user=" + "bblp" + "&passwd=" + accessCode;
                    CallAfter([=] {
                        boost::shared_ptr fs(wfs.lock());
                        if (!fs) return;
                        if (boost::algorithm::starts_with(tcp_url, "bambu:///")) {
                            fs->SetUrl(tcp_url);
                        } else {
                            fs->SetUrl("3");
                        }
                    });
                    break;
                } 
            case URL_TUTK: {
                    std::string protocols[] = {"", "\"tutk\"", "\"agora\"", "\"tutk\",\"agora\""};
                    agent->get_camera_url(obj->dev_id + "|" + dev_ver + "|" + protocols[1], [this, wfs, m = dev_id, v = agent->get_version(), dv = dev_ver](std::string url) {
                        if (boost::algorithm::starts_with(url, "bambu:///")) {
                            url += "&device=" + m;
                            url += "&net_ver=" + v;
                            url += "&dev_ver=" + dv;
                            url += "&refresh_url=" + boost::lexical_cast<std::string>(&refresh_agora_url);
                            url += "&cli_id=" + wxGetApp().app_config->get("slicer_uuid");
                            url += "&cli_ver=" + std::string(SLIC3R_VERSION);
                        }
                        BOOST_LOG_TRIVIAL(info) << "SendToPrinter::fetchUrl: camera_url: " << hide_passwd(url, {"?uid=", "authkey=", "passwd="});
                        std::cout << "SendToPrinter::fetchUrl: camera_url: " << hide_passwd(url, {"?uid=", "authkey=", "passwd="});
                        CallAfter([=] {
                            boost::shared_ptr fs(wfs.lock());
                            if (!fs) return;
                            if (boost::algorithm::starts_with(url, "bambu:///")) {
                                fs->SetUrl(url);
                            } else {
                                fs->SetUrl("3");
                            }
                        });
                    }); 
                    break;
                }
            default: break;
        }
    }
}

void PartSkipDialog::OnFileSystemEvent(wxCommandEvent &e)
{
    e.Skip();
    auto wfs = boost::weak_ptr(m_file_sys);
    boost::shared_ptr fs(wfs.lock());
    if (!fs) return;

    wxString msg;
    int      status = e.GetInt();
    int      extra  = e.GetExtraLong();

    switch (status) {
    case PrinterFileSystem::Initializing:
    case PrinterFileSystem::Connecting: break;
    case PrinterFileSystem::ListSyncing: {
        m_file_sys->GetPickImages(m_local_paths, m_target_paths);
        break;
    }
    case PrinterFileSystem::Failed: {
        m_file_sys->SendConnectFail();
        break;
    }
    case PrinterFileSystem::Reconnecting: break;
    }
    if (e.GetInt() == PrinterFileSystem::Initializing) {
        CallAfter([=] {
            boost::shared_ptr fs(wfs.lock());
            if (!fs) return;
            fetchUrl(boost::weak_ptr(fs));
        });
    }
}
void PartSkipDialog::OnFileSystemResult(wxCommandEvent &event){
    int result = event.GetInt();
    if (result == 0) {
        InitDialogUI();
        SetSimplebookPage(2);
    }else{
        switch (m_url_state) {
        case URL_TCP: {
            m_url_state = URL_TUTK;
            break;
        }
        case URL_TUTK: {
            m_url_state = URL_TCP;
            break;
        }
        }
        SetSimplebookPage(1);
    }
    // wxMilliSleep(1000);
    // m_loading_icon->Stop();
    Refresh();
}

void PartSkipDialog::InitSchedule(MachineObject *obj){
    m_obj = obj;
    SetSimplebookPage(0);
    // m_loading_icon->Play();
    DownloadPartsFile();
}

void PartSkipDialog::OnRetryButton(wxCommandEvent &event) {
    event.Skip();
    InitSchedule(m_obj);
    Refresh();
}


bool PartSkipDialog::is_drag_mode(){
    return m_is_drag == true;
}

PartsInfo PartSkipDialog::GetPartsInfo(){
    PartsInfo parts_info;
    for (auto [part_id, part_state] : this->m_parts_state) 
    { 
        parts_info.push_back(std::pair<int, PartState>(part_id, part_state)); 
    }
    return parts_info;
}

void PartSkipDialog::OnZoomIn(wxCommandEvent &event){
    m_canvas->ZoomIn(20);
    UpdateZoomPercent();
}

void PartSkipDialog::OnZoomOut(wxCommandEvent &event){ 
    m_canvas->ZoomOut(20);
    UpdateZoomPercent();
}

void PartSkipDialog::OnSwitchDrag(wxCommandEvent& event){
    if (this->is_drag_mode()) {
        m_is_drag = false;
        m_switch_drag_btn->SetBackgroundColor(*wxWHITE);
        m_switch_drag_btn->SetIcon("canvas_drag");
    } else {
        m_is_drag = true;
        m_switch_drag_btn->SetBackgroundColor(wxColour(0, 174, 66));
        m_switch_drag_btn->SetIcon("canvas_drag_active");
    }
    m_canvas->SwitchDrag(m_is_drag); 
}

void PartSkipDialog::OnZoomPercent(wxCommandEvent &event) {
    m_zoom_percent = event.GetInt();
    if (m_zoom_percent >= 1000) {
        m_zoom_percent = 1000;
        m_zoom_in_btn->Enable(false);
        m_zoom_in_btn->SetIcon("canvas_zoom_in_disable");
    } else if (m_zoom_percent <= 100) {
        m_zoom_percent = 100;
        m_zoom_out_btn->Enable(false);
        m_zoom_out_btn->SetIcon("canvas_zoom_out_disable");
    } else {
        m_zoom_in_btn->Enable(true);
        m_zoom_out_btn->Enable(true);
        m_zoom_in_btn->SetIcon("canvas_zoom_in");
        m_zoom_out_btn->SetIcon("canvas_zoom_out");
    }

    UpdateZoomPercent();
}

void PartSkipDialog::UpdatePartsStateFromCanvas(wxCommandEvent &event) {
    int part_id = event.GetExtraLong();
    PartState part_state = PartState(event.GetInt());

    m_parts_state[part_id] = part_state;
    if(part_state == psUnCheck){
        m_all_checkbox->SetValue(false);
    }
    if(IsAllChecked()){
        m_all_checkbox->SetValue(true);
    }
    UpdateDialogUI();
}

void PartSkipDialog::UpdateZoomPercent() { 
    m_percent_label->SetLabel(wxString::Format(_L("%d%%"), m_zoom_percent)); 
}

void PartSkipDialog::UpdateCountLabel() {
    int check_cnt = 0;
    int tot_cnt   = 0;
    for (auto [part_id, part_state] : m_parts_state) { 
        if (part_state == PartState::psChecked) check_cnt++;
        if (part_state != PartState::psSkipped) tot_cnt++;
    }
    m_cnt_label->SetLabel(wxString::Format(_L("%d"), check_cnt));
    m_tot_label->SetLabel(wxString::Format(_L("/%d Selected"), tot_cnt));
}

bool PartSkipDialog::Show(bool show)
{
    if (show) {
        wxGetApp().UpdateDlgDarkUI(this);
        CentreOnParent();

        Layout();
        Fit();
    }
    return DPIDialog::Show(show);
}

void PartSkipDialog::InitDialogUI() {
    m_print_lock = true;
    m_scroll_sizer->Clear(true);
    m_all_checkbox->SetValue(false);
    m_parts_state.clear();
    m_parts_name.clear();

    string pick_img = m_local_paths[0];
    string model_settings = m_local_paths[1];

    m_switch_drag_btn->SetIcon("canvas_drag");
    m_switch_drag_btn->SetBackgroundColor(*wxWHITE);
    m_is_drag = false;
    m_canvas->SwitchDrag(false);
    m_canvas->SetZoomPercent(100);
    m_canvas->SetOffset(wxPoint(0, 0));
    m_canvas->LoadPickImage(pick_img);
    ModelSettingHelper helper(model_settings);

    if (helper.Parse()) {
        auto parse_result = helper.GetResults();
        for (const auto& part : parse_result) {
            m_parts_state[part.identify_id] = part.state;
            m_parts_name[part.identify_id] = part.name;
        }
        if (m_obj) {
            std::vector<int> partskip_ids = m_obj->m_partskip_ids;
            for (auto part_id : partskip_ids) {
                m_parts_state[part_id] = PartState::psSkipped;
            }
        }
        
        for(const auto& [part_id, part_state] : m_parts_state) {
            auto line_sizer = new wxBoxSizer(wxHORIZONTAL);
            auto checkbox   = new CheckBox(m_list_view);
            auto label      = new Label(m_list_view, _L("file 0"));
            
            checkbox->Bind(wxEVT_TOGGLEBUTTON, [this, part_id=part_id](wxCommandEvent& event) { 
                m_parts_state[part_id] = event.IsChecked() ? PartState::psChecked : PartState::psUnCheck;
                if(!event.IsChecked()){
                    m_all_checkbox->SetValue(false);
                }else if(IsAllChecked()){
                    m_all_checkbox->SetValue(true);
                }
                m_canvas->UpdatePartsInfo(GetPartsInfo());
                UpdateCountLabel();
                event.Skip();
            }, checkbox->GetId());
        
            if (part_state == PartState::psChecked) {
                checkbox->SetValue(true);
                checkbox->Enable(true);
            } else if(part_state == PartState::psUnCheck) {
                checkbox->SetValue(false);
                checkbox->Enable(true);
            } else if (part_state == PartState::psSkipped) {
                checkbox->SetValue(true);
                checkbox->Enable(false);
            }
            label->SetLabel(wxString::FromUTF8(m_parts_name[part_id]));
            label->SetBackgroundColour(*wxWHITE);
            label->SetForegroundColour(wxColor(107,107,107));
            label->Wrap(-1);
            checkbox->SetBackgroundColour(*wxWHITE);
            
            line_sizer->Add(checkbox, 0, wxALL | wxALIGN_CENTER_VERTICAL, 0);
            line_sizer->Add(label, 1, wxALL | wxALIGN_CENTER_VERTICAL, 0);
            m_scroll_sizer->Add(line_sizer, 0, wxALL, FromDIP(5));
        }
        m_canvas->UpdatePartsInfo(GetPartsInfo());
    }
	
    m_scroll_sizer->Layout();
    UpdateCountLabel();
    Refresh();
    m_print_lock = false;
}

void PartSkipDialog::UpdatePartsStateFromPrinter(MachineObject *obj) {
    if (m_print_lock) return;
    m_obj = obj;
    if (m_obj) {
        std::vector<int> partskip_ids = m_obj->m_partskip_ids;
        for(auto part_id : partskip_ids) {
            m_parts_state[part_id] = PartState::psSkipped;
        }
        m_canvas->UpdatePartsInfo(GetPartsInfo());
        UpdateDialogUI();
    }
}


void PartSkipDialog::UpdateDialogUI(){
    if(m_parts_state.size() != m_scroll_sizer->GetItemCount()){
        BOOST_LOG_TRIVIAL(warning) << "m_parts_state and m_scroll_sizer mismatch.";
        return;
    }

    for (auto it = m_parts_state.begin(); it != m_parts_state.end(); ++it) { 
        int idx = std::distance(m_parts_state.begin(), it); 
        auto part_id = it->first;
        auto part_state = it->second;

        wxSizerItem *item = m_scroll_sizer->GetItem(idx);
        if (item && item->IsSizer()) {
            wxSizer* sizer  = item->GetSizer();
            auto check_item = sizer->GetItem((size_t)0);
           
            if (check_item && check_item->IsWindow()) 
            { 
                wxWindow *window = check_item->GetWindow(); 
                CheckBox *checkbox = dynamic_cast<CheckBox *>(window);
                if (part_state == PartState::psChecked) { 
                    checkbox->SetValue(true);
                } else if (part_state == PartState::psUnCheck) {
                    checkbox->SetValue(false);  
                } else {
                    checkbox->SetValue(true); 
                    checkbox->Enable(false);
                }
            }
        }
    }

    UpdateCountLabel();
    Refresh();
}

void PartSkipDialog::SetSimplebookPage(int page) {
    m_simplebook->SetSelection(page);
}

bool PartSkipDialog::IsAllChecked() {
    for (auto& [part_id, part_state] : m_parts_state) {
        if (part_state == PartState::psUnCheck)
            return false;
    }
    return true;
}

void PartSkipDialog::OnAllCheckbox(wxCommandEvent &event)
{
    if (m_all_checkbox->GetValue()) {
        for (auto& [part_id, part_state] : m_parts_state) {
            if (part_state == PartState::psUnCheck)
                part_state = PartState::psChecked; 
        }
    }
    else{
        for (auto& [part_id, part_state] : m_parts_state) {
            if (part_state == PartState::psChecked)
                part_state = PartState::psUnCheck; 
        }
    }
    m_canvas->UpdatePartsInfo(GetPartsInfo());
    UpdateDialogUI();
    event.Skip();
}


void PartSkipDialog::OnApplyDialog(wxCommandEvent &event)
{
    event.Skip();
    m_partskip_ids.clear();
    for (const auto& [part_id, part_state] : m_parts_state) {
        if (part_state == PartState::psChecked) {
            m_partskip_ids.push_back(part_id);
        }
    }
    PartSkipConfirmDialog confirm_dialog(this);
    confirm_dialog.SetMsgLabel(wxString::Format(_L("Skipping %d objects."), m_partskip_ids.size()));

    if(confirm_dialog.ShowModal() == wxID_OK){
        if (m_obj) {
            BOOST_LOG_TRIVIAL(info) << "monitor: skipping "<< m_partskip_ids.size() <<" parts dev_id =" << m_obj->dev_id;
            
            bool all_skipped = true;
            for (auto [part_id, part_state] : m_parts_state) { 
                if (part_state == PartState::psUnCheck) all_skipped = false;
            }

            if (all_skipped) { 
                m_obj->command_task_abort(); 
            } else {
                m_obj->command_task_partskip(m_partskip_ids);
            }
            EndModal(wxID_OK);
        } else {
            BOOST_LOG_TRIVIAL(warning) << "machine object is null.";
        }
    }
}


PartSkipConfirmDialog::PartSkipConfirmDialog(wxWindow *parent) : DPIDialog(parent, wxID_ANY, _L("Skip Objects"), wxDefaultPosition, wxDefaultSize, wxCAPTION | wxCLOSE_BOX)
{
    std::string icon_path = (boost::format("%1%/images/BambuStudioTitle.ico") % Slic3r::resources_dir()).str();
    SetIcon(wxIcon(Slic3r::encode_path(icon_path.c_str()), wxBITMAP_TYPE_ICO));
    SetBackgroundColour(*wxWHITE);
    SetSizeHints(wxDefaultSize, wxDefaultSize);

	wxBoxSizer* m_sizer;
	m_sizer = new wxBoxSizer( wxVERTICAL );

    auto m_line_top = new wxPanel(this, wxID_ANY, wxDefaultPosition, wxSize(-1, 1), wxTAB_TRAVERSAL);
    m_line_top->SetBackgroundColour(wxColour(0xA6, 0xa9, 0xAA));
    m_sizer->Add(m_line_top, 0, wxEXPAND, 0);
    m_sizer->Add(0, 0, 0, wxALL, FromDIP(15));

	m_msg_label = new Label( this, wxString(_L("Skipping objects.")));
	m_msg_label->Wrap( -1 );
    m_msg_label->SetBackgroundColour(*wxWHITE);

	m_sizer->Add(m_msg_label, 0, wxLEFT, FromDIP(15));

    auto m_tip_label = new Label(this, wxString(_("This action cannot be undone. Continue?")));
    m_tip_label->Wrap(-1);
    m_tip_label->SetBackgroundColour(*wxWHITE);
    m_tip_label->SetForegroundColour(wxColor(92,92,92));
    m_sizer->Add(m_tip_label, 0, wxLEFT | wxTOP, FromDIP(15));

	wxBoxSizer* m_button_sizer;
    m_button_sizer = new wxBoxSizer(wxHORIZONTAL);
    m_button_sizer->SetMinSize(wxSize(FromDIP(480), FromDIP(54)));
	m_button_sizer->Add(0, 0, 1, wxEXPAND, 0);


    StateColor btn_bg_white(std::pair<wxColour, int>(wxColour(206, 206, 206), StateColor::Pressed), std::pair<wxColour, int>(wxColour(238, 238, 238), StateColor::Hovered),
                            std::pair<wxColour, int>(*wxWHITE, StateColor::Normal));

    StateColor btn_bg_green(std::pair<wxColour, int>(wxColour(27, 136, 68), StateColor::Pressed), std::pair<wxColour, int>(wxColour(61, 203, 115), StateColor::Hovered),
                            std::pair<wxColour, int>(wxColour(0, 177, 66), StateColor::Normal));

    m_apply_button = new Button(this, wxString(_L("Continue")));
    m_apply_button->SetBackgroundColor(btn_bg_green);
    m_apply_button->SetTextColor(*wxWHITE);
    // m_apply_button->SetBorderColor(wxColour(38, 46, 48));
    m_apply_button->SetFont(Label::Body_14);
    m_apply_button->SetSize(wxSize(FromDIP(80), FromDIP(32)));
    m_apply_button->SetMinSize(wxSize(FromDIP(80), FromDIP(32)));
    m_apply_button->SetCornerRadius(FromDIP(16));
    m_apply_button->Bind(wxEVT_BUTTON, [this](auto& e){
        EndModal(wxID_OK);
        e.Skip();
    });

    m_button_sizer->Add(m_apply_button, 0, wxALL, FromDIP(15));
	m_sizer->Add(m_button_sizer, 1, wxEXPAND, FromDIP(5));
    m_sizer->Fit( this );

	SetSizer( m_sizer );
    Layout();
    Fit();
}

PartSkipConfirmDialog::~PartSkipConfirmDialog()
{
}

bool PartSkipConfirmDialog::Show(bool show)
{
    if (show) {
        wxGetApp().UpdateDlgDarkUI(this);
        CentreOnParent();

        Layout();
        Fit();
    }
    return DPIDialog::Show(show);
}

void PartSkipConfirmDialog::on_dpi_changed(const wxRect& suggested_rect) { Fit(); }


Button* PartSkipConfirmDialog::GetConfirmButton()
{
    return m_apply_button;
}

void PartSkipConfirmDialog::SetMsgLabel(wxString msg){
    m_msg_label->SetLabel(msg);
}


}} // namespace Slic3r::GUI