Doric/doric-iOS/Pod/Classes/Plugin/DoricNotificationPlugin.m

/*
 * Copyright [2019] [Doric.Pub]
 *
 * 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.
 */
//
// Created by pengfei.zhou on 2020/1/8.
//

#import <DoricCore/Doric.h>
#import "DoricNotificationPlugin.h"

@interface DoricNotificationPlugin ()

@property(nonatomic, strong) NSDictionary<NSString *, id> *observers;
@end

@implementation DoricNotificationPlugin

- (NSDictionary *)observers {
    if (!_observers) {
        _observers = [NSDictionary new];
    }
    return _observers;
}

- (void)publish:(NSDictionary *)dic withPromise:(DoricPromise *)promise {
    NSString *biz = [dic optString:@"biz"];
    NSString *name = [dic optString:@"name"];
    if (biz) {
        name = [NSString stringWithFormat:@"__doric__%@#%@", biz, name];
    }
    NSString *data = [dic optString:@"data"];
    NSDictionary *dataDic = nil;
    if (data) {
        NSData *jsonData = [data dataUsingEncoding:NSUTF8StringEncoding];
        NSError *err;
        dataDic = [NSJSONSerialization JSONObjectWithData:jsonData
                                                  options:NSJSONReadingMutableContainers
                                                    error:&err];
    }
    [[NSNotificationCenter defaultCenter] postNotificationName:name object:nil userInfo:dataDic];
    [promise resolve:nil];
}

- (void)subscribe:(NSDictionary *)dic withPromise:(DoricPromise *)promise {
    NSString *biz = [dic optString:@"biz"];
    NSString *name = [dic optString:@"name"];
    if (biz) {
        name = [NSString stringWithFormat:@"__doric__%@#%@", biz, name];
    }
    NSString *callbackId = [dic optString:@"callback"];
    __weak typeof(self) _self = self;
    id observer = [[NSNotificationCenter defaultCenter]
            addObserverForName:name
                        object:nil
                         queue:[NSOperationQueue mainQueue]
                    usingBlock:^(NSNotification *note) {
                        __strong typeof(_self) self = _self;
                        DoricPromise *currentPromise = [[DoricPromise alloc] initWithContext:self.doricContext callbackId:callbackId];
                        [currentPromise resolve:note.userInfo];
                    }];
    NSMutableDictionary *mutableDictionary = [self.observers mutableCopy];
    mutableDictionary[callbackId] = observer;
    self.observers = mutableDictionary;
    [promise resolve:callbackId];
}

- (void)unsubscribe:(NSString *)subscribeId withPromise:(DoricPromise *)promise {
    id observer = self.observers[subscribeId];
    if (observer) {
        [[NSNotificationCenter defaultCenter] removeObserver:observer];
        NSMutableDictionary *mutableDictionary = [self.observers mutableCopy];
        [mutableDictionary removeObjectForKey:subscribeId];
        self.observers = mutableDictionary;
    }
    [promise resolve:nil];
}

- (void)dealloc {
    NSDictionary *dictionary = self.observers;
    [dictionary enumerateKeysAndObjectsUsingBlock:^(NSString *key, id obj, BOOL *stop) {
        [[NSNotificationCenter defaultCenter] removeObserver:obj];
    }];
}

@end