2 files changed, 12 insertions, 1 deletions

@@ -22,6 +22,14 @@ module Wittle
       @puzzle = Puzzle.find(params[:id])
       @playable = @puzzle.latest? && !((session[:played_puzzles] || []).include? @puzzle.id)
       @already_started = ((session[:started_puzzles] || []).include? @puzzle.id)
+
+      unless @playable
+        if (session[:puzzle_solutions] || {}).has_key? @puzzle.id.to_s
+          @solution = session[:puzzle_solutions][@puzzle.id.to_s]
+        else
+          @solution = @puzzle.solved_data
+        end
+      end
     end
 
     def start
@@ -45,6 +53,9 @@ module Wittle
 
       session[:played_puzzles] ||= []
       session[:played_puzzles] << @puzzle.id
+
+      session[:puzzle_solutions] ||= {}
+      session[:puzzle_solutions][@puzzle.id] = params[:solved]
     end
 
     def submit
@@ -51,7 +51,7 @@ window.onload = function() {
   <% else %>
     var puzzle = window.deserializePuzzle("<%= @puzzle.data %>")
     draw(puzzle)
-    drawPath(puzzle, JSON.parse("<%= escape_javascript(sanitize @puzzle.solved_data) %>"))
+    drawPath(puzzle, JSON.parse("<%= escape_javascript(sanitize @solution) %>"))
     window.trace = function() {}
   <% end %>
 }