Files
2026-02-03 16:52:44 +08:00

67 lines
1.5 KiB
Objective-C

#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