espanso/src/ui/macos.rs

use std::fs::create_dir_all;
use std::{fs, io};
use std::io::{BufReader, Cursor};
use zip::ZipArchive;
use log::{info, debug, error};
use std::path::PathBuf;
use std::process::Command;

const NOTIFY_HELPER_BINARY : &'static [u8] = include_bytes!("../res/mac/EspansoNotifyHelper.zip");
const DEFAULT_NOTIFICATION_DELAY : f64 = 1.5;

pub struct MacUIManager {
    notify_helper_path: PathBuf
}

impl super::UIManager for MacUIManager {
    fn notify(&self, message: &str) {
        let executable_path = self.notify_helper_path.join("Contents");
        let executable_path = executable_path.join("MacOS");
        let executable_path = executable_path.join("EspansoNotifyHelper");

        let res = Command::new(executable_path)
            .args(&["espanso", message, &DEFAULT_NOTIFICATION_DELAY.to_string()])
            .spawn();
    }
}

impl MacUIManager {
    pub fn new() -> MacUIManager {
        let notify_helper_path = MacUIManager::initialize_notify_helper();

        MacUIManager{
            notify_helper_path
        }
    }

    fn initialize_notify_helper() -> PathBuf {
        let data_dir = dirs::data_dir().expect("Can't obtain data_dir(), terminating.");

        let espanso_dir = data_dir.join("espanso");

        let res = create_dir_all(&espanso_dir);

        info!("Initializing EspansoNotifyHelper in {}", espanso_dir.as_path().display());

        let espanso_target = espanso_dir.join("EspansoNotifyHelper.app");

        if espanso_target.exists() {
            info!("EspansoNotifyHelper already initialized, skipping.");
        }else{
            if let Ok(_) = res {
                // Extract zip file
                let reader = Cursor::new(NOTIFY_HELPER_BINARY);

                let mut archive = zip::ZipArchive::new(reader).unwrap();

                for i in 0..archive.len() {
                    let mut file = archive.by_index(i).unwrap();
                    let outpath = espanso_dir.join(file.sanitized_name());

                    {
                        let comment = file.comment();
                        if !comment.is_empty() {
                            debug!("File {} comment: {}", i, comment);
                        }
                    }

                    if (&*file.name()).ends_with('/') {
                        debug!("File {} extracted to \"{}\"", i, outpath.as_path().display());
                        fs::create_dir_all(&outpath).unwrap();
                    } else {
                        debug!("File {} extracted to \"{}\" ({} bytes)", i, outpath.as_path().display(), file.size());
                        if let Some(p) = outpath.parent() {
                            if !p.exists() {
                                fs::create_dir_all(&p).unwrap();
                            }
                        }
                        let mut outfile = fs::File::create(&outpath).unwrap();
                        io::copy(&mut file, &mut outfile).unwrap();
                    }

                    use std::os::unix::fs::PermissionsExt;

                    if let Some(mode) = file.unix_mode() {
                        fs::set_permissions(&outpath, fs::Permissions::from_mode(mode)).unwrap();
                    }
                }
            }
        }

        espanso_target
    }
}