初始化提交

WebDriverAgentLib/Vendor/RoutingHTTPServer/HTTPResponseProxy.m

@@ -0,0 +1,84 @@
+#import "HTTPResponseProxy.h"
+
+#pragma clang diagnostic ignored "-Wdirect-ivar-access"
+
+@implementation HTTPResponseProxy
+
+@synthesize response;
+@synthesize status;
+
+- (NSInteger)status {
+  if (status != 0) {
+    return status;
+  } else if ([response respondsToSelector:@selector(status)]) {
+    return [response status];
+  }
+  
+  return 200;
+}
+
+- (void)setStatus:(NSInteger)statusCode {
+  status = statusCode;
+}
+
+- (NSInteger)customStatus {
+  return status;
+}
+
+// Implement the required HTTPResponse methods
+- (UInt64)contentLength {
+  if (response) {
+    return [response contentLength];
+  } else {
+    return 0;
+  }
+}
+
+- (UInt64)offset {
+  if (response) {
+    return [response offset];
+  } else {
+    return 0;
+  }
+}
+
+- (void)setOffset:(UInt64)offset {
+  if (response) {
+    [response setOffset:offset];
+  }
+}
+
+- (NSData *)readDataOfLength:(NSUInteger)length {
+  if (response) {
+    return [response readDataOfLength:length];
+  } else {
+    return nil;
+  }
+}
+
+- (BOOL)isDone {
+  if (response) {
+    return [response isDone];
+  } else {
+    return YES;
+  }
+}
+
+// Forward all other invocations to the actual response object
+- (void)forwardInvocation:(NSInvocation *)invocation {
+  if ([response respondsToSelector:[invocation selector]]) {
+    [invocation invokeWithTarget:response];
+  } else {
+    [super forwardInvocation:invocation];
+  }
+}
+
+- (BOOL)respondsToSelector:(SEL)selector {
+  if ([super respondsToSelector:selector])
+    return YES;
+  
+  return [response respondsToSelector:selector];
+}
+
+@end
+