diff --git a/src/main.rs b/src/main.rs
index 4187575..67310db 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -39,9 +39,10 @@ const VERSION: &str = env!("CARGO_PKG_VERSION");
lazy_static! {
static ref CLIENT: reqwest::Client = reqwest::Client::new();
+ static ref OPT: Opt = Opt::from_args();
static ref INDEX: Vec = {
let mut buf = Vec::new();
- templates::index(&mut buf, COMMIT, VERSION).unwrap();
+ templates::index(&mut buf, COMMIT, VERSION, &OPT.domain).unwrap();
buf
};
static ref P404: Vec = {
@@ -87,6 +88,9 @@ struct Opt {
#[structopt(short = "h", long = "host", default_value = "0.0.0.0")]
/// Interface to listen on
host: String,
+ #[structopt(short = "d", long = "domain", default_value = "hitsofcode.com")]
+ /// Interface to listen on
+ domain: String,
}
fn pull(path: impl AsRef) -> Result<(), Error> {
@@ -248,11 +252,10 @@ fn main() -> std::io::Result<()> {
std::env::set_var("RUST_LOG", "actix_web=warn");
pretty_env_logger::init();
openssl_probe::init_ssl_cert_env_vars();
- let opt = Opt::from_args();
- let interface = format!("{}:{}", opt.host, opt.port);
+ let interface = format!("{}:{}", OPT.host, OPT.port);
let state = Arc::new(State {
- repos: opt.outdir.display().to_string(),
- cache: opt.cachedir.display().to_string(),
+ repos: OPT.outdir.display().to_string(),
+ cache: OPT.cachedir.display().to_string(),
});
HttpServer::new(move || {
App::new()
diff --git a/templates/index.rs.html b/templates/index.rs.html
index ae559e6..da129fe 100644
--- a/templates/index.rs.html
+++ b/templates/index.rs.html
@@ -1,6 +1,6 @@
@use super::base;
-@(commit: &str, version: &str)
+@(commit: &str, version: &str, domain: &str)
@:base("Hits-of-Code Badges", "Hits-of-Code Badges", {
@@ -23,7 +23,7 @@ used for GitHub, GitLab and Bitbucket repositories. Just put the following code
-[![Hits-of-Code](https://hitsofcode.com/<service>/<user>/<repo>)](https://hitsofcode.com/view/<service>/<user>/<repo>)
+[![Hits-of-Code](https://@domain/<service>/<user>/<repo>)](https://@domain/view/<service>/<user>/<repo>)
@@ -32,7 +32,7 @@ following Markdown
-[![Hits-of-Code](https://hitsofcode.com/github/vbrandl/hoc)](https://hitsofcode.com/view/github/vbrandl/hoc)
+[![Hits-of-Code](https://@domain/github/vbrandl/hoc)](https://@domain/view/github/vbrandl/hoc)
@@ -40,7 +40,7 @@ would render this badge:
-