问SnakeYAML scala中的YAML环境变量插值

EN

import java.io.{File, FileInputStream}
import scala.util.matching.Regex
import com.fasterxml.jackson.databind.{JsonNode, ObjectMapper}
import com.fasterxml.jackson.module.scala.DefaultScalaModule
import org.yaml.snakeyaml.{DumperOptions, LoaderOptions, Yaml}
import org.yaml.snakeyaml.constructor.{AbstractConstruct, Constructor}
import org.yaml.snakeyaml.error.MissingEnvironmentVariableException
import org.yaml.snakeyaml.nodes.Node
import org.yaml.snakeyaml.nodes.ScalarNode
import org.yaml.snakeyaml.nodes.Tag
import org.yaml.snakeyaml.representer.Representer

/**
 * Copyright (c) 2008, http://www.snakeyaml.org
 * Class dedicated for SnakeYAML Support for Environment Variables
 */


/**
 * Construct scalar for format ${VARIABLE} replacing the template with the value from the environment.
 *
 * @see <a href="https://bitbucket.org/asomov/snakeyaml/wiki/Variable%20substitution">Variable substitution</a>
 * @see <a href="https://docs.docker.com/compose/compose-file/#variable-substitution">Variable substitution</a>
 */

class EnvScalarConstructor() extends Constructor {
  val ENV_TAG = new Tag("!ENV")
  this.yamlConstructors.put(ENV_TAG, new ConstructEnv)

  val ENV_regex: Regex = "\\$\\{\\s*((?<name>\\w+)((?<separator>:?(-|\\?))(?<value>\\w+)?)?)\\s*\\}".r

  private class ConstructEnv extends AbstractConstruct {
    override def construct(node: Node) = {
      val matchValue = constructScalar(node.asInstanceOf[ScalarNode])
      val patternMatch = ENV_regex.findAllIn(matchValue)
      val eval = patternMatch.toString()
      val name = patternMatch.group(1)
      val value = patternMatch.group(2)
      val separator = null
      apply(name, separator, if (value != null) value else "", ENV_regex.replaceAllIn(matchValue, getEnv(name)))

    }
  }

  /**
   * Implement the logic for missing and unset variables
   *
   * @param name        - variable name in the template
   * @param separator   - separator in the template, can be :-, -, :?, ?
   * @param value       - default value or the error in the template
   * @param environment - the value from the environment for the provided variable
   * @return the value to apply in the template
   */
  def apply(name: String, separator: String, value: String, environment: String): String = {
    if (environment != null && !environment.isEmpty) return environment
    // variable is either unset or empty
    if (separator != null) { //there is a default value or error
      if (separator == "?") if (environment == null) throw new MissingEnvironmentVariableException("Missing mandatory variable " + name + ": " + value)
      if (separator == ":?") {
        if (environment == null) throw new MissingEnvironmentVariableException("Missing mandatory variable " + name + ": " + value)
        if (environment.isEmpty) throw new MissingEnvironmentVariableException("Empty mandatory variable " + name + ": " + value)
      }
      if (separator.startsWith(":")) if (environment == null || environment.isEmpty) return value
      else if (environment == null) return value
    }
    ""
  }

  /**
   * Get the value of the environment variable
   *
   * @param key - the name of the variable
   * @return value or null if not set
   */
  def getEnv(key: String) = sys.env.getOrElse(key, System.getProperty(key, s"UNKNOWN_ENV_VAR-$key"))
}

/**
 * Function that will be used to load the YAML file
 * @param yamlFilePath - String with YAML path to read
 * @return - FasterXML JsonNode
 */
def parseYaml(yamlFilePath: String): JsonNode = {
  val ios = new FileInputStream(new File(yamlFilePath))
  // Parsing the YAML file with SnakeYAML - since Jackson Parser does not have Anchors and reference support
  val loaderOptions = new LoaderOptions
  loaderOptions.setAllowDuplicateKeys(false)
  val yaml = new Yaml(
    new EnvScalarConstructor,
    new Representer,
    new DumperOptions,
    loaderOptions
  )
  val mapper = new ObjectMapper().registerModules(DefaultScalaModule)
  val yamlObj = yaml.loadAs(ios, classOf[Any])

  // Converting the YAML to Jackson YAML - since it has more flexibility
  val jsonString = mapper.writerWithDefaultPrettyPrinter().writeValueAsString(yamlObj)
  val jsonObj = mapper.readTree(jsonString)
  println(jsonString)

  jsonObj
}