Endpoints provided by lessons now work

webgoat-container/src/main/java/org/owasp/webgoat/LessonTemplateResolver.java

@@ -0,0 +1,60 @@
+package org.owasp.webgoat;
+
+import com.google.common.collect.Sets;
+import com.google.common.io.Files;
+import org.owasp.webgoat.session.WebSession;
+import org.thymeleaf.TemplateProcessingParameters;
+import org.thymeleaf.resourceresolver.IResourceResolver;
+import org.thymeleaf.templateresolver.TemplateResolver;
+
+import java.io.ByteArrayInputStream;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+
+/**
+ *
+ */
+public class LessonTemplateResolver extends TemplateResolver {
+
+
+    private final static String PREFIX = "lesson:";
+    private final File pluginTargetDirectory;
+    private final WebSession webSession;
+
+
+    public LessonTemplateResolver(File pluginTargetDirectory, WebSession webSession) {
+        this.pluginTargetDirectory = pluginTargetDirectory;
+        this.webSession = webSession;
+        setResourceResolver(new LessonResourceResolver());
+        setResolvablePatterns(Sets.newHashSet(PREFIX + "*"));
+    }
+
+    @Override
+    protected String computeResourceName(TemplateProcessingParameters params) {
+        String templateName = params.getTemplateName();
+        return templateName.substring(PREFIX.length());
+    }
+
+    private class LessonResourceResolver implements IResourceResolver {
+
+        @Override
+        public InputStream getResourceAsStream(TemplateProcessingParameters params, String resourceName) {
+            String lessonName = webSession.getCurrentLesson().getClass().getSimpleName();
+            File lesson = new File(pluginTargetDirectory, "/plugin/" + lessonName + "/html/" + lessonName + ".html");
+            if (lesson != null) {
+                try {
+                    return new ByteArrayInputStream(Files.toByteArray(lesson));
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+            }
+            return null;
+        }
+
+        @Override
+        public String getName() {
+            return "lessonResourceResolver";
+        }
+    }
+}