Refer to protocols with @protocol

Fixes #114
JonasGessner · Feb 13, 2015 · e558d2d · e558d2d
1 parent 40e2987
commit e558d2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bin/lib/Logos/Generator/MobileSubstrate/Subclass.pm b/bin/lib/Logos/Generator/MobileSubstrate/Subclass.pm
@@ -23,7 +23,7 @@ sub initializers {
 	}
 	# </ivars>
 	foreach(keys %{$class->protocols}) {
-		$return .= "class_addProtocol(".$self->variable($class).", objc_getProtocol(\"$_\")); ";
+		$return .= "class_addProtocol(".$self->variable($class).", \@protocol($_)); ";
 	}
 	$return .= "objc_registerClassPair(".$self->variable($class)."); ";
 	$return .= "}";

diff --git a/bin/lib/Logos/Generator/internal/Subclass.pm b/bin/lib/Logos/Generator/internal/Subclass.pm
@@ -23,7 +23,7 @@ sub initializers {
 	}
 	# </ivars>
 	foreach(keys %{$class->protocols}) {
-		$return .= "class_addProtocol(".$self->variable($class).", objc_getProtocol(\"$_\")); ";
+		$return .= "class_addProtocol(".$self->variable($class).", \@protocol($_)); ";
 	}
 	$return .= "objc_registerClassPair(".$self->variable($class)."); ";
 	$return .= "}";