2020-01-26 00:30:35 +00:00
|
|
|
// Contains an embedded version of livereload-js 3.2.1
|
2017-06-07 09:25:36 +00:00
|
|
|
//
|
|
|
|
|
// Copyright (c) 2010-2012 Andrey Tarantsov
|
|
|
|
|
//
|
|
|
|
|
// Permission is hereby granted, free of charge, to any person obtaining
|
|
|
|
|
// a copy of this software and associated documentation files (the
|
|
|
|
|
// "Software"), to deal in the Software without restriction, including
|
|
|
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
|
|
|
|
// distribute, sublicense, and/or sell copies of the Software, and to
|
|
|
|
|
// permit persons to whom the Software is furnished to do so, subject to
|
|
|
|
|
// the following conditions:
|
|
|
|
|
//
|
|
|
|
|
// The above copyright notice and this permission notice shall be
|
|
|
|
|
// included in all copies or substantial portions of the Software.
|
|
|
|
|
//
|
|
|
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
|
|
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
|
|
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
|
|
|
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
|
|
|
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
|
|
|
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
|
|
2017-03-06 10:35:56 +00:00
|
|
|
use std::env;
|
2019-12-31 15:20:28 +00:00
|
|
|
use std::fs::{read_dir, remove_dir_all};
|
2018-10-02 09:31:18 +00:00
|
|
|
use std::path::{Path, PathBuf, MAIN_SEPARATOR};
|
2017-03-06 10:35:56 +00:00
|
|
|
use std::sync::mpsc::channel;
|
|
|
|
|
use std::thread;
|
2018-10-31 07:18:57 +00:00
|
|
|
use std::time::{Duration, Instant};
|
2017-03-03 08:12:40 +00:00
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
use hyper::header;
|
|
|
|
|
use hyper::service::{make_service_fn, service_fn};
|
|
|
|
|
use hyper::{Body, Method, Request, Response, Server, StatusCode};
|
|
|
|
|
use hyper_staticfile::ResolveResult;
|
|
|
|
|
use tokio::io::AsyncReadExt;
|
|
|
|
|
|
2018-10-31 07:18:57 +00:00
|
|
|
use chrono::prelude::*;
|
|
|
|
|
use notify::{watcher, RecursiveMode, Watcher};
|
|
|
|
|
use ws::{Message, Sender, WebSocket};
|
2018-01-22 17:11:25 +00:00
|
|
|
|
2019-02-09 18:54:46 +00:00
|
|
|
use errors::{Error as ZolaError, Result};
|
2019-12-21 21:52:39 +00:00
|
|
|
use globset::GlobSet;
|
2018-10-31 07:18:57 +00:00
|
|
|
use site::Site;
|
2018-03-14 21:03:06 +00:00
|
|
|
use utils::fs::copy_file;
|
2017-03-06 10:35:56 +00:00
|
|
|
|
2019-12-21 21:52:39 +00:00
|
|
|
use crate::console;
|
2017-03-08 04:21:45 +00:00
|
|
|
|
|
|
|
|
#[derive(Debug, PartialEq)]
|
|
|
|
|
enum ChangeKind {
|
|
|
|
|
Content,
|
|
|
|
|
Templates,
|
2019-08-24 17:05:02 +00:00
|
|
|
Themes,
|
2017-03-08 04:21:45 +00:00
|
|
|
StaticFiles,
|
2017-07-06 13:19:15 +00:00
|
|
|
Sass,
|
2018-01-12 10:50:29 +00:00
|
|
|
Config,
|
2017-03-08 04:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
static INTERNAL_SERVER_ERROR_TEXT: &[u8] = b"Internal Server Error";
|
|
|
|
|
static METHOD_NOT_ALLOWED_TEXT: &[u8] = b"Method Not Allowed";
|
|
|
|
|
static NOT_FOUND_TEXT: &[u8] = b"Not Found";
|
|
|
|
|
|
2020-07-24 19:37:57 +00:00
|
|
|
// This is dist/livereload.min.js from the LiveReload.js v3.2.4 release
|
2018-09-30 19:15:09 +00:00
|
|
|
const LIVE_RELOAD: &str = include_str!("livereload.js");
|
2017-03-06 10:35:56 +00:00
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
async fn handle_request(req: Request<Body>, root: PathBuf) -> Result<Response<Body>> {
|
|
|
|
|
// livereload.js is served using the LIVE_RELOAD str, not a file
|
|
|
|
|
if req.uri().path() == "/livereload.js" {
|
|
|
|
|
if req.method() == Method::GET {
|
|
|
|
|
return Ok(livereload_js());
|
|
|
|
|
} else {
|
|
|
|
|
return Ok(method_not_allowed());
|
|
|
|
|
}
|
|
|
|
|
}
|
2018-04-26 21:14:37 +00:00
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
let result = hyper_staticfile::resolve(&root, &req).await.unwrap();
|
|
|
|
|
match result {
|
|
|
|
|
ResolveResult::MethodNotMatched => return Ok(method_not_allowed()),
|
|
|
|
|
ResolveResult::NotFound | ResolveResult::UriNotMatched => {
|
|
|
|
|
return Ok(not_found(Path::new(&root.join("404.html"))).await)
|
|
|
|
|
}
|
|
|
|
|
_ => (),
|
2019-06-18 23:05:00 +00:00
|
|
|
};
|
|
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
Ok(hyper_staticfile::ResponseBuilder::new().request(&req).build(result).unwrap())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn livereload_js() -> Response<Body> {
|
|
|
|
|
Response::builder()
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/javascript")
|
|
|
|
|
.status(StatusCode::OK)
|
|
|
|
|
.body(LIVE_RELOAD.into())
|
|
|
|
|
.expect("Could not build livereload.js response")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn internal_server_error() -> Response<Body> {
|
|
|
|
|
Response::builder()
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/plain")
|
|
|
|
|
.status(StatusCode::INTERNAL_SERVER_ERROR)
|
|
|
|
|
.body(INTERNAL_SERVER_ERROR_TEXT.into())
|
|
|
|
|
.expect("Could not build Internal Server Error response")
|
|
|
|
|
}
|
2019-06-18 23:05:00 +00:00
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
fn method_not_allowed() -> Response<Body> {
|
|
|
|
|
Response::builder()
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/plain")
|
|
|
|
|
.status(StatusCode::METHOD_NOT_ALLOWED)
|
|
|
|
|
.body(METHOD_NOT_ALLOWED_TEXT.into())
|
|
|
|
|
.expect("Could not build Method Not Allowed response")
|
2018-04-26 21:14:37 +00:00
|
|
|
}
|
2017-03-06 10:35:56 +00:00
|
|
|
|
2019-12-31 15:20:28 +00:00
|
|
|
async fn not_found(page_path: &Path) -> Response<Body> {
|
|
|
|
|
if let Ok(mut file) = tokio::fs::File::open(page_path).await {
|
|
|
|
|
let mut buf = Vec::new();
|
|
|
|
|
if file.read_to_end(&mut buf).await.is_ok() {
|
|
|
|
|
return Response::builder()
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/html")
|
|
|
|
|
.status(StatusCode::NOT_FOUND)
|
|
|
|
|
.body(buf.into())
|
|
|
|
|
.expect("Could not build Not Found response");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return internal_server_error();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Use a plain text response when page_path isn't available
|
|
|
|
|
Response::builder()
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/plain")
|
|
|
|
|
.status(StatusCode::NOT_FOUND)
|
|
|
|
|
.body(NOT_FOUND_TEXT.into())
|
|
|
|
|
.expect("Could not build Not Found response")
|
2017-03-06 10:35:56 +00:00
|
|
|
}
|
|
|
|
|
|
2019-01-25 00:47:30 +00:00
|
|
|
fn rebuild_done_handling(broadcaster: &Option<Sender>, res: Result<()>, reload_path: &str) {
|
2017-03-10 11:39:58 +00:00
|
|
|
match res {
|
|
|
|
|
Ok(_) => {
|
2019-01-25 00:47:30 +00:00
|
|
|
if let Some(broadcaster) = broadcaster.as_ref() {
|
|
|
|
|
broadcaster
|
|
|
|
|
.send(format!(
|
|
|
|
|
r#"
|
|
|
|
|
{{
|
|
|
|
|
"command": "reload",
|
|
|
|
|
"path": "{}",
|
|
|
|
|
"originalPath": "",
|
|
|
|
|
"liveCSS": true,
|
|
|
|
|
"liveImg": true,
|
|
|
|
|
"protocol": ["http://livereload.com/protocols/official-7"]
|
|
|
|
|
}}"#,
|
|
|
|
|
reload_path
|
|
|
|
|
))
|
|
|
|
|
.unwrap();
|
|
|
|
|
}
|
2018-10-31 07:18:57 +00:00
|
|
|
}
|
|
|
|
|
Err(e) => console::unravel_errors("Failed to build the site", &e),
|
2017-03-10 11:39:58 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-20 10:50:34 +00:00
|
|
|
fn create_new_site(
|
2020-01-21 19:52:24 +00:00
|
|
|
root_dir: &Path,
|
2018-10-31 07:18:57 +00:00
|
|
|
interface: &str,
|
|
|
|
|
port: u16,
|
2020-05-23 09:55:45 +00:00
|
|
|
output_dir: &Path,
|
2018-10-31 07:18:57 +00:00
|
|
|
base_url: &str,
|
2020-05-23 09:55:45 +00:00
|
|
|
config_file: &Path,
|
2019-08-24 20:23:08 +00:00
|
|
|
include_drafts: bool,
|
2018-10-31 07:18:57 +00:00
|
|
|
) -> Result<(Site, String)> {
|
2020-01-21 19:52:24 +00:00
|
|
|
let mut site = Site::new(root_dir, config_file)?;
|
2017-03-25 07:12:58 +00:00
|
|
|
|
2018-02-02 16:18:07 +00:00
|
|
|
let base_address = format!("{}:{}", base_url, port);
|
2017-03-20 10:00:00 +00:00
|
|
|
let address = format!("{}:{}", interface, port);
|
2018-02-02 20:35:04 +00:00
|
|
|
let base_url = if site.config.base_url.ends_with('/') {
|
2018-02-02 16:18:07 +00:00
|
|
|
format!("http://{}/", base_address)
|
2017-03-20 10:00:00 +00:00
|
|
|
} else {
|
2018-02-02 16:18:07 +00:00
|
|
|
format!("http://{}", base_address)
|
2017-03-20 10:00:00 +00:00
|
|
|
};
|
2018-02-02 16:18:07 +00:00
|
|
|
|
2019-07-12 21:47:15 +00:00
|
|
|
site.config.enable_serve_mode();
|
2018-02-02 20:35:04 +00:00
|
|
|
site.set_base_url(base_url);
|
2017-12-29 18:25:06 +00:00
|
|
|
site.set_output_path(output_dir);
|
2019-08-24 20:23:08 +00:00
|
|
|
if include_drafts {
|
|
|
|
|
site.include_drafts();
|
|
|
|
|
}
|
2017-03-21 07:57:00 +00:00
|
|
|
site.load()?;
|
2018-10-19 14:33:11 +00:00
|
|
|
site.enable_live_reload(port);
|
2017-05-12 14:10:21 +00:00
|
|
|
console::notify_site_size(&site);
|
|
|
|
|
console::warn_about_ignored_pages(&site);
|
2017-03-06 10:35:56 +00:00
|
|
|
site.build()?;
|
2018-01-12 10:50:29 +00:00
|
|
|
Ok((site, address))
|
|
|
|
|
}
|
|
|
|
|
|
2018-10-31 07:18:57 +00:00
|
|
|
pub fn serve(
|
2020-01-21 19:52:24 +00:00
|
|
|
root_dir: &Path,
|
2018-10-31 07:18:57 +00:00
|
|
|
interface: &str,
|
|
|
|
|
port: u16,
|
2020-05-23 09:55:45 +00:00
|
|
|
output_dir: &Path,
|
2018-10-31 07:18:57 +00:00
|
|
|
base_url: &str,
|
2020-05-23 09:55:45 +00:00
|
|
|
config_file: &Path,
|
2018-11-01 22:20:35 +00:00
|
|
|
watch_only: bool,
|
2019-07-04 21:42:37 +00:00
|
|
|
open: bool,
|
2019-08-24 20:23:08 +00:00
|
|
|
include_drafts: bool,
|
2018-10-31 07:18:57 +00:00
|
|
|
) -> Result<()> {
|
2018-01-12 10:50:29 +00:00
|
|
|
let start = Instant::now();
|
2020-02-10 19:48:52 +00:00
|
|
|
let (mut site, address) = create_new_site(
|
|
|
|
|
root_dir,
|
|
|
|
|
interface,
|
|
|
|
|
port,
|
|
|
|
|
output_dir,
|
|
|
|
|
base_url,
|
|
|
|
|
config_file,
|
|
|
|
|
include_drafts,
|
|
|
|
|
)?;
|
2017-05-12 14:10:21 +00:00
|
|
|
console::report_elapsed_time(start);
|
2017-03-06 10:35:56 +00:00
|
|
|
|
2017-05-01 09:11:18 +00:00
|
|
|
// Setup watchers
|
2018-01-22 17:11:25 +00:00
|
|
|
let mut watching_static = false;
|
2018-10-11 18:52:41 +00:00
|
|
|
let mut watching_templates = false;
|
2019-08-24 17:05:02 +00:00
|
|
|
let mut watching_themes = false;
|
2017-05-01 09:11:18 +00:00
|
|
|
let (tx, rx) = channel();
|
2019-01-05 10:02:46 +00:00
|
|
|
let mut watcher = watcher(tx, Duration::from_secs(1)).unwrap();
|
2018-10-31 07:18:57 +00:00
|
|
|
watcher
|
2019-04-20 10:50:34 +00:00
|
|
|
.watch("content/", RecursiveMode::Recursive)
|
2019-01-11 19:29:46 +00:00
|
|
|
.map_err(|e| ZolaError::chain("Can't watch the `content` folder. Does it exist?", e))?;
|
2018-10-31 07:18:57 +00:00
|
|
|
watcher
|
2019-04-20 10:50:34 +00:00
|
|
|
.watch(config_file, RecursiveMode::Recursive)
|
2019-01-11 19:29:46 +00:00
|
|
|
.map_err(|e| ZolaError::chain("Can't watch the `config` file. Does it exist?", e))?;
|
2017-05-01 09:11:18 +00:00
|
|
|
|
2019-04-20 10:50:34 +00:00
|
|
|
if Path::new("static").exists() {
|
2017-10-25 12:49:54 +00:00
|
|
|
watching_static = true;
|
2018-10-31 07:18:57 +00:00
|
|
|
watcher
|
2019-04-20 10:50:34 +00:00
|
|
|
.watch("static/", RecursiveMode::Recursive)
|
2019-01-11 19:29:46 +00:00
|
|
|
.map_err(|e| ZolaError::chain("Can't watch the `static` folder.", e))?;
|
2018-10-11 18:52:41 +00:00
|
|
|
}
|
|
|
|
|
|
2019-04-20 10:50:34 +00:00
|
|
|
if Path::new("templates").exists() {
|
2018-10-11 18:52:41 +00:00
|
|
|
watching_templates = true;
|
2018-10-31 07:18:57 +00:00
|
|
|
watcher
|
2019-04-20 10:50:34 +00:00
|
|
|
.watch("templates/", RecursiveMode::Recursive)
|
2019-01-11 19:29:46 +00:00
|
|
|
.map_err(|e| ZolaError::chain("Can't watch the `templates` folder.", e))?;
|
2017-10-25 12:49:54 +00:00
|
|
|
}
|
|
|
|
|
|
2019-08-24 17:05:02 +00:00
|
|
|
if Path::new("themes").exists() {
|
|
|
|
|
watching_themes = true;
|
|
|
|
|
watcher
|
|
|
|
|
.watch("themes/", RecursiveMode::Recursive)
|
|
|
|
|
.map_err(|e| ZolaError::chain("Can't watch the `themes` folder.", e))?;
|
|
|
|
|
}
|
|
|
|
|
|
2017-07-06 13:19:15 +00:00
|
|
|
// Sass support is optional so don't make it an error to no have a sass folder
|
2019-04-20 10:50:34 +00:00
|
|
|
let _ = watcher.watch("sass/", RecursiveMode::Recursive);
|
2017-07-06 13:19:15 +00:00
|
|
|
|
2018-05-11 11:54:16 +00:00
|
|
|
let ws_address = format!("{}:{}", interface, site.live_reload.unwrap());
|
2018-05-27 05:19:01 +00:00
|
|
|
let output_path = Path::new(output_dir).to_path_buf();
|
2017-03-06 10:35:56 +00:00
|
|
|
|
2018-05-27 05:19:01 +00:00
|
|
|
// output path is going to need to be moved later on, so clone it for the
|
|
|
|
|
// http closure to avoid contention.
|
|
|
|
|
let static_root = output_path.clone();
|
2018-11-01 22:20:35 +00:00
|
|
|
let broadcaster = if !watch_only {
|
|
|
|
|
thread::spawn(move || {
|
2019-12-31 15:20:28 +00:00
|
|
|
let addr = address.parse().unwrap();
|
|
|
|
|
|
|
|
|
|
let mut rt = tokio::runtime::Builder::new()
|
|
|
|
|
.enable_all()
|
|
|
|
|
.basic_scheduler()
|
|
|
|
|
.build()
|
|
|
|
|
.expect("Could not build tokio runtime");
|
|
|
|
|
|
|
|
|
|
rt.block_on(async {
|
|
|
|
|
let make_service = make_service_fn(move |_| {
|
|
|
|
|
let static_root = static_root.clone();
|
|
|
|
|
|
|
|
|
|
async {
|
|
|
|
|
Ok::<_, hyper::Error>(service_fn(move |req| {
|
|
|
|
|
handle_request(req, static_root.clone())
|
|
|
|
|
}))
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let server = Server::bind(&addr).serve(make_service);
|
|
|
|
|
|
|
|
|
|
println!("Web server is available at http://{}\n", &address);
|
|
|
|
|
if open {
|
|
|
|
|
if let Err(err) = open::that(format!("http://{}", &address)) {
|
|
|
|
|
eprintln!("Failed to open URL in your browser: {}", err);
|
|
|
|
|
}
|
2019-07-04 21:42:37 +00:00
|
|
|
}
|
2019-12-31 15:20:28 +00:00
|
|
|
|
|
|
|
|
server.await.expect("Could not start web server");
|
|
|
|
|
});
|
2018-11-01 22:20:35 +00:00
|
|
|
});
|
2019-12-31 15:20:28 +00:00
|
|
|
|
2018-11-01 22:20:35 +00:00
|
|
|
// The websocket for livereload
|
|
|
|
|
let ws_server = WebSocket::new(|output: Sender| {
|
|
|
|
|
move |msg: Message| {
|
|
|
|
|
if msg.into_text().unwrap().contains("\"hello\"") {
|
|
|
|
|
return output.send(Message::text(
|
|
|
|
|
r#"
|
|
|
|
|
{
|
|
|
|
|
"command": "hello",
|
|
|
|
|
"protocols": [ "http://livereload.com/protocols/official-7" ],
|
|
|
|
|
"serverName": "Zola"
|
|
|
|
|
}
|
|
|
|
|
"#,
|
|
|
|
|
));
|
|
|
|
|
}
|
|
|
|
|
Ok(())
|
2017-05-12 12:15:50 +00:00
|
|
|
}
|
2018-11-01 22:20:35 +00:00
|
|
|
})
|
|
|
|
|
.unwrap();
|
|
|
|
|
let broadcaster = ws_server.broadcaster();
|
|
|
|
|
thread::spawn(move || {
|
|
|
|
|
ws_server.listen(&*ws_address).unwrap();
|
|
|
|
|
});
|
|
|
|
|
Some(broadcaster)
|
|
|
|
|
} else {
|
|
|
|
|
println!("Watching in watch only mode, no web server will be started");
|
|
|
|
|
None
|
|
|
|
|
};
|
|
|
|
|
|
2019-04-20 10:50:34 +00:00
|
|
|
let pwd = env::current_dir().unwrap();
|
|
|
|
|
|
2018-10-11 18:52:41 +00:00
|
|
|
let mut watchers = vec!["content", "config.toml"];
|
2017-10-25 12:49:54 +00:00
|
|
|
if watching_static {
|
|
|
|
|
watchers.push("static");
|
|
|
|
|
}
|
2018-10-11 18:52:41 +00:00
|
|
|
if watching_templates {
|
|
|
|
|
watchers.push("templates");
|
|
|
|
|
}
|
2019-08-24 17:05:02 +00:00
|
|
|
if watching_themes {
|
|
|
|
|
watchers.push("themes");
|
|
|
|
|
}
|
2018-03-12 19:11:03 +00:00
|
|
|
if site.config.compile_sass {
|
2017-10-25 12:49:54 +00:00
|
|
|
watchers.push("sass");
|
2017-07-06 13:19:15 +00:00
|
|
|
}
|
2017-10-25 12:49:54 +00:00
|
|
|
|
2018-10-31 07:18:57 +00:00
|
|
|
println!(
|
|
|
|
|
"Listening for changes in {}{}{{{}}}",
|
2019-04-20 10:50:34 +00:00
|
|
|
pwd.display(),
|
2018-10-31 07:18:57 +00:00
|
|
|
MAIN_SEPARATOR,
|
|
|
|
|
watchers.join(", ")
|
|
|
|
|
);
|
2018-05-27 05:19:01 +00:00
|
|
|
|
2017-03-25 06:52:51 +00:00
|
|
|
println!("Press Ctrl+C to stop\n");
|
2018-01-22 17:11:25 +00:00
|
|
|
// Delete the output folder on ctrl+C
|
|
|
|
|
ctrlc::set_handler(move || {
|
2019-12-23 08:43:08 +00:00
|
|
|
match remove_dir_all(&output_path) {
|
|
|
|
|
Ok(()) => (),
|
|
|
|
|
Err(e) => println!("Errored while deleting output folder: {}", e),
|
|
|
|
|
}
|
2018-01-22 17:11:25 +00:00
|
|
|
::std::process::exit(0);
|
2018-10-31 07:18:57 +00:00
|
|
|
})
|
|
|
|
|
.expect("Error setting Ctrl-C handler");
|
2017-03-06 10:35:56 +00:00
|
|
|
|
|
|
|
|
use notify::DebouncedEvent::*;
|
|
|
|
|
|
2018-11-10 21:23:37 +00:00
|
|
|
let reload_templates = |site: &mut Site, path: &Path| {
|
|
|
|
|
let msg = if path.is_dir() {
|
|
|
|
|
format!("-> Directory in `templates` folder changed {}", path.display())
|
|
|
|
|
} else {
|
|
|
|
|
format!("-> Template changed {}", path.display())
|
|
|
|
|
};
|
|
|
|
|
console::info(&msg);
|
2019-01-25 00:47:30 +00:00
|
|
|
// Force refresh
|
2019-02-09 18:54:46 +00:00
|
|
|
rebuild_done_handling(&broadcaster, rebuild::after_template_change(site, &path), "/x.js");
|
2018-11-10 21:23:37 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let reload_sass = |site: &Site, path: &Path, partial_path: &Path| {
|
|
|
|
|
let msg = if path.is_dir() {
|
|
|
|
|
format!("-> Directory in `sass` folder changed {}", path.display())
|
|
|
|
|
} else {
|
|
|
|
|
format!("-> Sass file changed {}", path.display())
|
|
|
|
|
};
|
|
|
|
|
console::info(&msg);
|
2019-01-25 00:47:30 +00:00
|
|
|
rebuild_done_handling(
|
|
|
|
|
&broadcaster,
|
|
|
|
|
site.compile_sass(&site.base_path),
|
|
|
|
|
&partial_path.to_string_lossy(),
|
|
|
|
|
);
|
2018-11-10 21:23:37 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let copy_static = |site: &Site, path: &Path, partial_path: &Path| {
|
|
|
|
|
// Do nothing if the file/dir was deleted
|
|
|
|
|
if !path.exists() {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let msg = if path.is_dir() {
|
|
|
|
|
format!("-> Directory in `static` folder changed {}", path.display())
|
|
|
|
|
} else {
|
|
|
|
|
format!("-> Static file changed {}", path.display())
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
console::info(&msg);
|
2019-01-25 00:47:30 +00:00
|
|
|
if path.is_dir() {
|
|
|
|
|
rebuild_done_handling(
|
|
|
|
|
&broadcaster,
|
|
|
|
|
site.copy_static_directories(),
|
|
|
|
|
&path.to_string_lossy(),
|
|
|
|
|
);
|
|
|
|
|
} else {
|
|
|
|
|
rebuild_done_handling(
|
|
|
|
|
&broadcaster,
|
2019-07-19 09:10:28 +00:00
|
|
|
copy_file(
|
|
|
|
|
&path,
|
|
|
|
|
&site.output_path,
|
|
|
|
|
&site.static_path,
|
|
|
|
|
site.config.hard_link_static,
|
|
|
|
|
),
|
2019-01-25 00:47:30 +00:00
|
|
|
&partial_path.to_string_lossy(),
|
|
|
|
|
);
|
2018-11-10 21:23:37 +00:00
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
2020-02-05 08:13:14 +00:00
|
|
|
let recreate_site = || match create_new_site(
|
|
|
|
|
root_dir,
|
|
|
|
|
interface,
|
|
|
|
|
port,
|
|
|
|
|
output_dir,
|
|
|
|
|
base_url,
|
|
|
|
|
config_file,
|
|
|
|
|
include_drafts,
|
|
|
|
|
) {
|
|
|
|
|
Ok((s, _)) => {
|
|
|
|
|
rebuild_done_handling(&broadcaster, Ok(()), "/x.js");
|
|
|
|
|
Some(s)
|
|
|
|
|
}
|
|
|
|
|
Err(e) => {
|
|
|
|
|
console::error(&format!("{}", e));
|
|
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
2017-03-06 10:35:56 +00:00
|
|
|
loop {
|
|
|
|
|
match rx.recv() {
|
2017-03-08 04:21:45 +00:00
|
|
|
Ok(event) => {
|
|
|
|
|
match event {
|
2018-11-10 21:23:37 +00:00
|
|
|
Rename(old_path, path) => {
|
|
|
|
|
if path.is_file() && is_temp_file(&path) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2019-04-20 10:50:34 +00:00
|
|
|
let (change_kind, partial_path) = detect_change_kind(&pwd, &path);
|
2018-11-10 21:23:37 +00:00
|
|
|
|
|
|
|
|
// We only care about changes in non-empty folders
|
|
|
|
|
if path.is_dir() && is_folder_empty(&path) {
|
2017-03-08 04:21:45 +00:00
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2018-10-31 07:18:57 +00:00
|
|
|
println!(
|
|
|
|
|
"Change detected @ {}",
|
|
|
|
|
Local::now().format("%Y-%m-%d %H:%M:%S").to_string()
|
|
|
|
|
);
|
2018-11-10 21:23:37 +00:00
|
|
|
|
2017-03-10 11:39:58 +00:00
|
|
|
let start = Instant::now();
|
2019-08-01 08:18:42 +00:00
|
|
|
|
2018-11-10 21:23:37 +00:00
|
|
|
match change_kind {
|
|
|
|
|
ChangeKind::Content => {
|
|
|
|
|
console::info(&format!("-> Content renamed {}", path.display()));
|
2019-01-25 00:47:30 +00:00
|
|
|
// Force refresh
|
|
|
|
|
rebuild_done_handling(
|
|
|
|
|
&broadcaster,
|
|
|
|
|
rebuild::after_content_rename(&mut site, &old_path, &path),
|
|
|
|
|
"/x.js",
|
|
|
|
|
);
|
2018-10-31 07:18:57 +00:00
|
|
|
}
|
2018-11-10 21:23:37 +00:00
|
|
|
ChangeKind::Templates => reload_templates(&mut site, &path),
|
|
|
|
|
ChangeKind::StaticFiles => copy_static(&site, &path, &partial_path),
|
|
|
|
|
ChangeKind::Sass => reload_sass(&site, &path, &partial_path),
|
2019-08-24 17:05:02 +00:00
|
|
|
ChangeKind::Themes => {
|
2019-08-24 20:23:08 +00:00
|
|
|
console::info(
|
|
|
|
|
"-> Themes changed. The whole site will be reloaded.",
|
|
|
|
|
);
|
2020-02-05 08:13:14 +00:00
|
|
|
|
|
|
|
|
if let Some(s) = recreate_site() {
|
|
|
|
|
site = s;
|
|
|
|
|
}
|
2019-08-24 20:23:08 +00:00
|
|
|
}
|
2018-11-10 21:23:37 +00:00
|
|
|
ChangeKind::Config => {
|
|
|
|
|
console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible.");
|
2020-02-05 08:13:14 +00:00
|
|
|
|
|
|
|
|
if let Some(s) = recreate_site() {
|
|
|
|
|
site = s;
|
|
|
|
|
}
|
2018-11-10 21:23:37 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
console::report_elapsed_time(start);
|
|
|
|
|
}
|
2019-05-27 17:51:33 +00:00
|
|
|
// Intellij does weird things on edit, chmod is there to count those changes
|
|
|
|
|
// https://github.com/passcod/notify/issues/150#issuecomment-494912080
|
|
|
|
|
Create(path) | Write(path) | Remove(path) | Chmod(path) => {
|
2019-08-01 08:18:42 +00:00
|
|
|
if is_ignored_file(&site.config.ignored_content_globset, &path) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2018-11-10 21:23:37 +00:00
|
|
|
if is_temp_file(&path) || path.is_dir() {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
println!(
|
|
|
|
|
"Change detected @ {}",
|
|
|
|
|
Local::now().format("%Y-%m-%d %H:%M:%S").to_string()
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
let start = Instant::now();
|
2019-04-20 10:50:34 +00:00
|
|
|
match detect_change_kind(&pwd, &path) {
|
2018-11-10 21:23:37 +00:00
|
|
|
(ChangeKind::Content, _) => {
|
|
|
|
|
console::info(&format!("-> Content changed {}", path.display()));
|
2019-01-25 00:47:30 +00:00
|
|
|
// Force refresh
|
|
|
|
|
rebuild_done_handling(
|
|
|
|
|
&broadcaster,
|
|
|
|
|
rebuild::after_content_change(&mut site, &path),
|
|
|
|
|
"/x.js",
|
|
|
|
|
);
|
2018-10-31 07:18:57 +00:00
|
|
|
}
|
2018-11-10 21:23:37 +00:00
|
|
|
(ChangeKind::Templates, _) => reload_templates(&mut site, &path),
|
|
|
|
|
(ChangeKind::StaticFiles, p) => copy_static(&site, &path, &p),
|
|
|
|
|
(ChangeKind::Sass, p) => reload_sass(&site, &path, &p),
|
2019-08-24 17:05:02 +00:00
|
|
|
(ChangeKind::Themes, _) => {
|
2019-08-24 20:23:08 +00:00
|
|
|
console::info(
|
|
|
|
|
"-> Themes changed. The whole site will be reloaded.",
|
|
|
|
|
);
|
2020-02-05 08:13:14 +00:00
|
|
|
|
|
|
|
|
if let Some(s) = recreate_site() {
|
|
|
|
|
site = s;
|
|
|
|
|
}
|
2019-08-24 20:23:08 +00:00
|
|
|
}
|
2018-01-12 10:50:29 +00:00
|
|
|
(ChangeKind::Config, _) => {
|
2018-09-30 19:15:09 +00:00
|
|
|
console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible.");
|
2020-02-05 08:13:14 +00:00
|
|
|
|
|
|
|
|
if let Some(s) = recreate_site() {
|
|
|
|
|
site = s;
|
|
|
|
|
}
|
2018-01-12 10:50:29 +00:00
|
|
|
}
|
2017-03-08 04:21:45 +00:00
|
|
|
};
|
2017-05-12 14:10:21 +00:00
|
|
|
console::report_elapsed_time(start);
|
2017-03-06 10:35:56 +00:00
|
|
|
}
|
2017-03-08 04:21:45 +00:00
|
|
|
_ => {}
|
2017-03-06 10:35:56 +00:00
|
|
|
}
|
2018-10-31 07:18:57 +00:00
|
|
|
}
|
2017-03-25 06:52:51 +00:00
|
|
|
Err(e) => console::error(&format!("Watch error: {:?}", e)),
|
2017-03-06 10:35:56 +00:00
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2019-08-01 08:18:42 +00:00
|
|
|
fn is_ignored_file(ignored_content_globset: &Option<GlobSet>, path: &Path) -> bool {
|
|
|
|
|
match ignored_content_globset {
|
|
|
|
|
Some(gs) => gs.is_match(path),
|
2019-08-10 16:53:16 +00:00
|
|
|
None => false,
|
2019-08-01 08:18:42 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2017-03-25 06:52:51 +00:00
|
|
|
/// Returns whether the path we received corresponds to a temp file created
|
|
|
|
|
/// by an editor or the OS
|
2017-03-06 10:35:56 +00:00
|
|
|
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,
|
2017-03-08 04:21:45 +00:00
|
|
|
// vim
|
2017-03-10 13:19:36 +00:00
|
|
|
x if x.ends_with('~') => true,
|
2017-03-06 10:35:56 +00:00
|
|
|
_ => {
|
|
|
|
|
if let Some(filename) = path.file_stem() {
|
|
|
|
|
// emacs
|
2018-09-13 14:57:38 +00:00
|
|
|
let name = filename.to_str().unwrap();
|
|
|
|
|
name.starts_with('#') || name.starts_with(".#")
|
2017-03-06 10:35:56 +00:00
|
|
|
} else {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
},
|
2018-10-31 07:18:57 +00:00
|
|
|
None => true,
|
2017-03-08 04:21:45 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Detect what changed from the given path so we have an idea what needs
|
|
|
|
|
/// to be reloaded
|
2018-08-04 19:47:45 +00:00
|
|
|
fn detect_change_kind(pwd: &Path, path: &Path) -> (ChangeKind, PathBuf) {
|
|
|
|
|
let mut partial_path = PathBuf::from("/");
|
2018-08-05 05:59:56 +00:00
|
|
|
partial_path.push(path.strip_prefix(pwd).unwrap_or(path));
|
2018-06-20 15:43:24 +00:00
|
|
|
|
2018-08-04 19:47:45 +00:00
|
|
|
let change_kind = if partial_path.starts_with("/templates") {
|
2017-03-08 04:21:45 +00:00
|
|
|
ChangeKind::Templates
|
2019-08-24 17:05:02 +00:00
|
|
|
} else if partial_path.starts_with("/themes") {
|
|
|
|
|
ChangeKind::Themes
|
2018-08-04 19:47:45 +00:00
|
|
|
} else if partial_path.starts_with("/content") {
|
2017-03-08 04:21:45 +00:00
|
|
|
ChangeKind::Content
|
2018-08-04 19:47:45 +00:00
|
|
|
} else if partial_path.starts_with("/static") {
|
2017-03-08 04:21:45 +00:00
|
|
|
ChangeKind::StaticFiles
|
2018-08-04 19:47:45 +00:00
|
|
|
} else if partial_path.starts_with("/sass") {
|
2017-07-06 13:19:15 +00:00
|
|
|
ChangeKind::Sass
|
2018-08-04 19:47:45 +00:00
|
|
|
} else if partial_path == Path::new("/config.toml") {
|
2018-01-12 10:50:29 +00:00
|
|
|
ChangeKind::Config
|
2017-03-08 04:21:45 +00:00
|
|
|
} else {
|
2018-08-04 19:47:45 +00:00
|
|
|
unreachable!("Got a change in an unexpected path: {}", partial_path.display());
|
2017-03-08 04:21:45 +00:00
|
|
|
};
|
|
|
|
|
|
2018-08-04 19:47:45 +00:00
|
|
|
(change_kind, partial_path)
|
2017-03-08 04:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
2018-11-10 21:23:37 +00:00
|
|
|
/// Check if the directory at path contains any file
|
|
|
|
|
fn is_folder_empty(dir: &Path) -> bool {
|
|
|
|
|
// Can panic if we don't have the rights I guess?
|
2018-12-29 10:17:43 +00:00
|
|
|
let files: Vec<_> =
|
|
|
|
|
read_dir(dir).expect("Failed to read a directory to see if it was empty").collect();
|
|
|
|
|
files.is_empty()
|
2018-11-10 21:23:37 +00:00
|
|
|
}
|
|
|
|
|
|
2017-03-08 04:21:45 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
|
mod tests {
|
2018-08-04 19:47:45 +00:00
|
|
|
use std::path::{Path, PathBuf};
|
2017-03-08 04:21:45 +00:00
|
|
|
|
2018-10-31 07:18:57 +00:00
|
|
|
use super::{detect_change_kind, is_temp_file, ChangeKind};
|
2017-03-08 04:21:45 +00:00
|
|
|
|
|
|
|
|
#[test]
|
2017-05-20 15:00:41 +00:00
|
|
|
fn can_recognize_temp_files() {
|
|
|
|
|
let test_cases = vec![
|
2017-03-08 04:21:45 +00:00
|
|
|
Path::new("hello.swp"),
|
|
|
|
|
Path::new("hello.swx"),
|
|
|
|
|
Path::new(".DS_STORE"),
|
|
|
|
|
Path::new("hello.tmp"),
|
|
|
|
|
Path::new("hello.html.__jb_old___"),
|
|
|
|
|
Path::new("hello.html.__jb_tmp___"),
|
|
|
|
|
Path::new("hello.html.__jb_bak___"),
|
|
|
|
|
Path::new("hello.html~"),
|
|
|
|
|
Path::new("#hello.html"),
|
|
|
|
|
];
|
|
|
|
|
|
2017-05-20 15:00:41 +00:00
|
|
|
for t in test_cases {
|
2017-03-08 04:21:45 +00:00
|
|
|
assert!(is_temp_file(&t));
|
|
|
|
|
}
|
2017-03-06 10:35:56 +00:00
|
|
|
}
|
2017-03-08 04:21:45 +00:00
|
|
|
|
|
|
|
|
#[test]
|
2017-05-20 15:00:41 +00:00
|
|
|
fn can_detect_kind_of_changes() {
|
|
|
|
|
let test_cases = vec![
|
2017-03-09 07:34:12 +00:00
|
|
|
(
|
2018-08-04 19:47:45 +00:00
|
|
|
(ChangeKind::Templates, PathBuf::from("/templates/hello.html")),
|
2018-10-31 07:18:57 +00:00
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/templates/hello.html"),
|
2017-03-09 07:34:12 +00:00
|
|
|
),
|
2019-08-24 17:05:02 +00:00
|
|
|
(
|
|
|
|
|
(ChangeKind::Themes, PathBuf::from("/themes/hello.html")),
|
|
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/themes/hello.html"),
|
|
|
|
|
),
|
2017-03-09 07:34:12 +00:00
|
|
|
(
|
2018-08-04 19:47:45 +00:00
|
|
|
(ChangeKind::StaticFiles, PathBuf::from("/static/site.css")),
|
2018-10-31 07:18:57 +00:00
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/static/site.css"),
|
2017-03-09 07:34:12 +00:00
|
|
|
),
|
|
|
|
|
(
|
2018-08-04 19:47:45 +00:00
|
|
|
(ChangeKind::Content, PathBuf::from("/content/posts/hello.md")),
|
2018-10-31 07:18:57 +00:00
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/content/posts/hello.md"),
|
2017-03-09 07:34:12 +00:00
|
|
|
),
|
2017-11-20 23:05:37 +00:00
|
|
|
(
|
2018-08-04 19:47:45 +00:00
|
|
|
(ChangeKind::Sass, PathBuf::from("/sass/print.scss")),
|
2018-10-31 07:18:57 +00:00
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/sass/print.scss"),
|
2018-01-12 10:50:29 +00:00
|
|
|
),
|
|
|
|
|
(
|
2018-08-04 19:47:45 +00:00
|
|
|
(ChangeKind::Config, PathBuf::from("/config.toml")),
|
2018-10-31 07:18:57 +00:00
|
|
|
Path::new("/home/vincent/site"),
|
|
|
|
|
Path::new("/home/vincent/site/config.toml"),
|
2018-01-12 10:50:29 +00:00
|
|
|
),
|
2017-03-08 04:21:45 +00:00
|
|
|
];
|
|
|
|
|
|
2017-05-20 15:00:41 +00:00
|
|
|
for (expected, pwd, path) in test_cases {
|
2017-03-08 04:21:45 +00:00
|
|
|
assert_eq!(expected, detect_change_kind(&pwd, &path));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2018-08-04 19:47:45 +00:00
|
|
|
#[test]
|
2018-08-04 20:28:39 +00:00
|
|
|
#[cfg(windows)]
|
2018-08-04 19:47:45 +00:00
|
|
|
fn windows_path_handling() {
|
|
|
|
|
let expected = (ChangeKind::Templates, PathBuf::from("/templates/hello.html"));
|
|
|
|
|
let pwd = Path::new(r#"C:\\Users\johan\site"#);
|
|
|
|
|
let path = Path::new(r#"C:\\Users\johan\site\templates\hello.html"#);
|
|
|
|
|
assert_eq!(expected, detect_change_kind(pwd, path));
|
|
|
|
|
}
|
|
|
|
|
|
2018-08-05 05:59:56 +00:00
|
|
|
#[test]
|
|
|
|
|
fn relative_path() {
|
|
|
|
|
let expected = (ChangeKind::Templates, PathBuf::from("/templates/hello.html"));
|
|
|
|
|
let pwd = Path::new("/home/johan/site");
|
|
|
|
|
let path = Path::new("templates/hello.html");
|
|
|
|
|
assert_eq!(expected, detect_change_kind(pwd, path));
|
|
|
|
|
}
|
2017-03-03 08:12:40 +00:00
|
|
|
}
|
