在移动互联网的下半场，越来越多的 APP 更加注重用户体验，以期来打动用户。主题的切换就是可以增强用户体验、结合运营活动的一个点：譬如 QQ 的夜间模式，节日里电商 APP 的皮肤切换等等的这些小细节往往就是赢得用户尊重的根本。本文将从主题的动态切换出发，介绍下贝聊 iOS 客户端在实现主题动态所采用的方案，供读者参考。

从切换方案说起

让 APP 已有的控件能切换主题可以用子类化，swizzle 或 category 来实现，其中子类化和 category 实现起来差不多，都是让控件调特定的方法达到切换风格的效果，而 swizzle 的影响范围会比较广，使用的时候可以通过 Associated Object 添加一个标记值，让需要切换风格的控件设置这个标记值，让标记值来决定是否需要 swizzle。考虑到上述几种方案的复杂度，最后选择了 category 来实现。

@implementation UILabel (BLTheme)

- (void)bl_setThemeTextColor {
    NSString *hexColorString = [BLThemeManager sharedInstance].styleModel.navTextColor;
    UIColor *color;
    if (hexColorString) {
        color = [UIColor colorFromHexString:hexColorString];
    }

    if (color) {
        [self setTextColor:color];
    }

}

@end

简单说来就是通过已配置的描述文件，在 category 内部读取了下配置的样式数据（样式数据可能为默认样式或自定义样式）。

怎样实现一个主题管理类

主题管理类的核心功能就是负责主题的更新，切换。正如下图所示，想让主题管理类通知到这么多待切换的 category 并不是一件容易的事，因为觉得在 category 上添加观察者并不是太好的设计，你很难知道什么时机该把观察者移除了。

难道说得改成子类化的实现，然后 ovrride dealloc 方法，可惜这样做感觉就没那么纯粹了。

这也就意味着，可能需要自己动手来实现回调机制了，让切换主题相关的 category 通过主题管理类注册一个回调 block，主题类维护使用一个字典维护这些 block，待切换时由主题管理类统一回调，达到类似 Notification 的效果。

先来看看 categroy 使用此方案后的变化，仍然是刚才的 UILabel 类：

- (void)bl_setThemeTextColor {
    NSString *hexColorString = [BLThemeManager sharedInstance].styleModel.navTextColor;
    UIColor *color;
    if (hexColorString) {
        color = [UIColor colorFromHexString:hexColorString];
    }

    if (color) {
        [self setTextColor:color];
    }

    @weakify(self)
    SwitchThemeBlock switchThemeBlock = ^{
        @strongify(self)
        [self bl_setThemeTextColor];
    };
    [[BLThemeManager sharedInstance] addObserveKey:[self keyWithSelector:@selector(bl_setThemeTextColor)] withSwitchThemBlock:switchThemeBlock];
}

只是在方法的底部添加了注册 block 的方法，而注册 block 的方法也十分简单，只需依据 key 判断下是否需要将 block 加入代码中。

// BLThemeManager.m
- (void)addObserveKey:(BLThemeMapModel *)key withSwitchThemBlock:(SwitchThemeBlock)block {
    if (block) {
        NSArray *allKeys = [self.blockDictionary allKeys];

        if (![allKeys containsObject:key]) {
            self.blockDictionary[key] = block;
        }
    }
}

那么问题来了，到底该如何设计一个这样的 key 呢？

同一个控件的主题 category 有多个需要切换主题的方法（例如 UIButton 有setTitleColor:forState: 和 setImage:forState:）；
多个控件都是通过同一个 categroy 来切换主题（例如有多个 UIButton 需要切换主题）；

其实统筹来看，就是如何通过某个类的实例和所需定制主题的方法来确定一个 key。

一开始很自然的拼了一个类的地址和方法名来作为key [NSString stringWithFormat:@"%p#%@", class, NSStringFromSelector(selector)]

流程能跑起来了，但是问题也很明显，只知道一个对象的指针字符串，根据对象是否被释放而进行的字典清理将变得难以实现：

// BLThemeManager.m
- (void)updateTheme {
    for (BLThemeMapModel *key in [self.blockDictionary allKeys]) {
        id object = key.target;
        if (object) {
            if ([object respondsToSelector:NSSelectorFromString(key.selectorName)]) {
                SwitchThemeBlock block = self.blockDictionary[key];
                if (block) {
                    block();
                }
            }

        } else {
            [self.blockDictionary removeObjectForKey:key];
        }
    }
}

那么，应该怎样设计 block 对应的 key 呢？

能从 key 中获取到注册的类；
key 中也存有方法做 key 的唯一性和对象访问该方法安全性的校验 respondsToSelector；

为此，实现了一个辅助的 model，用以访问需要注册的对象实例和方法名，同时作为 Dictionary 的 key，它还需要实现 NSCoping 协议：

@interface BLThemeMapModel : NSObject <NSCopying>

@property (nonatomic, weak, readonly) id target;
@property (nonatomic, copy, readonly) NSString *selectorName;

- (instancetype)initWithTarget:(id)target selctorName:(NSString *)selectorName;

@end

weak 修饰的对象实例能够在对象被释放后自动置 nil，下面附上最初的 .m 文件实现。

@implementation BLThemeMapModel

- (instancetype)initWithTarget:(id)target selctorName:(NSString *)selectorName {
    if (self = [super init]) {
        _target = target;
        _selectorName = selectorName;
    }
    return self;
}

- (BOOL)isEqual:(id)object {
    BLThemeMapModel *model = (BLThemeMapModel *)object;

    if (model) {
        if([self.target isEqualToString:model.target] &&
           [self.selectorName isEqualToString:model.selectorName]){
            return YES;
        }

    }
    return NO;
}

- (NSUInteger)hash {
    NSUInteger hash = [self.target hash] ^ [self.selectorName hash];
    return hash;
}

- (id)copyWithZone:(NSZone *)zone {
    BLThemeMapModel *themeModel = [[BLThemeMapModel allocWithZone:zone] initWithTarget:self.target selctorName:self.selectorName];
    return themeModel;
}

@end

可惜自测后发现一个挺莫名的 bug，最后调试了好一会才解决。细心的读者可以先想想看～

因为 target 可能被置 nil，从而引起同一个 key 的 hash 值被修改了，然后在遍历字典时，就无法取到之前加入字典的对象了，即便对象是被释放了，但仍有个 dirty 的 BLThemeMapModel 对象在字典里。

定位问题后其实就很好办了，在初始化方法中添加两个用以 hash 的属性：

_pointerString = [NSString stringWithFormat:@"%p", target];
_targetTypeName = NSStringFromClass([target class]);

最后使用这两个属性完成 hash 和 -isEqual: 方法：

- (BOOL)isEqual:(id)object {
    BLThemeMapModel *model = (BLThemeMapModel *)object;

    if (model) {
        if([self.pointerString isEqualToString:model.pointerString] &&
           [self.selectorName isEqualToString:model.selectorName] &&
           [self.targetTypeName isEqualToString:model.targetTypeName]){
            return YES;
        }

    }
    return NO;
}

- (NSUInteger)hash {
    NSUInteger hash = [self.pointerString hash] ^ [self.selectorName hash] ^ [self.targetTypeName hash];
    return hash;
}

至此，动态切换主题的功能大致就实现了，而且没使用到 OC 的任何动态方法。

总结

本文描述了实现一个主题管理类的大致思路，希望能对读者有所帮助。后来笔者想到既然有了 target 和 selector，能不能通过 NSInvocation 来动态调用，就不借助 block 来回调了，在尝试中笔者 NSInvocation 的效率的确会低一点，而且没有 block 灵活：

@implementation UIViewController (BLTheme)

- (UIStatusBarStyle)bl_setPreferredStatusBarStyle {
    NSInteger statusValue = [BLThemeManager sharedInstance].styleModel.statusBarColor;

    @weakify(self)
    SwitchThemeBlock switchThemeBlock = ^{
        @strongify(self)
        [self setNeedsStatusBarAppearanceUpdate];
    };
    [[BLThemeManager sharedInstance] addObserveKey:[self keyWithSelector:@selector(bl_setPreferredStatusBarStyle)] withSwitchThemBlock:switchThemeBlock];

    if (statusValue == 1) {
        return UIStatusBarStyleLightContent;
    } else {
        return UIStatusBarStyleDefault;
    }
}

@end

就像这，逻辑上并不期望再调一次 bl_setPreferredStatusBarStyle，而是仅仅调用一下 [self setNeedsStatusBarAppearanceUpdate]; 用 block 可以很灵活的指定好需要调用什么方法。

或许，也可以通过实现一个 weak proxy 的方式使用 Notification 来实现，笔者就没有尝试了，感兴趣的读者可以试试。