feat(SPEL): new base64 encode and decode methods (#1339)

spinnaker · May 12, 2017 · 70e42a8 · 70e42a8
1 parent eaa5399
commit 70e42a8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/...src/main/groovy/com/netflix/spinnaker/orca/pipeline/util/ContextParameterProcessor.groovy b/...src/main/groovy/com/netflix/spinnaker/orca/pipeline/util/ContextParameterProcessor.groovy
@@ -40,7 +40,6 @@ import org.springframework.expression.spel.support.StandardEvaluationContext
 import org.springframework.expression.spel.support.StandardTypeLocator
 
 import java.lang.reflect.Method
-import java.text.DateFormat
 import java.text.SimpleDateFormat
 import java.util.concurrent.atomic.AtomicReference
 
@@ -153,6 +152,8 @@ class ContextParameterProcessor {
       evaluationContext.registerFunction('toInt', ContextUtilities.getDeclaredMethod("toInt", String))
       evaluationContext.registerFunction('toFloat', ContextUtilities.getDeclaredMethod("toFloat", String))
       evaluationContext.registerFunction('toBoolean', ContextUtilities.getDeclaredMethod("toBoolean", String))
+      evaluationContext.registerFunction('toBase64', ContextUtilities.getDeclaredMethod("toBase64", String))
+      evaluationContext.registerFunction('fromBase64', ContextUtilities.getDeclaredMethod("fromBase64", String))
 
       // only add methods that are context sensitive at stage evaluation time
       if (allowUnknownKeys) {
@@ -313,6 +314,14 @@ abstract class ContextUtilities {
     context.stages?.find { it.name == id && it.type == 'manualJudgment' }?.context?.judgmentInput
   }
 
+  static String toBase64(String text) {
+    Base64.getEncoder().encodeToString(text.getBytes())
+  }
+
+  static String fromBase64(String text) {
+    new String(Base64.getDecoder().decode(text), 'UTF-8')
+  }
+
 }
 
 class MapPropertyAccessor extends ReflectivePropertyAccessor {

diff --git a/...test/groovy/com/netflix/spinnaker/orca/pipeline/util/ContextParameterProcessorSpec.groovy b/...test/groovy/com/netflix/spinnaker/orca/pipeline/util/ContextParameterProcessorSpec.groovy
@@ -78,6 +78,12 @@ class ContextParameterProcessorSpec extends Specification {
     //successful case: 'http://captive.apple.com/' | 'this should work'
   }
 
+  def "base64 encode and decode"() {
+    expect:
+    contextParameterProcessor.process([test: '${#toBase64("Yo Dawg")}'], [:], true).test == 'WW8gRGF3Zw=='
+    contextParameterProcessor.process([test: '${#fromBase64("WW8gRGF3Zw==")}'], [:], true).test == 'Yo Dawg'
+  }
+
   @Unroll
   def "should not System.exit"() {
     when: