firebase-ios-sdk/Firebase/Database/Core/FRepoInfo.m

/*
 * Copyright 2017 Google
 *
 * 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 "FRepoInfo.h"
#import "FConstants.h"

@interface FRepoInfo ()

@property (nonatomic, strong) NSString *domain;

@end


@implementation FRepoInfo

@synthesize namespace;
@synthesize host;
@synthesize internalHost;
@synthesize secure;
@synthesize domain;

- (id) initWithHost:(NSString*)aHost isSecure:(bool)isSecure withNamespace:(NSString*)aNamespace {
    self = [super init];
    if (self) {
        host = aHost;
        domain = [host substringFromIndex:[host rangeOfString:@"."].location+1];
        secure = isSecure;
        namespace = aNamespace;

        // Get cached internal host if it exists
        NSString* internalHostKey = [NSString stringWithFormat:@"firebase:host:%@", self.host];
        NSString* cachedInternalHost = [[NSUserDefaults standardUserDefaults] stringForKey:internalHostKey];
        if (cachedInternalHost != nil) {
            internalHost = cachedInternalHost;
        } else {
            internalHost = self.host;
        }
    }
    return self;
}

- (NSString *)description {
    // The namespace is encoded in the hostname, so we can just return this.
    return [NSString stringWithFormat:@"http%@://%@", (self.secure ? @"s" : @""), self.host];
}

- (void) setInternalHost:(NSString *)newHost {
    if (![internalHost isEqualToString:newHost]) {
        internalHost = newHost;

        // Cache the internal host so we don't need to redirect later on
        NSString* internalHostKey = [NSString stringWithFormat:@"firebase:host:%@", self.host];
        NSUserDefaults* cache = [NSUserDefaults standardUserDefaults];
        [cache setObject:internalHost forKey:internalHostKey];
        [cache synchronize];
    }
}

- (void) clearInternalHostCache {
    internalHost = self.host;

    // Remove the cached entry
    NSString* internalHostKey = [NSString stringWithFormat:@"firebase:host:%@", self.host];
    NSUserDefaults* cache = [NSUserDefaults standardUserDefaults];
    [cache removeObjectForKey:internalHostKey];
    [cache synchronize];
}

- (BOOL) isDemoHost {
    return [self.domain isEqualToString:@"firebaseio-demo.com"];
}

- (BOOL) isCustomHost {
    return ![self.domain isEqualToString:@"firebaseio-demo.com"] && ![self.domain isEqualToString:@"firebaseio.com"];
}


- (NSString *) connectionURL {
    return [self connectionURLWithLastSessionID:nil];
}

- (NSString *) connectionURLWithLastSessionID:(NSString*)lastSessionID {
    NSString *scheme;
    if (self.secure) {
        scheme = @"wss";
    } else {
        scheme = @"ws";
    }
    NSString *url = [NSString stringWithFormat:@"%@://%@/.ws?%@=%@&ns=%@",
                     scheme,
                     self.internalHost,
                     kWireProtocolVersionParam,
                     kWebsocketProtocolVersion,
                     self.namespace];

    if (lastSessionID != nil) {
        url = [NSString stringWithFormat:@"%@&ls=%@", url, lastSessionID];
    }
    return url;
}

- (id)copyWithZone:(NSZone *)zone; {
    return self; // Immutable
}

- (NSUInteger)hash {
    NSUInteger result = host.hash;
    result = 31 * result + (secure ? 1 : 0);
    result = 31 * result + namespace.hash;
    result = 31 * result + host.hash;
    return result;
}

- (BOOL)isEqual:(id)anObject {
    if (![anObject isKindOfClass:[FRepoInfo class]]) return NO;
    FRepoInfo *other = (FRepoInfo *)anObject;
    return secure == other.secure && [host isEqualToString:other.host] &&
           [namespace isEqualToString:other.namespace];
}

@end