124 lines
3.9 KiB
Objective-C
124 lines
3.9 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "SignalsNavigationController.h"
|
|
#import "Session-Swift.h"
|
|
#import <SignalMessaging/UIUtil.h>
|
|
#import <SessionServiceKit/NSTimer+OWS.h>
|
|
#import <SessionServiceKit/OWSSignalService.h>
|
|
#import <SessionServiceKit/TSSocketManager.h>
|
|
|
|
static double const STALLED_PROGRESS = 0.9;
|
|
|
|
@interface SignalsNavigationController ()
|
|
|
|
@property (nonatomic) UIProgressView *socketStatusView;
|
|
@property (nonatomic) NSTimer *updateStatusTimer;
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation SignalsNavigationController
|
|
|
|
- (void)viewDidLoad {
|
|
[super viewDidLoad];
|
|
|
|
// Do any additional setup after loading the view.
|
|
[self initializeObserver];
|
|
[self updateSocketStatusView];
|
|
}
|
|
|
|
- (void)initializeSocketStatusBar {
|
|
if (!_socketStatusView) {
|
|
_socketStatusView = [[UIProgressView alloc] initWithProgressViewStyle:UIProgressViewStyleDefault];
|
|
}
|
|
|
|
CGRect bar = self.navigationBar.frame;
|
|
_socketStatusView.frame = CGRectMake(0, bar.size.height - 1.0f, self.view.frame.size.width, 1.0f);
|
|
_socketStatusView.progress = 0.0f;
|
|
_socketStatusView.progressTintColor = [UIColor ows_fadedBlueColor];
|
|
|
|
/** Loki: Original code
|
|
if (![_socketStatusView superview]) {
|
|
[self.navigationBar addSubview:_socketStatusView];
|
|
}
|
|
*/
|
|
}
|
|
|
|
- (void)dealloc
|
|
{
|
|
[[NSNotificationCenter defaultCenter] removeObserver:self];
|
|
}
|
|
|
|
#pragma mark - Socket Status Notifications
|
|
|
|
- (void)initializeObserver {
|
|
[[NSNotificationCenter defaultCenter] addObserver:self
|
|
selector:@selector(OWSWebSocketStateDidChange)
|
|
name:kNSNotification_OWSWebSocketStateDidChange
|
|
object:nil];
|
|
[[NSNotificationCenter defaultCenter] addObserver:self
|
|
selector:@selector(isCensorshipCircumventionActiveDidChange:)
|
|
name:kNSNotificationName_IsCensorshipCircumventionActiveDidChange
|
|
object:nil];
|
|
}
|
|
|
|
- (void)isCensorshipCircumventionActiveDidChange:(NSNotification *)notification
|
|
{
|
|
OWSAssertIsOnMainThread();
|
|
|
|
[self updateSocketStatusView];
|
|
}
|
|
|
|
- (void)OWSWebSocketStateDidChange
|
|
{
|
|
OWSAssertIsOnMainThread();
|
|
|
|
[self updateSocketStatusView];
|
|
}
|
|
|
|
- (void)updateSocketStatusView {
|
|
OWSAssertIsOnMainThread();
|
|
|
|
if ([OWSSignalService sharedInstance].isCensorshipCircumventionActive) {
|
|
[_updateStatusTimer invalidate];
|
|
[_socketStatusView removeFromSuperview];
|
|
_socketStatusView = nil;
|
|
return;
|
|
}
|
|
|
|
switch (TSSocketManager.shared.highestSocketState) {
|
|
case OWSWebSocketStateClosed:
|
|
if (_socketStatusView == nil) {
|
|
[self initializeSocketStatusBar];
|
|
[_updateStatusTimer invalidate];
|
|
_updateStatusTimer = [NSTimer weakScheduledTimerWithTimeInterval:0.5
|
|
target:self
|
|
selector:@selector(updateProgress)
|
|
userInfo:nil
|
|
repeats:YES];
|
|
|
|
} else if (_socketStatusView.progress >= STALLED_PROGRESS) {
|
|
[_updateStatusTimer invalidate];
|
|
}
|
|
break;
|
|
case OWSWebSocketStateConnecting:
|
|
// Do nothing.
|
|
break;
|
|
case OWSWebSocketStateOpen:
|
|
[_updateStatusTimer invalidate];
|
|
[_socketStatusView removeFromSuperview];
|
|
_socketStatusView = nil;
|
|
break;
|
|
}
|
|
}
|
|
|
|
- (void)updateProgress {
|
|
double progress = _socketStatusView.progress + 0.05;
|
|
_socketStatusView.progress = (float) MIN(progress, STALLED_PROGRESS);
|
|
}
|
|
|
|
@end
|