//
|
// NSObject+LHModel.m
|
// LHDBDemo
|
//
|
// Created by 3wchina01 on 16/3/21.
|
// Copyright © 2016年 李浩. All rights reserved.
|
//
|
|
#import "NSObject+LHModel.h"
|
#import "LHObjectInfo.h"
|
#import <objc/message.h>
|
#import <UIKit/UIKit.h>
|
#import <objc/runtime.h>
|
|
@implementation NSObject (LHModel)
|
|
static void ModelSetValueToProperty(const void *key, const void *value, void *context)
|
{
|
ModelSetContext* modelContext = context;
|
NSString* dicKey = (__bridge NSString *)(key);
|
id dicValue = (__bridge id)(value);
|
LHObjectInfo* objectInfo = [((__bridge LHClassInfo*)modelContext->classInfo) objectInfoWithName:dicKey];
|
NSObject* object = (__bridge NSObject*)modelContext->model;
|
if (objectInfo) {
|
if (objectInfo.cls) {
|
setNSTypePropertyValue(object, dicValue, objectInfo.nsTypeEcoding, objectInfo.set);
|
|
}else if (objectInfo.type.length>0) {
|
NSNumber* number = numberWithValue(dicValue);
|
setBaseTypePropertyValue(object, number, objectInfo.baseTypeEcoding,objectInfo.set);
|
}
|
}
|
}
|
|
static void ModelGetValueToDic(const void* key,const void* value,void* context)
|
{
|
ModelSetContext* modelContext = context;
|
NSMutableDictionary* dic = (__bridge NSMutableDictionary *)(modelContext->classInfo);
|
id object = (__bridge id)(modelContext->model);
|
NSString* dicKey = (__bridge NSString *)(key);
|
LHObjectInfo* objectInfo = (__bridge LHObjectInfo*)(value);
|
if (objectInfo) {
|
if (objectInfo.cls) {
|
[dic setValue:((id(*)(id,SEL))(void*) objc_msgSend)(object,objectInfo.get) forKey:dicKey];;
|
}else if (objectInfo.type.length>0) {
|
NSNumber* number = getBaseTypePropertyValue(object, objectInfo.baseTypeEcoding, objectInfo.get);
|
[dic setValue:number forKey:dicKey];
|
}
|
}
|
}
|
|
static NSNumber* numberWithValue(__unsafe_unretained id value)
|
{
|
if (!value) {
|
return nil;
|
}
|
if ([value isKindOfClass:[NSNumber class]]) return value;
|
if ([value isKindOfClass:[NSString class]]) {
|
if ([value containsString:@"."]) {
|
const char *cstring = ((NSString *)value).UTF8String;
|
if (!cstring) return nil;
|
double num = atof(cstring);
|
if (isnan(num) || isinf(num)) return nil;
|
return @(num);
|
}else {
|
const char *cstring = ((NSString*)value).UTF8String;
|
if (!cstring) return nil;
|
NSNumber* number = @(atoll(cstring));
|
if (!atoll(cstring)) {
|
number = [NSNumber numberWithChar:*(cstring+0)];
|
}
|
return number;
|
}
|
}
|
return nil;
|
|
}
|
|
static void setBaseTypePropertyValue(__unsafe_unretained NSObject* object,__unsafe_unretained NSNumber* value, NSUInteger type,SEL set)
|
{
|
switch (type) {
|
case LHBaseTypeEcodingINT:
|
((void (*)(id, SEL, int))(void *) objc_msgSend)(object, set, value.intValue);
|
break;
|
|
case LHBaseTypeEcodingLONG:
|
((void(*)(id,SEL,long))(void*) objc_msgSend)(object,set,value.integerValue);
|
break;
|
case LHBaseTypeEcodingULONG:
|
((void(*)(id,SEL,long))(void*) objc_msgSend)(object,set,value.unsignedIntegerValue);
|
break;
|
|
case LHBaseTypeEcodingFLOAT:
|
((void(*)(id,SEL,float))(void*) objc_msgSend)(object,set,value.floatValue);
|
break;
|
case LHBaseTypeEcodingDOUBLE:
|
((void(*)(id,SEL,double))(void*) objc_msgSend)(object,set,value.doubleValue);
|
break;
|
case LHBaseTypeEcodingBOOL:
|
((void(*)(id,SEL,BOOL))(void*) objc_msgSend)(object,set,value.boolValue);
|
break;
|
|
case LHBaseTypeEcodingCHAR:
|
((void(*)(id,SEL,char))(void*) objc_msgSend)(object,set,value.charValue);
|
break;
|
default:
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,nil);
|
break;
|
}
|
}
|
|
static void setNSTypePropertyValue(__unsafe_unretained id object,__unsafe_unretained id value,LHNSTypeEcoding typeEcoding,SEL set)
|
{
|
switch (typeEcoding) {
|
case LHNSTypeUNknow:
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,value);
|
break;
|
|
case LHNSTypeNSString:
|
if ([value isKindOfClass:[NSString class]]) {
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,value);
|
}else if ([value isKindOfClass:[NSNumber class]]) {
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,[value stringValue]);
|
}else if ([value isKindOfClass:[NSData class]]) {
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,[[NSString alloc] initWithData:value encoding:NSUTF8StringEncoding]);
|
}else if ([value isKindOfClass:[NSDate class]]) {
|
((void(*)(id,SEL,NSString*))(void*) objc_msgSend)(object,set,stringFormDate(value));
|
}else
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,value);
|
break;
|
|
case LHNSTypeNSNumber:
|
((void(*)(id,SEL,NSNumber*))(void*) objc_msgSend)(object,set,numberWithValue(value));
|
break;
|
|
case LHNSTypeNSDate:
|
if ([value isKindOfClass:[NSDate class]]) {
|
((void(*)(id,SEL,NSDate*))(void*) objc_msgSend)(object,set,value);
|
}else if ([value isKindOfClass:[NSString class]]) {
|
((void(*)(id,SEL,NSDate*))(void*) objc_msgSend)(object,set,dateFromString(value));
|
}else if ([value isKindOfClass:[NSData class]]) {
|
NSString* dateStr = [[NSString alloc] initWithData:value encoding:NSUTF8StringEncoding];
|
((void(*)(id,SEL,NSDate*))(void*) objc_msgSend)(object,set,dateFromString(dateStr));
|
}else
|
((void(*)(id,SEL,id))(void*) objc_msgSend)(object,set,value);
|
break;
|
|
case LHNSTypeNSData:
|
((void(*)(id,SEL,NSData*))(void*) objc_msgSend)(object,set,dataFromObject(value));
|
break;
|
|
case LHNSTypeNSURL:
|
((void(*)(id,SEL,NSURL*))(void*) objc_msgSend)(object,set,urlFromObject(value));
|
break;
|
|
case LHNSTypeNSArray:
|
((void(*)(id,SEL,NSArray*))(void*) objc_msgSend)(object,set,arrayFromObject(value));
|
break;
|
|
case LHNSTypeNSDictionary:
|
((void(*)(id,SEL,NSDictionary*))(void*) objc_msgSend)(object,set,dicFromObject(value));
|
break;
|
|
case LHNSTypeUIImage:
|
((void(*)(id,SEL,UIImage*))(void*) objc_msgSend)(object,set,imageFromObject(value));
|
break;
|
|
default:
|
break;
|
}
|
}
|
|
static NSNumber* getBaseTypePropertyValue(__unsafe_unretained NSObject* object, NSUInteger type,SEL get)
|
{
|
switch (type) {
|
case LHBaseTypeEcodingINT:
|
|
return @(((int (*)(id, SEL))(void *) objc_msgSend)(object, get));
|
|
case LHBaseTypeEcodingLONG:
|
|
return @(((long (*)(id, SEL))(void *) objc_msgSend)(object,get));
|
|
case LHBaseTypeEcodingULONG:
|
|
return @(((NSUInteger(*)(id,SEL))(void*) objc_msgSend)(object,get));
|
|
case LHBaseTypeEcodingFLOAT:
|
|
return @(((float(*)(id,SEL))(void*) objc_msgSend)(object,get));
|
|
case LHBaseTypeEcodingDOUBLE:
|
|
return @(((double(*)(id,SEL))(void*) objc_msgSend)(object,get));
|
|
case LHBaseTypeEcodingBOOL:
|
|
return @(((BOOL(*)(id,SEL))(void*) objc_msgSend)(object,get));
|
|
case LHBaseTypeEcodingCHAR:
|
|
return @(((char(*)(id,SEL))(void*) objc_msgSend)(object,get));
|
|
default:
|
return nil;
|
break;
|
}
|
}
|
|
typedef struct {
|
void *classInfo;
|
void *model;
|
} ModelSetContext;
|
|
+ (id)lh_ModelWithJSON:(id)json
|
{
|
if (!json) return nil;
|
if ([json isKindOfClass:[NSDictionary class]]) {
|
return [self lh_ModelWithDictionary:(NSDictionary*)json];
|
}
|
NSData* jsonData;
|
if ([json isKindOfClass:[NSString class]]) {
|
jsonData = [(NSString*)json dataUsingEncoding:NSUTF8StringEncoding];
|
}else if ([json isKindOfClass:[NSData class]])
|
jsonData = json;
|
if (jsonData) {
|
return [self lh_ModelWithDictionary:[NSJSONSerialization JSONObjectWithData:jsonData options:NSJSONReadingMutableContainers error:nil]];
|
}
|
return nil;
|
}
|
|
+ (id)lh_ModelWithDictionary:(NSDictionary*)dic
|
{
|
if (!dic ||![dic isKindOfClass:[NSDictionary class]]) return nil;
|
NSObject* object = [[self alloc] init];
|
ModelSetContext context = {0};
|
LHClassInfo* info;
|
//判断缓存中是否有这个类的信息
|
if ([LHClassInfo isCacheWithClass:self]) {
|
info = [LHClassInfo classInfoWithClass:self];
|
}else
|
info = [[LHClassInfo alloc] initWithClass:self];
|
context.classInfo = (__bridge void *)(info);
|
context.model = (__bridge void *)(object);
|
|
CFDictionaryApplyFunction((__bridge CFDictionaryRef)dic, ModelSetValueToProperty, &context);
|
return object;
|
}
|
|
- (NSDictionary*)lh_ModelToDictionary
|
{
|
if ([self isKindOfClass:[NSArray class]]) {
|
return nil;
|
}else if ([self isKindOfClass:[NSDictionary class]]){
|
return (NSDictionary*)self;
|
}else if ([self isKindOfClass:[NSString class]]||[self isKindOfClass:[NSData class]]) {
|
return [NSJSONSerialization JSONObjectWithData:dataFromObject(self) options:NSJSONReadingMutableContainers error:nil];
|
}else {
|
NSMutableDictionary* dic = [NSMutableDictionary dictionary];
|
ModelSetContext context = {0};
|
context.classInfo = (__bridge void *)(dic);
|
context.model = (__bridge void *)(self);
|
LHClassInfo* classInfo;
|
//判断缓存中是否有这个类的信息
|
if ([LHClassInfo isCacheWithClass:object_getClass(self)]) {
|
classInfo = [LHClassInfo classInfoWithClass:object_getClass(self)];
|
}else
|
classInfo = [[LHClassInfo alloc] initWithClass:object_getClass(self)];
|
CFDictionaryApplyFunction((__bridge CFMutableDictionaryRef)classInfo.objectInfoDic, ModelGetValueToDic, &context);
|
return dic;
|
}
|
return nil;
|
}
|
|
- (NSData*)lh_ModelToData
|
{
|
return [NSJSONSerialization dataWithJSONObject:[self lh_ModelToDictionary] options:NSJSONWritingPrettyPrinted error:nil];;
|
}
|
|
+ (NSDictionary*)getAllPropertyNameAndType
|
{
|
NSMutableDictionary* dic = [NSMutableDictionary dictionary];
|
unsigned int count = 0;
|
objc_property_t* property_t = class_copyPropertyList(self, &count);
|
for (int i=0; i<count; i++) {
|
objc_property_t propert = property_t[i];
|
NSString* propertyName = [NSString stringWithUTF8String:property_getName(propert)];
|
NSString* propertyType = [NSString stringWithUTF8String:property_getAttributes(propert)];
|
[dic setValue:objectType(propertyType) forKey:propertyName];
|
}
|
free(property_t);
|
return dic;
|
}
|
|
+ (NSString*)getTypeNameWith:(NSString*)propertyName
|
{
|
NSString* typeStr = [[self getAllPropertyNameAndType]valueForKey:propertyName];
|
if ([typeStr isEqualToString:@"i"]) {
|
return @"INT";
|
}else if ([typeStr isEqualToString:@"f"]) {
|
return @"FLOAT";
|
}else if ([typeStr isEqualToString:@"B"]) {
|
return @"BOOL";
|
}else if ([typeStr isEqualToString:@"d"]) {
|
return @"DOUBLE";
|
}else if ([typeStr isEqualToString:@"q"]) {
|
return @"LONG";
|
}else if ([typeStr isEqualToString:@"NSData"]||[typeStr isEqualToString:@"UIImage"]) {
|
return @"BLOB";
|
}else if ([typeStr isEqualToString:@"NSNumber"]){
|
return @"INT";
|
} else
|
return @"TEXT";
|
}
|
|
static id objectType(NSString* typeString)
|
{
|
if ([typeString containsString:@"@"]) {
|
NSArray* strArray = [typeString componentsSeparatedByString:@"\""];
|
if (strArray.count >= 1) {
|
return strArray[1];
|
}else
|
return nil;
|
}else
|
return [typeString substringWithRange:NSMakeRange(1, 1)];
|
}
|
|
|
static NSString* stringFormDate(NSDate* date)
|
{
|
NSDateFormatter* formatter = [[NSDateFormatter alloc] init];
|
formatter.dateFormat = @"yyyy-MM-dd HH:mm:ss";
|
return [formatter stringFromDate:date];
|
}
|
|
static NSDate* dateFromString(NSString* string)
|
{
|
NSDateFormatter* formatter = [[NSDateFormatter alloc] init];
|
formatter.dateFormat = @"yyyy-MM-dd HH:mm:ss";
|
return [formatter dateFromString:string];
|
}
|
|
static NSData* dataFromObject(id object)
|
{
|
if ([object isKindOfClass:[NSData class]]) {
|
return object;
|
}else if ([object isKindOfClass:[NSString class]]) {
|
return [object dataUsingEncoding:NSUTF8StringEncoding];
|
}else if ([object isKindOfClass:[NSDate class]]) {
|
return [stringFormDate(object) dataUsingEncoding:NSUTF8StringEncoding];
|
}else if ([object isKindOfClass:[NSArray class]]||[object isKindOfClass:[NSDictionary class]]) {
|
return [NSJSONSerialization dataWithJSONObject:object options:NSJSONWritingPrettyPrinted error:nil];
|
}else if ([object isKindOfClass:NSClassFromString(@"UIImage")]) {
|
return UIImageJPEGRepresentation(object, 1);
|
}else
|
return [object dataUsingEncoding:NSUTF8StringEncoding];
|
}
|
|
static NSURL* urlFromObject(id object)
|
{
|
if ([object isKindOfClass:[NSURL class]]) {
|
return object;
|
}else if ([object isKindOfClass:[NSString class]]) {
|
return [NSURL URLWithString:[object stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]];
|
}else {
|
return [NSURL URLWithString:[[[NSString alloc] initWithData:dataFromObject(object) encoding:NSUTF8StringEncoding] stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]];
|
}
|
}
|
|
static NSArray* arrayFromObject(id object)
|
{
|
if ([object isKindOfClass:[NSArray class]]) {
|
return object;
|
}else if ([object isKindOfClass:[NSDictionary class]]){
|
return nil;
|
}else {
|
id value = [NSJSONSerialization JSONObjectWithData:dataFromObject(object) options:NSJSONReadingMutableContainers error:nil];
|
if ([value isKindOfClass:[NSArray class]]) {
|
return value;
|
}
|
return nil;
|
}
|
}
|
|
static NSDictionary* dicFromObject(id object)
|
{
|
if ([object isKindOfClass:[NSDictionary class]]) {
|
return object;
|
}else {
|
NSData* data = dataFromObject(object);
|
id value = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingMutableContainers error:nil];
|
if ([value isKindOfClass:[NSDictionary class]]) {
|
return value;
|
}
|
return nil;
|
}
|
}
|
|
static UIImage* imageFromObject(id object)
|
{
|
if ([object isKindOfClass:[UIImage class]]) {
|
return object;
|
}else {
|
return [UIImage imageWithData:dataFromObject(object)];
|
}
|
}
|
|
@end
|
