Changeset 306 for trunk/src/org/expeditee/network/FrameShare.java

Timestamp:

09/16/08 09:55:34 (16 years ago)

Author:

ra33

Message:

Added some HELP actions
More mail stuff... and networking stuff
Can now save frames and send messages to peers
Also version control prevent versions from being lost

File:

• trunk/src/org/expeditee/network/FrameShare.java (modified) (8 diffs)

trunk/src/org/expeditee/network/FrameShare.java

@@ -4 +4 @@
 import java.io.IOException;
 import java.io.StringReader;
+import java.io.StringWriter;
 import java.net.DatagramPacket;
 import java.net.DatagramSocket;
+import java.net.InetAddress;
 import java.net.UnknownHostException;
+import java.util.Collection;
 import java.util.HashMap;
+import java.util.LinkedList;
 import java.util.Map;
 
 import org.expeditee.gui.AttributeValuePair;
 import org.expeditee.gui.Frame;
+import org.expeditee.gui.FrameIO;
 import org.expeditee.gui.MessageBay;
 import org.expeditee.io.ExpReader;
+import org.expeditee.io.ExpWriter;
 import org.expeditee.io.FrameReader;
+import org.expeditee.io.FrameWriter;
 import org.expeditee.items.Text;
 
 public class FrameShare {
 
-        private static FrameShare _theSession = null;
-
-        private static FrameServer _server = null;
+        private static final int SOCKET_TIMEOUT = 1000;
+
+        private static Collection<DefaultServer> _servers = new LinkedList<DefaultServer>();
+
+        private static FrameShare _theSession;
 
         private Map<String, Peer> _peers;
+
+        private int _port = -1;
 
         public static void init(Frame settingsFrame) {
@@ -48 +59 @@
                                 try {
                                         if (!avp.hasPair()) {
-                                                _server = new FrameServer();
+                                                _servers.add(new FrameServer());
+                                                _servers.add(new FrameSaver());
+                                                _servers.add(new MessageReciever());
                                         } else {
-                                                _server = new FrameServer("FrameServer", avp
-                                                                .getIntegerValue());
+                                                _port = avp.getIntegerValue();
+                                                _servers.add(new FrameServer("FrameServer", _port));
+                                                _servers.add(new FrameSaver("FrameSaver", _port + 1));
+                                                _servers
+                                                                .add(new MessageReciever("Messages", _port + 2));
                                         }
                                 } catch (Exception e) {
                                         e.printStackTrace();
-                                        MessageBay.errorMessage("Could not start frame server ["
+                                        MessageBay.errorMessage("Could not start servers ["
                                                         + avp.toString() + "]");
                                 }
@@ -73 +89 @@
 
                 try {
-                        if (_server != null) {
-                                _server.start();
-                        }
+                        for (DefaultServer server : _servers)
+                                server.start();
                 } catch (Exception e) {
                         MessageBay.errorMessage("Error in PeerToPeer setup");
@@ -82 +97 @@
 
         public void finalise() {
-                _server.close();
+                for (DefaultServer server : _servers)
+                        server.close();
         }
 
@@ -97 +113 @@
         public Frame loadFrame(String frameName) {
                 byte[] nameBuf = frameName.getBytes();
-                byte[] buf = new byte[2056];
+                byte[] buf = new byte[FrameServer.MAX_PACKET_LENGTH];
 
                 String result = null;
@@ -104 +120 @@
                         // get a datagram socket
                         DatagramSocket socket = new DatagramSocket();
-                        socket.setSoTimeout(1000);
+                        socket.setSoTimeout(SOCKET_TIMEOUT);
 
                         for (Peer peer : _peers.values()) {
@@ -129 +145 @@
 
                 }
-                
-                if(result == null)
+
+                if (result == null)
                         return null;
-                
+
                 // Now read the frame from the file contents
                 FrameReader reader = new ExpReader(frameName);
@@ -155 +171 @@
                 return frame;
         }
+
+        public boolean sendMessage(String message, String peerName) {
+                Peer peer = _peers.get(peerName.toLowerCase());
+                
+                if(peer == null){
+                        return false;
+                }
+
+                try {
+                        // get a datagram socket
+                        DatagramSocket socket = new DatagramSocket(_port - 1);
+                        socket.setSoTimeout(SOCKET_TIMEOUT);
+
+                        // message = peerName + " says " + message;
+                        byte[] contentsBuf = message.getBytes();
+
+                        try {
+                                // send save request
+                                DatagramPacket packet = new DatagramPacket(contentsBuf,
+                                                contentsBuf.length, peer.getAddress(),
+                                                peer.getPort() + 2);
+                                socket.send(packet);
+                        } catch (Exception e) {
+                                e.printStackTrace();
+                        }
+                        socket.close();
+
+                } catch (IOException e) {
+                        e.printStackTrace();
+                }
+                return true;
+        }
+
+        public String saveFrame(Frame toSave) {
+                
+                FrameIO.setSavedProperties(toSave);
+                
+                Peer peer = _peers.get(toSave.getPath().toLowerCase());
+                String fileContents = "";
+                // Now read the frame from the file contents
+                FrameWriter writer = new ExpWriter();
+                try {
+                        // String tempName = new Date().toString() + ".exp";
+                        // writer.setOutputLocation(tempName);
+                        // Write out the file to a StringBuffer
+                        StringWriter sw = new StringWriter();
+                        // First write out the name of the frame
+                        sw.write(toSave.getName() + "\n");
+                        //Then the version
+                        sw.write(toSave.getVersion() + "\n");
+                        // Write out the rest of the frame
+                        writer.writeFrame(toSave, sw);
+                        // Now send the packet
+                        fileContents = sw.getBuffer().toString();
+                        byte[] contentsBuf = fileContents.getBytes();
+
+                        // get a datagram socket
+                        DatagramSocket socket = new DatagramSocket(_port - 2);
+                        socket.setSoTimeout(SOCKET_TIMEOUT);
+
+                        try {
+                                // send save request
+                                DatagramPacket packet = new DatagramPacket(contentsBuf,
+                                                contentsBuf.length, peer.getAddress(),
+                                                peer.getPort() + 1);
+                                socket.send(packet);
+                        } catch (Exception e) {
+                                e.printStackTrace();
+                        }
+                        socket.close();
+
+                } catch (IOException e) {
+                        e.printStackTrace();
+                }
+                toSave.setSaved();
+                return fileContents;
+        }
+
+        public String getPeerName(int port, InetAddress address) {
+                for (Peer p : _peers.values()) {
+                        if(p.getPort() == port && p.getAddress().equals(address))
+                                return p.getName();
+                }
+                return null;
+        }
 }