add HHRouteType.

Add HHRouteType.
lightory · May 22, 2015 · a7e908a · a7e908a
1 parent f1e0774
commit a7e908a
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/HHRouter/HHRouter.h b/HHRouter/HHRouter.h
@@ -26,6 +26,12 @@
 /// @name HHRouter
 ///---------------
 
+typedef NS_ENUM (NSInteger, HHRouteType) {
+    HHRouteTypeNone = 0,
+    HHRouteTypeViewController = 1,
+    HHRouteTypeBlock = 2
+};
+
 typedef id (^HHRouterBlock)(NSDictionary *params);
 
 @interface HHRouter : NSObject
@@ -40,6 +46,8 @@ typedef id (^HHRouterBlock)(NSDictionary *params);
 - (HHRouterBlock)matchBlock:(NSString *)route;
 - (id)callBlock:(NSString *)route;
 
+- (HHRouteType)canRoute:(NSString *)route;
+
 @end
 
 ///--------------------------------

diff --git a/HHRouter/HHRouter.m b/HHRouter/HHRouter.m
@@ -230,6 +230,21 @@ - (void)map:(NSString *)route toControllerClass:(Class)controllerClass
     subRoutes[@"_"] = controllerClass;
 }
 
+- (HHRouteType)canRoute:(NSString *)route
+{
+    NSDictionary *params = [self paramsInRoute:route];
+
+    if (params[@"controller_class"]) {
+        return HHRouteTypeViewController;
+    }
+
+    if (params[@"block"]) {
+        return HHRouteTypeBlock;
+    }
+
+    return HHRouteTypeNone;
+}
+
 @end
 
 #pragma mark - UIViewController Category