mod config;
mod dialogs;
mod environment;
mod error;
#[allow(clippy::all, dead_code)]
mod icon;
mod options;
mod panes;
mod theme;
mod types;
mod values;
mod widget;

use std::path::PathBuf;
use std::sync::Arc;

use config::Config;
use dialogs::{
    cancel_button, error_dialog, ok_button, unsaved_changes_dialog,
    warning_dialog,
};
use error::Error;
use iced::advanced::widget::Id;
use iced::widget::{
    Column, container, pane_grid, pick_list, row, text, text_editor,
};
use iced::{Alignment, Element, Length, Task, Theme, clipboard, keyboard};
use iced_anim::transition::Easing;
use iced_anim::{Animated, Animation};
use iced_dialog::dialog::Dialog;
use panes::{code_view, designer_view, element_list};
use tokio::runtime;
use types::{
    Action, DesignerPane, DialogAction, DialogButtons, ElementName, Message,
    Project,
};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let version = std::env::args()
        .nth(1)
        .is_some_and(|s| s == "--version" || s == "-V");

    if version {
        println!("iced-builder {}", environment::formatted_version());
        println!("{}", env!("CARGO_PKG_REPOSITORY"));

        return Ok(());
    }

    let config_load = {
        let rt = runtime::Builder::new_current_thread()
            .enable_all()
            .build()?;

        rt.block_on(Config::load())
    };

    iced::application(App::title, App::update, App::view)
        .font(icon::FONT)
        .theme(|state| state.theme.value().clone())
        .subscription(App::subscription)
        .run_with(move || App::new(config_load))?;
    Ok(())
}

struct App {
    is_dirty: bool,
    is_loading: bool,
    project_path: Option<PathBuf>,
    project: Project,
    config: Arc<Config>,
    theme: Animated<Theme>,
    pane_state: pane_grid::State<Panes>,
    focus: Option<pane_grid::Pane>,
    designer_page: DesignerPane,
    dialog_is_open: bool,
    dialog_title: &'static str,
    dialog_content: String,
    dialog_buttons: DialogButtons,
    dialog_action: DialogAction,
    element_list: &'static [ElementName],
    editor_content: text_editor::Content,
}

#[derive(Clone, Copy, Debug)]
enum Panes {
    Designer,
    ElementList,
}

impl App {
    fn new(config_load: Result<Config, Error>) -> (Self, Task<Message>) {
        let state = pane_grid::State::with_configuration(
            pane_grid::Configuration::Split {
                axis: pane_grid::Axis::Vertical,
                ratio: 0.8,
                a: Box::new(pane_grid::Configuration::Pane(Panes::Designer)),
                b: Box::new(pane_grid::Configuration::Pane(Panes::ElementList)),
            },
        );

        let config = Arc::new(config_load.unwrap_or_default());
        let theme = config.selected_theme();

        let task = if let Some(path) = config.last_project.clone() {
            if path.exists() && path.is_file() {
                Task::perform(Project::from_path(path), Message::FileOpened)
            } else {
                warning_dialog(format!(
                    "The file {} does not exist, or isn't a file.",
                    path.to_string_lossy()
                ))
            }
        } else {
            Task::none()
        };

        (
            Self {
                is_dirty: false,
                is_loading: false,
                project_path: None,
                project: Project::new(),
                config,
                theme: Animated::new(theme, Easing::EASE_IN),
                pane_state: state,
                focus: None,
                designer_page: DesignerPane::DesignerView,
                dialog_is_open: false,
                dialog_title: "",
                dialog_content: String::new(),
                dialog_buttons: DialogButtons::None,
                dialog_action: DialogAction::None,
                element_list: ElementName::ALL,
                editor_content: text_editor::Content::new(),
            },
            task,
        )
    }

    fn title(&self) -> String {
        let saved_state = if self.is_dirty { " *" } else { "" };

        let project_name = match &self.project.title {
            Some(n) => {
                format!(
                    " - {}",
                    if n.len() > 60 {
                        format!("...{}", &n[n.len() - 40..])
                    } else {
                        n.to_owned()
                    }
                )
            }
            None => String::new(),
        };

        format!("iced Builder{project_name}{saved_state}")
    }

    fn update(&mut self, message: Message) -> Task<Message> {
        match message {
            Message::SwitchTheme(event) => self.theme.update(event),
            Message::CopyCode => {
                return clipboard::write(self.editor_content.text());
            }
            Message::SwitchPage(page) => self.designer_page = page,
            Message::EditorAction(action) => {
                if let text_editor::Action::Scroll { lines: _ } = action {
                    self.editor_content.perform(action);
                }
            }
            Message::RefreshEditorContent => {
                match self.project.app_code(&self.config) {
                    Ok(code) => {
                        self.editor_content =
                            text_editor::Content::with_text(&code);
                    }
                    Err(error) => return error_dialog(error),
                }
                Err(error) => Task::future(error_dialog(error)).discard(),
            },
            Message::DropNewElement(name, point, _) => {
                return iced_drop::zones_on_point(
                    move |zones| Message::HandleNew(name.clone(), zones),
                    point,
                    None,
                    None,
                );
            }
            Message::HandleNew(name, zones) => {
                let refresh_editor_task =
                    Task::done(Message::RefreshEditorContent);

                let ids: Vec<Id> = zones.into_iter().map(|z| z.0).collect();
                if !ids.is_empty() {
                    let eltree_clone = self.project.element_tree.clone();
                    let action = Action::new(&ids, eltree_clone.as_ref(), None);
                    let result = name.handle_action(
                        self.project.element_tree.as_mut(),
                        action,
                    );
                    self.is_dirty = true;
                    match result {
                        Ok(Some(ref element)) => {
                            self.project.element_tree = Some(element.clone());
                        }
                        Err(error) => {
                            return error_dialog(error)
                                .chain(refresh_editor_task);
                        }
                        _ => {}
                    }
                    return refresh_editor_task;
                }
            }
            Message::MoveElement(element, point, _) => {
                return iced_drop::zones_on_point(
                    move |zones| Message::HandleMove(element.clone(), zones),
                    point,
                    None,
                    None,
                );
            }
            Message::HandleMove(element, zones) => {
                let ids: Vec<Id> = zones.into_iter().map(|z| z.0).collect();
                if !ids.is_empty() {
                    let eltree_clone = self.project.element_tree.clone();
                    let action = Action::new(
                        &ids,
                        eltree_clone.as_ref(),
                        Some(element.id()),
                    );
                    let result = element.handle_action(
                        self.project.element_tree.as_mut(),
                        action,
                    );
                    if let Err(error) = result {
                        return error_dialog(error);
                    }

                    self.is_dirty = true;
                    return Task::done(Message::RefreshEditorContent);
                }
            }
            Message::PaneResized(pane_grid::ResizeEvent { split, ratio }) => {
                self.pane_state.resize(split, ratio);
            }
            Message::PaneClicked(pane) => self.focus = Some(pane),
            Message::PaneDragged(pane_grid::DragEvent::Dropped {
                pane,
                target,
            }) => self.pane_state.drop(pane, target),
            Message::PaneDragged(_) => {}
            Message::OpenDialog(title, content, buttons, action) => {
                self.dialog_title = title;
                self.dialog_content = content;
                self.dialog_buttons = buttons;
                self.dialog_action = action;
                self.dialog_is_open = true;
            }
            Message::CloseDialog => self.dialog_is_open = false,
            Message::DialogOk => {
                let close_dialog_task = Task::done(Message::CloseDialog);

                match self.dialog_action {
                    DialogAction::None => {}
                    DialogAction::NewFile => {
                        self.is_dirty = false;
                        self.project = Project::new();
                        self.project_path = None;
                        self.editor_content = text_editor::Content::new();
                    }
                    DialogAction::OpenFile => {
                        self.is_dirty = false;
                        self.is_loading = true;
                        return Task::perform(
                            Project::from_file(self.config.clone()),
                            Message::FileOpened,
                        )
                        .chain(close_dialog_task);
                    }
                }

                return close_dialog_task;
            }
            Message::DialogCancel => return Task::done(Message::CloseDialog),
            Message::NewFile => {
                if !self.is_loading {
                    if !self.is_dirty {
                        self.project = Project::new();
                        self.project_path = None;
                        self.editor_content = text_editor::Content::new();
                    } else {
                        return unsaved_changes_dialog(
                            "You have unsaved changes. Do you wish to discard these and create a new project?",
                            DialogAction::NewFile,
                        );
                    }
                }
            }
            Message::OpenFile => {
                if !self.is_loading {
                    if !self.is_dirty {
                        self.is_loading = true;

                        return Task::perform(
                            Project::from_file(self.config.clone()),
                            Message::FileOpened,
                        );
                    } else {
                        return unsaved_changes_dialog(
                            "You have unsaved changes. Do you wish to discard these and open another project?",
                            DialogAction::OpenFile,
                        );
                    }
                }
            }
            Message::FileOpened(result) => {
                self.is_loading = false;
                self.is_dirty = false;

                return match result {
                    Ok((path, project)) => {
                        self.project = project;
                        self.project_path = Some(path);
                        Task::done(Message::RefreshEditorContent)
                    }
                    Err(error) => return error_dialog(error),
                };
            }
            Message::SaveFile => {
                if !self.is_loading {
                    self.is_loading = true;

                    return Task::perform(
                        self.project
                            .clone()
                            .write_to_file(self.project_path.clone()),
                        Message::FileSaved,
                    );
                }
            }
            Message::SaveFileAs => {
                if !self.is_loading {
                    self.is_loading = true;

                    return Task::perform(
                        self.project.clone().write_to_file(None),
                        Message::FileSaved,
                    );
                }
            }
            Message::FileSaved(result) => {
                self.is_loading = false;

                match result {
                    Ok(path) => {
                        self.project_path = Some(path);
                        self.is_dirty = false;
                    }
                    Err(error) => return error_dialog(error),
                }
            }
        }

        Task::none()
    }

    fn subscription(&self) -> iced::Subscription<Message> {
        keyboard::on_key_press(|key, modifiers| {
            if modifiers.command() {
                match key.as_ref() {
                    keyboard::Key::Character("o") => Some(Message::OpenFile),
                    keyboard::Key::Character("s") => {
                        Some(if modifiers.shift() {
                            Message::SaveFileAs
                        } else {
                            Message::SaveFile
                        })
                    }
                    keyboard::Key::Character("n") => Some(Message::NewFile),
                    _ => None,
                }
            } else {
                None
            }
        })
    }

    fn view(&self) -> Element<'_, Message> {
        let header = row![pick_list(
            self.config.theme.all.clone(),
            Some(self.theme.target()),
            |theme| Message::SwitchTheme(theme.into())
        )]
        .width(200);
        let pane_grid = pane_grid::PaneGrid::new(
            &self.pane_state,
            |id, pane, _is_maximized| {
                let is_focused = Some(id) == self.focus;
                match pane {
                    Panes::Designer => match &self.designer_page {
                        DesignerPane::DesignerView => designer_view::view(
                            self.project.element_tree.as_ref(),
                            self.project.get_theme(),
                            is_focused,
                        ),
                        DesignerPane::CodeView => {
                            code_view::view(&self.editor_content, is_focused)
                        }
                    },
                    Panes::ElementList => {
                        element_list::view(self.element_list, is_focused)
                    }
                }
            },
        )
        .width(Length::Fill)
        .height(Length::Fill)
        .spacing(10)
        .on_resize(10, Message::PaneResized)
        .on_click(Message::PaneClicked)
        .on_drag(Message::PaneDragged);

        let base = Column::new()
            .push(header)
            .push(pane_grid)
            .spacing(5)
            .align_x(Alignment::Center)
            .width(Length::Fill);

        let content = Dialog::with_buttons(
            self.dialog_is_open,
            container(base).height(Length::Fill),
            text(&self.dialog_content),
            match self.dialog_buttons {
                DialogButtons::None => vec![],
                DialogButtons::Ok => vec![ok_button()],
                DialogButtons::OkCancel => vec![ok_button(), cancel_button()],
            },
        )
        .title(self.dialog_title);

        Animation::new(&self.theme, content)
            .on_update(Message::SwitchTheme)
            .into()
    }
}