Ta android + ios
This commit is contained in:
LYP
@@ -0,0 +1,491 @@
|
||||
//
|
||||
// TARouter.m
|
||||
// Pods
|
||||
//
|
||||
// Created by wwango on 2022/10/8.
|
||||
//
|
||||
|
||||
#import "TARouter.h"
|
||||
#import "TAServiceProtocol.h"
|
||||
#import "TAServiceManager.h"
|
||||
#import <objc/runtime.h>
|
||||
|
||||
@interface NSObject (TARetType)
|
||||
|
||||
+ (id)bh_getReturnFromInv:(NSInvocation *)inv withSig:(NSMethodSignature *)sig;
|
||||
|
||||
@end
|
||||
|
||||
@implementation NSObject (TARetType)
|
||||
|
||||
+ (id)bh_getReturnFromInv:(NSInvocation *)inv withSig:(NSMethodSignature *)sig {
|
||||
NSUInteger length = [sig methodReturnLength];
|
||||
if (length == 0) return nil;
|
||||
|
||||
char *type = (char *)[sig methodReturnType];
|
||||
while (*type == 'r' || // const
|
||||
*type == 'n' || // in
|
||||
*type == 'N' || // inout
|
||||
*type == 'o' || // out
|
||||
*type == 'O' || // bycopy
|
||||
*type == 'R' || // byref
|
||||
*type == 'V') { // oneway
|
||||
type++; // cutoff useless prefix
|
||||
}
|
||||
|
||||
#define ta_return_with_number(_type_) \
|
||||
do { \
|
||||
_type_ ret; \
|
||||
[inv getReturnValue:&ret]; \
|
||||
return @(ret); \
|
||||
} while (0)
|
||||
|
||||
switch (*type) {
|
||||
case 'v': return nil; // void
|
||||
case 'B': ta_return_with_number(bool);
|
||||
case 'c': ta_return_with_number(char);
|
||||
case 'C': ta_return_with_number(unsigned char);
|
||||
case 's': ta_return_with_number(short);
|
||||
case 'S': ta_return_with_number(unsigned short);
|
||||
case 'i': ta_return_with_number(int);
|
||||
case 'I': ta_return_with_number(unsigned int);
|
||||
case 'l': ta_return_with_number(int);
|
||||
case 'L': ta_return_with_number(unsigned int);
|
||||
case 'q': ta_return_with_number(long long);
|
||||
case 'Q': ta_return_with_number(unsigned long long);
|
||||
case 'f': ta_return_with_number(float);
|
||||
case 'd': ta_return_with_number(double);
|
||||
case 'D': { // long double
|
||||
long double ret;
|
||||
[inv getReturnValue:&ret];
|
||||
return [NSNumber numberWithDouble:ret];
|
||||
};
|
||||
|
||||
case '@': { // id
|
||||
id ret = nil;
|
||||
[inv getReturnValue:&ret];
|
||||
return ret;
|
||||
};
|
||||
|
||||
case '#': { // Class
|
||||
Class ret = nil;
|
||||
[inv getReturnValue:&ret];
|
||||
return ret;
|
||||
};
|
||||
|
||||
default: { // struct / union / SEL / void* / unknown
|
||||
const char *objCType = [sig methodReturnType];
|
||||
char *buf = calloc(1, length);
|
||||
if (!buf) return nil;
|
||||
[inv getReturnValue:buf];
|
||||
NSValue *value = [NSValue valueWithBytes:buf objCType:objCType];
|
||||
free(buf);
|
||||
return value;
|
||||
};
|
||||
}
|
||||
#undef ta_return_with_number
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
static NSString *const TARClassRegex = @"(?<=T@\")(.*)(?=\",)";
|
||||
|
||||
|
||||
typedef NS_ENUM(NSUInteger, TARUsage) {
|
||||
TARUsageUnknown,
|
||||
TARUsageCallService,
|
||||
};
|
||||
|
||||
|
||||
static NSMutableDictionary<NSString *, TARouter *> *routerByScheme = nil;
|
||||
|
||||
|
||||
@interface TARPathComponent : NSObject
|
||||
|
||||
@property (nonatomic, copy) NSString *key;
|
||||
@property (nonatomic, strong) Class mClass;
|
||||
@property (nonatomic, copy) NSDictionary<NSString *, id> *params;
|
||||
@property (nonatomic, copy) TARPathComponentCustomHandler handler;
|
||||
|
||||
@end
|
||||
|
||||
@implementation TARPathComponent
|
||||
|
||||
@end
|
||||
|
||||
static NSString *TARURLGlobalScheme = nil;
|
||||
|
||||
@interface TARouter ()
|
||||
|
||||
@property (nonatomic, strong) NSMutableDictionary<NSString *, TARPathComponent *> *pathComponentByKey;
|
||||
@property (nonatomic, copy) NSString *scheme;
|
||||
|
||||
@end
|
||||
|
||||
@implementation TARouter
|
||||
|
||||
#pragma mark - property init
|
||||
- (NSMutableDictionary<NSString *, TARPathComponent *> *)pathComponentByKey {
|
||||
if (!_pathComponentByKey) {
|
||||
_pathComponentByKey = @{}.mutableCopy;
|
||||
}
|
||||
return _pathComponentByKey;
|
||||
}
|
||||
|
||||
#pragma mark - router init
|
||||
|
||||
+ (instancetype)globalRouter
|
||||
{
|
||||
if (!TARURLGlobalScheme) {
|
||||
TARURLGlobalScheme = @"com.thinkingdata";
|
||||
}
|
||||
return [self routerForScheme:TARURLGlobalScheme];
|
||||
}
|
||||
+ (instancetype)routerForScheme:(NSString *)scheme
|
||||
{
|
||||
if (!scheme.length) {
|
||||
return nil;
|
||||
}
|
||||
|
||||
TARouter *router = nil;
|
||||
|
||||
static dispatch_once_t onceToken;
|
||||
dispatch_once(&onceToken, ^{
|
||||
routerByScheme = @{}.mutableCopy;
|
||||
});
|
||||
|
||||
if (!routerByScheme[scheme]) {
|
||||
router = [[self alloc] init];
|
||||
router.scheme = scheme;
|
||||
[routerByScheme setObject:router forKey:scheme];
|
||||
} else {
|
||||
router = [routerByScheme objectForKey:scheme];
|
||||
}
|
||||
|
||||
return router;
|
||||
}
|
||||
|
||||
+ (void)unRegisterRouterForScheme:(NSString *)scheme
|
||||
{
|
||||
if (!scheme.length) {
|
||||
return;
|
||||
}
|
||||
|
||||
[routerByScheme removeObjectForKey:scheme];
|
||||
}
|
||||
+ (void)unRegisterAllRouters
|
||||
{
|
||||
[routerByScheme removeAllObjects];
|
||||
}
|
||||
|
||||
- (void)addPathComponent:(NSString *)pathComponentKey
|
||||
forClass:(Class)mClass
|
||||
{
|
||||
[self addPathComponent:pathComponentKey forClass:mClass handler:nil];
|
||||
}
|
||||
//handler is a custom module or service init function
|
||||
- (void)addPathComponent:(NSString *)pathComponentKey
|
||||
forClass:(Class)mClass
|
||||
handler:(TARPathComponentCustomHandler)handler
|
||||
{
|
||||
TARPathComponent *pathComponent = [[TARPathComponent alloc] init];
|
||||
pathComponent.key = pathComponentKey;
|
||||
pathComponent.mClass = mClass;
|
||||
pathComponent.handler = handler;
|
||||
[self.pathComponentByKey setObject:pathComponent forKey:pathComponentKey];
|
||||
}
|
||||
- (void)removePathComponent:(NSString *)pathComponentKey
|
||||
{
|
||||
[self.pathComponentByKey removeObjectForKey:pathComponentKey];
|
||||
}
|
||||
|
||||
+ (BOOL)canOpenURL:(NSURL *)URL
|
||||
{
|
||||
if (!URL) {
|
||||
return NO;
|
||||
}
|
||||
NSString *scheme = URL.scheme;
|
||||
if (!scheme.length) {
|
||||
return NO;
|
||||
}
|
||||
|
||||
NSString *host = URL.host;
|
||||
TARUsage usage = [self usage:host];
|
||||
if (usage == TARUsageUnknown) {
|
||||
return NO;
|
||||
}
|
||||
|
||||
TARouter *router = [self routerForScheme:scheme];
|
||||
|
||||
NSArray<NSString *> *pathComponents = URL.pathComponents;
|
||||
|
||||
__block BOOL flag = YES;
|
||||
|
||||
[pathComponents enumerateObjectsUsingBlock:^(NSString * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
NSArray<NSString *> * subPaths = [obj componentsSeparatedByString:TARURLSubPathSplitPattern];
|
||||
|
||||
if ([subPaths.firstObject isEqualToString:@"/"]) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (!subPaths.count) {
|
||||
flag = NO;
|
||||
*stop = NO;
|
||||
return;
|
||||
}
|
||||
NSString *pathComponentKey = subPaths.firstObject;
|
||||
|
||||
|
||||
if (router.pathComponentByKey[pathComponentKey]) {
|
||||
return;
|
||||
}
|
||||
Class mClass = NSClassFromString(pathComponentKey);
|
||||
if (!mClass) {
|
||||
flag = NO;
|
||||
*stop = NO;
|
||||
return;
|
||||
}
|
||||
switch (usage) {
|
||||
case TARUsageCallService: {
|
||||
if (subPaths.count < 3) {
|
||||
flag = NO;
|
||||
*stop = NO;
|
||||
return;
|
||||
}
|
||||
NSString *protocolStr = subPaths[1];
|
||||
NSString *selectorStr = subPaths[2];
|
||||
Protocol *protocol = NSProtocolFromString(protocolStr);
|
||||
SEL selector = NSSelectorFromString(selectorStr);
|
||||
if (!protocol ||
|
||||
!selector ||
|
||||
![mClass conformsToProtocol:protocol] ||
|
||||
![mClass instancesRespondToSelector:selector]) {
|
||||
flag = NO;
|
||||
*stop = NO;
|
||||
return;
|
||||
}
|
||||
} break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}];
|
||||
|
||||
return flag;
|
||||
}
|
||||
|
||||
|
||||
+ (BOOL)openURL:(NSURL *)URL
|
||||
{
|
||||
return [self openURL:URL withParams:nil andThen:nil];
|
||||
}
|
||||
+ (BOOL)openURL:(NSURL *)URL
|
||||
withParams:(NSDictionary<NSString *, NSDictionary<NSString *, id> *> *)params
|
||||
{
|
||||
return [self openURL:URL withParams:params andThen:nil];
|
||||
}
|
||||
+ (BOOL)openURL:(NSURL *)URL
|
||||
withParams:(NSDictionary<NSString *, NSDictionary<NSString *, id> *> *)params
|
||||
andThen:(void(^)(NSString *pathComponentKey, id obj, id returnValue))then
|
||||
{
|
||||
if (![self canOpenURL:URL]) {
|
||||
return NO;
|
||||
}
|
||||
|
||||
NSString *scheme = URL.scheme;
|
||||
TARouter *router = [self routerForScheme:scheme];
|
||||
|
||||
NSString *host = URL.host;
|
||||
TARUsage usage = [self usage:host];
|
||||
|
||||
NSDictionary<NSString *, NSString *> *queryDic = [self queryDicFromURL:URL];
|
||||
NSString *paramsJson = [queryDic objectForKey:TARURLQueryParamsKey];
|
||||
NSDictionary<NSString *, NSDictionary<NSString *, id> *> *allURLParams = [self paramsFromJson:paramsJson];
|
||||
|
||||
NSArray<NSString *> *pathComponents = URL.pathComponents;
|
||||
|
||||
[pathComponents enumerateObjectsUsingBlock:^(NSString * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
if (![obj isEqualToString:@"/"]) {
|
||||
|
||||
NSArray<NSString *> * subPaths = [obj componentsSeparatedByString:TARURLSubPathSplitPattern];
|
||||
NSString *pathComponentKey = subPaths.firstObject;
|
||||
|
||||
Class mClass;
|
||||
TARPathComponentCustomHandler handler;
|
||||
TARPathComponent *pathComponent = [router.pathComponentByKey objectForKey:pathComponentKey];
|
||||
if (pathComponent) {
|
||||
mClass = pathComponent.mClass;
|
||||
handler = pathComponent.handler;
|
||||
} else {
|
||||
mClass = NSClassFromString(pathComponentKey);
|
||||
}
|
||||
|
||||
NSDictionary<NSString *, id> *URLParams = [allURLParams objectForKey:pathComponentKey];
|
||||
NSDictionary<NSString *, id> *funcParams = [params objectForKey:pathComponentKey];
|
||||
NSDictionary<NSString *, id> *finalParams = [self solveURLParams:URLParams withFuncParams:funcParams forClass:usage == TARUsageCallService ? nil : mClass];
|
||||
|
||||
if (handler) {
|
||||
handler(finalParams);
|
||||
return;
|
||||
}
|
||||
|
||||
NSString *protocolStr;
|
||||
Protocol *protocol;
|
||||
if (subPaths.count >= 2) {
|
||||
protocolStr = subPaths[1];
|
||||
protocol = NSProtocolFromString(protocolStr);
|
||||
}
|
||||
|
||||
id obj;
|
||||
id returnValue;
|
||||
|
||||
switch (usage) {
|
||||
case TARUsageCallService: {
|
||||
NSString *selectorStr = subPaths[2];
|
||||
SEL selector = NSSelectorFromString(selectorStr);
|
||||
obj = [[TAServiceManager sharedManager] createService:protocol];
|
||||
returnValue = [self safePerformAction:selector forTarget:obj withParams:finalParams];
|
||||
} break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
!then?:then(pathComponentKey, obj, returnValue);
|
||||
}
|
||||
}];
|
||||
|
||||
return YES;
|
||||
}
|
||||
|
||||
|
||||
#pragma mark - private
|
||||
+ (TARUsage)usage:(NSString *)usagePattern
|
||||
{
|
||||
usagePattern = usagePattern.lowercaseString;
|
||||
if ([usagePattern isEqualToString:TARURLHostCallService]) {
|
||||
return TARUsageCallService;
|
||||
}
|
||||
return TARUsageUnknown;
|
||||
}
|
||||
|
||||
+ (NSDictionary<NSString *, id> *)queryDicFromURL:(NSURL *)URL
|
||||
{
|
||||
if (!URL) {
|
||||
return nil;
|
||||
}
|
||||
if ([UIDevice currentDevice].systemVersion.floatValue < 8) {
|
||||
NSMutableDictionary *dic = @{}.mutableCopy;
|
||||
NSString *query = URL.query;
|
||||
NSArray<NSString *> *queryStrs = [query componentsSeparatedByString:@"&"];
|
||||
[queryStrs enumerateObjectsUsingBlock:^(NSString * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
NSArray *keyValue = [obj componentsSeparatedByString:@"="];
|
||||
if (keyValue.count >= 2) {
|
||||
NSString *key = keyValue[0];
|
||||
NSString *value = keyValue[1];
|
||||
[dic setObject:value forKey:key];
|
||||
}
|
||||
}];
|
||||
return dic;
|
||||
} else {
|
||||
NSURLComponents *URLComponents = [NSURLComponents componentsWithURL:URL
|
||||
resolvingAgainstBaseURL:NO];
|
||||
NSArray *queryItems = URLComponents.queryItems;
|
||||
NSMutableDictionary *dic = @{}.mutableCopy;
|
||||
for (NSURLQueryItem *item in queryItems) {
|
||||
if (item.name && item.value) {
|
||||
[dic setObject:item.value forKey:item.name];
|
||||
}
|
||||
}
|
||||
return dic;
|
||||
}
|
||||
}
|
||||
|
||||
+ (NSDictionary<NSString *, NSDictionary<NSString *, id> *> *)paramsFromJson:(NSString *)json
|
||||
{
|
||||
if (!json.length) {
|
||||
return nil;
|
||||
}
|
||||
NSError *error;
|
||||
NSDictionary *dic = [NSJSONSerialization JSONObjectWithData:[json dataUsingEncoding:NSUTF8StringEncoding] options:0 error:&error];
|
||||
if (error) {
|
||||
NSLog(@"TARouter [Error] Wrong URL Query Format: \n%@", error.description);
|
||||
}
|
||||
return dic;
|
||||
}
|
||||
|
||||
|
||||
+ (NSDictionary<NSString *, id> *)solveURLParams:(NSDictionary<NSString *, id> *)URLParams
|
||||
withFuncParams:(NSDictionary<NSString *, id> *)funcParams
|
||||
forClass:(Class)mClass
|
||||
{
|
||||
if (!URLParams) {
|
||||
URLParams = @{};
|
||||
}
|
||||
NSMutableDictionary<NSString *, id> *params = URLParams.mutableCopy;
|
||||
NSArray<NSString *> *funcParamKeys = funcParams.allKeys;
|
||||
[funcParamKeys enumerateObjectsUsingBlock:^(id _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
[params setObject:funcParams[obj] forKey:obj];
|
||||
}];
|
||||
if (mClass) {
|
||||
NSArray<NSString *> *paramKeys = params.allKeys;
|
||||
[paramKeys enumerateObjectsUsingBlock:^(NSString * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
objc_property_t prop = class_getProperty(mClass, obj.UTF8String);
|
||||
if (!prop) {
|
||||
[params removeObjectForKey:obj];
|
||||
} else {
|
||||
NSString *propAttr = [[NSString alloc] initWithCString:property_getAttributes(prop) encoding:NSUTF8StringEncoding];
|
||||
NSRange range = [propAttr rangeOfString:TARClassRegex options:NSRegularExpressionSearch];
|
||||
if (range.length != 0) {
|
||||
NSString *propClassName = [propAttr substringWithRange:range];
|
||||
Class propClass = objc_getClass([propClassName UTF8String]);
|
||||
if ([propClass isSubclassOfClass:[NSString class]] && [params[obj] isKindOfClass:[NSNumber class]]) {
|
||||
[params setObject:[NSString stringWithFormat:@"%@", params[obj]] forKey:obj];
|
||||
} else if ([propClass isSubclassOfClass:[NSNumber class]] && [params[obj] isKindOfClass:[NSString class]]) {
|
||||
[params setObject:@(((NSString *)params[obj]).doubleValue) forKey:obj];
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}];
|
||||
}
|
||||
return params;
|
||||
}
|
||||
|
||||
+ (void)setObject:(id)object
|
||||
withPropertys:(NSDictionary<NSString *, id> *)propertys
|
||||
{
|
||||
if (!object) {
|
||||
return;
|
||||
}
|
||||
[propertys enumerateKeysAndObjectsUsingBlock:^(NSString * _Nonnull key, id _Nonnull obj, BOOL * _Nonnull stop) {
|
||||
[object setValue:obj forKey:key];
|
||||
}];
|
||||
}
|
||||
|
||||
+ (id)safePerformAction:(SEL)action
|
||||
forTarget:(NSObject *)target
|
||||
withParams:(NSDictionary *)params
|
||||
{
|
||||
NSMethodSignature * sig = [target methodSignatureForSelector:action];
|
||||
if (!sig) { return nil; }
|
||||
NSInvocation *inv = [NSInvocation invocationWithMethodSignature:sig];
|
||||
if (!inv) { return nil; }
|
||||
[inv setTarget:target];
|
||||
[inv setSelector:action];
|
||||
NSArray<NSString *> *keys = params.allKeys;
|
||||
keys = [keys sortedArrayUsingComparator:^NSComparisonResult(NSString * _Nonnull obj1, NSString * _Nonnull obj2) {
|
||||
if (obj1.integerValue < obj2.integerValue) {
|
||||
return NSOrderedAscending;
|
||||
} else if (obj1.integerValue == obj2.integerValue) {
|
||||
return NSOrderedSame;
|
||||
} else {
|
||||
return NSOrderedDescending;
|
||||
}
|
||||
}];
|
||||
[keys enumerateObjectsUsingBlock:^(NSString * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
|
||||
id value = params[obj];
|
||||
[inv setArgument:&value atIndex:idx+2];
|
||||
}];
|
||||
[inv invoke];
|
||||
return [NSObject bh_getReturnFromInv:inv withSig:sig];
|
||||
}
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user