zola/src/cmd/serve.rs

use std::env;
use std::path::Path;
use std::sync::mpsc::channel;
use std::time::Duration;
use std::thread;

use iron::{Iron, Request, IronResult, Response, status};
use mount::Mount;
use staticfile::Static;
use notify::{Watcher, RecursiveMode, watcher};
use ws::{WebSocket};

use site::Site;
use errors::{Result};

const LIVE_RELOAD: &'static [u8; 37809] = include_bytes!("livereload.js");


fn livereload_handler(_: &mut Request) -> IronResult<Response> {
    Ok(Response::with((status::Ok, String::from_utf8(LIVE_RELOAD.to_vec()).unwrap())))
}


// Most of it taken from mdbook
pub fn serve(interface: &str, port: &str) -> Result<()> {
    let mut site = Site::new(true)?;
    site.build()?;

    let address = format!("{}:{}", interface, port);
    let ws_address = format!("{}:{}", interface, "1112");

    // Start a webserver that serves the `public` directory
    let mut mount = Mount::new();
    mount.mount("/", Static::new(Path::new("public/")));
    mount.mount("/livereload.js", livereload_handler);
    // Starts with a _ to not trigger the unused lint
    // we need to assign to a variable otherwise it will block
    let _iron = Iron::new(mount).http(address.clone()).unwrap();
    println!("Web server is available at http://{}", address);
    println!("Press CTRL+C to stop");

    // The websocket for livereload
    let ws_server = WebSocket::new(|_| {
        |_| {
            Ok(())
        }
    }).unwrap();
    let broadcaster = ws_server.broadcaster();
    thread::spawn(move || {
        ws_server.listen(&*ws_address).unwrap();
    });

    // And finally watching/reacting on file changes
    let (tx, rx) = channel();
    let mut watcher = watcher(tx, Duration::from_secs(2)).unwrap();
    watcher.watch("content/", RecursiveMode::Recursive).unwrap();
    watcher.watch("static/", RecursiveMode::Recursive).unwrap();
    watcher.watch("templates/", RecursiveMode::Recursive).unwrap();
    let pwd = env::current_dir().unwrap();
    println!("Listening for changes in {}/{{content, static, templates}}", pwd.display());

    use notify::DebouncedEvent::*;

    loop {
        // See https://github.com/spf13/hugo/blob/master/commands/hugo.go
        // for a more complete version of that
        match rx.recv() {
            Ok(event) => match event {
                NoticeWrite(path) |
                NoticeRemove(path) |
                Create(path) |
                Write(path) |
                Remove(path) |
                Rename(_, path) => {
                    if !is_temp_file(&path) {
                        println!("Change detected in {}", path.display());
                        match site.rebuild() {
                            Ok(_) => {
                                println!("Site rebuilt");
                                broadcaster.send(r#"
                                {
                                    "command": "reload",
                                    "path": "",
                                    "originalPath": "",
                                    "liveCSS": true,
                                    "liveImg": true,
                                    "protocol": ["http://livereload.com/protocols/official-7"]
                                }"#).unwrap();
                            },
                            Err(e) => {
                                println!("Failed to build the site");
                                println!("Error: {}", e);
                                for e in e.iter().skip(1) {
                                    println!("Reason: {}", e)
                                }
                            }
                        }
                    }
                }
                _ => {}
            },
            Err(e) => println!("Watch error: {:?}", e),
        };
    }
}


/// Returns whether the path we received corresponds to a temp file create
/// by an editor
fn is_temp_file(path: &Path) -> bool {
    let ext = path.extension();
    match ext {
        Some(ex) => match ex.to_str().unwrap() {
            "swp" | "swx" | "tmp" | ".DS_STORE" => true,
            // jetbrains IDE
            x if x.ends_with("jb_old___") => true,
            x if x.ends_with("jb_tmp___") => true,
            x if x.ends_with("jb_bak___") => true,
            // byword
            x if x.starts_with("sb-") => true,
            // gnome
            x if x.starts_with(".gooutputstream") => true,
            _ => {
                if let Some(filename) = path.file_stem() {
                    // emacs
                    filename.to_str().unwrap().starts_with("#")
                } else {
                    false
                }
            }
        },
        None => false,
    }
}