Added logic to inject country prefix if it's missing (eu-digital-gree…

…n-certificates#49)
ralexey · Jul 19, 2021 · eb92318 · eb92318
1 parent 864de48
commit eb92318
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/...rc/main/java/dgca/verifier/app/decoder/prefixvalidation/DefaultPrefixValidationService.kt b/...rc/main/java/dgca/verifier/app/decoder/prefixvalidation/DefaultPrefixValidationService.kt
@@ -33,4 +33,9 @@ class DefaultPrefixValidationService(private val prefix: String = "HC1:") : Pref
         input.startsWith(prefix) -> input.drop(prefix.length).also { verificationResult.contextPrefix = prefix }
         else -> input.also { verificationResult.contextPrefix = null }
     }
+
+    override fun encode(input: String): String = when {
+        input.startsWith(prefix) -> input
+        else -> "$prefix$input"
+    }
 }
diff --git a/decoder/src/main/java/dgca/verifier/app/decoder/prefixvalidation/PrefixValidationService.kt b/decoder/src/main/java/dgca/verifier/app/decoder/prefixvalidation/PrefixValidationService.kt
@@ -30,4 +30,6 @@ import dgca.verifier.app.decoder.model.VerificationResult
 interface PrefixValidationService {
 
     fun decode(input: String, verificationResult: VerificationResult): String
+
+    fun encode(input: String): String
 }