/* * 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 "DoricNotificationPlugin.h" @interface DoricNotificationPlugin () @property(nonatomic, copy) NSDictionary *observers; @end @implementation DoricNotificationPlugin - (NSDictionary *)observers { if (!_observers) { _observers = [NSDictionary new]; } return _observers; } - (void)publish:(NSDictionary *)dic withPromise:(DoricPromise *)promise { NSString *biz = dic[@"biz"]; NSString *name = dic[@"name"]; if (biz) { name = [NSString stringWithFormat:@"__doric__%@#%@", biz, name]; } NSString *data = dic[@"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[@"biz"]; NSString *name = dic[@"name"]; if (biz) { name = [NSString stringWithFormat:@"__doric__%@#%@", biz, name]; } NSString *callbackId = dic[@"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