encode token into url

3 files changed, 11 insertions, 24 deletions

diff --git a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLConnection.java b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLConnection.java
index 5335f7b3..56eab8c8 100644
--- a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLConnection.java
+++ b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLConnection.java
@@ -1,6 +1,7 @@
 package kr.syeyoung.dungeonsguide.customurl;
 
 import kr.syeyoung.dungeonsguide.Authenticator;
+import kr.syeyoung.dungeonsguide.DungeonsGuideMain;
 import lombok.SneakyThrows;
 
 import javax.crypto.Cipher;
@@ -17,30 +18,24 @@ import java.net.URLConnection;
 import java.net.URLEncoder;
 
 public class DGURLConnection extends URLConnection {
-    private Authenticator authenticator;
-    protected DGURLConnection(URL url, Authenticator authenticator) {
+    private static final String DOMAIN = "http://localhost:8080/";
+
+    protected DGURLConnection(URL url) {
         super(url);
-        this.authenticator = authenticator;
-        try {
-            connect();
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
     }
 
-    private static final String DOMAIN = "http://localhost:8080/";
-
     @SneakyThrows
     @Override
     public void connect() throws IOException {
+        if (!(url.getPath().contains("kr/syeyoung") ||url.getPath().contains("roomdata"))) return;
         this.connected = true;
 
-        System.out.println("loading "+url.getPath());
+        System.out.println("loading "+url.getPath().substring(1));
 
-        HttpURLConnection huc = (HttpURLConnection) new URL(DOMAIN+"resource/resource?class="+ URLEncoder.encode(url.getPath())).openConnection();
+        HttpURLConnection huc = (HttpURLConnection) new URL(DOMAIN+"resource/resource?class="+ URLEncoder.encode(url.getPath().substring(1))).openConnection();
         huc.setRequestProperty("User-Agent", "DungeonsGuide/1.0");
         huc.setRequestProperty("Content-Type", "application/json");
-        huc.setRequestProperty("Authorization", authenticator.getToken());
+        huc.setRequestProperty("Authorization", url.getUserInfo());
         huc.setDoInput(true);
         huc.setDoOutput(true);
 
@@ -57,7 +52,7 @@ public class DGURLConnection extends URLConnection {
 
         Cipher cipher = Cipher.getInstance("RSA/ECB/PKCS1Padding");
         byte[] byteEncrypted = pubKey;
-        cipher.init(Cipher.DECRYPT_MODE, authenticator.getKeyPair().getPrivate());
+        cipher.init(Cipher.DECRYPT_MODE, DungeonsGuideMain.getDungeonsGuideMain().getAuthenticator().getKeyPair().getPrivate());
         byte[] bytePlain = cipher.doFinal(byteEncrypted);
 
         cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandler.java b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandler.java
index 0a8f53a9..727e6e78 100644
--- a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandler.java
+++ b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandler.java
@@ -8,12 +8,8 @@ import java.net.URLConnection;
 import java.net.URLStreamHandler;
 
 public class DGURLStreamHandler extends URLStreamHandler {
-    private Authenticator authenticator;
-    public DGURLStreamHandler(Authenticator authenticator) {
-        this.authenticator = authenticator;
-    }
     @Override
     protected URLConnection openConnection(URL url) throws IOException {
-        return new DGURLConnection(url, authenticator);
+        return new DGURLConnection(url);
     }
 }
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandlerFactory.java b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandlerFactory.java
index 15defa5e..a27bda87 100644
--- a/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandlerFactory.java
+++ b/src/main/java/kr/syeyoung/dungeonsguide/customurl/DGURLStreamHandlerFactory.java
@@ -6,14 +6,10 @@ import java.net.URLStreamHandler;
 import java.net.URLStreamHandlerFactory;
 
 public class DGURLStreamHandlerFactory implements URLStreamHandlerFactory {
-    private Authenticator authenticator;
-    public DGURLStreamHandlerFactory(Authenticator authenticator) {
-        this.authenticator = authenticator;
-    }
     @Override
     public URLStreamHandler createURLStreamHandler(String s) {
         if ("dungeonsguide".equals(s)) {
-            return new DGURLStreamHandler(authenticator);
+            return new DGURLStreamHandler();
         }
 
         return null;