write 0.5.0

use std::collections::VecDeque;
use std::fs::{self, File, OpenOptions, TryLockError};
use std::io::{BufWriter, Write};
use std::path::PathBuf;
use std::time::{Duration, Instant};

use chrono::Local;
use eframe::egui;

mod render;

use render::{
    Glyph, PRIMARY, PRIMARY_DARKER, build_job, glyph_factor, shake_offset, shake_ready,
    start_byte_of_second_to_last_word, word_count,
};

const SYNC_INTERVAL: Duration = Duration::from_secs(30);
const FONT_SIZE: f32 = 28.0;

/// Result of acquiring today's session file: the locked handle, the day-total word
/// count already on disk, and the file's byte length at open (the hard delete floor).
type SessionFile = (File, usize, u64);

pub struct WriteApp {
    document: String,         // file mirror; retained for 1c word count
    visible: VecDeque<Glyph>, // render buffer; pruned each frame as glyphs fully fade
    writer: BufWriter<File>,  // append-only handle to the session .md
    visibility_head: f64,     // glyphs born before this fade out at once (visual clear)
    shake_start: f64,         // time of the last backspace-clear; drives the wiggle feedback
    enter_run: u8,            // 0 = none, 1 = one Enter written, 2+ = cleared / no-op
    boot_size: u64,           // file length at open; hard deletion floor (protects prior sessions)
    delete_floor: u64, // monotonic, non-decreasing byte floor; backspace never truncates below it
    menu_open: bool,
    fullscreen: bool,         // live window state; toggled from the menu (W)
    darker: bool,             // session-only render state; toggled from the menu (D)
    entered_fullscreen: bool, // one-shot: launch-time deferred fullscreen, applied once focused
    seed_words: usize,        // day-total words already on disk at open time
    last_sync: Instant,
}

/// Open today's append-only session file (`~/Documents/write/YYYY-MM-DD.md`),
/// creating it if needed, and take an exclusive advisory lock. Invocations on the
/// same day share one file; the lock ensures only one `write` process appends to
/// it at a time. Returns `Ok(None)` when another process already holds today's
/// lock — the caller should exit quietly.
pub fn open_session_file() -> Result<Option<SessionFile>, Box<dyn std::error::Error + Send + Sync>>
{
    let path = today_file_path()?;
    let file = OpenOptions::new().create(true).append(true).open(&path)?;
    // Advisory flock tied to this fd; released automatically when the process exits.
    match file.try_lock() {
        Ok(()) => {
            let existing = fs::read_to_string(&path).unwrap_or_default();
            // Count words BEFORE writing the header so its tokens aren't counted as the writer's.
            let seed = word_count(&existing);

            // Idempotent blank-line separation: cap at exactly two trailing newlines so
            // repeated launches don't grow blank-line runs. Empty file starts cleanly.
            let prefix = if existing.is_empty() || existing.ends_with("\n\n") {
                ""
            } else if existing.ends_with('\n') {
                "\n"
            } else {
                "\n\n"
            };
            let header = format!(
                "{prefix}## {}\n\n",
                Local::now().format("%Y-%m-%d %-I:%M:%S%P %Z")
            );

            // Write the header through the raw fd (append mode) BEFORE capturing boot_size,
            // so the existing delete-floor mechanism protects it and the TUI never shows it.
            let mut file = file;
            file.write_all(header.as_bytes())?;
            let boot_size = file.metadata()?.len();
            Ok(Some((file, seed, boot_size)))
        }
        Err(TryLockError::WouldBlock) => Ok(None),
        Err(TryLockError::Error(e)) => Err(Box::new(e)),
    }
}

/// Resolve today's session-file path (`~/Documents/write/YYYY-MM-DD.md`),
/// creating the `write/` directory if needed. Does not open or lock the file.
pub fn today_file_path() -> Result<PathBuf, Box<dyn std::error::Error + Send + Sync>> {
    let mut dir: PathBuf = dirs::document_dir()
        .or_else(dirs::home_dir)
        .ok_or("no Documents or home directory")?;
    dir.push("write");
    fs::create_dir_all(&dir)?;
    dir.push(format!("{}.md", Local::now().format("%Y-%m-%d")));
    Ok(dir)
}

/// Open today's session file in the writer's editor and wait for it to close.
/// Pre-creates the file so the editor opens an existing path. Resolves the editor
/// via `$VISUAL` → `$EDITOR` → `vi`, shell-splitting on whitespace so values like
/// `"code -w"` work. Propagates a non-zero editor exit code as our exit code.
pub fn run_edit() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    let path = today_file_path()?;
    // Touch so the editor opens an existing file (no advisory lock here).
    drop(OpenOptions::new().create(true).append(true).open(&path)?);

    let editor = std::env::var("VISUAL")
        .or_else(|_| std::env::var("EDITOR"))
        .unwrap_or_else(|_| "vi".to_string());

    let mut parts = editor.split_whitespace();
    let program = parts.next().ok_or("empty editor command")?;
    let status = std::process::Command::new(program)
        .args(parts)
        .arg(&path)
        .status()?; // inherits stdio by default

    if !status.success() {
        std::process::exit(status.code().unwrap_or(1));
    }
    Ok(())
}

/// Dump today's session file to stdout verbatim (including `## <datetime>` headers and
/// blank-line separators). A missing file is treated as an empty day: print nothing, exit 0.
/// Takes no advisory lock and creates nothing.
pub fn run_show() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    let path = today_file_path()?;
    match fs::read(&path) {
        Ok(bytes) => {
            use std::io::Write as _;
            std::io::stdout().write_all(&bytes)?;
            Ok(())
        }
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => Ok(()),
        Err(e) => Err(Box::new(e)),
    }
}

impl WriteApp {
    pub fn new(
        _cc: &eframe::CreationContext<'_>,
        file: File,
        seed_words: usize,
        boot_size: u64,
    ) -> Self {
        Self {
            document: String::new(),
            visible: VecDeque::new(),
            writer: BufWriter::new(file),
            visibility_head: 0.0,
            shake_start: f64::NEG_INFINITY,
            enter_run: 0,
            boot_size,
            delete_floor: boot_size,
            menu_open: false,
            fullscreen: false,
            darker: false,
            entered_fullscreen: false,
            seed_words,
            last_sync: Instant::now(),
        }
    }

    // Append raw text to the file mirror, the file, and the visible render buffer.
    fn append(&mut self, s: &str, now: f64) {
        let mut end = self.document.len();
        for ch in s.chars() {
            end += ch.len_utf8();
            self.visible.push_back(Glyph {
                ch,
                birth: now,
                end_offset: end,
            });
        }
        self.document.push_str(s);
        if let Err(e) = self.writer.write_all(s.as_bytes()) {
            eprintln!("write: failed to append to file: {e}");
        }
        let candidate = self.boot_size + start_byte_of_second_to_last_word(&self.document) as u64;
        self.delete_floor = self.delete_floor.max(candidate); // max() keeps it monotonic
    }

    // Trigger the "no" wiggle, debounced: ignore re-triggers while a wiggle is
    // still in flight so holding backspace at the floor produces one clean decay.
    fn start_shake(&mut self, now: f64) {
        if shake_ready(now, self.shake_start) {
            self.shake_start = now;
        }
    }

    // Bounded deletion: truncate the day file by one whole char, but never below
    // `delete_floor` (the monotonic "two words ago" floor). A blocked backspace
    // triggers a horizontal wiggle (via `shake_start`) as "no" feedback; a successful
    // deletion does not.
    fn backspace(&mut self, now: f64) {
        let current_end = self.boot_size + self.document.len() as u64;
        if self.document.is_empty() || current_end <= self.delete_floor {
            self.start_shake(now); // floor reached → wiggle feedback
            return;
        }
        let ch = self.document.chars().next_back().unwrap();
        if current_end - (ch.len_utf8() as u64) < self.delete_floor {
            self.start_shake(now); // would cross the floor → wiggle feedback
            return;
        }
        self.document.pop(); // UTF-8 safe: pops a whole char
        if let Err(e) = self.writer.flush() {
            eprintln!("write: flush before truncate failed: {e}");
            return; // do NOT set_len with bytes still buffered
        }
        let new_end = self.boot_size + self.document.len() as u64;
        if let Err(e) = self.writer.get_ref().set_len(new_end) {
            eprintln!("write: truncate failed: {e}");
        }
        let doc_len = self.document.len();
        while let Some(g) = self.visible.back() {
            if g.end_offset > doc_len {
                self.visible.pop_back();
            } else {
                break;
            }
        }
        self.enter_run = 0;
    }

    fn fsync(&mut self) {
        if let Err(e) = self.writer.flush() {
            eprintln!("write: flush failed: {e}");
        }
        if let Err(e) = self.writer.get_ref().sync_data() {
            eprintln!("write: sync_data failed: {e}");
        }
    }
}

impl eframe::App for WriteApp {
    fn ui(&mut self, ui: &mut egui::Ui, _frame: &mut eframe::Frame) {
        let ctx = ui.ctx().clone(); // cheap Arc clone; avoids borrow conflict with ui.painter()

        // Suppress Tab focus navigation.
        ctx.input_mut(|i| {
            let _ = i.consume_key(egui::Modifiers::NONE, egui::Key::Tab);
        });

        let now = ctx.input(|i| i.time);

        // Enter fullscreen only once the window has focus, to dodge the launch-time
        // activation race that otherwise bounces the Space back to the desktop. Keep
        // repainting until then so we actually observe the focus transition.
        if !self.entered_fullscreen {
            if ctx.input(|i| i.focused) {
                ctx.send_viewport_cmd(egui::ViewportCommand::Fullscreen(true));
                self.entered_fullscreen = true;
                self.fullscreen = true;
            } else {
                ctx.request_repaint();
            }
        }

        // Copy events out, keep the input closure short.
        let events = ctx.input(|i| i.events.clone());
        // Closing the menu is deferred to after this batch so the trailing Text
        // event a key emits (e.g. W's "w") is still swallowed by the menu gate
        // and doesn't leak into the document.
        let mut close_menu = false;
        for event in events {
            if self.menu_open {
                if let egui::Event::Key {
                    key, pressed: true, ..
                } = event
                {
                    match key {
                        egui::Key::Q => ctx.send_viewport_cmd(egui::ViewportCommand::Close),
                        egui::Key::W => {
                            self.fullscreen = !self.fullscreen;
                            ctx.send_viewport_cmd(egui::ViewportCommand::Fullscreen(
                                self.fullscreen,
                            ));
                            close_menu = true; // resume writing in the new window state
                        }
                        egui::Key::D => {
                            self.darker = !self.darker;
                            close_menu = true; // resume writing in the new brightness
                        }
                        egui::Key::Backslash => {
                            self.append("\\", now);
                            self.enter_run = 0; // match the normal Text-append path
                            close_menu = true; // resume writing after inserting
                        }
                        egui::Key::Escape => close_menu = true, // resume writing
                        _ => {}
                    }
                }
                continue; // ignore Text and everything else while the menu is open
            }
            match event {
                egui::Event::Text(t) => {
                    self.append(&t, now); // Text already handles shift/dead/accents
                    self.enter_run = 0;
                }
                egui::Event::Key {
                    key,
                    modifiers,
                    pressed: true,
                    ..
                } => {
                    if modifiers.command {
                        // ignore Cmd/Ctrl combos (paste, undo, ...)
                        continue;
                    }
                    match key {
                        egui::Key::Enter => {
                            match self.enter_run {
                                0 => {
                                    self.append("\n\n", now);
                                    self.enter_run = 1;
                                }
                                1 => {
                                    // Second consecutive Enter: clear the screen (no newlines).
                                    self.visibility_head = now;
                                    // Move the backspace floor beyond the just-inserted CR so a
                                    // post-clear backspace can't reach back into the now-cleared text.
                                    self.delete_floor = self.boot_size + self.document.len() as u64;
                                    self.enter_run = 2;
                                }
                                _ => {} // already cleared → no-op
                            }
                        }
                        egui::Key::Backspace => {
                            self.backspace(now);
                        }
                        egui::Key::Escape | egui::Key::Backslash => {
                            self.menu_open = true;
                            self.fsync(); // durable flush on menu open
                        }
                        _ => {} // arrows, Delete, Tab, Home, End, PageUp/Down → no-op
                    }
                }
                _ => {} // Event::Paste, Event::Ime, pointer, etc. → ignored
            }
        }
        if close_menu {
            self.menu_open = false;
        }

        // Periodic durability: flush + sync_data ~every 30s.
        if self.last_sync.elapsed() >= SYNC_INTERVAL {
            self.fsync();
            self.last_sync = Instant::now();
        }

        // Prune fully-faded glyphs from the front (oldest first; glyphs are in birth order).
        // Uses the combined factor so clear-faded glyphs drain too.
        while let Some(front) = self.visible.front() {
            if glyph_factor(front.birth, now, self.visibility_head) <= 0.0 {
                self.visible.pop_front();
            } else {
                break;
            }
        }

        // Render: full-rect black background + a center-anchored, centered fading galley.
        let rect = ctx.content_rect();
        let margin = 40.0_f32;
        let max_col = 760.0_f32;
        let col_width = (rect.width() - 2.0 * margin).clamp(0.0, max_col);
        let primary = if self.darker { PRIMARY_DARKER } else { PRIMARY };
        let floor_byte = (self.delete_floor - self.boot_size) as usize;
        let job = build_job(
            self.visible.make_contiguous(),
            now,
            self.visibility_head,
            egui::FontId::proportional(FONT_SIZE),
            primary,
            col_width,
            floor_byte,
        );
        let painter = ui.painter();
        painter.rect_filled(rect, 0.0, egui::Color32::BLACK);
        let galley = painter.layout_job(job);
        let dx = shake_offset(now - self.shake_start);
        let x = rect.left() + (rect.width() - col_width) * 0.5 + dx;
        let x_height = 0.5 * FONT_SIZE; // egui 0.34 has no x-height metric; 0.5em is close enough
        let center_y = rect.center().y;
        let y = center_y + x_height - galley.size().y; // newest line one x-height below center; older lines scroll up

        // Insertion-point marker: a passive, non-blinking block at the galley tail. Computed
        // here because painter.galley(...) below moves `galley`.
        let cursor_rect = (!self.visible.is_empty()).then(|| {
            let tail = galley.pos_from_cursor(galley.end()); // galley-local, zero-width
            let top_left = egui::pos2(x, y) + tail.min.to_vec2();
            let h = tail.height().max(FONT_SIZE); // never degenerate
            egui::Rect::from_min_size(top_left, egui::vec2(0.5 * FONT_SIZE, h))
        });

        painter.galley(egui::pos2(x, y), galley, primary); // existing line — galley moved here

        // Cursor fades with the newest visible glyph and inherits darker mode via `primary`.
        if let Some(cursor_rect) = cursor_rect
            && let Some(g) = self.visible.back()
        {
            let factor = glyph_factor(g.birth, now, self.visibility_head);
            let cursor_color = primary.gamma_multiply(factor);
            painter.rect_filled(cursor_rect, 0.0, cursor_color);
        }

        let words = self.seed_words + word_count(&self.document);
        if words >= 2 {
            let pos = egui::pos2(rect.right() - margin, rect.top() + margin);
            painter.text(
                pos,
                egui::Align2::RIGHT_TOP,
                format!("{words} words"),
                egui::FontId::proportional(0.5 * FONT_SIZE), // smaller than body
                egui::Color32::from_gray(110),               // dim gray so the HUD recedes
            );
        }

        if self.menu_open {
            let words = self.seed_words + word_count(&self.document);
            egui::Window::new("menu")
                .title_bar(false)
                .resizable(false)
                .collapsible(false)
                .anchor(egui::Align2::CENTER_CENTER, egui::Vec2::ZERO)
                .show(&ctx, |ui| {
                    ui.vertical_centered(|ui| {
                        ui.label(
                            egui::RichText::new(format!("{words} words"))
                                .size(24.0)
                                .color(PRIMARY),
                        );
                        ui.add_space(8.0);
                        let win_cmd = if self.fullscreen {
                            "windowed"
                        } else {
                            "fullscreen"
                        };
                        let dark_cmd = if self.darker { "normal" } else { "darker" };
                        let rows = [
                            ("quit", "Q"),
                            ("resume", "Esc"),
                            (win_cmd, "W"),
                            (dark_cmd, "D"),
                            ("backslash", "\\"),
                        ];
                        egui::Grid::new("menu_commands")
                            .num_columns(2)
                            .spacing([24.0, 6.0])
                            .show(ui, |ui| {
                                for (cmd, key) in rows {
                                    ui.label(egui::RichText::new(cmd).size(17.5).color(PRIMARY));
                                    ui.with_layout(
                                        egui::Layout::right_to_left(egui::Align::Center),
                                        |ui| {
                                            ui.label(
                                                egui::RichText::new(key).size(17.5).color(PRIMARY),
                                            );
                                        },
                                    );
                                    ui.end_row();
                                }
                            });
                    });
                });
        }

        if self.visible.is_empty() {
            // Idle: nothing to animate. Wake only to run the periodic fsync.
            ctx.request_repaint_after(SYNC_INTERVAL.saturating_sub(self.last_sync.elapsed()));
        } else {
            // Text still visible/fading: animate.
            ctx.request_repaint();
        }
    }

    fn clear_color(&self, _visuals: &egui::Visuals) -> [f32; 4] {
        [0.0, 0.0, 0.0, 1.0]
    }

    fn on_exit(&mut self) {
        self.fsync(); // final flush + sync on clean exit
    }
}