NSObject+White_YYModel.m 85 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840
  1. //
  2. // NSObject+YYModel.m
  3. // YYModel <https://github.com/ibireme/YYModel>
  4. //
  5. // Created by ibireme on 15/5/10.
  6. // Copyright (c) 2015 ibireme.
  7. //
  8. // This source code is licensed under the MIT-style license found in the
  9. // LICENSE file in the root directory of this source tree.
  10. //
  11. #import "NSObject+White_YYModel.h"
  12. #import "White_YYClassInfo.h"
  13. #import <objc/message.h>
  14. #define force_inline __inline__ __attribute__((always_inline))
  15. /// Foundation Class Type
  16. typedef NS_ENUM (NSUInteger, White_YYEncodingNSType) {
  17. White_YYEncodingTypeNSUnknown = 0,
  18. White_YYEncodingTypeNSString,
  19. White_YYEncodingTypeNSMutableString,
  20. White_YYEncodingTypeNSValue,
  21. White_YYEncodingTypeNSNumber,
  22. White_YYEncodingTypeNSDecimalNumber,
  23. White_YYEncodingTypeNSData,
  24. White_YYEncodingTypeNSMutableData,
  25. White_YYEncodingTypeNSDate,
  26. White_YYEncodingTypeNSURL,
  27. White_YYEncodingTypeNSArray,
  28. White_YYEncodingTypeNSMutableArray,
  29. White_YYEncodingTypeNSDictionary,
  30. White_YYEncodingTypeNSMutableDictionary,
  31. White_YYEncodingTypeNSSet,
  32. White_YYEncodingTypeNSMutableSet,
  33. };
  34. /// Get the Foundation class type from property info.
  35. static force_inline White_YYEncodingNSType White_YYClassGetNSType(Class cls) {
  36. if (!cls) return White_YYEncodingTypeNSUnknown;
  37. if ([cls isSubclassOfClass:[NSMutableString class]]) return White_YYEncodingTypeNSMutableString;
  38. if ([cls isSubclassOfClass:[NSString class]]) return White_YYEncodingTypeNSString;
  39. if ([cls isSubclassOfClass:[NSDecimalNumber class]]) return White_YYEncodingTypeNSDecimalNumber;
  40. if ([cls isSubclassOfClass:[NSNumber class]]) return White_YYEncodingTypeNSNumber;
  41. if ([cls isSubclassOfClass:[NSValue class]]) return White_YYEncodingTypeNSValue;
  42. if ([cls isSubclassOfClass:[NSMutableData class]]) return White_YYEncodingTypeNSMutableData;
  43. if ([cls isSubclassOfClass:[NSData class]]) return White_YYEncodingTypeNSData;
  44. if ([cls isSubclassOfClass:[NSDate class]]) return White_YYEncodingTypeNSDate;
  45. if ([cls isSubclassOfClass:[NSURL class]]) return White_YYEncodingTypeNSURL;
  46. if ([cls isSubclassOfClass:[NSMutableArray class]]) return White_YYEncodingTypeNSMutableArray;
  47. if ([cls isSubclassOfClass:[NSArray class]]) return White_YYEncodingTypeNSArray;
  48. if ([cls isSubclassOfClass:[NSMutableDictionary class]]) return White_YYEncodingTypeNSMutableDictionary;
  49. if ([cls isSubclassOfClass:[NSDictionary class]]) return White_YYEncodingTypeNSDictionary;
  50. if ([cls isSubclassOfClass:[NSMutableSet class]]) return White_YYEncodingTypeNSMutableSet;
  51. if ([cls isSubclassOfClass:[NSSet class]]) return White_YYEncodingTypeNSSet;
  52. return White_YYEncodingTypeNSUnknown;
  53. }
  54. /// Whether the type is c number.
  55. static force_inline BOOL White_YYEncodingTypeIsCNumber(White_YYEncodingType type) {
  56. switch (type & White_YYEncodingTypeMask) {
  57. case White_YYEncodingTypeBool:
  58. case White_YYEncodingTypeInt8:
  59. case White_YYEncodingTypeUInt8:
  60. case White_YYEncodingTypeInt16:
  61. case White_YYEncodingTypeUInt16:
  62. case White_YYEncodingTypeInt32:
  63. case White_YYEncodingTypeUInt32:
  64. case White_YYEncodingTypeInt64:
  65. case White_YYEncodingTypeUInt64:
  66. case White_YYEncodingTypeFloat:
  67. case White_YYEncodingTypeDouble:
  68. case White_YYEncodingTypeLongDouble: return YES;
  69. default: return NO;
  70. }
  71. }
  72. /// Parse a number value from 'id'.
  73. static force_inline NSNumber *White_YYNSNumberCreateFromID(__unsafe_unretained id value) {
  74. static NSCharacterSet *dot;
  75. static NSDictionary *dic;
  76. static dispatch_once_t onceToken;
  77. dispatch_once(&onceToken, ^{
  78. dot = [NSCharacterSet characterSetWithRange:NSMakeRange('.', 1)];
  79. dic = @{@"TRUE" : @(YES),
  80. @"True" : @(YES),
  81. @"true" : @(YES),
  82. @"FALSE" : @(NO),
  83. @"False" : @(NO),
  84. @"false" : @(NO),
  85. @"YES" : @(YES),
  86. @"Yes" : @(YES),
  87. @"yes" : @(YES),
  88. @"NO" : @(NO),
  89. @"No" : @(NO),
  90. @"no" : @(NO),
  91. @"NIL" : (id)kCFNull,
  92. @"Nil" : (id)kCFNull,
  93. @"nil" : (id)kCFNull,
  94. @"NULL" : (id)kCFNull,
  95. @"Null" : (id)kCFNull,
  96. @"null" : (id)kCFNull,
  97. @"(NULL)" : (id)kCFNull,
  98. @"(Null)" : (id)kCFNull,
  99. @"(null)" : (id)kCFNull,
  100. @"<NULL>" : (id)kCFNull,
  101. @"<Null>" : (id)kCFNull,
  102. @"<null>" : (id)kCFNull};
  103. });
  104. if (!value || value == (id)kCFNull) return nil;
  105. if ([value isKindOfClass:[NSNumber class]]) return value;
  106. if ([value isKindOfClass:[NSString class]]) {
  107. NSNumber *num = dic[value];
  108. if (num) {
  109. if (num == (id)kCFNull) return nil;
  110. return num;
  111. }
  112. if ([(NSString *)value rangeOfCharacterFromSet:dot].location != NSNotFound) {
  113. const char *cstring = ((NSString *)value).UTF8String;
  114. if (!cstring) return nil;
  115. double num = atof(cstring);
  116. if (isnan(num) || isinf(num)) return nil;
  117. return @(num);
  118. } else {
  119. const char *cstring = ((NSString *)value).UTF8String;
  120. if (!cstring) return nil;
  121. return @(atoll(cstring));
  122. }
  123. }
  124. return nil;
  125. }
  126. /// Parse string to date.
  127. static force_inline NSDate *White_YYNSDateFromString(__unsafe_unretained NSString *string) {
  128. typedef NSDate* (^White_YYNSDateParseBlock)(NSString *string);
  129. #define kParserNum 34
  130. static White_YYNSDateParseBlock blocks[kParserNum + 1] = {0};
  131. static dispatch_once_t onceToken;
  132. dispatch_once(&onceToken, ^{
  133. {
  134. /*
  135. 2014-01-20 // Google
  136. */
  137. NSDateFormatter *formatter = [[NSDateFormatter alloc] init];
  138. formatter.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  139. formatter.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
  140. formatter.dateFormat = @"yyyy-MM-dd";
  141. blocks[10] = ^(NSString *string) { return [formatter dateFromString:string]; };
  142. }
  143. {
  144. /*
  145. 2014-01-20 12:24:48
  146. 2014-01-20T12:24:48 // Google
  147. 2014-01-20 12:24:48.000
  148. 2014-01-20T12:24:48.000
  149. */
  150. NSDateFormatter *formatter1 = [[NSDateFormatter alloc] init];
  151. formatter1.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  152. formatter1.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
  153. formatter1.dateFormat = @"yyyy-MM-dd'T'HH:mm:ss";
  154. NSDateFormatter *formatter2 = [[NSDateFormatter alloc] init];
  155. formatter2.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  156. formatter2.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
  157. formatter2.dateFormat = @"yyyy-MM-dd HH:mm:ss";
  158. NSDateFormatter *formatter3 = [[NSDateFormatter alloc] init];
  159. formatter3.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  160. formatter3.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
  161. formatter3.dateFormat = @"yyyy-MM-dd'T'HH:mm:ss.SSS";
  162. NSDateFormatter *formatter4 = [[NSDateFormatter alloc] init];
  163. formatter4.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  164. formatter4.timeZone = [NSTimeZone timeZoneForSecondsFromGMT:0];
  165. formatter4.dateFormat = @"yyyy-MM-dd HH:mm:ss.SSS";
  166. blocks[19] = ^(NSString *string) {
  167. if ([string characterAtIndex:10] == 'T') {
  168. return [formatter1 dateFromString:string];
  169. } else {
  170. return [formatter2 dateFromString:string];
  171. }
  172. };
  173. blocks[23] = ^(NSString *string) {
  174. if ([string characterAtIndex:10] == 'T') {
  175. return [formatter3 dateFromString:string];
  176. } else {
  177. return [formatter4 dateFromString:string];
  178. }
  179. };
  180. }
  181. {
  182. /*
  183. 2014-01-20T12:24:48Z // Github, Apple
  184. 2014-01-20T12:24:48+0800 // Facebook
  185. 2014-01-20T12:24:48+12:00 // Google
  186. 2014-01-20T12:24:48.000Z
  187. 2014-01-20T12:24:48.000+0800
  188. 2014-01-20T12:24:48.000+12:00
  189. */
  190. NSDateFormatter *formatter = [NSDateFormatter new];
  191. formatter.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  192. formatter.dateFormat = @"yyyy-MM-dd'T'HH:mm:ssZ";
  193. NSDateFormatter *formatter2 = [NSDateFormatter new];
  194. formatter2.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  195. formatter2.dateFormat = @"yyyy-MM-dd'T'HH:mm:ss.SSSZ";
  196. blocks[20] = ^(NSString *string) { return [formatter dateFromString:string]; };
  197. blocks[24] = ^(NSString *string) { return [formatter dateFromString:string]?: [formatter2 dateFromString:string]; };
  198. blocks[25] = ^(NSString *string) { return [formatter dateFromString:string]; };
  199. blocks[28] = ^(NSString *string) { return [formatter2 dateFromString:string]; };
  200. blocks[29] = ^(NSString *string) { return [formatter2 dateFromString:string]; };
  201. }
  202. {
  203. /*
  204. Fri Sep 04 00:12:21 +0800 2015 // Weibo, Twitter
  205. Fri Sep 04 00:12:21.000 +0800 2015
  206. */
  207. NSDateFormatter *formatter = [NSDateFormatter new];
  208. formatter.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  209. formatter.dateFormat = @"EEE MMM dd HH:mm:ss Z yyyy";
  210. NSDateFormatter *formatter2 = [NSDateFormatter new];
  211. formatter2.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  212. formatter2.dateFormat = @"EEE MMM dd HH:mm:ss.SSS Z yyyy";
  213. blocks[30] = ^(NSString *string) { return [formatter dateFromString:string]; };
  214. blocks[34] = ^(NSString *string) { return [formatter2 dateFromString:string]; };
  215. }
  216. });
  217. if (!string) return nil;
  218. if (string.length > kParserNum) return nil;
  219. White_YYNSDateParseBlock parser = blocks[string.length];
  220. if (!parser) return nil;
  221. return parser(string);
  222. #undef kParserNum
  223. }
  224. /// Get the 'NSBlock' class.
  225. static force_inline Class White_YYNSBlockClass(void) {
  226. static Class cls;
  227. static dispatch_once_t onceToken;
  228. dispatch_once(&onceToken, ^{
  229. void (^block)(void) = ^{};
  230. cls = ((NSObject *)block).class;
  231. while (class_getSuperclass(cls) != [NSObject class]) {
  232. cls = class_getSuperclass(cls);
  233. }
  234. });
  235. return cls; // current is "NSBlock"
  236. }
  237. /**
  238. Get the ISO date formatter.
  239. ISO8601 format example:
  240. 2010-07-09T16:13:30+12:00
  241. 2011-01-11T11:11:11+0000
  242. 2011-01-26T19:06:43Z
  243. length: 20/24/25
  244. */
  245. static force_inline NSDateFormatter *White_YYISODateFormatter(void) {
  246. static NSDateFormatter *formatter = nil;
  247. static dispatch_once_t onceToken;
  248. dispatch_once(&onceToken, ^{
  249. formatter = [[NSDateFormatter alloc] init];
  250. formatter.locale = [[NSLocale alloc] initWithLocaleIdentifier:@"en_US_POSIX"];
  251. formatter.dateFormat = @"yyyy-MM-dd'T'HH:mm:ssZ";
  252. });
  253. return formatter;
  254. }
  255. /// Get the value with key paths from dictionary
  256. /// The dic should be NSDictionary, and the keyPath should not be nil.
  257. static force_inline id White_YYValueForKeyPath(__unsafe_unretained NSDictionary *dic, __unsafe_unretained NSArray *keyPaths) {
  258. id value = nil;
  259. for (NSUInteger i = 0, max = keyPaths.count; i < max; i++) {
  260. value = dic[keyPaths[i]];
  261. if (i + 1 < max) {
  262. if ([value isKindOfClass:[NSDictionary class]]) {
  263. dic = value;
  264. } else {
  265. return nil;
  266. }
  267. }
  268. }
  269. return value;
  270. }
  271. /// Get the value with multi key (or key path) from dictionary
  272. /// The dic should be NSDictionary
  273. static force_inline id YYValueForMultiKeys(__unsafe_unretained NSDictionary *dic, __unsafe_unretained NSArray *multiKeys) {
  274. id value = nil;
  275. for (NSString *key in multiKeys) {
  276. if ([key isKindOfClass:[NSString class]]) {
  277. value = dic[key];
  278. if (value) break;
  279. } else {
  280. value = White_YYValueForKeyPath(dic, (NSArray *)key);
  281. if (value) break;
  282. }
  283. }
  284. return value;
  285. }
  286. /// A property info in object model.
  287. @interface _White_YYModelPropertyMeta : NSObject {
  288. @package
  289. NSString *_name; ///< property's name
  290. White_YYEncodingType _type; ///< property's type
  291. White_YYEncodingNSType _nsType; ///< property's Foundation type
  292. BOOL _isCNumber; ///< is c number type
  293. Class _cls; ///< property's class, or nil
  294. Class _genericCls; ///< container's generic class, or nil if threr's no generic class
  295. SEL _getter; ///< getter, or nil if the instances cannot respond
  296. SEL _setter; ///< setter, or nil if the instances cannot respond
  297. BOOL _isKVCCompatible; ///< YES if it can access with key-value coding
  298. BOOL _isStructAvailableForKeyedArchiver; ///< YES if the struct can encoded with keyed archiver/unarchiver
  299. BOOL _hasCustomClassFromDictionary; ///< class/generic class implements +modelCustomClassForDictionary:
  300. /*
  301. property->key: _mappedToKey:key _mappedToKeyPath:nil _mappedToKeyArray:nil
  302. property->keyPath: _mappedToKey:keyPath _mappedToKeyPath:keyPath(array) _mappedToKeyArray:nil
  303. property->keys: _mappedToKey:keys[0] _mappedToKeyPath:nil/keyPath _mappedToKeyArray:keys(array)
  304. */
  305. NSString *_mappedToKey; ///< the key mapped to
  306. NSArray *_mappedToKeyPath; ///< the key path mapped to (nil if the name is not key path)
  307. NSArray *_mappedToKeyArray; ///< the key(NSString) or keyPath(NSArray) array (nil if not mapped to multiple keys)
  308. White_YYClassPropertyInfo *_info; ///< property's info
  309. _White_YYModelPropertyMeta *_next; ///< next meta if there are multiple properties mapped to the same key.
  310. }
  311. @end
  312. @implementation _White_YYModelPropertyMeta
  313. + (instancetype)metaWithClassInfo:(White_YYClassInfo *)classInfo propertyInfo:(White_YYClassPropertyInfo *)propertyInfo generic:(Class)generic {
  314. // support pseudo generic class with protocol name
  315. if (!generic && propertyInfo.protocols) {
  316. for (NSString *protocol in propertyInfo.protocols) {
  317. Class cls = objc_getClass(protocol.UTF8String);
  318. if (cls) {
  319. generic = cls;
  320. break;
  321. }
  322. }
  323. }
  324. _White_YYModelPropertyMeta *meta = [self new];
  325. meta->_name = propertyInfo.name;
  326. meta->_type = propertyInfo.type;
  327. meta->_info = propertyInfo;
  328. meta->_genericCls = generic;
  329. if ((meta->_type & White_YYEncodingTypeMask) == White_YYEncodingTypeObject) {
  330. meta->_nsType = White_YYClassGetNSType(propertyInfo.cls);
  331. } else {
  332. meta->_isCNumber = White_YYEncodingTypeIsCNumber(meta->_type);
  333. }
  334. if ((meta->_type & White_YYEncodingTypeMask) == White_YYEncodingTypeStruct) {
  335. /*
  336. It seems that NSKeyedUnarchiver cannot decode NSValue except these structs:
  337. */
  338. static NSSet *types = nil;
  339. static dispatch_once_t onceToken;
  340. dispatch_once(&onceToken, ^{
  341. NSMutableSet *set = [NSMutableSet new];
  342. // 32 bit
  343. [set addObject:@"{CGSize=ff}"];
  344. [set addObject:@"{CGPoint=ff}"];
  345. [set addObject:@"{CGRect={CGPoint=ff}{CGSize=ff}}"];
  346. [set addObject:@"{CGAffineTransform=ffffff}"];
  347. [set addObject:@"{UIEdgeInsets=ffff}"];
  348. [set addObject:@"{UIOffset=ff}"];
  349. // 64 bit
  350. [set addObject:@"{CGSize=dd}"];
  351. [set addObject:@"{CGPoint=dd}"];
  352. [set addObject:@"{CGRect={CGPoint=dd}{CGSize=dd}}"];
  353. [set addObject:@"{CGAffineTransform=dddddd}"];
  354. [set addObject:@"{UIEdgeInsets=dddd}"];
  355. [set addObject:@"{UIOffset=dd}"];
  356. types = set;
  357. });
  358. if ([types containsObject:propertyInfo.typeEncoding]) {
  359. meta->_isStructAvailableForKeyedArchiver = YES;
  360. }
  361. }
  362. meta->_cls = propertyInfo.cls;
  363. if (generic) {
  364. meta->_hasCustomClassFromDictionary = [generic respondsToSelector:@selector(modelCustomClassForDictionary:)];
  365. } else if (meta->_cls && meta->_nsType == White_YYEncodingTypeNSUnknown) {
  366. meta->_hasCustomClassFromDictionary = [meta->_cls respondsToSelector:@selector(modelCustomClassForDictionary:)];
  367. }
  368. if (propertyInfo.getter) {
  369. if ([classInfo.cls instancesRespondToSelector:propertyInfo.getter]) {
  370. meta->_getter = propertyInfo.getter;
  371. }
  372. }
  373. if (propertyInfo.setter) {
  374. if ([classInfo.cls instancesRespondToSelector:propertyInfo.setter]) {
  375. meta->_setter = propertyInfo.setter;
  376. }
  377. }
  378. if (meta->_getter && meta->_setter) {
  379. /*
  380. KVC invalid type:
  381. long double
  382. pointer (such as SEL/CoreFoundation object)
  383. */
  384. switch (meta->_type & White_YYEncodingTypeMask) {
  385. case White_YYEncodingTypeBool:
  386. case White_YYEncodingTypeInt8:
  387. case White_YYEncodingTypeUInt8:
  388. case White_YYEncodingTypeInt16:
  389. case White_YYEncodingTypeUInt16:
  390. case White_YYEncodingTypeInt32:
  391. case White_YYEncodingTypeUInt32:
  392. case White_YYEncodingTypeInt64:
  393. case White_YYEncodingTypeUInt64:
  394. case White_YYEncodingTypeFloat:
  395. case White_YYEncodingTypeDouble:
  396. case White_YYEncodingTypeObject:
  397. case White_YYEncodingTypeClass:
  398. case White_YYEncodingTypeBlock:
  399. case White_YYEncodingTypeStruct:
  400. case White_YYEncodingTypeUnion: {
  401. meta->_isKVCCompatible = YES;
  402. } break;
  403. default: break;
  404. }
  405. }
  406. return meta;
  407. }
  408. @end
  409. /// A class info in object model.
  410. @interface _White_YYModelMeta : NSObject {
  411. @package
  412. White_YYClassInfo *_classInfo;
  413. /// Key:mapped key and key path, Value:_YYModelPropertyMeta.
  414. NSDictionary *_mapper;
  415. /// Array<_YYModelPropertyMeta>, all property meta of this model.
  416. NSArray *_allPropertyMetas;
  417. /// Array<_YYModelPropertyMeta>, property meta which is mapped to a key path.
  418. NSArray *_keyPathPropertyMetas;
  419. /// Array<_YYModelPropertyMeta>, property meta which is mapped to multi keys.
  420. NSArray *_multiKeysPropertyMetas;
  421. /// The number of mapped key (and key path), same to _mapper.count.
  422. NSUInteger _keyMappedCount;
  423. /// Model class type.
  424. White_YYEncodingNSType _nsType;
  425. BOOL _hasCustomWillTransformFromDictionary;
  426. BOOL _hasCustomTransformFromDictionary;
  427. BOOL _hasCustomTransformToDictionary;
  428. BOOL _hasCustomClassFromDictionary;
  429. }
  430. @end
  431. @implementation _White_YYModelMeta
  432. - (instancetype)initWithClass:(Class)cls {
  433. White_YYClassInfo *classInfo = [White_YYClassInfo classInfoWithClass:cls];
  434. if (!classInfo) return nil;
  435. self = [super init];
  436. // Get black list
  437. NSSet *blacklist = nil;
  438. if ([cls respondsToSelector:@selector(modelPropertyBlacklist)]) {
  439. NSArray *properties = [(id<White_YYModel>)cls modelPropertyBlacklist];
  440. if (properties) {
  441. blacklist = [NSSet setWithArray:properties];
  442. }
  443. }
  444. // Get white list
  445. NSSet *whitelist = nil;
  446. if ([cls respondsToSelector:@selector(modelPropertyWhitelist)]) {
  447. NSArray *properties = [(id<White_YYModel>)cls modelPropertyWhitelist];
  448. if (properties) {
  449. whitelist = [NSSet setWithArray:properties];
  450. }
  451. }
  452. // Get container property's generic class
  453. NSDictionary *genericMapper = nil;
  454. if ([cls respondsToSelector:@selector(modelContainerPropertyGenericClass)]) {
  455. genericMapper = [(id<White_YYModel>)cls modelContainerPropertyGenericClass];
  456. if (genericMapper) {
  457. NSMutableDictionary *tmp = [NSMutableDictionary new];
  458. [genericMapper enumerateKeysAndObjectsUsingBlock:^(id key, id obj, BOOL *stop) {
  459. if (![key isKindOfClass:[NSString class]]) return;
  460. Class meta = object_getClass(obj);
  461. if (!meta) return;
  462. if (class_isMetaClass(meta)) {
  463. tmp[key] = obj;
  464. } else if ([obj isKindOfClass:[NSString class]]) {
  465. Class cls = NSClassFromString(obj);
  466. if (cls) {
  467. tmp[key] = cls;
  468. }
  469. }
  470. }];
  471. genericMapper = tmp;
  472. }
  473. }
  474. // Create all property metas.
  475. NSMutableDictionary *allPropertyMetas = [NSMutableDictionary new];
  476. White_YYClassInfo *curClassInfo = classInfo;
  477. while (curClassInfo && curClassInfo.superCls != nil) { // recursive parse super class, but ignore root class (NSObject/NSProxy)
  478. for (White_YYClassPropertyInfo *propertyInfo in curClassInfo.propertyInfos.allValues) {
  479. if (!propertyInfo.name) continue;
  480. if (blacklist && [blacklist containsObject:propertyInfo.name]) continue;
  481. if (whitelist && ![whitelist containsObject:propertyInfo.name]) continue;
  482. _White_YYModelPropertyMeta *meta = [_White_YYModelPropertyMeta metaWithClassInfo:classInfo
  483. propertyInfo:propertyInfo
  484. generic:genericMapper[propertyInfo.name]];
  485. if (!meta || !meta->_name) continue;
  486. if (!meta->_getter || !meta->_setter) continue;
  487. if (allPropertyMetas[meta->_name]) continue;
  488. allPropertyMetas[meta->_name] = meta;
  489. }
  490. curClassInfo = curClassInfo.superClassInfo;
  491. }
  492. if (allPropertyMetas.count) _allPropertyMetas = allPropertyMetas.allValues.copy;
  493. // create mapper
  494. NSMutableDictionary *mapper = [NSMutableDictionary new];
  495. NSMutableArray *keyPathPropertyMetas = [NSMutableArray new];
  496. NSMutableArray *multiKeysPropertyMetas = [NSMutableArray new];
  497. if ([cls respondsToSelector:@selector(modelCustomPropertyMapper)]) {
  498. NSDictionary *customMapper = [(id <White_YYModel>)cls modelCustomPropertyMapper];
  499. [customMapper enumerateKeysAndObjectsUsingBlock:^(NSString *propertyName, NSString *mappedToKey, BOOL *stop) {
  500. _White_YYModelPropertyMeta *propertyMeta = allPropertyMetas[propertyName];
  501. if (!propertyMeta) return;
  502. [allPropertyMetas removeObjectForKey:propertyName];
  503. if ([mappedToKey isKindOfClass:[NSString class]]) {
  504. if (mappedToKey.length == 0) return;
  505. propertyMeta->_mappedToKey = mappedToKey;
  506. NSArray *keyPath = [mappedToKey componentsSeparatedByString:@"."];
  507. for (NSString *onePath in keyPath) {
  508. if (onePath.length == 0) {
  509. NSMutableArray *tmp = keyPath.mutableCopy;
  510. [tmp removeObject:@""];
  511. keyPath = tmp;
  512. break;
  513. }
  514. }
  515. if (keyPath.count > 1) {
  516. propertyMeta->_mappedToKeyPath = keyPath;
  517. [keyPathPropertyMetas addObject:propertyMeta];
  518. }
  519. propertyMeta->_next = mapper[mappedToKey] ?: nil;
  520. mapper[mappedToKey] = propertyMeta;
  521. } else if ([mappedToKey isKindOfClass:[NSArray class]]) {
  522. NSMutableArray *mappedToKeyArray = [NSMutableArray new];
  523. for (NSString *oneKey in ((NSArray *)mappedToKey)) {
  524. if (![oneKey isKindOfClass:[NSString class]]) continue;
  525. if (oneKey.length == 0) continue;
  526. NSArray *keyPath = [oneKey componentsSeparatedByString:@"."];
  527. if (keyPath.count > 1) {
  528. [mappedToKeyArray addObject:keyPath];
  529. } else {
  530. [mappedToKeyArray addObject:oneKey];
  531. }
  532. if (!propertyMeta->_mappedToKey) {
  533. propertyMeta->_mappedToKey = oneKey;
  534. propertyMeta->_mappedToKeyPath = keyPath.count > 1 ? keyPath : nil;
  535. }
  536. }
  537. if (!propertyMeta->_mappedToKey) return;
  538. propertyMeta->_mappedToKeyArray = mappedToKeyArray;
  539. [multiKeysPropertyMetas addObject:propertyMeta];
  540. propertyMeta->_next = mapper[mappedToKey] ?: nil;
  541. mapper[mappedToKey] = propertyMeta;
  542. }
  543. }];
  544. }
  545. [allPropertyMetas enumerateKeysAndObjectsUsingBlock:^(NSString *name, _White_YYModelPropertyMeta *propertyMeta, BOOL *stop) {
  546. propertyMeta->_mappedToKey = name;
  547. propertyMeta->_next = mapper[name] ?: nil;
  548. mapper[name] = propertyMeta;
  549. }];
  550. if (mapper.count) _mapper = mapper;
  551. if (keyPathPropertyMetas) _keyPathPropertyMetas = keyPathPropertyMetas;
  552. if (multiKeysPropertyMetas) _multiKeysPropertyMetas = multiKeysPropertyMetas;
  553. _classInfo = classInfo;
  554. _keyMappedCount = _allPropertyMetas.count;
  555. _nsType = White_YYClassGetNSType(cls);
  556. _hasCustomWillTransformFromDictionary = ([cls instancesRespondToSelector:@selector(modelCustomWillTransformFromDictionary:)]);
  557. _hasCustomTransformFromDictionary = ([cls instancesRespondToSelector:@selector(modelCustomTransformFromDictionary:)]);
  558. _hasCustomTransformToDictionary = ([cls instancesRespondToSelector:@selector(modelCustomTransformToDictionary:)]);
  559. _hasCustomClassFromDictionary = ([cls respondsToSelector:@selector(modelCustomClassForDictionary:)]);
  560. return self;
  561. }
  562. /// Returns the cached model class meta
  563. + (instancetype)metaWithClass:(Class)cls {
  564. if (!cls) return nil;
  565. static CFMutableDictionaryRef cache;
  566. static dispatch_once_t onceToken;
  567. static dispatch_semaphore_t lock;
  568. dispatch_once(&onceToken, ^{
  569. cache = CFDictionaryCreateMutable(CFAllocatorGetDefault(), 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
  570. lock = dispatch_semaphore_create(1);
  571. });
  572. dispatch_semaphore_wait(lock, DISPATCH_TIME_FOREVER);
  573. _White_YYModelMeta *meta = CFDictionaryGetValue(cache, (__bridge const void *)(cls));
  574. dispatch_semaphore_signal(lock);
  575. if (!meta || meta->_classInfo.needUpdate) {
  576. meta = [[_White_YYModelMeta alloc] initWithClass:cls];
  577. if (meta) {
  578. dispatch_semaphore_wait(lock, DISPATCH_TIME_FOREVER);
  579. CFDictionarySetValue(cache, (__bridge const void *)(cls), (__bridge const void *)(meta));
  580. dispatch_semaphore_signal(lock);
  581. }
  582. }
  583. return meta;
  584. }
  585. @end
  586. /**
  587. Get number from property.
  588. @discussion Caller should hold strong reference to the parameters before this function returns.
  589. @param model Should not be nil.
  590. @param meta Should not be nil, meta.isCNumber should be YES, meta.getter should not be nil.
  591. @return A number object, or nil if failed.
  592. */
  593. static force_inline NSNumber *White_ModelCreateNumberFromProperty(__unsafe_unretained id model,
  594. __unsafe_unretained _White_YYModelPropertyMeta *meta) {
  595. switch (meta->_type & White_YYEncodingTypeMask) {
  596. case White_YYEncodingTypeBool: {
  597. return @(((bool (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  598. }
  599. case White_YYEncodingTypeInt8: {
  600. return @(((int8_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  601. }
  602. case White_YYEncodingTypeUInt8: {
  603. return @(((uint8_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  604. }
  605. case White_YYEncodingTypeInt16: {
  606. return @(((int16_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  607. }
  608. case White_YYEncodingTypeUInt16: {
  609. return @(((uint16_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  610. }
  611. case White_YYEncodingTypeInt32: {
  612. return @(((int32_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  613. }
  614. case White_YYEncodingTypeUInt32: {
  615. return @(((uint32_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  616. }
  617. case White_YYEncodingTypeInt64: {
  618. return @(((int64_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  619. }
  620. case White_YYEncodingTypeUInt64: {
  621. return @(((uint64_t (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter));
  622. }
  623. case White_YYEncodingTypeFloat: {
  624. float num = ((float (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter);
  625. if (isnan(num) || isinf(num)) return nil;
  626. return @(num);
  627. }
  628. case White_YYEncodingTypeDouble: {
  629. double num = ((double (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter);
  630. if (isnan(num) || isinf(num)) return nil;
  631. return @(num);
  632. }
  633. case White_YYEncodingTypeLongDouble: {
  634. double num = ((long double (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter);
  635. if (isnan(num) || isinf(num)) return nil;
  636. return @(num);
  637. }
  638. default: return nil;
  639. }
  640. }
  641. /**
  642. Set number to property.
  643. @discussion Caller should hold strong reference to the parameters before this function returns.
  644. @param model Should not be nil.
  645. @param num Can be nil.
  646. @param meta Should not be nil, meta.isCNumber should be YES, meta.setter should not be nil.
  647. */
  648. static force_inline void White_ModelSetNumberToProperty(__unsafe_unretained id model,
  649. __unsafe_unretained NSNumber *num,
  650. __unsafe_unretained _White_YYModelPropertyMeta *meta) {
  651. switch (meta->_type & White_YYEncodingTypeMask) {
  652. case White_YYEncodingTypeBool: {
  653. ((void (*)(id, SEL, bool))(void *) objc_msgSend)((id)model, meta->_setter, num.boolValue);
  654. } break;
  655. case White_YYEncodingTypeInt8: {
  656. ((void (*)(id, SEL, int8_t))(void *) objc_msgSend)((id)model, meta->_setter, (int8_t)num.charValue);
  657. } break;
  658. case White_YYEncodingTypeUInt8: {
  659. ((void (*)(id, SEL, uint8_t))(void *) objc_msgSend)((id)model, meta->_setter, (uint8_t)num.unsignedCharValue);
  660. } break;
  661. case White_YYEncodingTypeInt16: {
  662. ((void (*)(id, SEL, int16_t))(void *) objc_msgSend)((id)model, meta->_setter, (int16_t)num.shortValue);
  663. } break;
  664. case White_YYEncodingTypeUInt16: {
  665. ((void (*)(id, SEL, uint16_t))(void *) objc_msgSend)((id)model, meta->_setter, (uint16_t)num.unsignedShortValue);
  666. } break;
  667. case White_YYEncodingTypeInt32: {
  668. ((void (*)(id, SEL, int32_t))(void *) objc_msgSend)((id)model, meta->_setter, (int32_t)num.intValue);
  669. }
  670. case White_YYEncodingTypeUInt32: {
  671. ((void (*)(id, SEL, uint32_t))(void *) objc_msgSend)((id)model, meta->_setter, (uint32_t)num.unsignedIntValue);
  672. } break;
  673. case White_YYEncodingTypeInt64: {
  674. if ([num isKindOfClass:[NSDecimalNumber class]]) {
  675. ((void (*)(id, SEL, int64_t))(void *) objc_msgSend)((id)model, meta->_setter, (int64_t)num.stringValue.longLongValue);
  676. } else {
  677. ((void (*)(id, SEL, uint64_t))(void *) objc_msgSend)((id)model, meta->_setter, (uint64_t)num.longLongValue);
  678. }
  679. } break;
  680. case White_YYEncodingTypeUInt64: {
  681. if ([num isKindOfClass:[NSDecimalNumber class]]) {
  682. ((void (*)(id, SEL, int64_t))(void *) objc_msgSend)((id)model, meta->_setter, (int64_t)num.stringValue.longLongValue);
  683. } else {
  684. ((void (*)(id, SEL, uint64_t))(void *) objc_msgSend)((id)model, meta->_setter, (uint64_t)num.unsignedLongLongValue);
  685. }
  686. } break;
  687. case White_YYEncodingTypeFloat: {
  688. float f = num.floatValue;
  689. if (isnan(f) || isinf(f)) f = 0;
  690. ((void (*)(id, SEL, float))(void *) objc_msgSend)((id)model, meta->_setter, f);
  691. } break;
  692. case White_YYEncodingTypeDouble: {
  693. double d = num.doubleValue;
  694. if (isnan(d) || isinf(d)) d = 0;
  695. ((void (*)(id, SEL, double))(void *) objc_msgSend)((id)model, meta->_setter, d);
  696. } break;
  697. case White_YYEncodingTypeLongDouble: {
  698. long double d = num.doubleValue;
  699. if (isnan(d) || isinf(d)) d = 0;
  700. ((void (*)(id, SEL, long double))(void *) objc_msgSend)((id)model, meta->_setter, (long double)d);
  701. } // break; commented for code coverage in next line
  702. default: break;
  703. }
  704. }
  705. /**
  706. Set value to model with a property meta.
  707. @discussion Caller should hold strong reference to the parameters before this function returns.
  708. @param model Should not be nil.
  709. @param value Should not be nil, but can be NSNull.
  710. @param meta Should not be nil, and meta->_setter should not be nil.
  711. */
  712. static void White_ModelSetValueForProperty(__unsafe_unretained id model,
  713. __unsafe_unretained id value,
  714. __unsafe_unretained _White_YYModelPropertyMeta *meta) {
  715. if (meta->_isCNumber) {
  716. NSNumber *num = White_YYNSNumberCreateFromID(value);
  717. White_ModelSetNumberToProperty(model, num, meta);
  718. if (num) [num class]; // hold the number
  719. } else if (meta->_nsType) {
  720. if (value == (id)kCFNull) {
  721. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, (id)nil);
  722. } else {
  723. switch (meta->_nsType) {
  724. case White_YYEncodingTypeNSString:
  725. case White_YYEncodingTypeNSMutableString: {
  726. if ([value isKindOfClass:[NSString class]]) {
  727. if (meta->_nsType == White_YYEncodingTypeNSString) {
  728. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  729. } else {
  730. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, ((NSString *)value).mutableCopy);
  731. }
  732. } else if ([value isKindOfClass:[NSNumber class]]) {
  733. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  734. meta->_setter,
  735. (meta->_nsType == White_YYEncodingTypeNSString) ?
  736. ((NSNumber *)value).stringValue :
  737. ((NSNumber *)value).stringValue.mutableCopy);
  738. } else if ([value isKindOfClass:[NSData class]]) {
  739. NSMutableString *string = [[NSMutableString alloc] initWithData:value encoding:NSUTF8StringEncoding];
  740. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, string);
  741. } else if ([value isKindOfClass:[NSURL class]]) {
  742. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  743. meta->_setter,
  744. (meta->_nsType == White_YYEncodingTypeNSString) ?
  745. ((NSURL *)value).absoluteString :
  746. ((NSURL *)value).absoluteString.mutableCopy);
  747. } else if ([value isKindOfClass:[NSAttributedString class]]) {
  748. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  749. meta->_setter,
  750. (meta->_nsType == White_YYEncodingTypeNSString) ?
  751. ((NSAttributedString *)value).string :
  752. ((NSAttributedString *)value).string.mutableCopy);
  753. }
  754. } break;
  755. case White_YYEncodingTypeNSValue:
  756. case White_YYEncodingTypeNSNumber:
  757. case White_YYEncodingTypeNSDecimalNumber: {
  758. if (meta->_nsType == White_YYEncodingTypeNSNumber) {
  759. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, White_YYNSNumberCreateFromID(value));
  760. } else if (meta->_nsType == White_YYEncodingTypeNSDecimalNumber) {
  761. if ([value isKindOfClass:[NSDecimalNumber class]]) {
  762. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  763. } else if ([value isKindOfClass:[NSNumber class]]) {
  764. NSDecimalNumber *decNum = [NSDecimalNumber decimalNumberWithDecimal:[((NSNumber *)value) decimalValue]];
  765. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, decNum);
  766. } else if ([value isKindOfClass:[NSString class]]) {
  767. NSDecimalNumber *decNum = [NSDecimalNumber decimalNumberWithString:value];
  768. NSDecimal dec = decNum.decimalValue;
  769. if (dec._length == 0 && dec._isNegative) {
  770. decNum = nil; // NaN
  771. }
  772. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, decNum);
  773. }
  774. } else { // YYEncodingTypeNSValue
  775. if ([value isKindOfClass:[NSValue class]]) {
  776. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  777. }
  778. }
  779. } break;
  780. case White_YYEncodingTypeNSData:
  781. case White_YYEncodingTypeNSMutableData: {
  782. if ([value isKindOfClass:[NSData class]]) {
  783. if (meta->_nsType == White_YYEncodingTypeNSData) {
  784. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  785. } else {
  786. NSMutableData *data = ((NSData *)value).mutableCopy;
  787. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, data);
  788. }
  789. } else if ([value isKindOfClass:[NSString class]]) {
  790. NSData *data = [(NSString *)value dataUsingEncoding:NSUTF8StringEncoding];
  791. if (meta->_nsType == White_YYEncodingTypeNSMutableData) {
  792. data = ((NSData *)data).mutableCopy;
  793. }
  794. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, data);
  795. }
  796. } break;
  797. case White_YYEncodingTypeNSDate: {
  798. if ([value isKindOfClass:[NSDate class]]) {
  799. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  800. } else if ([value isKindOfClass:[NSString class]]) {
  801. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, White_YYNSDateFromString(value));
  802. }
  803. } break;
  804. case White_YYEncodingTypeNSURL: {
  805. if ([value isKindOfClass:[NSURL class]]) {
  806. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  807. } else if ([value isKindOfClass:[NSString class]]) {
  808. NSCharacterSet *set = [NSCharacterSet whitespaceAndNewlineCharacterSet];
  809. NSString *str = [value stringByTrimmingCharactersInSet:set];
  810. if (str.length == 0) {
  811. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, nil);
  812. } else {
  813. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, [[NSURL alloc] initWithString:str]);
  814. }
  815. }
  816. } break;
  817. case White_YYEncodingTypeNSArray:
  818. case White_YYEncodingTypeNSMutableArray: {
  819. if (meta->_genericCls) {
  820. NSArray *valueArr = nil;
  821. if ([value isKindOfClass:[NSArray class]]) valueArr = value;
  822. else if ([value isKindOfClass:[NSSet class]]) valueArr = ((NSSet *)value).allObjects;
  823. if (valueArr) {
  824. NSMutableArray *objectArr = [NSMutableArray new];
  825. for (id one in valueArr) {
  826. if ([one isKindOfClass:meta->_genericCls]) {
  827. [objectArr addObject:one];
  828. } else if ([one isKindOfClass:[NSDictionary class]]) {
  829. Class cls = meta->_genericCls;
  830. if (meta->_hasCustomClassFromDictionary) {
  831. cls = [cls modelCustomClassForDictionary:one];
  832. if (!cls) cls = meta->_genericCls; // for xcode code coverage
  833. }
  834. NSObject *newOne = [cls new];
  835. [newOne _white_yy_modelSetWithDictionary:one];
  836. if (newOne) [objectArr addObject:newOne];
  837. }
  838. }
  839. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, objectArr);
  840. }
  841. } else {
  842. if ([value isKindOfClass:[NSArray class]]) {
  843. if (meta->_nsType == White_YYEncodingTypeNSArray) {
  844. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  845. } else {
  846. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  847. meta->_setter,
  848. ((NSArray *)value).mutableCopy);
  849. }
  850. } else if ([value isKindOfClass:[NSSet class]]) {
  851. if (meta->_nsType == White_YYEncodingTypeNSArray) {
  852. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, ((NSSet *)value).allObjects);
  853. } else {
  854. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  855. meta->_setter,
  856. ((NSSet *)value).allObjects.mutableCopy);
  857. }
  858. }
  859. }
  860. } break;
  861. case White_YYEncodingTypeNSDictionary:
  862. case White_YYEncodingTypeNSMutableDictionary: {
  863. if ([value isKindOfClass:[NSDictionary class]]) {
  864. if (meta->_genericCls) {
  865. NSMutableDictionary *dic = [NSMutableDictionary new];
  866. [((NSDictionary *)value) enumerateKeysAndObjectsUsingBlock:^(NSString *oneKey, id oneValue, BOOL *stop) {
  867. if ([oneValue isKindOfClass:[NSDictionary class]]) {
  868. Class cls = meta->_genericCls;
  869. if (meta->_hasCustomClassFromDictionary) {
  870. cls = [cls modelCustomClassForDictionary:oneValue];
  871. if (!cls) cls = meta->_genericCls; // for xcode code coverage
  872. }
  873. NSObject *newOne = [cls new];
  874. [newOne _white_yy_modelSetWithDictionary:(id)oneValue];
  875. if (newOne) dic[oneKey] = newOne;
  876. }
  877. }];
  878. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, dic);
  879. } else {
  880. if (meta->_nsType == White_YYEncodingTypeNSDictionary) {
  881. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, value);
  882. } else {
  883. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  884. meta->_setter,
  885. ((NSDictionary *)value).mutableCopy);
  886. }
  887. }
  888. }
  889. } break;
  890. case White_YYEncodingTypeNSSet:
  891. case White_YYEncodingTypeNSMutableSet: {
  892. NSSet *valueSet = nil;
  893. if ([value isKindOfClass:[NSArray class]]) valueSet = [NSMutableSet setWithArray:value];
  894. else if ([value isKindOfClass:[NSSet class]]) valueSet = ((NSSet *)value);
  895. if (meta->_genericCls) {
  896. NSMutableSet *set = [NSMutableSet new];
  897. for (id one in valueSet) {
  898. if ([one isKindOfClass:meta->_genericCls]) {
  899. [set addObject:one];
  900. } else if ([one isKindOfClass:[NSDictionary class]]) {
  901. Class cls = meta->_genericCls;
  902. if (meta->_hasCustomClassFromDictionary) {
  903. cls = [cls modelCustomClassForDictionary:one];
  904. if (!cls) cls = meta->_genericCls; // for xcode code coverage
  905. }
  906. NSObject *newOne = [cls new];
  907. [newOne _white_yy_modelSetWithDictionary:one];
  908. if (newOne) [set addObject:newOne];
  909. }
  910. }
  911. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, set);
  912. } else {
  913. if (meta->_nsType == White_YYEncodingTypeNSSet) {
  914. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, valueSet);
  915. } else {
  916. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model,
  917. meta->_setter,
  918. ((NSSet *)valueSet).mutableCopy);
  919. }
  920. }
  921. } // break; commented for code coverage in next line
  922. default: break;
  923. }
  924. }
  925. } else {
  926. BOOL isNull = (value == (id)kCFNull);
  927. switch (meta->_type & White_YYEncodingTypeMask) {
  928. case White_YYEncodingTypeObject: {
  929. if (isNull) {
  930. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, (id)nil);
  931. } else if ([value isKindOfClass:meta->_cls] || !meta->_cls) {
  932. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, (id)value);
  933. } else if ([value isKindOfClass:[NSDictionary class]]) {
  934. NSObject *one = nil;
  935. if (meta->_getter) {
  936. one = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, meta->_getter);
  937. }
  938. if (one) {
  939. [one _white_yy_modelSetWithDictionary:value];
  940. } else {
  941. Class cls = meta->_cls;
  942. if (meta->_hasCustomClassFromDictionary) {
  943. cls = [cls modelCustomClassForDictionary:value];
  944. if (!cls) cls = meta->_genericCls; // for xcode code coverage
  945. }
  946. one = [cls new];
  947. [one _white_yy_modelSetWithDictionary:value];
  948. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)model, meta->_setter, (id)one);
  949. }
  950. }
  951. } break;
  952. case White_YYEncodingTypeClass: {
  953. if (isNull) {
  954. ((void (*)(id, SEL, Class))(void *) objc_msgSend)((id)model, meta->_setter, (Class)NULL);
  955. } else {
  956. Class cls = nil;
  957. if ([value isKindOfClass:[NSString class]]) {
  958. cls = NSClassFromString(value);
  959. if (cls) {
  960. ((void (*)(id, SEL, Class))(void *) objc_msgSend)((id)model, meta->_setter, (Class)cls);
  961. }
  962. } else {
  963. cls = object_getClass(value);
  964. if (cls) {
  965. if (class_isMetaClass(cls)) {
  966. ((void (*)(id, SEL, Class))(void *) objc_msgSend)((id)model, meta->_setter, (Class)value);
  967. }
  968. }
  969. }
  970. }
  971. } break;
  972. case White_YYEncodingTypeSEL: {
  973. if (isNull) {
  974. ((void (*)(id, SEL, SEL))(void *) objc_msgSend)((id)model, meta->_setter, (SEL)NULL);
  975. } else if ([value isKindOfClass:[NSString class]]) {
  976. SEL sel = NSSelectorFromString(value);
  977. if (sel) ((void (*)(id, SEL, SEL))(void *) objc_msgSend)((id)model, meta->_setter, (SEL)sel);
  978. }
  979. } break;
  980. case White_YYEncodingTypeBlock: {
  981. if (isNull) {
  982. ((void (*)(id, SEL, void (^)(void)))(void *) objc_msgSend)((id)model, meta->_setter, (void (^)(void))NULL);
  983. } else if ([value isKindOfClass:White_YYNSBlockClass()]) {
  984. ((void (*)(id, SEL, void (^)(void)))(void *) objc_msgSend)((id)model, meta->_setter, (void (^)(void))value);
  985. }
  986. } break;
  987. case White_YYEncodingTypeStruct:
  988. case White_YYEncodingTypeUnion:
  989. case White_YYEncodingTypeCArray: {
  990. if ([value isKindOfClass:[NSValue class]]) {
  991. const char *valueType = ((NSValue *)value).objCType;
  992. const char *metaType = meta->_info.typeEncoding.UTF8String;
  993. if (valueType && metaType && strcmp(valueType, metaType) == 0) {
  994. [model setValue:value forKey:meta->_name];
  995. }
  996. }
  997. } break;
  998. case White_YYEncodingTypePointer:
  999. case White_YYEncodingTypeCString: {
  1000. if (isNull) {
  1001. ((void (*)(id, SEL, void *))(void *) objc_msgSend)((id)model, meta->_setter, (void *)NULL);
  1002. } else if ([value isKindOfClass:[NSValue class]]) {
  1003. NSValue *nsValue = value;
  1004. if (nsValue.objCType && strcmp(nsValue.objCType, "^v") == 0) {
  1005. ((void (*)(id, SEL, void *))(void *) objc_msgSend)((id)model, meta->_setter, nsValue.pointerValue);
  1006. }
  1007. }
  1008. } // break; commented for code coverage in next line
  1009. default: break;
  1010. }
  1011. }
  1012. }
  1013. typedef struct {
  1014. void *modelMeta; ///< _YYModelMeta
  1015. void *model; ///< id (self)
  1016. void *dictionary; ///< NSDictionary (json)
  1017. } White_ModelSetContext;
  1018. /**
  1019. Apply function for dictionary, to set the key-value pair to model.
  1020. @param _key should not be nil, NSString.
  1021. @param _value should not be nil.
  1022. @param _context _context.modelMeta and _context.model should not be nil.
  1023. */
  1024. static void White_ModelSetWithDictionaryFunction(const void *_key, const void *_value, void *_context) {
  1025. White_ModelSetContext *context = _context;
  1026. __unsafe_unretained _White_YYModelMeta *meta = (__bridge _White_YYModelMeta *)(context->modelMeta);
  1027. __unsafe_unretained _White_YYModelPropertyMeta *propertyMeta = [meta->_mapper objectForKey:(__bridge id)(_key)];
  1028. __unsafe_unretained id model = (__bridge id)(context->model);
  1029. while (propertyMeta) {
  1030. if (propertyMeta->_setter) {
  1031. White_ModelSetValueForProperty(model, (__bridge __unsafe_unretained id)_value, propertyMeta);
  1032. }
  1033. propertyMeta = propertyMeta->_next;
  1034. };
  1035. }
  1036. /**
  1037. Apply function for model property meta, to set dictionary to model.
  1038. @param _propertyMeta should not be nil, _YYModelPropertyMeta.
  1039. @param _context _context.model and _context.dictionary should not be nil.
  1040. */
  1041. static void White_ModelSetWithPropertyMetaArrayFunction(const void *_propertyMeta, void *_context) {
  1042. White_ModelSetContext *context = _context;
  1043. __unsafe_unretained NSDictionary *dictionary = (__bridge NSDictionary *)(context->dictionary);
  1044. __unsafe_unretained _White_YYModelPropertyMeta *propertyMeta = (__bridge _White_YYModelPropertyMeta *)(_propertyMeta);
  1045. if (!propertyMeta->_setter) return;
  1046. id value = nil;
  1047. if (propertyMeta->_mappedToKeyArray) {
  1048. value = YYValueForMultiKeys(dictionary, propertyMeta->_mappedToKeyArray);
  1049. } else if (propertyMeta->_mappedToKeyPath) {
  1050. value = White_YYValueForKeyPath(dictionary, propertyMeta->_mappedToKeyPath);
  1051. } else {
  1052. value = [dictionary objectForKey:propertyMeta->_mappedToKey];
  1053. }
  1054. if (value) {
  1055. __unsafe_unretained id model = (__bridge id)(context->model);
  1056. White_ModelSetValueForProperty(model, value, propertyMeta);
  1057. }
  1058. }
  1059. /**
  1060. Returns a valid JSON object (NSArray/NSDictionary/NSString/NSNumber/NSNull),
  1061. or nil if an error occurs.
  1062. @param model Model, can be nil.
  1063. @return JSON object, nil if an error occurs.
  1064. */
  1065. static id White_ModelToJSONObjectRecursive(NSObject *model) {
  1066. if (!model || model == (id)kCFNull) return model;
  1067. if ([model isKindOfClass:[NSString class]]) return model;
  1068. if ([model isKindOfClass:[NSNumber class]]) return model;
  1069. if ([model isKindOfClass:[NSDictionary class]]) {
  1070. if ([NSJSONSerialization isValidJSONObject:model]) return model;
  1071. NSMutableDictionary *newDic = [NSMutableDictionary new];
  1072. [((NSDictionary *)model) enumerateKeysAndObjectsUsingBlock:^(NSString *key, id obj, BOOL *stop) {
  1073. NSString *stringKey = [key isKindOfClass:[NSString class]] ? key : key.description;
  1074. if (!stringKey) return;
  1075. id jsonObj = White_ModelToJSONObjectRecursive(obj);
  1076. if (!jsonObj) jsonObj = (id)kCFNull;
  1077. newDic[stringKey] = jsonObj;
  1078. }];
  1079. return newDic;
  1080. }
  1081. if ([model isKindOfClass:[NSSet class]]) {
  1082. NSArray *array = ((NSSet *)model).allObjects;
  1083. if ([NSJSONSerialization isValidJSONObject:array]) return array;
  1084. NSMutableArray *newArray = [NSMutableArray new];
  1085. for (id obj in array) {
  1086. if ([obj isKindOfClass:[NSString class]] || [obj isKindOfClass:[NSNumber class]]) {
  1087. [newArray addObject:obj];
  1088. } else {
  1089. id jsonObj = White_ModelToJSONObjectRecursive(obj);
  1090. if (jsonObj && jsonObj != (id)kCFNull) [newArray addObject:jsonObj];
  1091. }
  1092. }
  1093. return newArray;
  1094. }
  1095. if ([model isKindOfClass:[NSArray class]]) {
  1096. if ([NSJSONSerialization isValidJSONObject:model]) return model;
  1097. NSMutableArray *newArray = [NSMutableArray new];
  1098. for (id obj in (NSArray *)model) {
  1099. if ([obj isKindOfClass:[NSString class]] || [obj isKindOfClass:[NSNumber class]]) {
  1100. [newArray addObject:obj];
  1101. } else {
  1102. id jsonObj = White_ModelToJSONObjectRecursive(obj);
  1103. if (jsonObj && jsonObj != (id)kCFNull) [newArray addObject:jsonObj];
  1104. }
  1105. }
  1106. return newArray;
  1107. }
  1108. if ([model isKindOfClass:[NSURL class]]) return ((NSURL *)model).absoluteString;
  1109. if ([model isKindOfClass:[NSAttributedString class]]) return ((NSAttributedString *)model).string;
  1110. if ([model isKindOfClass:[NSDate class]]) return [White_YYISODateFormatter() stringFromDate:(id)model];
  1111. if ([model isKindOfClass:[NSData class]]) return nil;
  1112. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:[model class]];
  1113. if (!modelMeta || modelMeta->_keyMappedCount == 0) return nil;
  1114. NSMutableDictionary *result = [[NSMutableDictionary alloc] initWithCapacity:64];
  1115. __unsafe_unretained NSMutableDictionary *dic = result; // avoid retain and release in block
  1116. [modelMeta->_mapper enumerateKeysAndObjectsUsingBlock:^(NSString *propertyMappedKey, _White_YYModelPropertyMeta *propertyMeta, BOOL *stop) {
  1117. if (!propertyMeta->_getter) return;
  1118. id value = nil;
  1119. if (propertyMeta->_isCNumber) {
  1120. value = White_ModelCreateNumberFromProperty(model, propertyMeta);
  1121. } else if (propertyMeta->_nsType) {
  1122. id v = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, propertyMeta->_getter);
  1123. value = White_ModelToJSONObjectRecursive(v);
  1124. } else {
  1125. switch (propertyMeta->_type & White_YYEncodingTypeMask) {
  1126. case White_YYEncodingTypeObject: {
  1127. id v = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, propertyMeta->_getter);
  1128. value = White_ModelToJSONObjectRecursive(v);
  1129. if (value == (id)kCFNull) value = nil;
  1130. } break;
  1131. case White_YYEncodingTypeClass: {
  1132. Class v = ((Class (*)(id, SEL))(void *) objc_msgSend)((id)model, propertyMeta->_getter);
  1133. value = v ? NSStringFromClass(v) : nil;
  1134. } break;
  1135. case White_YYEncodingTypeSEL: {
  1136. SEL v = ((SEL (*)(id, SEL))(void *) objc_msgSend)((id)model, propertyMeta->_getter);
  1137. value = v ? NSStringFromSelector(v) : nil;
  1138. } break;
  1139. default: break;
  1140. }
  1141. }
  1142. if (!value) return;
  1143. if (propertyMeta->_mappedToKeyPath) {
  1144. NSMutableDictionary *superDic = dic;
  1145. NSMutableDictionary *subDic = nil;
  1146. for (NSUInteger i = 0, max = propertyMeta->_mappedToKeyPath.count; i < max; i++) {
  1147. NSString *key = propertyMeta->_mappedToKeyPath[i];
  1148. if (i + 1 == max) { // end
  1149. if (!superDic[key]) superDic[key] = value;
  1150. break;
  1151. }
  1152. subDic = superDic[key];
  1153. if (subDic) {
  1154. if ([subDic isKindOfClass:[NSDictionary class]]) {
  1155. subDic = subDic.mutableCopy;
  1156. superDic[key] = subDic;
  1157. } else {
  1158. break;
  1159. }
  1160. } else {
  1161. subDic = [NSMutableDictionary new];
  1162. superDic[key] = subDic;
  1163. }
  1164. superDic = subDic;
  1165. subDic = nil;
  1166. }
  1167. } else {
  1168. if (!dic[propertyMeta->_mappedToKey]) {
  1169. dic[propertyMeta->_mappedToKey] = value;
  1170. }
  1171. }
  1172. }];
  1173. if (modelMeta->_hasCustomTransformToDictionary) {
  1174. BOOL suc = [((id<White_YYModel>)model) modelCustomTransformToDictionary:dic];
  1175. if (!suc) return nil;
  1176. }
  1177. return result;
  1178. }
  1179. /// Add indent to string (exclude first line)
  1180. static NSMutableString *White_ModelDescriptionAddIndent(NSMutableString *desc, NSUInteger indent) {
  1181. for (NSUInteger i = 0, max = desc.length; i < max; i++) {
  1182. unichar c = [desc characterAtIndex:i];
  1183. if (c == '\n') {
  1184. for (NSUInteger j = 0; j < indent; j++) {
  1185. [desc insertString:@" " atIndex:i + 1];
  1186. }
  1187. i += indent * 4;
  1188. max += indent * 4;
  1189. }
  1190. }
  1191. return desc;
  1192. }
  1193. /// Generaate a description string
  1194. static NSString *White_ModelDescription(NSObject *model) {
  1195. static const int kDescMaxLength = 100;
  1196. if (!model) return @"<nil>";
  1197. if (model == (id)kCFNull) return @"<null>";
  1198. if (![model isKindOfClass:[NSObject class]]) return [NSString stringWithFormat:@"%@",model];
  1199. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:model.class];
  1200. switch (modelMeta->_nsType) {
  1201. case White_YYEncodingTypeNSString: case White_YYEncodingTypeNSMutableString: {
  1202. return [NSString stringWithFormat:@"\"%@\"",model];
  1203. }
  1204. case White_YYEncodingTypeNSValue:
  1205. case White_YYEncodingTypeNSData: case White_YYEncodingTypeNSMutableData: {
  1206. NSString *tmp = model.description;
  1207. if (tmp.length > kDescMaxLength) {
  1208. tmp = [tmp substringToIndex:kDescMaxLength];
  1209. tmp = [tmp stringByAppendingString:@"..."];
  1210. }
  1211. return tmp;
  1212. }
  1213. case White_YYEncodingTypeNSNumber:
  1214. case White_YYEncodingTypeNSDecimalNumber:
  1215. case White_YYEncodingTypeNSDate:
  1216. case White_YYEncodingTypeNSURL: {
  1217. return [NSString stringWithFormat:@"%@",model];
  1218. }
  1219. case White_YYEncodingTypeNSSet: case White_YYEncodingTypeNSMutableSet: {
  1220. model = ((NSSet *)model).allObjects;
  1221. } // no break
  1222. case White_YYEncodingTypeNSArray: case White_YYEncodingTypeNSMutableArray: {
  1223. NSArray *array = (id)model;
  1224. NSMutableString *desc = [NSMutableString new];
  1225. if (array.count == 0) {
  1226. return [desc stringByAppendingString:@"[]"];
  1227. } else {
  1228. [desc appendFormat:@"[\n"];
  1229. for (NSUInteger i = 0, max = array.count; i < max; i++) {
  1230. NSObject *obj = array[i];
  1231. [desc appendString:@" "];
  1232. [desc appendString:White_ModelDescriptionAddIndent(White_ModelDescription(obj).mutableCopy, 1)];
  1233. [desc appendString:(i + 1 == max) ? @"\n" : @";\n"];
  1234. }
  1235. [desc appendString:@"]"];
  1236. return desc;
  1237. }
  1238. }
  1239. case White_YYEncodingTypeNSDictionary: case White_YYEncodingTypeNSMutableDictionary: {
  1240. NSDictionary *dic = (id)model;
  1241. NSMutableString *desc = [NSMutableString new];
  1242. if (dic.count == 0) {
  1243. return [desc stringByAppendingString:@"{}"];
  1244. } else {
  1245. NSArray *keys = dic.allKeys;
  1246. [desc appendFormat:@"{\n"];
  1247. for (NSUInteger i = 0, max = keys.count; i < max; i++) {
  1248. NSString *key = keys[i];
  1249. NSObject *value = dic[key];
  1250. [desc appendString:@" "];
  1251. [desc appendFormat:@"%@ = %@",key, White_ModelDescriptionAddIndent(White_ModelDescription(value).mutableCopy, 1)];
  1252. [desc appendString:(i + 1 == max) ? @"\n" : @";\n"];
  1253. }
  1254. [desc appendString:@"}"];
  1255. }
  1256. return desc;
  1257. }
  1258. default: {
  1259. NSMutableString *desc = [NSMutableString new];
  1260. [desc appendFormat:@"<%@: %p>", model.class, model];
  1261. if (modelMeta->_allPropertyMetas.count == 0) return desc;
  1262. // sort property names
  1263. NSArray *properties = [modelMeta->_allPropertyMetas
  1264. sortedArrayUsingComparator:^NSComparisonResult(_White_YYModelPropertyMeta *p1, _White_YYModelPropertyMeta *p2) {
  1265. return [p1->_name compare:p2->_name];
  1266. }];
  1267. [desc appendFormat:@" {\n"];
  1268. for (NSUInteger i = 0, max = properties.count; i < max; i++) {
  1269. _White_YYModelPropertyMeta *property = properties[i];
  1270. NSString *propertyDesc;
  1271. if (property->_isCNumber) {
  1272. NSNumber *num = White_ModelCreateNumberFromProperty(model, property);
  1273. propertyDesc = num.stringValue;
  1274. } else {
  1275. switch (property->_type & White_YYEncodingTypeMask) {
  1276. case White_YYEncodingTypeObject: {
  1277. id v = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, property->_getter);
  1278. propertyDesc = White_ModelDescription(v);
  1279. if (!propertyDesc) propertyDesc = @"<nil>";
  1280. } break;
  1281. case White_YYEncodingTypeClass: {
  1282. id v = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, property->_getter);
  1283. propertyDesc = ((NSObject *)v).description;
  1284. if (!propertyDesc) propertyDesc = @"<nil>";
  1285. } break;
  1286. case White_YYEncodingTypeSEL: {
  1287. SEL sel = ((SEL (*)(id, SEL))(void *) objc_msgSend)((id)model, property->_getter);
  1288. if (sel) propertyDesc = NSStringFromSelector(sel);
  1289. else propertyDesc = @"<NULL>";
  1290. } break;
  1291. case White_YYEncodingTypeBlock: {
  1292. id block = ((id (*)(id, SEL))(void *) objc_msgSend)((id)model, property->_getter);
  1293. propertyDesc = block ? ((NSObject *)block).description : @"<nil>";
  1294. } break;
  1295. case White_YYEncodingTypeCArray: case White_YYEncodingTypeCString: case White_YYEncodingTypePointer: {
  1296. void *pointer = ((void* (*)(id, SEL))(void *) objc_msgSend)((id)model, property->_getter);
  1297. propertyDesc = [NSString stringWithFormat:@"%p",pointer];
  1298. } break;
  1299. case White_YYEncodingTypeStruct: case White_YYEncodingTypeUnion: {
  1300. NSValue *value = [model valueForKey:property->_name];
  1301. propertyDesc = value ? value.description : @"{unknown}";
  1302. } break;
  1303. default: propertyDesc = @"<unknown>";
  1304. }
  1305. }
  1306. propertyDesc = White_ModelDescriptionAddIndent(propertyDesc.mutableCopy, 1);
  1307. [desc appendFormat:@" %@ = %@",property->_name, propertyDesc];
  1308. [desc appendString:(i + 1 == max) ? @"\n" : @";\n"];
  1309. }
  1310. [desc appendFormat:@"}"];
  1311. return desc;
  1312. }
  1313. }
  1314. }
  1315. @implementation NSObject (White_YYModel)
  1316. + (NSDictionary *)_yy_dictionaryWithJSON:(id)json {
  1317. if (!json || json == (id)kCFNull) return nil;
  1318. NSDictionary *dic = nil;
  1319. NSData *jsonData = nil;
  1320. if ([json isKindOfClass:[NSDictionary class]]) {
  1321. dic = json;
  1322. } else if ([json isKindOfClass:[NSString class]]) {
  1323. jsonData = [(NSString *)json dataUsingEncoding : NSUTF8StringEncoding];
  1324. } else if ([json isKindOfClass:[NSData class]]) {
  1325. jsonData = json;
  1326. }
  1327. if (jsonData) {
  1328. dic = [NSJSONSerialization JSONObjectWithData:jsonData options:kNilOptions error:NULL];
  1329. if (![dic isKindOfClass:[NSDictionary class]]) dic = nil;
  1330. }
  1331. return dic;
  1332. }
  1333. + (instancetype)_white_yy_modelWithJSON:(id)json {
  1334. NSDictionary *dic = [self _yy_dictionaryWithJSON:json];
  1335. return [self _white_yy_modelWithDictionary:dic];
  1336. }
  1337. + (instancetype)_white_yy_modelWithDictionary:(NSDictionary *)dictionary {
  1338. if (!dictionary || dictionary == (id)kCFNull) return nil;
  1339. if (![dictionary isKindOfClass:[NSDictionary class]]) return nil;
  1340. Class cls = [self class];
  1341. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:cls];
  1342. if (modelMeta->_hasCustomClassFromDictionary) {
  1343. cls = [cls modelCustomClassForDictionary:dictionary] ?: cls;
  1344. }
  1345. NSObject *one = [cls new];
  1346. if ([one _white_yy_modelSetWithDictionary:dictionary]) return one;
  1347. return nil;
  1348. }
  1349. - (BOOL)_white_yy_modelSetWithJSON:(id)json {
  1350. NSDictionary *dic = [NSObject _yy_dictionaryWithJSON:json];
  1351. return [self _white_yy_modelSetWithDictionary:dic];
  1352. }
  1353. - (BOOL)_white_yy_modelSetWithDictionary:(NSDictionary *)dic {
  1354. if (!dic || dic == (id)kCFNull) return NO;
  1355. if (![dic isKindOfClass:[NSDictionary class]]) return NO;
  1356. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:object_getClass(self)];
  1357. if (modelMeta->_keyMappedCount == 0) return NO;
  1358. if (modelMeta->_hasCustomWillTransformFromDictionary) {
  1359. dic = [((id<White_YYModel>)self) modelCustomWillTransformFromDictionary:dic];
  1360. if (![dic isKindOfClass:[NSDictionary class]]) return NO;
  1361. }
  1362. White_ModelSetContext context = {0};
  1363. context.modelMeta = (__bridge void *)(modelMeta);
  1364. context.model = (__bridge void *)(self);
  1365. context.dictionary = (__bridge void *)(dic);
  1366. if (modelMeta->_keyMappedCount >= CFDictionaryGetCount((CFDictionaryRef)dic)) {
  1367. CFDictionaryApplyFunction((CFDictionaryRef)dic, White_ModelSetWithDictionaryFunction, &context);
  1368. if (modelMeta->_keyPathPropertyMetas) {
  1369. CFArrayApplyFunction((CFArrayRef)modelMeta->_keyPathPropertyMetas,
  1370. CFRangeMake(0, CFArrayGetCount((CFArrayRef)modelMeta->_keyPathPropertyMetas)),
  1371. White_ModelSetWithPropertyMetaArrayFunction,
  1372. &context);
  1373. }
  1374. if (modelMeta->_multiKeysPropertyMetas) {
  1375. CFArrayApplyFunction((CFArrayRef)modelMeta->_multiKeysPropertyMetas,
  1376. CFRangeMake(0, CFArrayGetCount((CFArrayRef)modelMeta->_multiKeysPropertyMetas)),
  1377. White_ModelSetWithPropertyMetaArrayFunction,
  1378. &context);
  1379. }
  1380. } else {
  1381. CFArrayApplyFunction((CFArrayRef)modelMeta->_allPropertyMetas,
  1382. CFRangeMake(0, modelMeta->_keyMappedCount),
  1383. White_ModelSetWithPropertyMetaArrayFunction,
  1384. &context);
  1385. }
  1386. if (modelMeta->_hasCustomTransformFromDictionary) {
  1387. return [((id<White_YYModel>)self) modelCustomTransformFromDictionary:dic];
  1388. }
  1389. return YES;
  1390. }
  1391. - (id)_white_yy_modelToJSONObject {
  1392. /*
  1393. Apple said:
  1394. The top level object is an NSArray or NSDictionary.
  1395. All objects are instances of NSString, NSNumber, NSArray, NSDictionary, or NSNull.
  1396. All dictionary keys are instances of NSString.
  1397. Numbers are not NaN or infinity.
  1398. */
  1399. id jsonObject = White_ModelToJSONObjectRecursive(self);
  1400. if ([jsonObject isKindOfClass:[NSArray class]]) return jsonObject;
  1401. if ([jsonObject isKindOfClass:[NSDictionary class]]) return jsonObject;
  1402. return nil;
  1403. }
  1404. - (NSData *)_white_yy_modelToJSONData {
  1405. id jsonObject = [self _white_yy_modelToJSONObject];
  1406. if (!jsonObject) return nil;
  1407. return [NSJSONSerialization dataWithJSONObject:jsonObject options:0 error:NULL];
  1408. }
  1409. - (NSString *)_white_yy_modelToJSONString {
  1410. NSData *jsonData = [self _white_yy_modelToJSONData];
  1411. if (jsonData.length == 0) return nil;
  1412. return [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
  1413. }
  1414. - (id)_white_yy_modelCopy{
  1415. if (self == (id)kCFNull) return self;
  1416. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:self.class];
  1417. if (modelMeta->_nsType) return [self copy];
  1418. NSObject *one = [self.class new];
  1419. for (_White_YYModelPropertyMeta *propertyMeta in modelMeta->_allPropertyMetas) {
  1420. if (!propertyMeta->_getter || !propertyMeta->_setter) continue;
  1421. if (propertyMeta->_isCNumber) {
  1422. switch (propertyMeta->_type & White_YYEncodingTypeMask) {
  1423. case White_YYEncodingTypeBool: {
  1424. bool num = ((bool (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1425. ((void (*)(id, SEL, bool))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1426. } break;
  1427. case White_YYEncodingTypeInt8:
  1428. case White_YYEncodingTypeUInt8: {
  1429. uint8_t num = ((bool (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1430. ((void (*)(id, SEL, uint8_t))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1431. } break;
  1432. case White_YYEncodingTypeInt16:
  1433. case White_YYEncodingTypeUInt16: {
  1434. uint16_t num = ((uint16_t (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1435. ((void (*)(id, SEL, uint16_t))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1436. } break;
  1437. case White_YYEncodingTypeInt32:
  1438. case White_YYEncodingTypeUInt32: {
  1439. uint32_t num = ((uint32_t (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1440. ((void (*)(id, SEL, uint32_t))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1441. } break;
  1442. case White_YYEncodingTypeInt64:
  1443. case White_YYEncodingTypeUInt64: {
  1444. uint64_t num = ((uint64_t (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1445. ((void (*)(id, SEL, uint64_t))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1446. } break;
  1447. case White_YYEncodingTypeFloat: {
  1448. float num = ((float (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1449. ((void (*)(id, SEL, float))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1450. } break;
  1451. case White_YYEncodingTypeDouble: {
  1452. double num = ((double (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1453. ((void (*)(id, SEL, double))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1454. } break;
  1455. case White_YYEncodingTypeLongDouble: {
  1456. long double num = ((long double (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1457. ((void (*)(id, SEL, long double))(void *) objc_msgSend)((id)one, propertyMeta->_setter, num);
  1458. } // break; commented for code coverage in next line
  1459. default: break;
  1460. }
  1461. } else {
  1462. switch (propertyMeta->_type & White_YYEncodingTypeMask) {
  1463. case White_YYEncodingTypeObject:
  1464. case White_YYEncodingTypeClass:
  1465. case White_YYEncodingTypeBlock: {
  1466. id value = ((id (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1467. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)one, propertyMeta->_setter, value);
  1468. } break;
  1469. case White_YYEncodingTypeSEL:
  1470. case White_YYEncodingTypePointer:
  1471. case White_YYEncodingTypeCString: {
  1472. size_t value = ((size_t (*)(id, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_getter);
  1473. ((void (*)(id, SEL, size_t))(void *) objc_msgSend)((id)one, propertyMeta->_setter, value);
  1474. } break;
  1475. case White_YYEncodingTypeStruct:
  1476. case White_YYEncodingTypeUnion: {
  1477. @try {
  1478. NSValue *value = [self valueForKey:NSStringFromSelector(propertyMeta->_getter)];
  1479. if (value) {
  1480. [one setValue:value forKey:propertyMeta->_name];
  1481. }
  1482. } @catch (NSException *exception) {}
  1483. } // break; commented for code coverage in next line
  1484. default: break;
  1485. }
  1486. }
  1487. }
  1488. return one;
  1489. }
  1490. - (void)_white_yy_modelEncodeWithCoder:(NSCoder *)aCoder {
  1491. if (!aCoder) return;
  1492. if (self == (id)kCFNull) {
  1493. [((id<NSCoding>)self)encodeWithCoder:aCoder];
  1494. return;
  1495. }
  1496. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:self.class];
  1497. if (modelMeta->_nsType) {
  1498. [((id<NSCoding>)self)encodeWithCoder:aCoder];
  1499. return;
  1500. }
  1501. for (_White_YYModelPropertyMeta *propertyMeta in modelMeta->_allPropertyMetas) {
  1502. if (!propertyMeta->_getter) return;
  1503. if (propertyMeta->_isCNumber) {
  1504. NSNumber *value = White_ModelCreateNumberFromProperty(self, propertyMeta);
  1505. if (value) [aCoder encodeObject:value forKey:propertyMeta->_name];
  1506. } else {
  1507. switch (propertyMeta->_type & White_YYEncodingTypeMask) {
  1508. case White_YYEncodingTypeObject: {
  1509. id value = ((id (*)(id, SEL))(void *)objc_msgSend)((id)self, propertyMeta->_getter);
  1510. if (value && (propertyMeta->_nsType || [value respondsToSelector:@selector(encodeWithCoder:)])) {
  1511. if ([value isKindOfClass:[NSValue class]]) {
  1512. if ([value isKindOfClass:[NSNumber class]]) {
  1513. [aCoder encodeObject:value forKey:propertyMeta->_name];
  1514. }
  1515. } else {
  1516. [aCoder encodeObject:value forKey:propertyMeta->_name];
  1517. }
  1518. }
  1519. } break;
  1520. case White_YYEncodingTypeSEL: {
  1521. SEL value = ((SEL (*)(id, SEL))(void *)objc_msgSend)((id)self, propertyMeta->_getter);
  1522. if (value) {
  1523. NSString *str = NSStringFromSelector(value);
  1524. [aCoder encodeObject:str forKey:propertyMeta->_name];
  1525. }
  1526. } break;
  1527. case White_YYEncodingTypeStruct:
  1528. case White_YYEncodingTypeUnion: {
  1529. if (propertyMeta->_isKVCCompatible && propertyMeta->_isStructAvailableForKeyedArchiver) {
  1530. @try {
  1531. NSValue *value = [self valueForKey:NSStringFromSelector(propertyMeta->_getter)];
  1532. [aCoder encodeObject:value forKey:propertyMeta->_name];
  1533. } @catch (NSException *exception) {}
  1534. }
  1535. } break;
  1536. default:
  1537. break;
  1538. }
  1539. }
  1540. }
  1541. }
  1542. - (id)_white_yy_modelInitWithCoder:(NSCoder *)aDecoder {
  1543. if (!aDecoder) return self;
  1544. if (self == (id)kCFNull) return self;
  1545. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:self.class];
  1546. if (modelMeta->_nsType) return self;
  1547. for (_White_YYModelPropertyMeta *propertyMeta in modelMeta->_allPropertyMetas) {
  1548. if (!propertyMeta->_setter) continue;
  1549. if (propertyMeta->_isCNumber) {
  1550. NSNumber *value = [aDecoder decodeObjectForKey:propertyMeta->_name];
  1551. if ([value isKindOfClass:[NSNumber class]]) {
  1552. White_ModelSetNumberToProperty(self, value, propertyMeta);
  1553. [value class];
  1554. }
  1555. } else {
  1556. White_YYEncodingType type = propertyMeta->_type & White_YYEncodingTypeMask;
  1557. switch (type) {
  1558. case White_YYEncodingTypeObject: {
  1559. id value = [aDecoder decodeObjectForKey:propertyMeta->_name];
  1560. ((void (*)(id, SEL, id))(void *) objc_msgSend)((id)self, propertyMeta->_setter, value);
  1561. } break;
  1562. case White_YYEncodingTypeSEL: {
  1563. NSString *str = [aDecoder decodeObjectForKey:propertyMeta->_name];
  1564. if ([str isKindOfClass:[NSString class]]) {
  1565. SEL sel = NSSelectorFromString(str);
  1566. ((void (*)(id, SEL, SEL))(void *) objc_msgSend)((id)self, propertyMeta->_setter, sel);
  1567. }
  1568. } break;
  1569. case White_YYEncodingTypeStruct:
  1570. case White_YYEncodingTypeUnion: {
  1571. if (propertyMeta->_isKVCCompatible) {
  1572. @try {
  1573. NSValue *value = [aDecoder decodeObjectForKey:propertyMeta->_name];
  1574. if (value) [self setValue:value forKey:propertyMeta->_name];
  1575. } @catch (NSException *exception) {}
  1576. }
  1577. } break;
  1578. default:
  1579. break;
  1580. }
  1581. }
  1582. }
  1583. return self;
  1584. }
  1585. - (NSUInteger)_white_yy_modelHash {
  1586. if (self == (id)kCFNull) return [self hash];
  1587. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:self.class];
  1588. if (modelMeta->_nsType) return [self hash];
  1589. NSUInteger value = 0;
  1590. NSUInteger count = 0;
  1591. for (_White_YYModelPropertyMeta *propertyMeta in modelMeta->_allPropertyMetas) {
  1592. if (!propertyMeta->_isKVCCompatible) continue;
  1593. value ^= [[self valueForKey:NSStringFromSelector(propertyMeta->_getter)] hash];
  1594. count++;
  1595. }
  1596. if (count == 0) value = (long)((__bridge void *)self);
  1597. return value;
  1598. }
  1599. - (BOOL)_white_yy_modelIsEqual:(id)model {
  1600. if (self == model) return YES;
  1601. if (![model isMemberOfClass:self.class]) return NO;
  1602. _White_YYModelMeta *modelMeta = [_White_YYModelMeta metaWithClass:self.class];
  1603. if (modelMeta->_nsType) return [self isEqual:model];
  1604. if ([self hash] != [model hash]) return NO;
  1605. for (_White_YYModelPropertyMeta *propertyMeta in modelMeta->_allPropertyMetas) {
  1606. if (!propertyMeta->_isKVCCompatible) continue;
  1607. id this = [self valueForKey:NSStringFromSelector(propertyMeta->_getter)];
  1608. id that = [model valueForKey:NSStringFromSelector(propertyMeta->_getter)];
  1609. if (this == that) continue;
  1610. if (this == nil || that == nil) return NO;
  1611. if (![this isEqual:that]) return NO;
  1612. }
  1613. return YES;
  1614. }
  1615. - (NSString *)_white_yy_modelDescription {
  1616. return White_ModelDescription(self);
  1617. }
  1618. @end
  1619. @implementation NSArray (White_YYModel)
  1620. + (NSArray *)_white_yy_modelArrayWithClass:(Class)cls json:(id)json {
  1621. if (!json) return nil;
  1622. NSArray *arr = nil;
  1623. NSData *jsonData = nil;
  1624. if ([json isKindOfClass:[NSArray class]]) {
  1625. arr = json;
  1626. } else if ([json isKindOfClass:[NSString class]]) {
  1627. jsonData = [(NSString *)json dataUsingEncoding : NSUTF8StringEncoding];
  1628. } else if ([json isKindOfClass:[NSData class]]) {
  1629. jsonData = json;
  1630. }
  1631. if (jsonData) {
  1632. arr = [NSJSONSerialization JSONObjectWithData:jsonData options:kNilOptions error:NULL];
  1633. if (![arr isKindOfClass:[NSArray class]]) arr = nil;
  1634. }
  1635. return [self _white_yy_modelArrayWithClass:cls array:arr];
  1636. }
  1637. + (NSArray *)_white_yy_modelArrayWithClass:(Class)cls array:(NSArray *)arr {
  1638. if (!cls || !arr) return nil;
  1639. NSMutableArray *result = [NSMutableArray new];
  1640. for (NSDictionary *dic in arr) {
  1641. if (![dic isKindOfClass:[NSDictionary class]]) continue;
  1642. NSObject *obj = [cls _white_yy_modelWithDictionary:dic];
  1643. if (obj) [result addObject:obj];
  1644. }
  1645. return result;
  1646. }
  1647. @end
  1648. @implementation NSDictionary (White_YYModel)
  1649. + (NSDictionary *)_white_yy_modelDictionaryWithClass:(Class)cls json:(id)json {
  1650. if (!json) return nil;
  1651. NSDictionary *dic = nil;
  1652. NSData *jsonData = nil;
  1653. if ([json isKindOfClass:[NSDictionary class]]) {
  1654. dic = json;
  1655. } else if ([json isKindOfClass:[NSString class]]) {
  1656. jsonData = [(NSString *)json dataUsingEncoding : NSUTF8StringEncoding];
  1657. } else if ([json isKindOfClass:[NSData class]]) {
  1658. jsonData = json;
  1659. }
  1660. if (jsonData) {
  1661. dic = [NSJSONSerialization JSONObjectWithData:jsonData options:kNilOptions error:NULL];
  1662. if (![dic isKindOfClass:[NSDictionary class]]) dic = nil;
  1663. }
  1664. return [self _white_yy_modelDictionaryWithClass:cls dictionary:dic];
  1665. }
  1666. + (NSDictionary *)_white_yy_modelDictionaryWithClass:(Class)cls dictionary:(NSDictionary *)dic {
  1667. if (!cls || !dic) return nil;
  1668. NSMutableDictionary *result = [NSMutableDictionary new];
  1669. for (NSString *key in dic.allKeys) {
  1670. if (![key isKindOfClass:[NSString class]]) continue;
  1671. NSObject *obj = [cls _white_yy_modelWithDictionary:dic[key]];
  1672. if (obj) result[key] = obj;
  1673. }
  1674. return result;
  1675. }
  1676. @end