mod appearance;
mod auth;
mod runtime;

use cosmic::iced::platform_specific::shell::commands::session_lock;
use cosmic::{
    Element,
    app::{Core, Settings, Task},
    iced::{
        self, Event, Length, Subscription,
        event::{self, PlatformSpecific, wayland},
        window,
    },
    theme,
    widget::{self, column, container, image, row},
};
use std::{
    collections::HashMap,
    time::{Duration, Instant},
};
use wayland_client::protocol::wl_output::WlOutput;
use zeroize::{Zeroize, Zeroizing};

fn main() -> iced::Result {
    let args: Vec<_> = std::env::args().skip(1).collect();
    if args == ["--help"] {
        println!(
            "Usage: ferese-lock [--preview | --foreground]\nWithout arguments, lock the session using PAM.\n--preview opens an ordinary window; it never locks or authenticates."
        );
        return Ok(());
    }
    let preview = args == ["--preview"]
        || (args.len() == 3 && args[0] == "--preview" && args[1] == "--config");
    let config = if preview && args.len() == 3 {
        Some(std::path::PathBuf::from(&args[2]))
    } else {
        ferese_config::config_path()
    };
    if !args.is_empty() && !preview && args != ["--foreground"] {
        eprintln!("Usage: ferese-lock [--preview | --foreground]");
        std::process::exit(2);
    }
    if !preview && !auth::available() {
        eprintln!("ferese-lock: install the ferese-lock PAM policy before locking");
        std::process::exit(1);
    }
    if !preview {
        if let Err(error) = runtime::check_protocol() {
            eprintln!("ferese-lock: {error}");
            std::process::exit(1);
        }
        if args.is_empty() {
            if let Err(error) = runtime::daemonize() {
                eprintln!("ferese-lock: {error}");
                std::process::exit(1);
            }
            return Ok(());
        }
    }
    let user = auth::username().unwrap_or_else(|error| {
        eprintln!("ferese-lock: {error}");
        std::process::exit(1)
    });
    // Password-bearing UI processes must not create core dumps.
    unsafe {
        libc::prctl(libc::PR_SET_DUMPABLE, 0);
    }
    let appearance = appearance::Appearance::load(&user, config.as_deref());
    cosmic::app::run::<Locker>(
        Settings::default()
            .no_main_window(!preview)
            .is_daemon(!preview)
            .client_decorations(false)
            .default_text_size(14.)
            .default_font(appearance.font)
            .theme(appearance.theme())
            .antialiasing(true),
        (preview, user, appearance),
    )
}

#[derive(Clone)]
enum Message {
    Event(Event),
    Input(String),
    Submit,
    Authenticated(bool),
    Tick,
}
// Never derive Debug for password-bearing messages.
impl std::fmt::Debug for Message {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str("LockerMessage")
    }
}
#[derive(Default, Debug, PartialEq)]
enum AuthState {
    #[default]
    Ready,
    Checking,
    Rejected(Instant),
    Unlocking,
}
impl AuthState {
    fn may_submit(&self) -> bool {
        matches!(self, Self::Ready)
    }
    fn complete(&mut self, success: bool) -> bool {
        if *self != Self::Checking {
            return false;
        }
        *self = if success {
            Self::Unlocking
        } else {
            Self::Rejected(Instant::now())
        };
        success
    }
}
struct Locker {
    core: Core,
    preview: bool,
    user: String,
    appearance: appearance::Appearance,
    password: Zeroizing<String>,
    state: AuthState,
    status: &'static str,
    outputs: HashMap<WlOutput, window::Id>,
    started: bool,
    clock: String,
    date: String,
    confirmation: runtime::Confirmation,
    caps_lock: bool,
}
impl cosmic::Application for Locker {
    type Executor = cosmic::executor::Default;
    type Flags = (bool, String, appearance::Appearance);
    type Message = Message;
    const APP_ID: &'static str = "dev.ferese.Lock";
    fn core(&self) -> &Core {
        &self.core
    }
    fn core_mut(&mut self) -> &mut Core {
        &mut self.core
    }
    fn init(mut core: Core, (preview, user, appearance): Self::Flags) -> (Self, Task<Message>) {
        core.window.show_headerbar = false;
        let now = jiff::Zoned::now();
        let clock = now.strftime(appearance.clock_format()).to_string();
        let title_task = core
            .main_window_id()
            .map(|id| cosmic::command::set_title(id, "Ferese Lock Preview".to_owned()))
            .unwrap_or_else(Task::none);
        (
            Self {
                core,
                preview,
                user,
                appearance,
                password: Zeroizing::new(String::new()),
                state: AuthState::Ready,
                status: "Press Enter to unlock",
                outputs: HashMap::new(),
                started: false,
                confirmation: Default::default(),
                caps_lock: false,
                clock,
                date: now.strftime("%A, %B %-d").to_string(),
            },
            if preview {
                Task::batch([title_task, focus()])
            } else {
                session_lock::lock()
            },
        )
    }
    fn subscription(&self) -> Subscription<Message> {
        Subscription::batch([
            event::listen_with(|event, _, _| match event {
                Event::PlatformSpecific(PlatformSpecific::Wayland(
                    wayland::Event::Output(..) | wayland::Event::SessionLock(..),
                ))
                | Event::Window(window::Event::Opened { .. })
                | Event::Keyboard(iced::keyboard::Event::ModifiersChanged(_)) => {
                    Some(Message::Event(event))
                }
                _ => None,
            }),
            iced::time::every(Duration::from_secs(1)).map(|_| Message::Tick),
        ])
    }
    fn update(&mut self, message: Message) -> Task<Message> {
        match message {
            Message::Event(Event::Keyboard(iced::keyboard::Event::ModifiersChanged(modifiers))) => {
                self.caps_lock = modifiers.contains(iced::keyboard::Modifiers::CAPS_LOCK);
            }
            Message::Tick => {
                let now = jiff::Zoned::now();
                self.clock = now.strftime(self.appearance.clock_format()).to_string();
                self.date = now.strftime("%A, %B %-d").to_string();
                if matches!(self.state, AuthState::Rejected(at) if at.elapsed() >= Duration::from_secs(2))
                {
                    self.state = AuthState::Ready;
                }
            }
            Message::Input(mut value) => {
                if self.state.may_submit() && value.len() <= 4096 {
                    self.password.zeroize();
                    *self.password = std::mem::take(&mut value);
                    self.status = "Press Enter to unlock";
                }
                value.zeroize();
            }
            Message::Submit => {
                if self.preview {
                    self.password.zeroize();
                    self.status = "Preview only · your session is not locked";
                    return focus();
                }
                if !self.confirmation.confirmed()
                    || !self.state.may_submit()
                    || self.password.is_empty()
                {
                    return Task::none();
                }
                self.state = AuthState::Checking;
                self.status = "Checking password…";
                let password = Zeroizing::new(std::mem::take(&mut *self.password));
                let user = self.user.clone();
                return Task::perform(
                    async move {
                        tokio::task::spawn_blocking(move || auth::authenticate(user, password))
                            .await
                            .unwrap_or(false)
                    },
                    |accepted| cosmic::Action::App(Message::Authenticated(accepted)),
                );
            }
            Message::Authenticated(accepted) => {
                if self.state.complete(accepted) && !self.preview {
                    return session_lock::unlock();
                }
                self.status = "Password not accepted. Try again.";
                return focus();
            }
            Message::Event(Event::PlatformSpecific(PlatformSpecific::Wayland(
                wayland::Event::Output(event, output),
            ))) if !self.preview => {
                if matches!(event, wayland::OutputEvent::Removed) {
                    if let Some(id) = self.outputs.remove(&output) {
                        return session_lock::destroy_lock_surface(id);
                    }
                } else if !self.outputs.contains_key(&output) {
                    let id = window::Id::unique();
                    self.outputs.insert(output.clone(), id);
                    if self.started {
                        return session_lock::get_lock_surface(id, output);
                    }
                }
            }
            Message::Event(Event::PlatformSpecific(PlatformSpecific::Wayland(
                wayland::Event::SessionLock(event),
            ))) if !self.preview => match event {
                wayland::SessionLockEvent::Locked => {
                    if self.confirmation.observe() {
                        use std::io::Write;
                        if std::env::var_os("FERESE_LOCK_READY").is_some() {
                            let _ = writeln!(std::io::stdout().lock(), "FERESE_LOCKED");
                        }
                    }
                    if !self.started {
                        self.started = true;
                        return Task::batch(self.outputs.iter().map(|(output, id)| {
                            session_lock::get_lock_surface(*id, output.clone())
                        }));
                    }
                }
                wayland::SessionLockEvent::Focused(..) => return focus(),
                wayland::SessionLockEvent::Unlocked if self.state == AuthState::Unlocking => {
                    return iced::exit();
                }
                wayland::SessionLockEvent::Finished | wayland::SessionLockEvent::NotSupported => {
                    eprintln!("ferese-lock: compositor refused the lock");
                    return iced::exit();
                }
                _ => {}
            },
            Message::Event(Event::Window(window::Event::Opened { .. })) => return focus(),
            _ => {}
        }
        Task::none()
    }
    fn on_escape(&mut self) -> Task<Message> {
        self.password.zeroize();
        focus()
    }
    fn view(&self) -> Element<'_, Message> {
        self.screen()
    }
    fn view_window(&self, _: window::Id) -> Element<'_, Message> {
        self.screen()
    }
}
fn focus() -> Task<Message> {
    widget::text_input::focus(widget::Id::new("password"))
}
impl Locker {
    fn label<'a>(
        &self,
        label: impl Into<std::borrow::Cow<'a, str>> + 'a,
        size: u16,
    ) -> widget::Text<'a, cosmic::Theme, cosmic::Renderer> {
        widget::text(label).font(self.appearance.font).size(size)
    }
    fn screen(&self) -> Element<'_, Message> {
        iced::widget::responsive(move |size| self.screen_content(size)).into()
    }
    fn screen_content(&self, size: iced::Size) -> Element<'_, Message> {
        let avatar_size = (size.height * 0.09).clamp(64., 112.);
        let clock_size = (size.height * 0.165)
            .min(
                size.width
                    * if self.appearance.twelve_hour {
                        0.075
                    } else {
                        0.12
                    },
            )
            .clamp(40., 180.) as u16;
        let margin = if size.width < 640. { 24. } else { 36. };
        let top = (size.height * 0.09).max(24.);
        let a = &self.appearance;
        let mut input = widget::text_input("Password", self.password.as_str())
            .password()
            .padding([8, 18])
            .style(password_style(a.radius, a.accent, a.panel))
            .font(a.font)
            .id(widget::Id::new("password"));
        if self.state.may_submit() {
            input = input.on_input(Message::Input).on_submit(|mut value| {
                value.zeroize();
                Message::Submit
            });
        }
        let submit = widget::button::custom(widget::icon::from_name("go-next-symbolic").size(20))
            .class(theme::Button::Icon)
            .padding(6)
            .on_press_maybe(self.state.may_submit().then_some(Message::Submit));
        input = input.trailing_icon(container(submit).padding([0, 6]).into());
        let avatar: Element<'_, Message> = if let Some(photo) = &a.avatar {
            image(photo.clone())
                .width(avatar_size)
                .height(avatar_size)
                .content_fit(iced::ContentFit::Cover)
                .border_radius(avatar_size / 2.)
                .into()
        } else {
            container(line_icon(
                "M16 8a4 4 0 1 1-8 0 4 4 0 0 1 8 0 M4 21v-2a8 8 0 0 1 16 0v2",
                46,
            ))
            .center_x(avatar_size)
            .center_y(avatar_size)
            .class(theme::Container::custom(move |_| {
                let fill = ferese_theme::composite(a.accent.scale_alpha(0.18), a.panel);
                let on = ferese_theme::foreground(fill, a.accent);
                widget::container::Style {
                    background: Some(iced::Background::Color(fill)),
                    text_color: Some(on),
                    icon_color: Some(on),
                    border: iced::Border {
                        radius: (avatar_size / 2.).into(),
                        width: 1.,
                        color: a.accent.scale_alpha(0.3),
                    },
                    ..Default::default()
                }
            }))
            .into()
        };
        let identity = column![
            avatar,
            self.label(&self.user, 16).class(iced::Color::WHITE),
            widget::Space::new().height(6),
            input,
            self.label(
                if self.caps_lock {
                    "Caps Lock is on"
                } else {
                    self.status
                },
                12
            )
            .class(iced::Color::WHITE.scale_alpha(0.68)),
        ]
        .spacing(12)
        .align_x(iced::Alignment::Center)
        .width(
            (size.height * 0.37)
                .clamp(260., 420.)
                .min((size.width - margin * 2.).max(160.)),
        );
        let mut clock = column([]).spacing(0).align_x(iced::Alignment::Center);
        if a.show_clock {
            clock = clock.push(
                self.label(&self.clock, clock_size)
                    .font(cosmic::font::Font {
                        weight: cosmic::iced::font::Weight::Light,
                        ..a.font
                    })
                    .class(iced::Color::WHITE),
            );
        }
        if a.show_date {
            clock = clock.push(
                self.label(&self.date, 18)
                    .class(iced::Color::WHITE.scale_alpha(0.72)),
            );
        }
        let clock_height = if a.show_clock {
            clock_size as f32 * 1.3
        } else {
            0.
        } + if a.show_date { 24. } else { 0. };
        let gap = (size.height * 0.44 - top - clock_height).max(24.);
        let foreground = container(
            column![
                container(clock).center_x(Length::Fill),
                widget::Space::new().height(gap),
                container(identity).center_x(Length::Fill),
                widget::Space::new().height(Length::Fill),
                row![
                    row![ferese_symbol(), self.label("Ferese", 13)]
                        .spacing(5)
                        .align_y(iced::Alignment::Center),
                    widget::Space::new().width(Length::Fill),
                    self.label(
                        if self.preview {
                            "Preview · your session is not locked"
                        } else {
                            ""
                        },
                        11
                    )
                ]
                .align_y(iced::Alignment::Center),
            ]
            .height(Length::Fill),
        )
        .padding(iced::Padding {
            top,
            right: margin,
            bottom: margin,
            left: margin,
        })
        .width(Length::Fill)
        .height(Length::Fill)
        .class(theme::Container::custom(move |_| {
            widget::container::Style {
                text_color: Some(iced::Color::WHITE.scale_alpha(0.65)),
                background: Some(iced::Background::Color(iced::Color::from_rgba(
                    0.025, 0.035, 0.06, a.dim,
                ))),
                ..Default::default()
            }
        }));
        let background: Element<'_, Message> = image(a.wallpaper.clone())
            .width(Length::Fill)
            .height(Length::Fill)
            .content_fit(iced::ContentFit::Cover)
            .into();
        iced::widget::stack![background, foreground].into()
    }
}
#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn only_an_active_successful_attempt_unlocks() {
        let mut state = AuthState::Ready;
        assert!(!state.complete(true));
        state = AuthState::Checking;
        assert!(!state.complete(false));
        assert!(!state.may_submit());
        assert!(!state.complete(true));
        state = AuthState::Checking;
        assert!(state.complete(true));
        assert!(!state.complete(true));
    }
    #[test]
    fn password_messages_are_redacted() {
        assert!(!format!("{:?}", Message::Input("secret".into())).contains("secret"));
    }
}

fn line_icon(path: &str, size: u16) -> widget::icon::Icon {
    let svg = format!(
        r#"<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><path d="{path}" fill="none" stroke="white" stroke-width="1.6" stroke-linecap="round" stroke-linejoin="round"/></svg>"#
    );
    widget::icon(widget::icon::from_svg_bytes(svg.into_bytes()).symbolic(true)).size(size)
}
fn ferese_symbol() -> widget::icon::Icon {
    widget::icon(
        widget::icon::from_svg_bytes(
            include_bytes!("../../../packaging/icons/ferese.svg").as_slice(),
        )
        .symbolic(true),
    )
    .size(30)
}

fn password_style(radius: f32, accent: iced::Color, surface: iced::Color) -> theme::TextInput {
    let appearance = move |focused: bool| widget::text_input::Appearance {
        background: iced::Color::from_rgba(
            (surface.r + 1.) * 0.5,
            (surface.g + 1.) * 0.5,
            (surface.b + 1.) * 0.5,
            0.24,
        )
        .into(),
        border_radius: (radius * 2.).min(26.).into(),
        border_width: 1.,
        border_offset: None,
        border_color: if focused {
            accent.scale_alpha(0.7)
        } else {
            iced::Color::WHITE.scale_alpha(0.15)
        },
        icon_color: Some(iced::Color::WHITE.scale_alpha(0.8)),
        text_color: Some(iced::Color::WHITE),
        placeholder_color: iced::Color::WHITE.scale_alpha(0.55),
        selected_text_color: iced::Color::WHITE,
        selected_fill: accent.scale_alpha(0.5),
        label_color: iced::Color::WHITE,
    };
    theme::TextInput::Custom {
        active: Box::new(move |_| appearance(false)),
        hovered: Box::new(move |_| appearance(true)),
        focused: Box::new(move |_| appearance(true)),
        error: Box::new(move |_| appearance(false)),
        disabled: Box::new(move |_| appearance(false)),
    }
}