#import "TGMessageViewController.h"

#import <WatchCommonWatch/WatchCommonWatch.h>

#import "TGWatchCommon.h"
#import "TGBridgeSendMessageSignals.h"
#import "TGBridgeRemoteSignals.h"
#import "TGBridgeAudioSignals.h"

#import "TGBridgeUserCache.h"

#import "WKInterfaceTable+TGDataDrivenTable.h"

#import "TGInputController.h"

#import "TGUserRowController.h"
#import "TGMessageViewMessageRowController.h"
#import "TGMessageViewWebPageRowController.h"
#import "TGMessageViewFooterController.h"

#import "TGUserInfoController.h"
#import "TGNeoChatsController.h"

#import "TGExtensionDelegate.h"

NSString *const TGMessageViewControllerIdentifier = @"TGMessageViewController";

@implementation TGMessageViewControllerContext

- (instancetype)initWithMessage:(TGBridgeMessage *)message peerId:(int64_t)peerId
{
    self = [super init];
    if (self != nil)
    {
        _message = message;
        _peerId = peerId;
    }
    return self;
}

- (instancetype)initWithMessage:(TGBridgeMessage *)message channel:(TGBridgeChat *)channel
{
    self = [super init];
    if (self != nil)
    {
        _message = message;
        _peerId = channel.identifier;
        _channel = channel;
    }
    return self;
}

@end

@interface TGMessageViewController () <TGTableDataSource>
{
    TGMessageViewControllerContext *_context;
    
    SMetaDisposable *_sendMessageDisposable;
    SMetaDisposable *_remoteActionDisposable;
    SMetaDisposable *_playAudioDisposable;
    
    TGBridgeMediaAttachment *_pendingAudioAttachment;
}
@end

@implementation TGMessageViewController

- (instancetype)init
{
    self = [super init];
    if (self != nil)
    {
        _sendMessageDisposable = [[SMetaDisposable alloc] init];
        _remoteActionDisposable = [[SMetaDisposable alloc] init];
        _playAudioDisposable = [[SMetaDisposable alloc] init];
        
        [self.table _setInitialHidden:true];
        self.table.tableDataSource = self;
    }
    return self;
}

- (void)dealloc
{
    [_sendMessageDisposable dispose];
    [_remoteActionDisposable dispose];
    [_playAudioDisposable dispose];
}

- (void)configureWithContext:(TGMessageViewControllerContext *)context
{
    _context = context;
    
    [self configureHandoff];
    
    self.title = TGLocalized(@"Watch.MessageView.Title");

    __weak TGMessageViewController *weakSelf = self;
    [self performInterfaceUpdate:^(bool animated)
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        strongSelf.table.hidden = false;
        strongSelf.activityIndicator.hidden = true;
        [strongSelf.table reloadData];
    }];
}

- (void)configureHandoff
{
//    int64_t peerId = _context.peerId;
//    bool isGroup = _context.peerId < 0;
//    
//    if (isGroup)
//        peerId = -peerId;
//    
//    NSMutableDictionary *peerDict = [[NSMutableDictionary alloc] init];
//    peerDict[@"type"] = isGroup ? @"group" : @"user";
//    peerDict[@"id"] = @(peerId);
//    
//    NSMutableDictionary *messageDict = [[NSMutableDictionary alloc] init];
//    messageDict[@"autoplay"] = @false;
//    messageDict[@"id"] = @(_context.message.identifier);
//    
//    NSDictionary *userInfo = @{@"user_id": @(_context.authorizedContext.userId), @"peer": peerDict, @"message": messageDict};
//    [self updateUserActivity:@"org.telegram.message" userInfo:userInfo webpageURL:[NSURL URLWithString:@"https://telegram.org/dl"]];
}

- (void)willActivate
{
    [super willActivate];
    
    [self configureHandoff];
    
    [self.table notifyVisiblityChange];
}

- (void)didDeactivate
{
    [super didDeactivate];
}

#pragma mark -

- (Class)headerControllerClassForTable:(WKInterfaceTable *)table
{
    return [TGUserRowController class];
}

- (void)table:(WKInterfaceTable *)table updateHeaderController:(TGUserRowController *)controller
{
    if (_context.channel != nil)
    {
        [controller updateWithChannel:_context.channel context:_context.context];
    }
    else
    {
        TGBridgeUser *user = [[TGBridgeUserCache instance] userWithId:(int64_t)_context.message.fromUid];
        [controller updateWithUser:user context:_context.context];
    }
}

- (void)tableDidSelectHeader:(WKInterfaceTable *)table
{
    if (_context.channel != nil)
    {
        TGUserInfoControllerContext *context = [[TGUserInfoControllerContext alloc] initWithChannel:_context.channel];
        context.disallowCompose = true;
        [self pushControllerWithClass:[TGUserInfoController class] context:context];
    }
    else
    {
        TGUserInfoControllerContext *context = [[TGUserInfoControllerContext alloc] initWithUserId:(int64_t)_context.message.fromUid];
        [self pushControllerWithClass:[TGUserInfoController class] context:context];
    }
}

- (Class)footerControllerClassForTable:(WKInterfaceTable *)table
{
    return [TGMessageViewFooterController class];
}

- (void)table:(WKInterfaceTable *)table updateFooterController:(TGMessageViewFooterController *)controller
{
    __weak TGMessageViewController *weakSelf = self;
    controller.forwardPressed = ^
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        TGNeoChatsControllerContext *context = [[TGNeoChatsControllerContext alloc] init];
        context.context = strongSelf->_context.context;
        context.initialChats = [[TGExtensionDelegate instance] chatsController].chats;
        context.completionBlock = ^(TGBridgeChat *peer)
        {
            __strong TGMessageViewController *strongSelf = weakSelf;
            if (strongSelf == nil)
                return;
            
            [strongSelf->_sendMessageDisposable setDisposable:[[TGBridgeSendMessageSignals forwardMessageWithPeerId:strongSelf->_context.message.cid mid:strongSelf->_context.message.identifier targetPeerId:peer.identifier] startWithNext:^(TGBridgeMessage *message)
            {
                
            }]];
        };
        [strongSelf presentControllerWithClass:[TGNeoChatsController class] context:context];
    };
    controller.replyPressed = ^
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        [TGInputController presentInputControllerForInterfaceController:strongSelf suggestionsForText:strongSelf->_context.message.text completion:^(NSString *text)
        {
            [strongSelf->_sendMessageDisposable setDisposable:[[TGBridgeSendMessageSignals sendMessageWithPeerId:strongSelf->_context.peerId text:text replyToMid:strongSelf->_context.message.identifier] startWithNext:^(TGBridgeMessage *message)
            {
                
            }]];
        }];
    };
    // MARK: Swiftgram
    controller.replyWithoutPresetPressed = ^
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        [TGInputController presentPlainInputControllerForInterfaceController:strongSelf completion:^(NSString *text)
        {
            [strongSelf->_sendMessageDisposable setDisposable:[[TGBridgeSendMessageSignals sendMessageWithPeerId:strongSelf->_context.peerId text:text replyToMid:strongSelf->_context.message.identifier] startWithNext:^(TGBridgeMessage *message)
            {
                
            }]];
        }];
    };
    //
    controller.viewPressed = ^
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        [strongSelf->_remoteActionDisposable setDisposable:[[TGBridgeRemoteSignals openRemoteMessageWithPeerId:strongSelf->_context.peerId messageId:strongSelf->_context.message.identifier type:0 autoPlay:false] startWithNext:^(id next)
        {
                                                                
        }]];
    };
    
    [controller updateWithMessage:_context.message channel:(_context.channel != nil)];
}

- (NSUInteger)numberOfRowsInTable:(WKInterfaceTable *)table section:(NSUInteger)section
{
    return 1 + [self _messageHasWebPage];
}

- (Class)table:(WKInterfaceTable *)table rowControllerClassAtIndexPath:(TGIndexPath *)indexPath
{
    if (indexPath.row == 1)
        return [TGMessageViewWebPageRowController class];
    
    return [TGMessageViewMessageRowController class];
}

- (void)table:(WKInterfaceTable *)table updateRowController:(TGTableRowController *)rowController forIndexPath:(NSIndexPath *)indexPath
{
    __weak TGMessageViewController *weakSelf = self;
    rowController.isVisible = ^bool
    {
        __strong TGMessageViewController *strongSelf = weakSelf;
        if (strongSelf == nil)
            return false;
        
        return strongSelf.isVisible;
    };
    
    TGBridgeMessage *message = _context.message;
    if ([rowController isKindOfClass:[TGMessageViewMessageRowController class]])
    {
        TGMessageViewMessageRowController *controller = (TGMessageViewMessageRowController *)rowController;
        [controller updateWithMessage:message context:_context.context additionalPeers:_context.additionalPeers];
        
        void (^openUserInfo)(int64_t) = ^(int64_t peerId)
        {
            __strong TGMessageViewController *strongSelf = weakSelf;
            if (strongSelf == nil)
                return;
            
            if (peerId != 0)
            {
                TGUserInfoControllerContext *context = nil;
                if (TGPeerIdIsChannel(peerId))
                    context = [[TGUserInfoControllerContext alloc] initWithChannel:_context.additionalPeers[@(peerId)]];
                else
                    context = [[TGUserInfoControllerContext alloc] initWithUserId:(int64_t)peerId];
                
                [strongSelf pushControllerWithClass:[TGUserInfoController class] context:context];
            }
        };
        
        void (^openRemote)(void) = ^
        {
            __strong TGMessageViewController *strongSelf = weakSelf;
            if (strongSelf == nil)
                return;
            
            [strongSelf->_remoteActionDisposable setDisposable:[[TGBridgeRemoteSignals openRemoteMessageWithPeerId:strongSelf->_context.peerId messageId:message.identifier type:0 autoPlay:true] startWithNext:^(id next)
            {
                
            }]];
        };
        
        for (TGBridgeMediaAttachment *attachment in message.media)
        {
            if ([attachment isKindOfClass:[TGBridgeForwardedMessageMediaAttachment class]])
            {
                TGBridgeForwardedMessageMediaAttachment *forwardAttachment = (TGBridgeForwardedMessageMediaAttachment *)attachment;
                
                controller.forwardPressed = ^
                {
                    openUserInfo(forwardAttachment.peerId);
                };
            }
            else if ([attachment isKindOfClass:[TGBridgeContactMediaAttachment class]])
            {
                TGBridgeContactMediaAttachment *contactAttachment = (TGBridgeContactMediaAttachment *)attachment;
                
                controller.contactPressed = ^
                {
                    openUserInfo(contactAttachment.uid);
                };
            }
            else if ([attachment isKindOfClass:[TGBridgeVideoMediaAttachment class]])
            {
                controller.playPressed = ^
                {
                    openRemote();
                };
            }
            else if ([attachment isKindOfClass:[TGBridgeAudioMediaAttachment class]])
            {
                __weak TGMessageViewMessageRowController *weakMessageRow = controller;
                controller.playPressed = ^
                {
                    __strong TGMessageViewController *strongSelf = weakSelf;
                    if (strongSelf == nil)
                        return;
                    
                    TGBridgeMediaAttachment *audioAttachment = nil;
                    for (TGBridgeMediaAttachment *attachment in message.media)
                    {
                        if ([attachment isKindOfClass:[TGBridgeAudioMediaAttachment class]])
                        {
                            audioAttachment = (TGBridgeAudioMediaAttachment *)attachment;
                        }
                        else if ([attachment isKindOfClass:[TGBridgeDocumentMediaAttachment class]])
                        {
                            TGBridgeDocumentMediaAttachment *documentAttachment = (TGBridgeDocumentMediaAttachment *)attachment;
                            if (documentAttachment.isVoice)
                                audioAttachment = documentAttachment;
                        }
                    }
                    
                    if (audioAttachment != nil)
                    {
                        __strong TGMessageViewMessageRowController *strongConversationRow = weakMessageRow;
                        if ([strongSelf->_pendingAudioAttachment isEqual:audioAttachment])
                        {
                            if (strongConversationRow != nil)
                                [strongConversationRow setProcessingState:false];
                            
                            strongSelf->_pendingAudioAttachment = nil;
                            
                            [strongSelf->_playAudioDisposable setDisposable:nil];
                        }
                        else
                        {
                            if (strongConversationRow != nil)
                                [strongConversationRow setProcessingState:true];
                            
                            strongSelf->_pendingAudioAttachment = audioAttachment;
                            
                            [strongSelf->_playAudioDisposable setDisposable:[[[TGBridgeAudioSignals audioForAttachment:audioAttachment conversationId:message.cid messageId:message.identifier] deliverOn:[SQueue mainQueue]] startWithNext:^(NSURL *url)
                            {
                                if (url == nil)
                                    return;
                                
                                __strong TGMessageViewController *strongSelf = weakSelf;
                                if (strongSelf == nil)
                                    return;
                                
                                __strong TGMessageViewMessageRowController *strongConversationRow = weakMessageRow;
                                if (strongConversationRow != nil)
                                    [strongConversationRow setProcessingState:false];
                                
                                strongSelf->_pendingAudioAttachment = nil;
                                
                                [strongSelf presentMediaPlayerControllerWithURL:url options:@{ WKMediaPlayerControllerOptionsAutoplayKey: @true } completion:^(BOOL didPlayToEnd, NSTimeInterval endTime, NSError *error) {}];
                            }]];
                        }
                    }
                };
            }
        }
    }
    else if ([rowController isKindOfClass:[TGMessageViewWebPageRowController class]])
    {
        TGMessageViewWebPageRowController *controller = (TGMessageViewWebPageRowController *)rowController;
        
        TGBridgeWebPageMediaAttachment *pageAttachment = nil;
        for (TGBridgeMediaAttachment *attachment in _context.message.media)
        {
            if ([attachment isKindOfClass:[TGBridgeWebPageMediaAttachment class]])
            {
                pageAttachment = (TGBridgeWebPageMediaAttachment *)attachment;
                break;
            }
        }
        
        [controller updateWithAttachment:pageAttachment message:message];
    }
}

- (bool)_messageHasWebPage
{
    for (TGBridgeMediaAttachment *attachment in _context.message.media)
    {
        if ([attachment isKindOfClass:[TGBridgeWebPageMediaAttachment class]])
        {
            TGBridgeWebPageMediaAttachment *webAttachment = (TGBridgeWebPageMediaAttachment *)attachment;
            if (webAttachment.title.length == 0 && webAttachment.pageDescription.length == 0)
                return false;
            
            return true;
        }
    }
    
    return false;
}

+ (NSString *)identifier
{
    return TGMessageViewControllerIdentifier;
}

@end