aboutsummaryrefslogtreecommitdiffhomepage
path: root/Firebase/Database/Core/FRepoInfo.m
blob: 6b15fe5db840caa2b73a0226e18c0c37bceba037 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*
 * 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;
}

@end