feat: add get_resolved_docker_registry

src/main.rs

@@ -7,17 +7,18 @@ mod docker_util;
 
 use std::fs;
 use std::env;
-use std::path::Path;
+use std::path::{Path, PathBuf};
 use rust_util::util_file;
 pub use docker_util::DockerCmd;
 
-const DOCKER_REGISTRY: &str = "~/.dockerbuild/register/rust_1_47";
 const MIRROR_USTC: &str = "git://mirrors.ustc.edu.cn/crates.io-index";
 
 fn main() {
     information!("dockerbuild v0.1");
 
-    let docker_registry_path_buf = util_file::get_absolute_path(DOCKER_REGISTRY).expect("Cannot find $HOME !");
+    let docker_image = "rust:1.47";
+
+    let docker_registry_path_buf =get_resolved_docker_registry(docker_image).expect("Cannot find $HOME !");
     let docker_registry = docker_registry_path_buf.to_str().expect("Cannot find $HOME !!");
 
     let p = Path::new(&docker_registry);
@@ -30,9 +31,28 @@ fn main() {
 
     let args_iter = env::args().skip(1);
     let args = args_iter.map(|a| a.to_owned()).collect::<Vec<_>>();
-    let mut docker_cmd = DockerCmd::new("rust:1.47");
+    let mut docker_cmd = DockerCmd::new(docker_image);
     docker_cmd.add_volumn(docker_registry, "/usr/local/cargo/registry")
               .mirror(MIRROR_USTC);
     docker_cmd.exec(&args).unwrap();
 }
 
+fn get_resolved_docker_registry(image: &str) -> Option<PathBuf> {
+    util_file::get_absolute_path(&get_docker_registry(image))
+}
+
+fn get_docker_registry(image: &str) -> String {
+    let mut registry = "~/.dockerbuild/register/".to_owned();
+    for c in image.chars() {
+        match c {
+            'a'..='z' | 'A'..='Z' | '0'..='9' => registry.push(c),
+            _ => registry.push('_'),
+        }
+    }
+    registry
+}
+
+#[test]
+fn test_get_docker_registry() {
+    assert_eq!("~/.dockerbuild/register/rust_1_47", get_docker_registry("rust:1.47"));
+}