codesnap.nvim/generator/src/save.rs

use crate::{config::TakeSnapshotParams, path::parse_save_path, snapshot::take_snapshot};
use nvim_oxi::{lua::Error::RuntimeError, Error, Result};

pub fn save_snapshot(config: TakeSnapshotParams) -> Result<()> {
    match &config.save_path {
        Some(path) => {
            if !path.ends_with(".png") {
                return Err(Error::Lua(RuntimeError(
                    "The save_path must ends with .png".to_string(),
                )));
            }

            let pixmap = take_snapshot(config.clone())?;
            let path = parse_save_path(path.to_string())
                .map_err(|err| Error::Lua(RuntimeError(err.to_string())))?;

            pixmap
                .save_png(path)
                .map_err(|err| Error::Lua(RuntimeError(err.to_string())))
        }
        None => Err(Error::Lua(RuntimeError(
            "Cannot find 'save_path' in config".to_string(),
        ))),
    }
}