//
//  UIAlertController+RQExtension.m
//  RQCommon
//
//  Created by 张嵘 on 2018/11/23.
//  Copyright © 2018 张嵘. All rights reserved.
//

#import "UIAlertController+RQExtension.h"
#import <objc/runtime.h>

@implementation UIAlertController (RQExtension)

- (BOOL)shouldAutorotate {
    return NO;
}

- (NSUInteger)supportedInterfaceOrientations {
    return UIInterfaceOrientationMaskPortrait;
}

- (void)viewWillLayoutSubviews {
	[super viewWillLayoutSubviews];
	
	//按钮统一颜色
	if (self.tintColor) {
		for (UIAlertAction *action in self.actions) {
			if (!action.textColor || action.style != UIAlertActionStyleDestructive) {
				action.textColor = self.tintColor;
			}
		}
	}
}

- (UIColor *)tintColor {
	return objc_getAssociatedObject(self, @selector(tintColor));
}

- (void)setTintColor:(UIColor *)tintColor {
	
	objc_setAssociatedObject(self, @selector(tintColor), tintColor, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (UIColor *)titleColor {
	return objc_getAssociatedObject(self, @selector(titleColor));
}

- (void)setTitleColor:(UIColor *)titleColor {
	unsigned int count = 0;
	Ivar *ivars = class_copyIvarList([UIAlertController class], &count);
	for(int i = 0;i < count;i ++){
		
		Ivar ivar = ivars[i];
		NSString *ivarName = [NSString stringWithCString:ivar_getName(ivar) encoding:NSUTF8StringEncoding];
		
		//标题颜色
		if ([ivarName isEqualToString:@"_attributedTitle"] && self.title && titleColor) {
			
			NSMutableAttributedString *attr = [[NSMutableAttributedString alloc]initWithString:self.title attributes:@{NSForegroundColorAttributeName:titleColor,NSFontAttributeName:[UIFont systemFontOfSize:16.0]}];
			[self setValue:attr forKey:@"attributedTitle"];
		}
	}
	
	free(ivars);
	
	objc_setAssociatedObject(self, @selector(titleColor), titleColor, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (UIColor *)messageColor {
	return objc_getAssociatedObject(self, @selector(messageColor));
}

- (void)setMessageColor:(UIColor *)messageColor {
	unsigned int count = 0;
	Ivar *ivars = class_copyIvarList([UIAlertController class], &count);
	for(int i = 0;i < count;i ++){
		
		Ivar ivar = ivars[i];
		NSString *ivarName = [NSString stringWithCString:ivar_getName(ivar) encoding:NSUTF8StringEncoding];
		
		//描述颜色
		if ([ivarName isEqualToString:@"_attributedMessage"] && self.message && messageColor) {
			
			NSMutableAttributedString *attr = [[NSMutableAttributedString alloc] initWithString:self.message attributes:@{NSForegroundColorAttributeName:messageColor,NSFontAttributeName:[UIFont systemFontOfSize:12.0]}];
			[self setValue:attr forKey:@"attributedMessage"];
		}
	}
	
	free(ivars);
	
	objc_setAssociatedObject(self, @selector(messageColor), messageColor, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

@end


@implementation UIAlertAction (RQExtension)
- (UIColor *)textColor {
	return objc_getAssociatedObject(self, @selector(textColor));
}

//按钮标题的字体颜色
- (void)setTextColor:(UIColor *)textColor {
	if (self.style == UIAlertActionStyleDestructive) {
		return;
	}
	
	unsigned int count = 0;
	Ivar *ivars = class_copyIvarList([UIAlertAction class], &count);
	for(int i =0; i<count; i++){
		
		Ivar ivar = ivars[i];
		NSString *ivarName = [NSString stringWithCString:ivar_getName(ivar) encoding:NSUTF8StringEncoding];
		
		if ([ivarName isEqualToString:@"_titleTextColor"]) {
			
			[self setValue:textColor forKey:@"titleTextColor"];
		}
	}
	free(ivars);
	
	objc_setAssociatedObject(self, @selector(textColor), textColor, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}


@end