/*!
 * iOS SDK
 *
 * Tencent is pleased to support the open source community by making
 * Hippy available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company.
 * All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import "HippyDevInfo.h"
#import <UIKit/UIDevice.h>
#import "HippyBundleURLProvider.h"

NSString *const HippyDevWebSocketSchemeWs = @"ws";
NSString *const HippyDevWebSocketSchemeWss = @"wss";
NSString *const HippyDevWebSocketInfoDebugURL = @"debugUrl=";

static NSString *const kHippyClientRoleForJSC = @"ios_client";
static NSString *const kHippyClientRoleForHermes = @"android_client"; // temporarily uses android as its identity

@implementation HippyDevInfo

- (void)setScheme:(NSString *)scheme {
    _scheme = HippyDevWebSocketSchemeWs;
    if ([scheme hasPrefix:@"https"]) {
        _scheme = HippyDevWebSocketSchemeWss;
    }
}

- (void)parseWsURLWithURLQuery:(NSString *)query {
    if (query.length <= 0) {
        return;
    }
    NSArray<NSString *> *queryItems = [query componentsSeparatedByString:@"&"];
    if (queryItems.count <= 0) {
        return;
    }
    NSString *debugWsURL = @"";
    for (NSString *item in queryItems) {
        if ([item hasPrefix:HippyDevWebSocketInfoDebugURL]) {
            debugWsURL = [item stringByRemovingPercentEncoding];
            break;
        }

    }
    if (debugWsURL.length <= 0) {
        return;
    }
    NSRange range = [debugWsURL rangeOfString:HippyDevWebSocketInfoDebugURL];
    if (range.location + range.length > debugWsURL.length) {
        return;
    }
    _wsURL = [debugWsURL substringFromIndex:range.location + range.length];
}

- (NSString *)assembleFullWSURLWithClientId:(NSString *)clientId 
                                contextName:(NSString *)contextName
                             isHermesEngine:(BOOL)usingHermes {
  if (self.port.length <= 0) {
    self.port = [self.scheme isEqualToString:HippyDevWebSocketSchemeWs] ? @"80" : @"443";
  }
  NSCharacterSet *allowedChar = [[NSCharacterSet characterSetWithCharactersInString:@"?!@#$^&%*+,:;='\"`<>()[]{}/\\| "] invertedSet];
  NSString *deviceName = [[UIDevice currentDevice] name];
  NSString *encodedDeviceName = [deviceName stringByAddingPercentEncodingWithAllowedCharacters:allowedChar];
  NSString *addressPrefix = [NSString stringWithFormat:@"%@://%@:%@/debugger-proxy", self.scheme, self.ipAddress, self.port];
  if (self.wsURL.length > 0) {
    // wsURL has a high priority
    addressPrefix = self.wsURL;
  }
  if ([addressPrefix containsString:@"?"]) {
    addressPrefix = [NSString stringWithFormat:@"%@&", addressPrefix];
  } else {
    addressPrefix = [NSString stringWithFormat:@"%@?", addressPrefix];
  }
  NSString *devAddress = [NSString stringWithFormat:@"%@clientId=%@&platform=1&role=%@&deviceName=%@",
                          addressPrefix, clientId, (usingHermes ? kHippyClientRoleForHermes : kHippyClientRoleForJSC), encodedDeviceName];
  if (self.versionId.length > 0) {
    devAddress = [NSString stringWithFormat:@"%@&hash=%@", devAddress, self.versionId];
  }
  contextName = [NSString stringWithFormat:@"HippyContext: %@", contextName];
  NSString *encodedContextName = [contextName stringByAddingPercentEncodingWithAllowedCharacters:allowedChar];
  devAddress = [NSString stringWithFormat:@"%@&contextName=%@", devAddress, encodedContextName];
  return devAddress;
}

@end