treefile: allow ${releasever} in more keys

Fixes #1809

Signed-off-by: Rafael Fonseca <[email protected]>

rust/Cargo.toml

@@ -23,6 +23,7 @@ c_utf8 = "0.1.0"
 systemd = "0.4.0"
 indicatif = "0.11.0"
 lazy_static = "1.1.0"
+envsubst = "0.1.0"
 
 [lib]
 name = "rpmostree_rust"

rust/src/treefile.rs

@@ -130,19 +130,6 @@ fn treefile_parse_stream<R: io::Read>(
         .into());
     }
 
-    // Substitute ${basearch}
-    treefile.treeref = match (basearch, treefile.treeref.take()) {
-        (Some(basearch), Some(treeref)) => {
-            let mut varsubsts = HashMap::new();
-            varsubsts.insert("basearch".to_string(), basearch.to_string());
-            Some(
-                utils::varsubst(&treeref, &varsubsts)
-                    .map_err(|e| io::Error::new(io::ErrorKind::InvalidInput, e.to_string()))?,
-            )
-        }
-        (_, v) => v,
-    };
-
     // Special handling for packages, since we allow whitespace within items.
     // We also canonicalize bootstrap_packages to packages here so it's
     // easier to append the basearch packages after.
@@ -410,7 +397,8 @@ impl Treefile {
         basearch: Option<&str>,
         workdir: openat::Dir,
     ) -> Fallible<Box<Treefile>> {
-        let parsed = treefile_parse_recurse(filename, basearch, 0)?;
+        let mut parsed = treefile_parse_recurse(filename, basearch, 0)?;
+        parsed.config = parsed.config.substitute_vars()?;
         Treefile::validate_config(&parsed.config)?;
         let dfd = openat::Dir::open(filename.parent().unwrap())?;
         let (rojig_name, rojig_spec) = if let Some(rojig) = parsed.config.rojig.as_ref() {
@@ -718,6 +706,39 @@ impl TreeComposeConfig {
 
         Ok(self)
     }
+
+    /// Look for use of ${variable} and replace it by its proper value
+    fn substitute_vars(mut self) -> Fallible<Self> {
+        let mut substvars: collections::HashMap<String, String> = collections::HashMap::new();
+        if let Some(arch) = &self.basearch {
+            substvars.insert("basearch".to_string(), arch.clone());
+        }
+        if let Some(releasever) = &self.releasever {
+            substvars.insert("releasever".to_string(), releasever.clone());
+        }
+        assert!(envsubst::validate_vars(&substvars).is_ok());
+
+        // Substitute ${basearch} and ${releasever}
+        macro_rules! substitute_var {
+            ( $field:ident ) => {{
+                if let Some(value) = self.$field.take() {
+                    self.$field = if envsubst::is_templated(&value) {
+                        match envsubst::substitute(value.clone(), &substvars) {
+                            Ok(s) => Some(s),
+                            Err(e) => return Err(e),
+                        }
+                    } else {
+                        Some(value)
+                    }
+                }
+            }};
+        };
+        substitute_var!(treeref);
+        substitute_var!(automatic_version_prefix);
+        substitute_var!(mutate_os_release);
+
+        Ok(self)
+    }
 }
 
 #[cfg(test)]
@@ -753,8 +774,9 @@ packages-s390x:
     #[test]
     fn basic_valid() {
         let mut input = io::BufReader::new(VALID_PRELUDE.as_bytes());
-        let treefile =
+        let mut treefile =
             treefile_parse_stream(InputFormat::YAML, &mut input, Some(ARCH_X86_64)).unwrap();
+        treefile = treefile.substitute_vars().unwrap();
         assert!(treefile.treeref.unwrap() == "exampleos/x86_64/blah");
         assert!(treefile.packages.unwrap().len() == 5);
     }
@@ -785,24 +807,28 @@ remove-files:
     #[test]
     fn basic_js_valid() {
         let mut input = io::BufReader::new(VALID_PRELUDE_JS.as_bytes());
-        let treefile =
+        let mut treefile =
             treefile_parse_stream(InputFormat::JSON, &mut input, Some(ARCH_X86_64)).unwrap();
+        treefile = treefile.substitute_vars().unwrap();
         assert!(treefile.treeref.unwrap() == "exampleos/x86_64/blah");
         assert!(treefile.packages.unwrap().len() == 5);
     }
 
     #[test]
     fn basic_valid_noarch() {
         let mut input = io::BufReader::new(VALID_PRELUDE.as_bytes());
-        let treefile = treefile_parse_stream(InputFormat::YAML, &mut input, None).unwrap();
+        let mut treefile = treefile_parse_stream(InputFormat::YAML, &mut input, None).unwrap();
+        treefile = treefile.substitute_vars().unwrap();
         assert!(treefile.treeref.unwrap() == "exampleos/x86_64/blah");
         assert!(treefile.packages.unwrap().len() == 3);
     }
 
     fn append_and_parse(append: &'static str) -> TreeComposeConfig {
         let buf = VALID_PRELUDE.to_string() + append;
         let mut input = io::BufReader::new(buf.as_bytes());
-        treefile_parse_stream(InputFormat::YAML, &mut input, Some(ARCH_X86_64)).unwrap()
+        let treefile =
+            treefile_parse_stream(InputFormat::YAML, &mut input, Some(ARCH_X86_64)).unwrap();
+        treefile.substitute_vars().unwrap()
     }
 
     fn test_invalid(data: &'static str) {
@@ -817,6 +843,31 @@ remove-files:
         }
     }
 
+    #[test]
+    fn basic_valid_releasever() {
+        let buf = r###"
+ref: "exampleos/${basearch}/${releasever}"
+releasever: 30
+automatic-version-prefix: ${releasever}
+mutate-os-release: ${releasever}
+"###;
+        let mut input = io::BufReader::new(buf.as_bytes());
+        let mut treefile =
+            treefile_parse_stream(InputFormat::YAML, &mut input, Some(ARCH_X86_64)).unwrap();
+        treefile = treefile.substitute_vars().unwrap();
+        assert!(treefile.treeref.unwrap() == "exampleos/x86_64/30");
+        assert!(treefile.releasever.unwrap() == "30");
+        assert!(treefile.automatic_version_prefix.unwrap() == "30");
+        assert!(treefile.mutate_os_release.unwrap() == "30");
+    }
+
+    #[test]
+    fn test_valid_no_releasever() {
+        let treefile = append_and_parse("automatic_version_prefix: ${releasever}");
+        assert!(treefile.releasever == None);
+        assert!(treefile.automatic_version_prefix.unwrap() == "${releasever}");
+    }
+
     #[test]
     fn basic_valid_legacy() {
         let treefile = append_and_parse(