From 9edd770911a9dfd5b350d579e9092a62873e1d4a Mon Sep 17 00:00:00 2001
From: Kyle Gunger <corechg@gmail.com>
Date: Thu, 23 Jul 2020 19:31:09 -0400
Subject: Proper type checking

---
 .../net/corechg/transit/type/group/GroupRegistry.java     | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

(limited to 'src/main/java')

diff --git a/src/main/java/net/corechg/transit/type/group/GroupRegistry.java b/src/main/java/net/corechg/transit/type/group/GroupRegistry.java
index 3929973..0b04913 100644
--- a/src/main/java/net/corechg/transit/type/group/GroupRegistry.java
+++ b/src/main/java/net/corechg/transit/type/group/GroupRegistry.java
@@ -30,20 +30,27 @@ public class GroupRegistry {
 		return true;
 	}
 	
-	public static final TypeGroup<?> groupByID(String groupID)
+	@SuppressWarnings("unchecked")
+	public static final <T> TypeGroup<T> groupByID(String groupID)
 	{
 		for(TypeGroup<?> g : GROUPS)
 		{
-			if(g.getGroup().equals(groupID)) return g;
+			if(g.getGroup().equals(groupID)){
+				try{
+					return (TypeGroup<T>) g;
+				}catch(ClassCastException e) {
+					return null;
+				}
+			}
 		}
 		
 		return null;
 		
 	}
 	
-	public static final Type<?> typeByIdentity(String groupID, String typeID)
+	public static final <T> Type<T> typeByIdentity(String groupID, String typeID)
 	{
-		return groupByID(groupID).getType(typeID);
+		return GroupRegistry.<T>groupByID(groupID).getType(typeID);
 	}
 	
 }
-- 
cgit v1.2.3