初始化提交

WebDriverAgentLib/Vendor/RoutingHTTPServer/RouteResponse.m

@@ -0,0 +1,66 @@
+#import "RouteResponse.h"
+#import "HTTPConnection.h"
+#import "HTTPDataResponse.h"
+#import "HTTPResponseProxy.h"
+
+#pragma clang diagnostic ignored "-Wdirect-ivar-access"
+#pragma clang diagnostic ignored "-Widiomatic-parentheses"
+
+@implementation RouteResponse {
+  NSMutableDictionary *headers;
+  HTTPResponseProxy *proxy;
+}
+
+@synthesize connection;
+@synthesize headers;
+
+- (id)initWithConnection:(HTTPConnection *)theConnection {
+  if (self = [super init]) {
+    connection = theConnection;
+    headers = [[NSMutableDictionary alloc] init];
+    proxy = [[HTTPResponseProxy alloc] init];
+  }
+  return self;
+}
+
+- (NSObject <HTTPResponse>*)response {
+  return proxy.response;
+}
+
+- (void)setResponse:(NSObject <HTTPResponse>*)response {
+  proxy.response = response;
+}
+
+- (NSObject <HTTPResponse>*)proxiedResponse {
+  if (proxy.response != nil || proxy.customStatus != 0 || [headers count] > 0) {
+    return proxy;
+  }
+  
+  return nil;
+}
+
+- (NSInteger)statusCode {
+  return proxy.status;
+}
+
+- (void)setStatusCode:(NSInteger)status {
+  proxy.status = status;
+}
+
+- (void)setHeader:(NSString *)field value:(NSString *)value {
+  [headers setObject:value forKey:field];
+}
+
+- (void)respondWithString:(NSString *)string {
+  [self respondWithString:string encoding:NSUTF8StringEncoding];
+}
+
+- (void)respondWithString:(NSString *)string encoding:(NSStringEncoding)encoding {
+  [self respondWithData:[string dataUsingEncoding:encoding]];
+}
+
+- (void)respondWithData:(NSData *)data {
+  self.response = [[HTTPDataResponse alloc] initWithData:data];
+}
+
+@end