#import <WSMTS/WSMTS.h>
#import "ViewController.h"

@interface ViewController ()<WSMTSDelegate>

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];

    // Initialize SDK
    BOOL result = [WSMTS initWithAppID:@"sdk.wekt.net" port:60000 appid:996];
    if (result) {
      	// Setup callback
      	[WSMTS setDelegate:delegate:self]
        // Enter room
        result = [WSMTS enterRoomWithID:1001 authority:@""];
        // Send barrage
        result = [WSMTS sendBarrageWithPriority:self.loginInfo.roomID message:messageData];
    }
}

#pragma - WSMTSDelegate

- (void)wsmtsConnectWithResult:(NSInteger)roomID result:(BOOL)result {
    NSString *msg = [NSString stringWithFormat:@"roomid: %ld connect result: %@", (long)roomID, result ? @"sucess" : @"failed"];
    [self displayMessage:msg];
    NSLog(@"%@", msg);
}

- (void)wsmtsDisconnectWithRoomID:(NSInteger)roomID {
    NSString *msg = [NSString stringWithFormat:@"disconnect roomid: %ld", (long)roomID];
    [self displayMessage:msg];
    NSLog(@"%@", msg);
}

- (void)wsmtsEnterRoom:(NSInteger)roomID withErrorCode:(NSInteger)errorCode {
    NSString *message;
    switch (errorCode) {
        case 0:
            message = @"sucess";
            break;
        case 1:
            message = @"get room network address failed";
            break;
        case 2:
            message = @"socket disconnect failed";
            break;
        case 3:
            message = @"authentication failed";
            break;
	case 4:
	    message = @"domain name parsing failed";
            break;
	case 5:
	    message = @"incorrect room ID";
            break;
        case 10:
            message = @"others error";
            break;
        default:
            message = @"unknown error";
            break;
    }
    NSString *msg = [NSString stringWithFormat:@"enter roomid: %ld, error code: %ld, description: %@", (long)roomID, (long)errorCode, message];
    [self displayMessage:msg];
    NSLog(@"%@", msg);

}

- (void)wsmtsReceiveBarrage:(NSInteger)roomID message:(NSData *)message {
    NSString *string = [[NSString alloc] initWithData:message encoding:NSUTF8StringEncoding];
    NSString *msg = [NSString stringWithFormat:@"room %ld receive barrage: %@", (long)roomID, string];
    [self displayMessage:msg];
    NSLog(@"%@", msg);
}

- (void)wsmtsReceivePrivateChatWithCID:(NSInteger)roomID cid:(NSString*)cid message:(NSData *)message {
    NSString *string = [[NSString alloc] initWithData:message encoding:NSUTF8StringEncoding];
    NSString *msg = [NSString stringWithFormat:@"room %ld receive private chat: %@, %@", (long)roomID, cid, string];
    [self displayMessage:msg];
    NSLog(@"%@", msg);
}

@end