Add output option for wrangler publish

Cargo.toml

@@ -39,6 +39,7 @@ log = "0.4.11"
 notify = "4.0.15"
 number_prefix = "0.4.0"
 openssl = { version = "0.10.29", optional = true }
+once_cell = "1.4.1"
 percent-encoding = "2.1.0"
 predicates = "1.0.5"
 prettytable-rs = "0.8.0"

src/commands/build.rs

@@ -2,10 +2,10 @@ use std::path::Path;
 
 use clap::ArgMatches;
 
+use super::DEFAULT_CONFIG_PATH;
 use crate::build;
 use crate::settings::toml::Manifest;
-
-use super::DEFAULT_CONFIG_PATH;
+use crate::terminal;
 
 pub fn run(matches: &ArgMatches) -> Result<(), failure::Error> {
     log::info!("Getting project settings");
@@ -14,5 +14,25 @@ pub fn run(matches: &ArgMatches) -> Result<(), failure::Error> {
     let manifest = Manifest::new(&config_path)?;
     let env = matches.value_of("env");
     let target = &manifest.get_target(env, false)?;
-    build(&target)
+    if matches.is_present("output") {
+        if matches.value_of("output") == Some("json") {
+            terminal::message::set_output_type(terminal::message::OutputType::Json)
+        } else {
+            terminal::message::user_error("json is the only valid value for output flag");
+        }
+        let result = build(&target);
+        let jsonoutput = terminal::message::PublishOutput {
+            name: Some(target.name.clone()),
+            success: match result {
+                Ok(_) => Some("true".to_string()),
+                Err(_) => Some("false".to_string()),
+            },
+            url: None,
+        };
+        terminal::message::jsonout(&jsonoutput);
+        result
+    } else {
+        terminal::message::set_output_type(terminal::message::OutputType::Human);
+        build(&target)
+    }
 }

src/main.rs

@@ -440,6 +440,13 @@ fn run() -> Result<(), failure::Error> {
                         .long("env")
                         .takes_value(true)
                 )
+                .arg(
+                    Arg::with_name("output")
+                    .short("o")
+                    .long("output")
+                    .takes_value(true)
+                    .help("Machine or human readable output")
+                )
                 .arg(wrangler_file.clone())
                 .arg(silent_verbose_arg.clone()),
         )

src/terminal/message.rs

@@ -1,9 +1,50 @@
 use super::emoji;
+use once_cell::sync::OnceCell;
 
 use billboard::{Billboard, BorderColor, BorderStyle};
+use serde::{Deserialize, Serialize};
+
+pub enum OutputType {
+    Json,
+    Human,
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct PublishOutput {
+    pub success: Option<String>,
+    pub name: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub url: Option<String>,
+}
+
+static OUTPUT_TYPE: OnceCell<OutputType> = OnceCell::new();
+
+pub fn set_output_type(typ: OutputType) {
+    match OUTPUT_TYPE.set(typ) {
+        Ok(_) => {}
+        Err(_) => {
+            message(&"Output type already set".to_string());
+        }
+    }
+}
+
+// Always goes to stdout
+pub fn jsonout<T>(value: &T)
+where
+    T: ?Sized + Serialize,
+{
+    println!("{}", &serde_json::to_string(value).unwrap());
+}
 
 fn message(msg: &str) {
-    println!("{}", msg);
+    match OUTPUT_TYPE.get() {
+        Some(OutputType::Json) => {
+            eprintln!("{}", msg);
+        }
+        _ => {
+            println!("{}", msg);
+        }
+    }
 }
 
 pub fn billboard(msg: &str) {