145 lines
4 KiB
Objective-C
145 lines
4 KiB
Objective-C
#import "EVEFrameStreamClient.h"
|
|
|
|
@interface EVEFrameStreamClient ()
|
|
|
|
@property(nonatomic, strong) NSURL *url;
|
|
@property(nonatomic, weak) id<EVEFrameStreamClientDelegate> delegate;
|
|
@property(nonatomic, strong) NSURLSession *session;
|
|
@property(nonatomic, strong) NSURLSessionWebSocketTask *task;
|
|
@property(nonatomic, assign) BOOL shouldReconnect;
|
|
|
|
@end
|
|
|
|
@implementation EVEFrameStreamClient
|
|
|
|
- (instancetype)initWithURL:(NSURL *)url delegate:(id<EVEFrameStreamClientDelegate>)delegate {
|
|
self = [super init];
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
self.url = url;
|
|
self.delegate = delegate;
|
|
self.session = [NSURLSession sessionWithConfiguration:NSURLSessionConfiguration.defaultSessionConfiguration];
|
|
self.shouldReconnect = YES;
|
|
return self;
|
|
}
|
|
|
|
- (void)connect {
|
|
self.shouldReconnect = YES;
|
|
[self.task cancelWithCloseCode:NSURLSessionWebSocketCloseCodeNormalClosure reason:nil];
|
|
self.task = [self.session webSocketTaskWithURL:self.url];
|
|
[self.task resume];
|
|
[self publishStatus:@"connecting"];
|
|
[self receiveNextMessage];
|
|
}
|
|
|
|
- (void)disconnect {
|
|
self.shouldReconnect = NO;
|
|
[self.task cancelWithCloseCode:NSURLSessionWebSocketCloseCodeNormalClosure reason:nil];
|
|
self.task = nil;
|
|
}
|
|
|
|
- (void)sendPointerPhase:(NSString *)phase x:(CGFloat)x y:(CGFloat)y {
|
|
if (!self.task) {
|
|
return;
|
|
}
|
|
|
|
NSDictionary *payload = @{
|
|
@"type": @"pointer",
|
|
@"phase": phase,
|
|
@"x": @(x),
|
|
@"y": @(y),
|
|
};
|
|
NSData *json = [NSJSONSerialization dataWithJSONObject:payload options:0 error:nil];
|
|
if (!json) {
|
|
return;
|
|
}
|
|
|
|
NSString *text = [[NSString alloc] initWithData:json encoding:NSUTF8StringEncoding];
|
|
NSURLSessionWebSocketMessage *message = [[NSURLSessionWebSocketMessage alloc] initWithString:text];
|
|
[self.task sendMessage:message completionHandler:^(NSError *error) {
|
|
if (error) {
|
|
[self publishStatus:@"input send failed"];
|
|
}
|
|
}];
|
|
}
|
|
|
|
- (void)receiveNextMessage {
|
|
__weak typeof(self) weakSelf = self;
|
|
[self.task receiveMessageWithCompletionHandler:^(NSURLSessionWebSocketMessage *message, NSError *error) {
|
|
__strong typeof(weakSelf) self = weakSelf;
|
|
if (!self) {
|
|
return;
|
|
}
|
|
|
|
if (error) {
|
|
[self publishStatus:@"stream disconnected"];
|
|
[self reconnectSoon];
|
|
return;
|
|
}
|
|
|
|
if (message.type == NSURLSessionWebSocketMessageTypeData) {
|
|
UIImage *image = [UIImage imageWithData:message.data];
|
|
if (image) {
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
[self.delegate frameStreamClient:self didReceiveImage:image];
|
|
});
|
|
}
|
|
} else if (message.type == NSURLSessionWebSocketMessageTypeString) {
|
|
[self handleTextMessage:message.string];
|
|
}
|
|
|
|
[self receiveNextMessage];
|
|
}];
|
|
}
|
|
|
|
- (void)handleTextMessage:(NSString *)text {
|
|
NSData *data = [text dataUsingEncoding:NSUTF8StringEncoding];
|
|
if (!data) {
|
|
return;
|
|
}
|
|
|
|
NSDictionary *message = [NSJSONSerialization JSONObjectWithData:data options:0 error:nil];
|
|
if (![message isKindOfClass:NSDictionary.class]) {
|
|
return;
|
|
}
|
|
|
|
if ([message[@"type"] isEqualToString:@"config"]) {
|
|
CGFloat width = [message[@"width"] doubleValue];
|
|
CGFloat height = [message[@"height"] doubleValue];
|
|
CGFloat scale = [message[@"DeviceScaleFactor"] doubleValue];
|
|
if (scale <= 0) {
|
|
scale = [message[@"deviceScaleFactor"] doubleValue];
|
|
}
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
[self.delegate frameStreamClient:self didReceiveViewportWidth:width height:height scale:scale];
|
|
[self.delegate frameStreamClient:self didChangeStatus:@"stream live"];
|
|
});
|
|
}
|
|
}
|
|
|
|
- (void)reconnectSoon {
|
|
if (!self.shouldReconnect) {
|
|
return;
|
|
}
|
|
|
|
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(2.0 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
|
|
if (self.shouldReconnect) {
|
|
[self connect];
|
|
}
|
|
});
|
|
}
|
|
|
|
- (void)publishStatus:(NSString *)status {
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
[self.delegate frameStreamClient:self didChangeStatus:status];
|
|
});
|
|
}
|
|
|
|
- (void)dealloc {
|
|
[self disconnect];
|
|
[self.session invalidateAndCancel];
|
|
}
|
|
|
|
@end
|