small fix for k8senv

OWASP · Nov 26, 2021 · 506dee0 · 506dee0
1 parent 7a1c43e
commit 506dee0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/java/org/owasp/wrongsecrets/ChallengesController.java b/src/main/java/org/owasp/wrongsecrets/ChallengesController.java
@@ -19,11 +19,13 @@ public class ChallengesController {
     private final String version;
     private final ScoreCard scoreCard;
     private final List<Challenge> challenges;
+    private final String k8sEnvironment;
 
-    public ChallengesController(@Value("${APP_VERSION}") String version, ScoreCard scoreCard, List<Challenge> challenges) {
+    public ChallengesController(@Value("${APP_VERSION}") String version, ScoreCard scoreCard, List<Challenge> challenges, @Value("${K8S_ENV}") String k8sEnvironment) {
         this.version = version;
         this.scoreCard = scoreCard;
         this.challenges = challenges;
+        this.k8sEnvironment = k8sEnvironment;
     }
 
     /**
@@ -58,7 +60,7 @@ public String challenge(Model model, @PathVariable String id) {
         model.addAttribute("answerIncorrect", null);
         model.addAttribute("solution", null);
         model.addAttribute("challengeNumber", challengeNumber(challenge));
-
+        model.addAttribute("k8sEnvironment", k8sEnvironment);
         includeScoringStatus(model, challenge);
         addWarning(challenge, model);