The Game page now links back to the Menu page

scourgemancer · Aug 27, 2017 · 7746f93 · 7746f93
1 parent 3f98a96
commit 7746f93
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/gui/SudokuGUI.java b/src/gui/SudokuGUI.java
@@ -7,12 +7,9 @@
 import javafx.stage.Screen;
 import javafx.application.Application;
 import javafx.scene.image.Image;
-import javafx.scene.layout.*;
 import javafx.scene.text.Text;
 import javafx.stage.Stage;
 
-import java.io.File;
-import java.util.ArrayList;
 import java.util.Observable;
 import java.util.Observer;
 

diff --git a/src/gui/state/GameState.java b/src/gui/state/GameState.java
@@ -102,6 +102,10 @@ public void setPage(SudokuGUI gui){
 
     @Override
     public void nextPage(SudokuGUI gui, String name){
-
+        switch(name){
+            case "menu":                gui.setState(new MenuState());                break;
+            case "difficultyselection": gui.setState(new DifficultySelectionState()); break;
+            case "puzzleselection":     gui.setState(new PuzzleSelectionState());     break;
+        }
     }
 }