use anyhow::bail;
use clap::Arg;
use clap::ArgAction::SetTrue;
use clap::ArgMatches;
use reqwest::{StatusCode, Url};
use spacetimedb_client_api_messages::name::PublishOp;
use spacetimedb_client_api_messages::name::{is_address, parse_domain_name, PublishResult};
use std::fs;
use std::io::Write;
use std::path::PathBuf;
use crate::common_args;
use crate::config::Config;
use crate::util::unauth_error_context;
use crate::util::{add_auth_header_opt, get_auth_header};
pub fn cli() -> clap::Command {
clap::Command::new("publish")
.about("Create and update a SpacetimeDB database")
.arg(
Arg::new("host_type")
.long("host-type")
.short('t')
.value_parser(["wasm"])
.default_value("wasm")
.help("The type of host that should be for hosting this module"),
)
.arg(
Arg::new("clear_database")
.long("clear-database")
.short('c')
.action(SetTrue)
.requires("name|address")
.help("When publishing to an existing address, first DESTROY all data associated with the module"),
)
.arg(
Arg::new("project_path")
.value_parser(clap::value_parser!(PathBuf))
.default_value(".")
.long("project-path")
.short('p')
.help("The system path (absolute or relative) to the module project")
)
.arg(
Arg::new("wasm_file")
.value_parser(clap::value_parser!(PathBuf))
.long("wasm-file")
.short('w')
.conflicts_with("project_path")
.help("The system path (absolute or relative) to the wasm file we should publish, instead of building the project."),
)
.arg(
Arg::new("trace_log")
.long("trace_log")
.help("Turn on diagnostic/performance tracing for this project")
.action(SetTrue),
)
.arg(
common_args::identity()
.help("The identity that should own the database")
.long_help("The identity that should own the database. If no identity is provided, your default identity will be used.")
.required(false)
.conflicts_with("anon_identity")
)
.arg(
Arg::new("anon_identity")
.long("anon-identity")
.short('a')
.action(SetTrue)
.help("Instruct SpacetimeDB to allocate a new identity to own this database"),
)
.arg(
Arg::new("skip_clippy")
.long("skip_clippy")
.short('S')
.action(SetTrue)
.env("SPACETIME_SKIP_CLIPPY")
.value_parser(clap::builder::FalseyValueParser::new())
.help("Skips running clippy on the module before publishing (intended to speed up local iteration, not recommended for CI)"),
)
.arg(
Arg::new("debug")
.long("debug")
.short('d')
.action(SetTrue)
.help("Builds the module using debug instead of release (intended to speed up local iteration, not recommended for CI)"),
)
.arg(
Arg::new("name|address")
.help("A valid domain or address for this database"),
)
.arg(common_args::server()
.help("The nickname, domain name or URL of the server to host the database."),
)
.arg(
Arg::new("force")
.long("force")
.action(SetTrue)
.help("DANGEROUS - Proceed with all actions without waiting for user confirmation")
)
.after_help("Run `spacetime help publish` for more detailed information.")
}
pub async fn exec(mut config: Config, args: &ArgMatches) -> Result<(), anyhow::Error> {
let server = args.get_one::<String>("server").map(|s| s.as_str());
let identity = args.get_one::<String>("identity").map(String::as_str);
let name_or_address = args.get_one::<String>("name|address");
let path_to_project = args.get_one::<PathBuf>("project_path").unwrap();
let host_type = args.get_one::<String>("host_type").unwrap();
let clear_database = args.get_flag("clear_database");
let force = args.get_flag("force");
let trace_log = args.get_flag("trace_log");
let anon_identity = args.get_flag("anon_identity");
let skip_clippy = args.get_flag("skip_clippy");
let build_debug = args.get_flag("debug");
let wasm_file = args.get_one::<PathBuf>("wasm_file");
let database_host = config.get_host_url(server)?;
let (auth_header, identity) = get_auth_header(&mut config, anon_identity, identity, server)
.await?
.unzip();
let mut query_params = Vec::<(&str, &str)>::new();
query_params.push(("host_type", host_type.as_str()));
query_params.push(("register_tld", "true"));
if let Some(name_or_address) = name_or_address {
if !is_address(name_or_address) {
parse_domain_name(name_or_address)?;
}
query_params.push(("name_or_address", name_or_address.as_str()));
}
if !path_to_project.exists() {
return Err(anyhow::anyhow!(
"Project path does not exist: {}",
path_to_project.display()
));
}
if trace_log {
query_params.push(("trace_log", "true"));
}
let path_to_wasm = if !path_to_project.is_dir() && path_to_project.extension().map_or(false, |ext| ext == "wasm") {
println!("Note: Using --project-path to provide a wasm file is deprecated, and will be");
println!("removed in a future release. Please use --wasm-file instead.");
path_to_project.clone()
} else if let Some(path) = wasm_file {
println!("Skipping build. Instead we are publishing {}", path.display());
path.clone()
} else {
crate::tasks::build(path_to_project, skip_clippy, build_debug)?
};
let program_bytes = fs::read(path_to_wasm)?;
println!(
"Uploading to {} => {}",
server.unwrap_or(config.default_server_name().unwrap_or("<default>")),
database_host
);
if clear_database {
if force {
println!("Skipping confirmation due to --force.");
} else {
println!(
"This will DESTROY the current {} module, and ALL corresponding data.",
name_or_address.unwrap()
);
print!(
"Are you sure you want to proceed? (y/N) [deleting {}] ",
name_or_address.unwrap()
);
std::io::stdout().flush()?;
let mut input = String::new();
std::io::stdin().read_line(&mut input)?;
if input.trim().to_lowercase() != "y" && input.trim().to_lowercase() != "yes" {
println!("Aborting");
return Ok(());
}
}
query_params.push(("clear", "true"));
}
println!("Publishing module...");
let mut builder = reqwest::Client::new().post(Url::parse_with_params(
format!("{}/database/publish", database_host).as_str(),
query_params,
)?);
builder = add_auth_header_opt(builder, &auth_header);
let res = builder.body(program_bytes).send().await?;
if res.status() == StatusCode::UNAUTHORIZED && !anon_identity {
if let Some(identity) = &identity {
let err = res.text().await?;
return unauth_error_context(
Err(anyhow::anyhow!(err)),
&identity.to_hex(),
config.server_nick_or_host(server)?,
);
}
}
if res.status().is_client_error() || res.status().is_server_error() {
let err = res.text().await?;
bail!(err)
}
let bytes = res.bytes().await.unwrap();
let response: PublishResult = serde_json::from_slice(&bytes[..]).unwrap();
match response {
PublishResult::Success { domain, address, op } => {
let op = match op {
PublishOp::Created => "Created new",
PublishOp::Updated => "Updated",
};
if let Some(domain) = domain {
println!("{} database with domain: {}, address: {}", op, domain, address);
} else {
println!("{} database with address: {}", op, address);
}
}
PublishResult::TldNotRegistered { domain } => {
return Err(anyhow::anyhow!(
"The top level domain that you provided is not registered.\n\
This tld is not yet registered to any identity. You can register this domain with the following command:\n\
\n\
\tspacetime dns register-tld {}\n",
domain.tld()
));
}
PublishResult::PermissionDenied { domain } => {
return match identity {
Some(identity) => {
let suggested_tld: String = identity.to_hex().chars().take(12).collect();
if let Some(sub_domain) = domain.sub_domain() {
Err(anyhow::anyhow!(
"The top level domain {} is not registered to the identity you provided.\n\
We suggest you publish to a domain that starts with a TLD owned by you, or publish to a new domain like:\n\
\tspacetime publish {}/{}\n",
domain.tld(),
suggested_tld,
sub_domain
))
} else {
Err(anyhow::anyhow!(
"The top level domain {} is not registered to the identity you provided.\n\
We suggest you push to either a domain owned by you, or a new domain like:\n\
\tspacetime publish {}\n",
domain.tld(),
suggested_tld
))
}
}
None => Err(anyhow::anyhow!(
"The domain {} is not registered to the identity you provided.",
domain
)),
};
}
}
Ok(())
}