upd
This commit is contained in:
parent
ca7f14c3a8
commit
15e8f9f738
1 changed files with 13 additions and 9 deletions
22
src/main.rs
22
src/main.rs
|
@ -244,10 +244,10 @@ struct PastebinConfig {
|
|||
tls_key: Option<String>,
|
||||
|
||||
#[structopt(
|
||||
long = "domain",
|
||||
help = "Domain name used to generate URL",
|
||||
long = "uri",
|
||||
help = "Override default URI",
|
||||
)]
|
||||
domain: Option<String>,
|
||||
uri: Option<String>,
|
||||
}
|
||||
|
||||
fn get_url(cfg: &PastebinConfig) -> String {
|
||||
|
@ -262,12 +262,16 @@ fn get_url(cfg: &PastebinConfig) -> String {
|
|||
"http"
|
||||
};
|
||||
|
||||
format!(
|
||||
"{scheme}://{address}{port}",
|
||||
scheme = scheme,
|
||||
port = port,
|
||||
address = cfg.domain.clone().unwrap_or_else(|| cfg.address.clone()),
|
||||
)
|
||||
if cfg.uri.is_some() {
|
||||
cfg.uri.clone().unwrap()
|
||||
} else {
|
||||
format!(
|
||||
"{scheme}://{address}{port}",
|
||||
scheme = scheme,
|
||||
port = port,
|
||||
address = cfg.address,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fn get_error_response(html: String, status: Status, cfg: &PastebinConfig) -> Response {
|
||||
|
|
Loading…
Reference in a new issue