Added a lesson restart call to allow lesson specific restart functionality

webgoat-container/src/main/java/org/owasp/webgoat/lessons/AbstractLesson.java

@@ -201,6 +201,11 @@ public abstract class AbstractLesson extends Screen implements Comparable<Object
 
     protected abstract Category getDefaultCategory();
 
+    /**
+     * Initiates lesson restart functionality
+     */
+    public abstract void restartLesson();
+
     protected abstract boolean getDefaultHidden();
 
     /**

webgoat-container/src/main/java/org/owasp/webgoat/lessons/LessonAdapter.java

@@ -108,6 +108,14 @@ public abstract class LessonAdapter extends AbstractLesson {
         return false;
     }
 
+    /**
+     * Initiates lesson restart functionality. Lessons should override this for 
+     * lesson specific actions 
+     */
+    public void restartLesson() {
+        // Do Nothing - called when restart lesson is pressed. Each lesson can do something
+    }
+    
     private final static Integer DEFAULT_RANKING = new Integer(1000);
 
     protected Integer getDefaultRanking() {

webgoat-container/src/main/java/org/owasp/webgoat/session/WebSession.java

@@ -949,6 +949,7 @@ public class WebSession {
     public void restartLesson(int lessonId) {
         AbstractLesson al = getLesson(lessonId);
         System.out.println("Restarting lesson: " + al);
+        al.restartLesson();
         al.getLessonTracker(this).setCompleted(false);
         if (al instanceof SequentialLessonAdapter) {
             SequentialLessonAdapter sla = (SequentialLessonAdapter) al;