6 files changed, 276 insertions, 0 deletions

@@ -0,0 +1,8 @@
+package com.fourisland.instadisc.DownloadItem;
+
+public class DeinitalizeModeThread implements Runnable
+{
+    public void run() {
+        ModeControl.INSTANCE.modeDeinitalize();
+    }
+}
\ No newline at end of file
@@ -0,0 +1,14 @@
+package com.fourisland.instadisc.DownloadItem;
+
+import java.util.Timer;
+
+public interface DownloadItemMode
+{
+    public void modeInitalize();
+    public void modeDeinitalize();
+    
+    public void requestRetained();
+    
+    public int setTimer();
+    public void timerTick();
+}
\ No newline at end of file
@@ -0,0 +1,45 @@
+package com.fourisland.instadisc.DownloadItem;
+
+import java.util.Timer;
+
+public class ModeControl implements DownloadItemMode
+{
+    public static final ModeControl INSTANCE = new ModeControl();
+    private DownloadItemMode dim;
+    
+    public void initalize(String dim) throws UnknownDownloadItemModeException
+    {
+        if (dim.equals("Push"))
+        {
+            this.dim = new PushMode();
+        } else if (dim.equals("Pull"))
+        {
+            this.dim = new PullMode();
+        } else {
+            throw new UnknownDownloadItemModeException();
+        }
+        
+        Runtime.getRuntime().addShutdownHook(new Thread(new DeinitalizeModeThread()));
+    }
+
+    public void modeInitalize() {
+        dim.modeInitalize();
+    }
+
+    public void modeDeinitalize() {
+        dim.modeDeinitalize();
+    }
+    
+    public void requestRetained()
+    {
+        dim.requestRetained();
+    }
+
+    public int setTimer() {
+        return dim.setTimer();
+    }
+
+    public void timerTick() {
+        dim.timerTick();
+    }
+}
\ No newline at end of file
@@ -0,0 +1,24 @@
+package com.fourisland.instadisc.DownloadItem;
+
+public class PullMode implements DownloadItemMode
+{
+    public void modeInitalize() {
+        throw new UnsupportedOperationException("Not supported yet.");
+    }
+
+    public void modeDeinitalize() {
+        throw new UnsupportedOperationException("Not supported yet.");
+    }
+
+    public void requestRetained() {
+        throw new UnsupportedOperationException("Not supported yet.");
+    }
+
+    public int setTimer() {
+        throw new UnsupportedOperationException("Not supported yet.");
+    }
+
+    public void timerTick() {
+        throw new UnsupportedOperationException("Not supported yet.");
+    }
+}
\ No newline at end of file
@@ -0,0 +1,180 @@
+package com.fourisland.instadisc.DownloadItem;
+
+import com.fourisland.instadisc.Database.Wrapper;
+import com.fourisland.instadisc.Item.Item;
+import com.fourisland.instadisc.XmlRpc;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.net.SocketException;
+import java.util.HashMap;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+
+public class PushMode implements DownloadItemMode
+{
+    InstaDiscThread idt;
+    
+    public void modeInitalize()
+    {
+        idt = new InstaDiscThread();
+        new Thread(idt).start();
+    }
+    
+    public void modeDeinitalize()
+    {
+        idt.kill();
+    }
+
+    public void requestRetained() {
+        XmlRpc xmlrpc = new XmlRpc("requestRetained");
+        xmlrpc.execute();
+    }
+
+    public int setTimer() {
+        int delay = (1000 * 60 * 60);
+        if (Wrapper.getConfig("ipCheckUnit").equals("day")) {
+            delay *= (24 * Integer.decode(Wrapper.getConfig("ipCheckValue")));
+        } else if (Wrapper.getConfig("ipCheckUnit").equals("hour")) {
+            delay *= Integer.decode(Wrapper.getConfig("ipCheckValue"));
+        }
+        
+        return delay;
+    }
+
+    public void timerTick() {
+        XmlRpc xmlrpc = new XmlRpc("checkRegistration");
+        xmlrpc.execute();
+    }
+}
+class InstaDiscThread implements Runnable {
+
+    boolean cancelled = false;
+    ServerSocket svr;
+
+    public void cancel() {
+        cancelled = true;
+    }
+
+    public void run() {
+        try
+        {
+            svr = new ServerSocket();
+            java.net.InetSocketAddress addr = new java.net.InetSocketAddress(1204);
+            svr.bind(addr);
+            while (!cancelled)
+            {
+                try
+                {
+                    Socket s = svr.accept();
+                    HandleItemThread hit = new HandleItemThread(s);
+                    Thread hitt = new Thread(hit);
+                    hitt.start();
+                } catch (SocketException ex)
+                {
+                    cancel();
+                } catch (Exception ex)
+                {
+                    cancel();
+                    Logger.getLogger(InstaDiscThread.class.getName()).log(Level.SEVERE, null, ex);
+                }
+            }
+            svr.close();
+        } catch (IOException ex)
+        {
+            Logger.getLogger(InstaDiscThread.class.getName()).log(Level.SEVERE, null, ex);
+        }
+    }
+    
+    public void kill()
+    {
+        try
+        {
+            svr.close();
+        } catch (IOException ex)
+        {
+            Logger.getLogger(InstaDiscThread.class.getName()).log(Level.SEVERE, null, ex);
+        }
+    }
+}
+
+class HandleItemThread implements Runnable {
+
+    Socket s;
+
+    public HandleItemThread(Socket s) {
+        this.s = s;
+    }
+
+    public void run() {
+        try
+        {
+            InputStream is = s.getInputStream();
+            int buffer[] = new int[1000];
+            int rs = 0;
+            int i = 0;
+
+            while (rs != -1)
+            {
+                try
+                {
+                    rs = is.read();
+
+                    if (rs != -1)
+                    {
+                        buffer[i] = rs;
+                    }
+                    
+                    i++;
+                } catch (SocketException ex)
+                {
+                    return;
+                } catch (IOException ex)
+                {
+                    Logger.getLogger(HandleItemThread.class.getName()).log(Level.SEVERE, null, ex);
+                }
+            }
+
+            StringBuilder result = new StringBuilder();
+            int j = 0;
+            for (j = 0; j < i; j++)
+            {
+                result.append(Character.toString((char) buffer[j]));
+            }
+
+            String[] headers = result.toString().split("\n");
+            HashMap<String, String> headerMap = new HashMap<String, String>();
+            i = 0;
+            while (1 == 1)
+            {
+                try
+                {
+                    String[] nameVal = headers[i].split(": ");
+                    String name = nameVal[0];
+                    String value = nameVal[1].trim().replace("__INSTADISC__", ": ");
+                    headerMap.put(name, value);
+                } catch (Exception ex)
+                {
+                    break;
+                }
+                i++;
+            }
+
+            try
+            {
+                s.close();
+            } catch (IOException ex)
+            {
+                Logger.getLogger(HandleItemThread.class.getName()).log(Level.SEVERE, null, ex);
+            }
+
+            Item idI = new Item(headerMap);
+            idI.start();
+        } catch (IOException ex)
+        {
+            Logger.getLogger(HandleItemThread.class.getName()).log(Level.SEVERE, null, ex);
+        }
+    }
+}
@@ -0,0 +1,5 @@
+package com.fourisland.instadisc.DownloadItem;
+
+public class UnknownDownloadItemModeException extends Exception
+{
+}
\ No newline at end of file