Add exit codes; I sure hope no memory leaks happened

This commit is contained in:
Sofia 2018-04-18 21:06:02 +03:00
parent 208f1eb254
commit 1bad6a2309
5 changed files with 78 additions and 55 deletions

View File

@ -1,6 +1,6 @@
[website] [website]
website_name = "TestWebsite" website_name = "TestWebsite"
built_pages = ["test_page/otus_test.toml"] built_pages = ["test_page/about.toml", "test_page/other.toml"]
use_default_css = true use_default_css = true
javascript = [] javascript = []
css = [] css = []

View File

@ -5,7 +5,7 @@ use template::Template;
use renderer; use renderer;
use logger::{LogLevel, Logger}; use logger::{LogLevel, Logger};
use error::Error; use error::Error;
use options::BuildOps; use options::{BuildOpt, Opt};
use file_writer; use file_writer;
const DEFAULT_CSS: &'static str = include_str!("templates/default-css.css"); const DEFAULT_CSS: &'static str = include_str!("templates/default-css.css");
@ -20,14 +20,14 @@ fn fetch_config() -> Result<Config, Error> {
} }
} }
pub fn build(logger: &Logger, ops: &BuildOps) -> Result<(), Error> { pub fn build(logger: &Logger, opt: &Opt, _: &BuildOpt) -> Result<(), Error> {
logger.log(LogLevel::INFO, "Starting build"); logger.log(LogLevel::INFO, "Starting build");
let config = fetch_config()?; let config = fetch_config()?;
if config.global_config.website.use_default_css { if config.global_config.website.use_default_css {
let css_path = file_writer::add_to_beginning(PathBuf::from("css/default.css"), "public")?; let css_path = file_writer::add_to_beginning(PathBuf::from("css/default.css"), "public")?;
logger.log(LogLevel::DETAIL, format!("Adding {:?}", css_path)); logger.log(LogLevel::DETAIL, format!("Adding {:?}", css_path));
file_writer::write_file(&css_path, DEFAULT_CSS, ops.overwrite)?; file_writer::write_file(&css_path, DEFAULT_CSS, opt.overwrite)?;
} }
logger.log(LogLevel::INFO, "Generating page templates"); logger.log(LogLevel::INFO, "Generating page templates");
@ -62,7 +62,7 @@ pub fn build(logger: &Logger, ops: &BuildOps) -> Result<(), Error> {
let html_path = file_writer::add_to_beginning(PathBuf::from(html_path), "public")?; let html_path = file_writer::add_to_beginning(PathBuf::from(html_path), "public")?;
logger.log(LogLevel::DETAILER, format!("Writing {:?}", html_path)); logger.log(LogLevel::DETAILER, format!("Writing {:?}", html_path));
if let Some(render) = renders.get(idx) { if let Some(render) = renders.get(idx) {
file_writer::write_file(&html_path, render.clone(), ops.overwrite)?; file_writer::write_file(&html_path, render.clone(), opt.overwrite)?;
} else { } else {
return Err(Error::new( return Err(Error::new(
LogLevel::SEVERE, LogLevel::SEVERE,

View File

@ -22,6 +22,7 @@ mod file_writer;
use std::collections::HashMap; use std::collections::HashMap;
use std::path::PathBuf; use std::path::PathBuf;
use std::process::exit;
use structopt::StructOpt; use structopt::StructOpt;
@ -30,34 +31,49 @@ use options::{Opt, Subcommands};
use config_toml::{GlobalConfigToml, NavbarConfig, NavbarItem, WebsiteConfig}; use config_toml::{GlobalConfigToml, NavbarConfig, NavbarItem, WebsiteConfig};
fn main() { fn main() {
exit(match run() {
Ok(()) => 0,
Err(()) => 1,
})
}
fn run() -> Result<(), ()> {
let opt = Opt::from_args(); let opt = Opt::from_args();
let log_level = (3 - opt.quiet as i8 + opt.verbose as i8).max(0).min(5); let log_level = (3 - opt.quiet as i8 + opt.verbose as i8).max(0).min(5);
let logger = Logger::new(LogLevel::from(log_level as u8)); let logger = Logger::new(LogLevel::from(log_level as u8));
match opt.cmd { match opt.clone().cmd {
Subcommands::Build(ops) => match builder::build(&logger, &ops) { Subcommands::Build(build_opt) => match builder::build(&logger, &opt, &build_opt) {
Ok(_) => logger.log(LogLevel::DETAILER, "Building finished successfully."), Ok(_) => {
logger.log(LogLevel::DETAILER, "Building finished successfully.");
Ok(())
}
Err(err) => { Err(err) => {
logger.log(err.severity(), err.description()); logger.log(err.severity(), err.description());
logger.log(LogLevel::SEVERE, "Aborting building due to error."); logger.log(LogLevel::SEVERE, "Aborting building due to error.");
Err(())
} }
}, },
Subcommands::New(ops) => match new_page::generate_new_page(ops, &logger) { Subcommands::New(new_opt) => match new_page::generate_new_page(&logger, &opt, &new_opt) {
Ok(_) => logger.log( Ok(_) => {
logger.log(
LogLevel::DETAILER, LogLevel::DETAILER,
"Generating the new page finished successfully.", "Generating the new page finished successfully.",
), );
Ok(())
}
Err(err) => { Err(err) => {
logger.log(err.severity(), err.description()); logger.log(err.severity(), err.description());
logger.log(LogLevel::SEVERE, "Aborting building due to error."); logger.log(LogLevel::SEVERE, "Aborting building due to error.");
Err(())
} }
}, },
Subcommands::Initialize(ops) => { Subcommands::Initialize(init_opt) => {
logger.log(LogLevel::DETAIL, "Generating config.toml"); logger.log(LogLevel::DETAIL, "Generating config.toml");
let mut navbar = None; let mut navbar = None;
if !ops.no_default_navbar { if !init_opt.no_default_navbar {
let mut hashmap = HashMap::new(); let mut hashmap = HashMap::new();
hashmap.insert( hashmap.insert(
"test".to_owned(), "test".to_owned(),
@ -83,28 +99,34 @@ fn main() {
let config = GlobalConfigToml { let config = GlobalConfigToml {
website: WebsiteConfig { website: WebsiteConfig {
website_name: ops.name, website_name: init_opt.name,
built_pages: Vec::new(), built_pages: Vec::new(),
use_default_css: true, use_default_css: true,
javascript: Vec::new(), javascript: Vec::new(),
css: Vec::new(), css: Vec::new(),
favicon: ops.favicon, favicon: init_opt.favicon,
before_navbar_url: None, before_navbar_url: None,
before_content_url: None, before_content_url: None,
after_content_url: None, after_content_url: None,
twitter_author: ops.twitter_author, twitter_author: init_opt.twitter_author,
google_robots: None, google_robots: None,
google_site_verification: None, google_site_verification: None,
}, },
navbar: navbar, navbar: navbar,
}; };
match file_writer::write_toml(&PathBuf::from("config.toml"), &config, ops.overwrite) { match file_writer::write_toml(&PathBuf::from("config.toml"), &config, opt.overwrite) {
Ok(_) => logger.log(LogLevel::INFO, "Done!"), Ok(_) => {
Err(err) => logger.log( logger.log(LogLevel::INFO, "Done!");
Ok(())
}
Err(err) => {
logger.log(
err.severity(), err.severity(),
format!("Initialization failed: {}", err.description()), format!("Initialization failed: {}", err.description()),
), );
Err(())
}
} }
} }
} }

View File

@ -1,6 +1,6 @@
use std::path::PathBuf; use std::path::PathBuf;
use options::NewOps; use options::{NewOpt, Opt};
use logger::{LogLevel, Logger}; use logger::{LogLevel, Logger};
use file_writer; use file_writer;
use error::Error; use error::Error;
@ -17,16 +17,16 @@ This is the markdown file, where you will insert this page's contents.
See [CommonMark](http://commonmark.org/help/) for reference."#; See [CommonMark](http://commonmark.org/help/) for reference."#;
pub fn generate_new_page(ops: NewOps, logger: &Logger) -> Result<(), Error> { pub fn generate_new_page(logger: &Logger, opt: &Opt, new_opt: &NewOpt) -> Result<(), Error> {
logger.log( logger.log(
LogLevel::DETAILER, LogLevel::DETAILER,
format!("Starting to create paths for given files"), format!("Starting to create paths for given files"),
); );
let toml_path = ensure_extension(ops.toml_path, "toml", false)?; let toml_path = ensure_extension(new_opt.toml_path.clone(), "toml", false)?;
let markdown_path; let markdown_path;
if let Some(markdown) = ops.markdown_path { if let Some(markdown) = new_opt.markdown_path.clone() {
markdown_path = ensure_extension(markdown, "md", false)?; markdown_path = ensure_extension(markdown, "md", false)?;
} else { } else {
markdown_path = ensure_extension(toml_path.clone(), "md", true)?; markdown_path = ensure_extension(toml_path.clone(), "md", true)?;
@ -46,14 +46,14 @@ pub fn generate_new_page(ops: NewOps, logger: &Logger) -> Result<(), Error> {
); );
let html_path; let html_path;
if let Some(path) = ops.html_path { if let Some(path) = new_opt.html_path.clone() {
html_path = ensure_extension(path, "html", false)?; html_path = ensure_extension(path, "html", false)?;
} else { } else {
html_path = ensure_extension(toml_path.clone(), "html", true)?; html_path = ensure_extension(toml_path.clone(), "html", true)?;
} }
let title: String; let title: String;
if let Some(t) = ops.title { if let Some(t) = new_opt.title.clone() {
title = t; title = t;
} else { } else {
let mut parts = toml_path.file_name().unwrap().to_str().unwrap().split("."); let mut parts = toml_path.file_name().unwrap().to_str().unwrap().split(".");
@ -87,7 +87,7 @@ pub fn generate_new_page(ops: NewOps, logger: &Logger) -> Result<(), Error> {
}, },
}; };
match file_writer::write_toml(&toml_path, &page_config, ops.overwrite) { match file_writer::write_toml(&toml_path, &page_config, opt.overwrite) {
Ok(_) => logger.log( Ok(_) => logger.log(
LogLevel::INFO, LogLevel::INFO,
format!("{:?} created successfully", toml_path), format!("{:?} created successfully", toml_path),
@ -105,9 +105,9 @@ pub fn generate_new_page(ops: NewOps, logger: &Logger) -> Result<(), Error> {
format!("Creating a new .md file at {:?}", markdown_path), format!("Creating a new .md file at {:?}", markdown_path),
); );
file_writer::write_file(&markdown_path, PLACEHOLDER_MARKDOWN, ops.overwrite)?; file_writer::write_file(&markdown_path, PLACEHOLDER_MARKDOWN, opt.overwrite)?;
if !ops.no_modify_config { if !new_opt.no_modify_config {
logger.log( logger.log(
LogLevel::INFO, LogLevel::INFO,
format!("Adding page config path to config.toml"), format!("Adding page config path to config.toml"),
@ -123,7 +123,7 @@ pub fn generate_new_page(ops: NewOps, logger: &Logger) -> Result<(), Error> {
format!("Re-serializing config and writing it"), format!("Re-serializing config and writing it"),
); );
match file_writer::write_toml(&PathBuf::from("config.toml"), &config, ops.overwrite) { match file_writer::write_toml(&PathBuf::from("config.toml"), &config, opt.overwrite) {
Ok(_) => {} Ok(_) => {}
Err(err) => { Err(err) => {
return Err(Error::new( return Err(Error::new(

View File

@ -1,6 +1,6 @@
use std::path::PathBuf; use std::path::PathBuf;
#[derive(StructOpt, Debug)] #[derive(StructOpt, Debug, Clone)]
#[structopt(name = "Teasca.de Generator", #[structopt(name = "Teasca.de Generator",
about = "A website generator, used mainly for generating teasca.de")] about = "A website generator, used mainly for generating teasca.de")]
pub struct Opt { pub struct Opt {
@ -12,66 +12,67 @@ pub struct Opt {
#[structopt(short = "q", long = "quiet", parse(from_occurrences))] #[structopt(short = "q", long = "quiet", parse(from_occurrences))]
pub quiet: u8, pub quiet: u8,
/// Overwrites necessary files to create new ones.
#[structopt(short = "o", long = "overwrite")]
pub overwrite: bool,
#[structopt(subcommand)] #[structopt(subcommand)]
pub cmd: Subcommands, pub cmd: Subcommands,
} }
#[derive(StructOpt, Debug)] #[derive(StructOpt, Debug, Clone)]
pub enum Subcommands { pub enum Subcommands {
/// Build the website /// Build the website
#[structopt(name = "build")] #[structopt(name = "build")]
Build(BuildOps), Build(BuildOpt),
/// Create a new page /// Create a new page
#[structopt(name = "new")] #[structopt(name = "new")]
New(NewOps), New(NewOpt),
/// Initializes a config.toml /// Initializes a config.toml
#[structopt(name = "init")] #[structopt(name = "init")]
Initialize(InitializeOps), Initialize(InitializeOpt),
} }
#[derive(StructOpt, Debug)] #[derive(StructOpt, Debug, Clone)]
pub struct BuildOps { pub struct BuildOpt {}
/// Overwrites necessary files to create new ones.
#[structopt(short = "o", long = "overwrite")]
pub overwrite: bool,
}
#[derive(StructOpt, Debug)] #[derive(StructOpt, Debug, Clone)]
pub struct NewOps { pub struct NewOpt {
/// File path for the .toml file created /// File path for the .toml file created
#[structopt(parse(from_os_str))] #[structopt(parse(from_os_str))]
pub toml_path: PathBuf, pub toml_path: PathBuf,
/// File path for the markdown file created /// File path for the markdown file created
#[structopt(short = "m", long = "markdown", parse(from_os_str))] #[structopt(short = "m", long = "markdown", parse(from_os_str))]
pub markdown_path: Option<PathBuf>, pub markdown_path: Option<PathBuf>,
/// Outpuh html file path of the generated page /// Outpuh html file path of the generated page
#[structopt(short = "h", long = "html", parse(from_os_str))] #[structopt(short = "h", long = "html", parse(from_os_str))]
pub html_path: Option<PathBuf>, pub html_path: Option<PathBuf>,
/// Sets the title of the generated page /// Sets the title of the generated page
#[structopt(short = "t", long = "title")] #[structopt(short = "t", long = "title")]
pub title: Option<String>, pub title: Option<String>,
/// Overwrites existing .toml / .md files
#[structopt(short = "o", long = "overwrite")]
pub overwrite: bool,
/// Don't modify config.toml automatically /// Don't modify config.toml automatically
#[structopt(short = "n", long = "no-modify-config")] #[structopt(short = "n", long = "no-modify-config")]
pub no_modify_config: bool, pub no_modify_config: bool,
} }
#[derive(StructOpt, Debug)] #[derive(StructOpt, Debug, Clone)]
pub struct InitializeOps { pub struct InitializeOpt {
/// Name of the webpage /// Name of the webpage
#[structopt()] #[structopt()]
pub name: String, pub name: String,
/// Favicon /// Favicon
#[structopt(short = "f", long = "favicon")] #[structopt(short = "f", long = "favicon")]
pub favicon: Option<String>, pub favicon: Option<String>,
/// Twitter author /// Twitter author
#[structopt(short = "t", long = "twitter")] #[structopt(short = "t", long = "twitter")]
pub twitter_author: Option<String>, pub twitter_author: Option<String>,
/// Overwrites existing config.toml
#[structopt(short = "o", long = "overwrite")]
pub overwrite: bool,
/// Disables the default navbar /// Disables the default navbar
#[structopt(short = "n", long = "no-navbar")] #[structopt(short = "n", long = "no-navbar")]
pub no_default_navbar: bool, pub no_default_navbar: bool,