/*
* Copyright (c) 2020, Jonathan Schleifer <js@nil.im>
*
* https://fossil.nil.im/objmatrix
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice is present in all copies.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#import "MTXRequest.h"
@implementation MTXRequest
{
OFData *_body;
mtx_request_block_t _block;
}
+ (instancetype)requestWithPath: (OFString *)path
accessToken: (OFString *)accessToken
homeserver: (OFURL *)homeserver
{
return [[[self alloc] initWithPath: path
accessToken: accessToken
homeserver: homeserver] autorelease];
}
- (instancetype)initWithPath: (OFString *)path
accessToken: (OFString *)accessToken
homeserver: (OFURL *)homeserver
{
self = [super init];
@try {
_accessToken = [accessToken copy];
_homeserver = [homeserver copy];
_path = [path copy];
_method = OF_HTTP_REQUEST_METHOD_GET;
} @catch (id e) {
[self release];
@throw e;
}
return self;
}
- (void)dealloc
{
[_accessToken release];
[_homeserver release];
[_path release];
[_body release];
[super dealloc];
}
- (void)setBody: (OFDictionary<OFString *, id> *)body
{
void *pool = objc_autoreleasePoolPush();
[_body release];
OFString *JSONString = [body JSONRepresentation];
_body = [[OFData alloc]
initWithItems: JSONString.UTF8String
count: JSONString.UTF8StringLength];
objc_autoreleasePoolPop(pool);
}
- (OFDictionary<OFString *, id> *)body
{
return [OFString stringWithUTF8String: _body.items
length: _body.count]
.objectByParsingJSON;
}
- (void)performWithBlock: (mtx_request_block_t)block
{
void *pool = objc_autoreleasePoolPush();
if (_block != nil)
/* Not the best exception to indicate it's already in-flight. */
@throw [OFAlreadyConnectedException exception];
OFMutableURL *requestURL = [[_homeserver mutableCopy] autorelease];
requestURL.path = _path;
requestURL.queryDictionary = _query;
OFMutableDictionary *headers = [OFMutableDictionary dictionary];
headers[@"User-Agent"] = @"ObjMatrix";
if (_accessToken != nil)
headers[@"Authorization"] =
[OFString stringWithFormat: @"Bearer %@", _accessToken];
if (_body != nil)
headers[@"Content-Length"] = @(_body.count).stringValue;
OFHTTPRequest *request = [OFHTTPRequest requestWithURL: requestURL];
request.method = _method;
request.headers = headers;
OFHTTPClient *client = [OFHTTPClient client];
client.delegate = self;
_block = [block copy];
[self retain];
[client asyncPerformRequest: request];
objc_autoreleasePoolPop(pool);
}
- (void)client: (OFHTTPClient *)client
didPerformRequest: (OFHTTPRequest *)request
response: (OFHTTPResponse *)response
{
/* Reset to nil first, so that another one can be performed. */
mtx_request_block_t block = _block;
_block = nil;
@try {
OFMutableData *responseData = [OFMutableData data];
while (!response.atEndOfStream) {
char buffer[512];
size_t length = [response readIntoBuffer: buffer
length: 512];
[responseData addItems: buffer
count: length];
}
mtx_response_t responseJSON = [OFString
stringWithUTF8String: responseData.items
length: responseData.count]
.objectByParsingJSON;
block(responseJSON, response.statusCode, nil);
} @catch (id e) {
block(nil, response.statusCode, e);
}
[block release];
[self release];
}
- (void)client: (OFHTTPClient *)client
didFailWithException: (id)exception
request: (OFHTTPRequest *)request
{
/*
* Convert OFHTTPRequestFailedException into a response, so that we
* still get the JSON for the failed request.
*/
if ([exception isKindOfClass: OFHTTPRequestFailedException.class]) {
[self client: client
didPerformRequest: request
response: [exception response]];
return;
}
/* Reset to nil first, so that another one can be performed. */
mtx_request_block_t block = _block;
_block = nil;
block(nil, 0, exception);
[block release];
[self release];
}
- (void)client: (OFHTTPClient *)client
wantsRequestBody: (OFStream *)body
request: (OFHTTPRequest *)request
{
[body writeData: _body];
}
@end