X-Git-Url: https://git.openstreetmap.org/potlatch2.git/blobdiff_plain/98b1ead1d7c6c50e938e2d70dd7bbb98d2465a04..HEAD:/net/systemeD/potlatch2/utils/SnapshotConnection.as?ds=sidebyside

diff --git a/net/systemeD/potlatch2/utils/SnapshotConnection.as b/net/systemeD/potlatch2/utils/SnapshotConnection.as
index 4910cf5f..a374a065 100644
--- a/net/systemeD/potlatch2/utils/SnapshotConnection.as
+++ b/net/systemeD/potlatch2/utils/SnapshotConnection.as
@@ -4,33 +4,74 @@ package net.systemeD.potlatch2.utils {
     import flash.events.Event;
     import flash.net.*;
 
+    /**
+    * A connection to a Snapshot server. A Snapshot server serves OSM map requests and can also
+    * track the "status" of an entity. Most other types of XMLConnection requests will fail. See
+    * http://www.github.com/gravitystorm/snapshot-server for example code based on the database
+    * structure created by osmosis pgsnapshot schema.
+    */
+
     public class SnapshotConnection extends XMLConnection {
 
         public function SnapshotConnection(cname:String,api:String,policy:String,initparams:Object=null) {
             super(cname,api,policy,initparams);
+            inlineStatus = true;
         }
 
-        /** Send a "complete" call to the server, and remove it from the current layer */
+        // As it stands, the following two functions could be refactored further.
+
+        /**
+        * Post a status update call to the server and update entity.status if successful.
+        */
         public function markComplete(entity:Entity):void {
+            var urlReq:URLRequest;
+
+            if (entity is Node) {
+                var node:Node = Node(entity);
+                if (node == getNode(node.id)) { // confirm it's from this connection
+                    makeRequest(entity, 'complete');
+                }
+
+            } else if (entity is Way) {
+                var way:Way = Way(entity);
+                if (way == getWay(way.id)) { // confirm it's from this connection
+                    makeRequest(entity, 'complete');
+                }
+            }
+        }
+
+        /**
+        * Send a "complete" call to the server and update entity.status if successful.
+        */
+        public function markNotComplete(entity:Entity):void {
+            var urlReq:URLRequest;
+
             if (entity is Node) {
-              var node:Node = Node(entity);
-              if (node == getNode(node.id)) { // confirm it's from this connection
-                  var urlReq:URLRequest = new URLRequest(apiBaseURL+"node/"+node.id+"/complete");
-                  urlReq.method = "POST";
-                  urlReq.data = '   ';
-                  urlReq.contentType = "application/xml";
-                  urlReq.requestHeaders = [ new URLRequestHeader("X_HTTP_METHOD_OVERRIDE", "PUT"),
-                                            new URLRequestHeader("X-Error-Format", "XML") ];
-                  var loader:URLLoader = new URLLoader();
-                  loader.addEventListener(Event.COMPLETE, function(e:Event):void { killNode(node.id) });
-                  loader.load(urlReq);
-              }
+                var node:Node = Node(entity);
+                if (node == getNode(node.id)) { // confirm it's from this connection
+                    makeRequest(entity, 'incomplete');
+                }
 
             } else if (entity is Way) {
-              var way:Way = Way(entity);
-              trace("not implemented");
+                var way:Way = Way(entity);
+                if (way == getWay(way.id)) { // confirm it's from this connection
+                    makeRequest(entity, 'incomplete');
+                }
             }
         }
 
+        private function makeRequest(entity:Entity, status:String):void {
+            var urlReq:URLRequest = new URLRequest(apiBaseURL+entity.getType()+"/"+entity.id+"/status");
+            urlReq.method = "POST";
+            urlReq.data = status;
+            var loader:URLLoader = new URLLoader();
+            loader.addEventListener(Event.COMPLETE, function(e:Event):void { updateStatus(entity, status) });
+            loader.load(urlReq);
+        }
+
+        private function updateStatus(e:Entity, s:String):void {
+            e.setStatus(s);
+        }
+
     }
 }
\ No newline at end of file