emelie/zola
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

add --watch-only flag

Browse source
This commit is contained in:
Robert Masen 2018-11-01 17:20:35 -05:00
parent cb3c42078a
commit b7c3d7199d
3 changed files with 86 additions and 65 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/cli.rs
View file

@ -62,6 +62,10 @@ pub fn build_cli() -> App<'static, 'static> {
.default_value("127.0.0.1") .default_value("127.0.0.1")
.takes_value(true) .takes_value(true)
.help("Changes the base_url"), .help("Changes the base_url"),
Arg::with_name("watch_only")
.long("watch-only")
.takes_value(false)
.help("Do not start a server, just re-build project on changes")
]), ]),
]) ])
} }

142
src/cmd/serve.rs
View file

@ -166,6 +166,7 @@ pub fn serve(
output_dir: &str, output_dir: &str,
base_url: &str, base_url: &str,
config_file: &str, config_file: &str,
watch_only: bool,
) -> Result<()> { ) -> Result<()> {
let start = Instant::now(); let start = Instant::now();
let (mut site, address) = create_new_site(interface, port, output_dir, base_url, config_file)?; let (mut site, address) = create_new_site(interface, port, output_dir, base_url, config_file)?;
@ -206,49 +207,56 @@ pub fn serve(
// output path is going to need to be moved later on, so clone it for the // output path is going to need to be moved later on, so clone it for the
// http closure to avoid contention. // http closure to avoid contention.
let static_root = output_path.clone(); let static_root = output_path.clone();
thread::spawn(move || { let broadcaster = if !watch_only {
let s = server::new(move || {
App::new()
.middleware(NotFoundHandler { rendered_template: static_root.join("404.html") })
.resource(r"/livereload.js", |r| r.f(livereload_handler))
// Start a webserver that serves the `output_dir` directory
.handler(
r"/",
fs::StaticFiles::new(&static_root)
.unwrap()
.show_files_listing()
.files_listing_renderer(handle_directory),
)
})
.bind(&address)
.expect("Can't start the webserver")
.shutdown_timeout(20);
println!("Web server is available at http://{}", &address);
s.run();
});
// The websocket for livereload thread::spawn(move || {
let ws_server = WebSocket::new(|output: Sender| { let s = server::new(move || {
move |msg: Message| { App::new()
if msg.into_text().unwrap().contains("\"hello\"") { .middleware(NotFoundHandler { rendered_template: static_root.join("404.html") })
return output.send(Message::text( .resource(r"/livereload.js", |r| r.f(livereload_handler))
r#" // Start a webserver that serves the `output_dir` directory
{ .handler(
"command": "hello", r"/",
"protocols": [ "http://livereload.com/protocols/official-7" ], fs::StaticFiles::new(&static_root)
"serverName": "Zola" .unwrap()
} .show_files_listing()
"#, .files_listing_renderer(handle_directory),
)); )
})
.bind(&address)
.expect("Can't start the webserver")
.shutdown_timeout(20);
println!("Web server is available at http://{}", &address);
s.run();
});
// 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(())
} }
Ok(()) })
} .unwrap();
}) let broadcaster = ws_server.broadcaster();
.unwrap(); thread::spawn(move || {
let broadcaster = ws_server.broadcaster(); ws_server.listen(&*ws_address).unwrap();
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
};
let pwd = env::current_dir().unwrap(); let pwd = env::current_dir().unwrap();
@ -297,21 +305,25 @@ pub fn serve(
match detect_change_kind(&pwd, &path) { match detect_change_kind(&pwd, &path) {
(ChangeKind::Content, _) => { (ChangeKind::Content, _) => {
console::info(&format!("-> Content changed {}", path.display())); console::info(&format!("-> Content changed {}", path.display()));
// Force refresh if let Some(ref broadcaster) = broadcaster {
rebuild_done_handling( // Force refresh
&broadcaster, rebuild_done_handling(
rebuild::after_content_change(&mut site, &path), broadcaster,
"/x.js", rebuild::after_content_change(&mut site, &path),
); "/x.js",
);
}
} }
(ChangeKind::Templates, _) => { (ChangeKind::Templates, _) => {
console::info(&format!("-> Template changed {}", path.display())); console::info(&format!("-> Template changed {}", path.display()));
// Force refresh if let Some(ref broadcaster) = broadcaster {
rebuild_done_handling( // Force refresh
&broadcaster, rebuild_done_handling(
rebuild::after_template_change(&mut site, &path), broadcaster,
"/x.js", rebuild::after_template_change(&mut site, &path),
); "/x.js",
);
}
} }
(ChangeKind::StaticFiles, p) => { (ChangeKind::StaticFiles, p) => {
if path.is_file() { if path.is_file() {
@ -319,20 +331,24 @@ pub fn serve(
"-> Static file changes detected {}", "-> Static file changes detected {}",
path.display() path.display()
)); ));
rebuild_done_handling( if let Some(ref broadcaster) = broadcaster {
&broadcaster, rebuild_done_handling(
copy_file(&path, &site.output_path, &site.static_path), broadcaster,
&p.to_string_lossy(), copy_file(&path, &site.output_path, &site.static_path),
); &p.to_string_lossy(),
);
}
} }
} }
(ChangeKind::Sass, p) => { (ChangeKind::Sass, p) => {
console::info(&format!("-> Sass file changed {}", path.display())); console::info(&format!("-> Sass file changed {}", path.display()));
rebuild_done_handling( if let Some(ref broadcaster) = broadcaster {
&broadcaster, rebuild_done_handling(
site.compile_sass(&site.base_path), &broadcaster,
&p.to_string_lossy(), site.compile_sass(&site.base_path),
); &p.to_string_lossy(),
);
}
} }
(ChangeKind::Config, _) => { (ChangeKind::Config, _) => {
console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible."); console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible.");

5
src/main.rs
View file

@ -77,11 +77,12 @@ fn main() {
::std::process::exit(1); ::std::process::exit(1);
} }
} }
let watch_only = matches.is_present("watch_only");
let output_dir = matches.value_of("output_dir").unwrap(); let output_dir = matches.value_of("output_dir").unwrap();
let base_url = matches.value_of("base_url").unwrap(); let base_url = matches.value_of("base_url").unwrap();
println!("watch_only: {}", watch_only);
console::info("Building site..."); console::info("Building site...");
match cmd::serve(interface, port, output_dir, base_url, config_file) { match cmd::serve(interface, port, output_dir, base_url, config_file, watch_only) {
Ok(()) => (), Ok(()) => (),
Err(e) => { Err(e) => {
console::unravel_errors("", &e); console::unravel_errors("", &e);