tuigram-cli 0.7.0

//! Files waiting in the composer to be sent, and the ways they get there:
//! a typed path (`a`), the clipboard (`p`), or a paste of file paths, which
//! is what dropping files on a terminal window types.

use std::path::{Path, PathBuf};

use crate::text;
use crate::tg::Upload;

/// Telegram's limits for photos. Bigger ones go as files, uncompressed.
const PHOTO_MAX_BYTES: u64 = 10 * 1024 * 1024;
/// Width plus height.
const PHOTO_MAX_SIDES: u32 = 10_000;
/// Long side over short side.
const PHOTO_MAX_RATIO: u32 = 20;
/// Image types that go as photos; anything else (GIFs, HEIC, RAW…) as a file.
const PHOTO_TYPES: &[&str] = &["jpg", "jpeg", "png", "webp"];
/// Messages Telegram groups into one album at most.
const ALBUM_MAX: usize = 10;

/// How a file is sent.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum Kind {
    /// Compressed and shown in the chat, like Telegram's "Send as photo".
    Photo { width: u32, height: u32 },
    /// As it is, with its name.
    File,
}

/// A file to go with the next message sent.
#[derive(Clone, Debug)]
pub struct Attachment {
    /// Absolute, as TDLib wants it.
    pub path: PathBuf,
    /// What the composer calls it: the file name, without control
    /// characters, which any file name can hold.
    pub name: String,
    pub size: u64,
    pub kind: Kind,
}

impl Attachment {
    /// Checks that `path` is a file Telegram can take, and works out whether
    /// it goes as a photo.
    pub fn new(path: &Path) -> Result<Self, String> {
        let name = path
            .file_name()
            .map(|n| text::clean(&n.to_string_lossy()))
            .unwrap_or_else(|| text::clean(&path.display().to_string()));
        let meta = std::fs::metadata(path).map_err(|e| match e.kind() {
            std::io::ErrorKind::NotFound => format!("No such file: {name}"),
            _ => format!("Can't read {name}: {e}"),
        })?;
        if meta.is_dir() {
            return Err(format!("{name} is a folder"));
        }
        if meta.len() == 0 {
            return Err(format!("{name} is empty"));
        }
        let path = std::fs::canonicalize(path).map_err(|e| format!("Can't read {name}: {e}"))?;
        // TDLib takes paths as JSON text.
        if path.to_str().is_none() {
            return Err(format!("Can't send {name}: its path isn't valid UTF-8"));
        }
        let kind = match photo_size(&path, meta.len()) {
            Some((width, height)) => Kind::Photo { width, height },
            None => Kind::File,
        };
        Ok(Self {
            path,
            name,
            size: meta.len(),
            kind,
        })
    }
}

impl Attachment {
    /// How it goes: as [`Kind`] says, unless `as_file` (Ctrl-t) sends
    /// photos as files, uncompressed.
    pub fn sent_as(&self, as_file: bool) -> Kind {
        if as_file { Kind::File } else { self.kind }
    }

    /// What TDLib is asked to send; see [`Self::sent_as`].
    pub fn upload(&self, as_file: bool) -> Upload {
        let path = self.path.to_string_lossy().into_owned();
        match self.sent_as(as_file) {
            Kind::Photo { width, height } => Upload::Photo {
                path,
                width: width as i32,
                height: height as i32,
            },
            Kind::File => Upload::File { path },
        }
    }
}

/// The size of an image that Telegram takes as a photo, read from its header.
fn photo_size(path: &Path, bytes: u64) -> Option<(u32, u32)> {
    let extension = path.extension()?.to_str()?.to_ascii_lowercase();
    if !PHOTO_TYPES.contains(&extension.as_str()) || bytes > PHOTO_MAX_BYTES {
        return None;
    }
    let (width, height) = image::image_dimensions(path).ok()?;
    let (long, short) = (width.max(height), width.min(height));
    let fits = short > 0 && width + height <= PHOTO_MAX_SIDES && long <= short * PHOTO_MAX_RATIO;
    fits.then_some((width, height))
}

/// A paste that became attachments, which Ctrl-z turns back into the text.
pub struct Dropped {
    pub text: String,
    /// How many attachments it added, at the end of the list.
    pub count: usize,
}

/// Groups attachments the way Telegram sends them together, keeping their
/// order: up to ten photos make one album, and so do up to ten files. A
/// photo and a file can't share one. `as_files` sends photos as files.
pub fn albums(attachments: &[Attachment], as_files: bool) -> Vec<&[Attachment]> {
    let photo = |a: &Attachment| a.sent_as(as_files).is_photo();
    attachments
        .chunk_by(|a, b| photo(a) == photo(b))
        .flat_map(|run| run.chunks(ALBUM_MAX))
        .collect()
}

impl Kind {
    pub fn is_photo(self) -> bool {
        matches!(self, Kind::Photo { .. })
    }
}

/// The files a paste names, if it names nothing else. Dropping files on a
/// terminal types their paths, quoted or with `\` before spaces, or as
/// `file://` URLs, depending on the terminal. Only absolute paths count, so
/// pasting a word that happens to be a file's name stays text.
pub fn pasted_paths(text: &str) -> Option<Vec<PathBuf>> {
    let text = text.trim();
    if text.is_empty() {
        return None;
    }
    // One path with spaces in it, pasted as it is.
    if let Some(path) = existing_file(text) {
        return Some(vec![path]);
    }
    words(text)?.iter().map(|w| existing_file(w)).collect()
}

fn existing_file(word: &str) -> Option<PathBuf> {
    let path = match word.strip_prefix("file://") {
        Some(url) => from_file_url(url)?,
        None => expand_home(word),
    };
    (path.is_absolute() && path.is_file()).then_some(path)
}

/// Splits `text` into words the way a shell does, which is how terminals
/// quote dropped paths. `None` if a quote is left open.
fn words(text: &str) -> Option<Vec<String>> {
    // Backslashes separate folders on Windows, where paths with spaces are
    // put in double quotes instead.
    let escapes = !cfg!(windows);
    let mut words = Vec::new();
    let mut word = String::new();
    let mut in_word = false;
    let mut chars = text.chars();
    while let Some(c) = chars.next() {
        match c {
            '\'' => {
                in_word = true;
                loop {
                    match chars.next()? {
                        '\'' => break,
                        c => word.push(c),
                    }
                }
            }
            '"' => {
                in_word = true;
                loop {
                    match chars.next()? {
                        '"' => break,
                        '\\' if escapes => word.push(chars.next()?),
                        c => word.push(c),
                    }
                }
            }
            '\\' if escapes => {
                in_word = true;
                word.push(chars.next()?);
            }
            c if c.is_whitespace() => {
                if in_word {
                    words.push(std::mem::take(&mut word));
                    in_word = false;
                }
            }
            c => {
                in_word = true;
                word.push(c);
            }
        }
    }
    if in_word {
        words.push(word);
    }
    Some(words)
}

/// The path in a `file://` URL (after the `file://`), percent-decoded.
fn from_file_url(url: &str) -> Option<PathBuf> {
    let path = url.strip_prefix("localhost").unwrap_or(url);
    let bytes = path.as_bytes();
    let mut decoded = Vec::with_capacity(bytes.len());
    let mut i = 0;
    while i < bytes.len() {
        if bytes[i] == b'%' {
            let hex = std::str::from_utf8(bytes.get(i + 1..i + 3)?).ok()?;
            decoded.push(u8::from_str_radix(hex, 16).ok()?);
            i += 3;
        } else {
            decoded.push(bytes[i]);
            i += 1;
        }
    }
    let path = String::from_utf8(decoded).ok()?;
    // `file:///C:/Users/…` on Windows.
    if cfg!(windows) && path.as_bytes().get(2) == Some(&b':') {
        return Some(PathBuf::from(&path[1..]));
    }
    Some(PathBuf::from(path))
}

/// `~` and `~/…` mean the home folder, as in a shell.
pub fn expand_home(path: &str) -> PathBuf {
    let rest = match path.strip_prefix('~') {
        Some("") => "",
        Some(rest) if rest.starts_with(['/', std::path::MAIN_SEPARATOR]) => &rest[1..],
        _ => return PathBuf::from(path),
    };
    match dirs::home_dir() {
        Some(home) => home.join(rest),
        None => PathBuf::from(path),
    }
}

/// What Tab does to a path typed in the attach prompt.
#[derive(Debug, PartialEq, Eq)]
pub struct Completion {
    /// The path, with its last part completed as far as the matching names
    /// agree, and a `/` after a lone folder.
    pub text: String,
    /// Every name that matches, folders with a `/`, when there's more than one.
    pub matches: Vec<String>,
}

/// Completes the last part of `typed` from the names in its folder, like a
/// shell's Tab. Hidden files only match once a `.` is typed.
pub fn complete(typed: &str) -> Completion {
    let split = typed
        .rfind(['/', std::path::MAIN_SEPARATOR])
        .map_or(0, |i| i + 1);
    let (folder, start) = typed.split_at(split);
    let dir = if folder.is_empty() {
        PathBuf::from(".")
    } else {
        expand_home(folder)
    };
    let mut names: Vec<(String, bool)> = std::fs::read_dir(&dir)
        .into_iter()
        .flatten()
        .flatten()
        .filter_map(|entry| {
            let name = entry.file_name().into_string().ok()?;
            let wanted =
                name.starts_with(start) && (start.starts_with('.') || !name.starts_with('.'));
            // Follows links, so a link to a folder completes like one.
            wanted.then(|| (name, entry.path().is_dir()))
        })
        .collect();
    names.sort();
    let unchanged = || Completion {
        text: typed.to_string(),
        matches: Vec::new(),
    };
    match names.as_slice() {
        [] => unchanged(),
        [(name, is_dir)] => Completion {
            text: format!("{folder}{name}{}", if *is_dir { "/" } else { "" }),
            matches: Vec::new(),
        },
        [(first, _), rest @ ..] => {
            let common = rest.iter().fold(first.as_str(), |common, (name, _)| {
                let same = common
                    .char_indices()
                    .zip(name.chars())
                    .find(|((_, a), b)| a != b)
                    .map_or(common.len().min(name.len()), |((i, _), _)| i);
                &common[..same]
            });
            Completion {
                text: format!("{folder}{common}"),
                matches: names
                    .iter()
                    .map(|(name, is_dir)| format!("{name}{}", if *is_dir { "/" } else { "" }))
                    .collect(),
            }
        }
    }
}

/// A file size the way Telegram shows one: "340 KB", "2.1 MB".
pub fn size_label(bytes: u64) -> String {
    const UNITS: [&str; 4] = ["KB", "MB", "GB", "TB"];
    if bytes < 1024 {
        return format!("{bytes} B");
    }
    let mut size = bytes as f64 / 1024.0;
    let mut unit = 0;
    while size >= 1024.0 && unit < UNITS.len() - 1 {
        size /= 1024.0;
        unit += 1;
    }
    if size < 10.0 {
        format!("{size:.1} {}", UNITS[unit])
    } else {
        format!("{size:.0} {}", UNITS[unit])
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    /// A folder of its own under the system temp folder, removed when dropped.
    struct TempDir(PathBuf);

    impl TempDir {
        fn new(name: &str) -> Self {
            let dir =
                std::env::temp_dir().join(format!("tuigram-test-{name}-{}", std::process::id()));
            let _ = std::fs::remove_dir_all(&dir);
            std::fs::create_dir_all(&dir).unwrap();
            // Canonical, so it compares equal to what `Attachment::new` gives
            // (on macOS the temp folder is behind a link).
            Self(std::fs::canonicalize(&dir).unwrap())
        }

        fn file(&self, name: &str, contents: &[u8]) -> PathBuf {
            let path = self.0.join(name);
            std::fs::write(&path, contents).unwrap();
            path
        }
    }

    impl Drop for TempDir {
        fn drop(&mut self) {
            let _ = std::fs::remove_dir_all(&self.0);
        }
    }

    fn png(width: u32, height: u32) -> Vec<u8> {
        let mut bytes = std::io::Cursor::new(Vec::new());
        image::RgbImage::new(width, height)
            .write_to(&mut bytes, image::ImageFormat::Png)
            .unwrap();
        bytes.into_inner()
    }

    #[test]
    fn images_go_as_photos_unless_telegram_would_refuse_them() {
        let dir = TempDir::new("photos");
        let kind =
            |name: &str, contents: &[u8]| Attachment::new(&dir.file(name, contents)).unwrap().kind;
        assert_eq!(
            kind("cat.png", &png(40, 30)),
            Kind::Photo {
                width: 40,
                height: 30
            }
        );
        assert_eq!(
            kind("strip.png", &png(420, 20)),
            Kind::File,
            "too long and thin"
        );
        assert_eq!(kind("cat.gif", &png(40, 30)), Kind::File, "GIFs are files");
        assert_eq!(kind("notes.txt", b"hello"), Kind::File);
        assert_eq!(
            kind("fake.jpg", b"not a jpeg"),
            Kind::File,
            "unreadable images too"
        );
    }

    #[test]
    fn folders_empty_files_and_missing_ones_cant_be_attached() {
        let dir = TempDir::new("refused");
        assert_eq!(
            Attachment::new(&dir.0).unwrap_err(),
            format!(
                "{} is a folder",
                dir.0.file_name().unwrap().to_string_lossy()
            )
        );
        assert_eq!(
            Attachment::new(&dir.file("empty.txt", b"")).unwrap_err(),
            "empty.txt is empty"
        );
        assert_eq!(
            Attachment::new(&dir.0.join("gone.txt")).unwrap_err(),
            "No such file: gone.txt"
        );
    }

    #[test]
    fn file_names_are_shown_without_control_characters() {
        let dir = TempDir::new("names");
        let attachment = Attachment::new(&dir.file("a\u{1b}[2Jb.txt", b"x")).unwrap();
        assert_eq!(attachment.name, "a[2Jb.txt");
    }

    #[test]
    fn dropped_paths_are_read_however_the_terminal_quotes_them() {
        let dir = TempDir::new("dropped");
        let plain = dir.file("plain.txt", b"x");
        let spaced = dir.file("my photo.png", b"x");
        let (p, s) = (plain.display(), spaced.display());
        let escaped = s.to_string().replace(' ', "\\ ");
        let both = vec![plain.clone(), spaced.clone()];
        // macOS Terminal, iTerm2 and Ghostty escape spaces and add a space.
        assert_eq!(
            pasted_paths(&format!("{escaped} ")),
            Some(vec![spaced.clone()])
        );
        // GNOME Terminal quotes.
        assert_eq!(pasted_paths(&format!("'{p}' '{s}' ")), Some(both.clone()));
        assert_eq!(
            pasted_paths(&format!("\"{s}\"\n{p}")),
            Some(vec![spaced.clone(), plain.clone()])
        );
        // A path pasted as it is, spaces and all.
        assert_eq!(pasted_paths(&s.to_string()), Some(vec![spaced.clone()]));
        let url = format!("file://{}", s.to_string().replace(' ', "%20"));
        assert_eq!(pasted_paths(&url), Some(vec![spaced]));
    }

    #[test]
    fn a_paste_with_anything_but_existing_files_stays_text() {
        let dir = TempDir::new("text");
        let file = dir.file("notes.txt", b"x");
        let f = file.display();
        assert_eq!(pasted_paths(&format!("see {f}")), None);
        assert_eq!(pasted_paths(&format!("{f} {f}.missing")), None);
        assert_eq!(pasted_paths(&dir.0.display().to_string()), None, "a folder");
        assert_eq!(pasted_paths("notes.txt"), None, "relative paths are words");
        assert_eq!(pasted_paths(&format!("'{f}")), None, "an open quote");
        assert_eq!(pasted_paths("  \n"), None);
    }

    #[test]
    fn tab_completes_as_far_as_the_names_agree() {
        let dir = TempDir::new("complete");
        dir.file("report-2025.pdf", b"x");
        dir.file("report-2026.pdf", b"x");
        dir.file(".hidden", b"x");
        std::fs::create_dir(dir.0.join("photos")).unwrap();
        let d = format!("{}/", dir.0.display());
        assert_eq!(
            complete(&format!("{d}rep")),
            Completion {
                text: format!("{d}report-202"),
                matches: vec!["report-2025.pdf".into(), "report-2026.pdf".into()],
            }
        );
        assert_eq!(
            complete(&format!("{d}report-2026")).text,
            format!("{d}report-2026.pdf")
        );
        assert_eq!(
            complete(&format!("{d}ph")).text,
            format!("{d}photos/"),
            "a folder gets a /"
        );
        assert_eq!(complete(&format!("{d}.h")).text, format!("{d}.hidden"));
        let everything = complete(&d);
        assert!(
            !everything.matches.contains(&".hidden".to_string()),
            "until a . is typed"
        );
        assert_eq!(complete(&format!("{d}nothing")).text, format!("{d}nothing"));
    }

    #[test]
    fn photos_and_files_go_in_separate_albums_of_up_to_ten() {
        let file = |kind| Attachment {
            path: PathBuf::new(),
            name: String::new(),
            size: 1,
            kind,
        };
        let photo = Kind::Photo {
            width: 1,
            height: 1,
        };
        let mut attachments = vec![file(photo); 12];
        attachments.push(file(Kind::File));
        attachments.push(file(photo));
        let sizes = |as_files| -> Vec<usize> {
            albums(&attachments, as_files)
                .iter()
                .map(|a| a.len())
                .collect()
        };
        assert_eq!(sizes(false), [10, 2, 1, 1]);
        assert_eq!(sizes(true), [10, 4], "all files, sent as files");
    }

    #[test]
    fn home_is_expanded_only_at_the_start() {
        let home = dirs::home_dir().unwrap();
        assert_eq!(expand_home("~/a.txt"), home.join("a.txt"));
        assert_eq!(expand_home("~"), home);
        assert_eq!(expand_home("~bob/a.txt"), PathBuf::from("~bob/a.txt"));
        assert_eq!(expand_home("/tmp/~"), PathBuf::from("/tmp/~"));
    }

    #[test]
    fn sizes_read_like_telegram_shows_them() {
        assert_eq!(size_label(512), "512 B");
        assert_eq!(size_label(340 * 1024), "340 KB");
        assert_eq!(size_label(2_200_000), "2.1 MB");
        assert_eq!(size_label(3 * 1024 * 1024 * 1024), "3.0 GB");
    }
}