From 26ab0818754fa3e62d61f9e6889e2c0a645769ff Mon Sep 17 00:00:00 2001
From: Gabriel Simmer <gsimmer@protonmail.com>
Date: Mon, 12 Dec 2022 18:46:49 +0000
Subject: [PATCH] Environment variable controlled container registry

---
 src/main.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 912ac0d..e861939 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,6 @@
 mod css;
 
-use std::{boxed, error};
+use std::{boxed, error, env};
 use dkregistry::v2::Client;
 use maud::{html, Markup, DOCTYPE};
 use axum::{Router, routing::get, response::IntoResponse};
@@ -57,10 +57,10 @@ fn footer() -> Markup {
 async fn root() -> Markup {
 	let c = get_images().await.unwrap_or_else(|e| panic!("{}", e));
     html! {
-		(header("/"))
+		(header("/DREF"))
 			body {
 				main {
-					h1 { "DREF" }
+					h1 { "/DREF" }
 					hr;
 					@for image in &c {
 						div {
@@ -81,10 +81,10 @@ async fn root() -> Markup {
 }
 
 async fn get_images() -> Result<Vec<Image>, boxed::Box<dyn error::Error>> {
-	let host = "icr.gmem.ca";
+	let host = env::var("DREF_REGISTRY").unwrap();
 	let dclient = Client::configure()
         .insecure_registry(false)
-        .registry(host)
+        .registry(&host)
         .build()?;
 	let catalog = dclient.get_catalog(None).collect::<Vec<_>>().await
 		.into_iter()
@@ -96,10 +96,10 @@ async fn get_images() -> Result<Vec<Image>, boxed::Box<dyn error::Error>> {
 }
 
 async fn get_tags(image: String) -> Result<Vec<Tag>, boxed::Box<dyn error::Error>> {
-	let host = "icr.gmem.ca";
+	let host = env::var("DREF_REGISTRY").unwrap();
 	let dclient = Client::configure()
         .insecure_registry(false)
-        .registry(host)
+        .registry(&host)
         .build()?;
 	let tags = dclient.get_tags(image.as_str(), None).collect::<Vec<_>>().await
 		.into_iter()