Merge pull request #14 from lincode/master

增加 PUT
douban · Feb 7, 2013 · cdb55ef · cdb55ef
2 parents b40d0f5 + ea17bad
commit cdb55ef
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/DoubanAPIEngine/DoubanAPIEngine/Sources/Network/DOUService.h b/DoubanAPIEngine/DoubanAPIEngine/Sources/Network/DOUService.h
@@ -52,6 +52,11 @@
                  callback:(DOUReqBlock)block
    uploadProgressDelegate:(id<ASIProgressDelegate>)progressDelegate;
 
+
+- (DOUHttpRequest *)put:(DOUQuery *)query
+                postBody:(NSString *)body
+                callback:(DOUReqBlock)block;
+
 - (DOUHttpRequest *)delete:(DOUQuery *)query callback:(DOUReqBlock)block;
 
 #endif

diff --git a/DoubanAPIEngine/DoubanAPIEngine/Sources/Network/DOUService.m b/DoubanAPIEngine/DoubanAPIEngine/Sources/Network/DOUService.m
@@ -387,6 +387,41 @@ - (DOUHttpRequest *)post:(DOUQuery *)query
 }
 
 
+- (DOUHttpRequest *)put:(DOUQuery *)query postBody:(NSString *)body callback:(DOUReqBlock)block {
+  query.apiBaseUrlString = self.apiBaseUrlString;
+  __block DOUHttpRequest * req = [DOUHttpRequest requestWithQuery:query completionBlock:^{
+    if (block != NULL) {
+      block(req);
+    }
+  }];
+
+  [req setRequestMethod:@"PUT"];
+  [req addRequestHeader:@"Content-Type" value:@"application/x-www-form-urlencoded; charset=UTF-8"];
+
+  if (body && [body length] > 0) {
+
+    NSError *error = nil;
+    GDataXMLElement *element = [[[GDataXMLElement alloc] initWithXMLString:body error:&error] autorelease];
+    if (!error && element) {
+      // if body is XML, Content-Type must be application/atom+xml
+      [req addRequestHeader:@"Content-Type" value:@"application/atom+xml"];
+    }
+
+    NSData *objectData = [body dataUsingEncoding:NSUTF8StringEncoding];
+    NSString *length = [NSString stringWithFormat:@"%d", [objectData length]];
+    [req appendPostData:objectData];
+    [req addRequestHeader:@"Content-Length" value:length];
+  }
+  else {
+    [req addRequestHeader:@"Content-Length" value:@"0"];
+  }
+
+  [req setResponseEncoding:NSUTF8StringEncoding];
+  [self addRequest:req];
+  return req;
+}
+
+
 - (DOUHttpRequest *)delete:(DOUQuery *)query callback:(DOUReqBlock)block {
 
   query.apiBaseUrlString = self.apiBaseUrlString;