feat : EndGamePrompt 리팩토링

AgainGameCommand.java

@@ -0,0 +1,9 @@
+public class AgainGameCommand implements ButtonCommand {
+    EndGamePrompt endGamePrompt;
+    public AgainGameCommand(EndGamePrompt endGamePrompt){
+        this.endGamePrompt = endGamePrompt;
+    }
+    public void execute(){
+        endGamePrompt.setResult(1);
+    }
+}

ButtonCommand.java

@@ -0,0 +1,3 @@
+public interface ButtonCommand {
+    public void execute();
+}

EndGameCommand.java

@@ -0,0 +1,9 @@
+public class EndGameCommand implements ButtonCommand {
+    EndGamePrompt endGamePrompt;
+    public EndGameCommand(EndGamePrompt endGamePrompt){
+        this.endGamePrompt = endGamePrompt;
+    }
+    public void execute(){
+        endGamePrompt.setResult(2);
+    }
+}

EndGamePrompt.java

@@ -9,14 +9,9 @@
 import java.awt.*;
 import java.awt.event.*;
 import javax.swing.*;
-import javax.swing.border.*;
-import javax.swing.event.*;
-
-import java.util.*;
-import java.text.*;
-
-public class EndGamePrompt implements ActionListener {
 
+public class EndGamePrompt {
+	private ButtonCommand command;
 	private JFrame win;
 	private JButton yesButton, noButton;
 
@@ -49,17 +44,19 @@ public EndGamePrompt( String partyName ) {
 		buttonPanel.setLayout(new GridLayout(1, 2));
 
 		Insets buttonMargin = new Insets(4, 4, 4, 4);
+
+		EndGamePromptClickEvent clickEvent = new EndGamePromptClickEvent(this);
 
 		yesButton = new JButton("Yes");
 		JPanel yesButtonPanel = new JPanel();
 		yesButtonPanel.setLayout(new FlowLayout());
-		yesButton.addActionListener(this);
+		yesButton.addActionListener(clickEvent);
 		yesButtonPanel.add(yesButton);
 
 		noButton = new JButton("No");
 		JPanel noButtonPanel = new JPanel();
 		noButtonPanel.setLayout(new FlowLayout());
-		noButton.addActionListener(this);
+		noButton.addActionListener(clickEvent);
 		noButtonPanel.add(noButton);
 
 		buttonPanel.add(yesButton);
@@ -82,16 +79,6 @@ public EndGamePrompt( String partyName ) {
 
 	}
 
-	public void actionPerformed(ActionEvent e) {
-		if (e.getSource().equals(yesButton)) {		
-			result=1;
-		}
-		if (e.getSource().equals(noButton)) {		
-			result=2;
-		}
-
-	}
-
 	public int getResult() {
 		while ( result == 0 ) {
 			try {
@@ -102,10 +89,25 @@ public int getResult() {
 		}
 		return result;	
 	}
-
+	public void setResult(int i) {
+		result = i;
+	}
 	public void distroy() {
 		win.hide();
 	}
-
+	public JButton getYesButton() {
+		return yesButton;
+	}
+	public JButton getNoButton() {
+		return noButton;
+	}
+
+	public void setCommand(ButtonCommand command) {
+        this.command = command;
+    }
+	public void buttonPressed() {
+        command.execute();
+    }
+
 }
 

EndGamePromptClickEvent.java

@@ -0,0 +1,18 @@
+import java.awt.event.ActionListener;
+import java.awt.event.ActionEvent;
+
+public class EndGamePromptClickEvent implements ActionListener {
+    private EndGamePrompt endGamePrompt;
+    EndGamePromptClickEvent(EndGamePrompt endGamePrompt){
+        this.endGamePrompt = endGamePrompt;
+    }
+    public void actionPerformed(ActionEvent e) {
+        if (e.getSource().equals(endGamePrompt.getYesButton())) {		
+            endGamePrompt.setCommand(new AgainGameCommand(endGamePrompt));
+        }
+        if (e.getSource().equals(endGamePrompt.getNoButton())) {		
+            endGamePrompt.setCommand(new EndGameCommand(endGamePrompt));
+        }
+        endGamePrompt.buttonPressed();
+    }
+}