Merge pull request #404 from epage/rename

feat: `rename` subcommand

src/bin/cobalt/main.rs

@@ -43,6 +43,7 @@ fn run() -> Result<()> {
         .args(&args::get_logging_args())
         .subcommand(new::init_command_args())
         .subcommand(new::new_command_args())
+        .subcommand(new::rename_command_args())
         .subcommand(new::publish_command_args())
         .subcommand(build::build_command_args())
         .subcommand(build::clean_command_args())
@@ -64,6 +65,7 @@ fn run() -> Result<()> {
     match command {
         "init" => new::init_command(matches),
         "new" => new::new_command(matches),
+        "rename" => new::rename_command(matches),
         "publish" => new::publish_command(matches),
         "build" => build::build_command(matches),
         "clean" => build::clean_command(matches),

src/bin/cobalt/new.rs

@@ -78,6 +78,52 @@ pub fn new_command(matches: &clap::ArgMatches) -> Result<()> {
     Ok(())
 }
 
+pub fn rename_command_args() -> clap::App<'static, 'static> {
+    clap::SubCommand::with_name("rename")
+        .about("Rename a document")
+        .args(&args::get_config_args())
+        .arg(
+            clap::Arg::with_name("SRC")
+                .required(true)
+                .help("File to rename")
+                .takes_value(true),
+        )
+        .arg(
+            clap::Arg::with_name("TITLE")
+                .required(true)
+                .help("Title of the post")
+                .takes_value(true),
+        )
+        .arg(
+            clap::Arg::with_name("file")
+                .short("f")
+                .long("file")
+                .value_name("DIR_OR_FILE")
+                .help("New document's parent directory or file (default: `<CWD>/title.ext`)")
+                .takes_value(true),
+        )
+}
+
+pub fn rename_command(matches: &clap::ArgMatches) -> Result<()> {
+    let config = args::get_config(matches)?;
+    let config = config.build()?;
+
+    let source = path::PathBuf::from(matches.value_of("SRC").unwrap());
+
+    let title = matches.value_of("TITLE").unwrap();
+
+    let mut file = env::current_dir().expect("How does this fail?");
+    if let Some(rel_file) = matches.value_of("file") {
+        file.push(path::Path::new(rel_file))
+    }
+    let file = file;
+
+    rename_document(&config, source, title, file)
+        .chain_err(|| format!("Could not rename `{}`", title))?;
+
+    Ok(())
+}
+
 pub fn publish_command_args() -> clap::App<'static, 'static> {
     clap::SubCommand::with_name("publish")
         .about("Publish a document")
@@ -282,6 +328,78 @@ fn create_file_for_path(path: &path::Path, content: &str) -> Result<()> {
     Ok(())
 }
 
+pub fn rename_document(
+    config: &cobalt_model::Config,
+    source: path::PathBuf,
+    title: &str,
+    file: path::PathBuf,
+) -> Result<()> {
+    let target = if file.extension().is_none() || file.is_dir() {
+        let extension = source.extension().and_then(|s| s.to_str()).unwrap_or("md");
+        let file_name = format!("{}.{}", cobalt_model::slug::slugify(title), extension);
+        let mut file = file;
+        file.push(path::Path::new(&file_name));
+        file
+    } else {
+        file
+    };
+
+    let doc = cobalt_model::files::read_file(&source)?;
+    let doc = cobalt_model::DocumentBuilder::<cobalt_model::FrontmatterBuilder>::parse(&doc)?;
+    let (front, content) = doc.parts();
+
+    let pages = config.pages.clone().build()?;
+    let posts = config.posts.clone().build()?;
+    let full_front = if posts.pages.includes_file(&target)
+        || posts
+            .drafts
+            .map(|d| d.includes_file(&target))
+            .unwrap_or_default()
+    {
+        // Can't rely on this for drafts atm
+        let rel_src = target
+            .strip_prefix(&config.source)
+            .expect("file was found under the root");
+        front
+            .clone()
+            .merge_path(rel_src)
+            .merge(posts.default.clone())
+    } else if pages.pages.includes_file(&target)
+        || pages
+            .drafts
+            .map(|d| d.includes_file(&target))
+            .unwrap_or_default()
+    {
+        // Can't rely on this for drafts atm
+        let rel_src = target
+            .strip_prefix(&config.source)
+            .expect("file was found under the root");
+        front
+            .clone()
+            .merge_path(rel_src)
+            .merge(pages.default.clone())
+    } else {
+        bail!(
+            "Target file wouldn't be a member of any collection: {:?}",
+            target
+        );
+    };
+    let full_front = full_front.build()?;
+
+    let new_front = front.set_title(Some(title.to_string()));
+    let doc =
+        cobalt_model::DocumentBuilder::<cobalt_model::FrontmatterBuilder>::new(new_front, content);
+    let doc = doc.to_string();
+    cobalt_model::files::write_document_file(doc, target)?;
+
+    if !full_front.is_draft {
+        warn!("Renaming a published page might invalidate links");
+    }
+    fs::remove_file(source)?;
+
+    Ok(())
+}
+
 pub fn publish_document(file: &path::Path) -> Result<()> {
     let doc = cobalt_model::files::read_file(file)?;
     let doc = cobalt_model::DocumentBuilder::<cobalt_model::FrontmatterBuilder>::parse(&doc)?;