Added support for @class protocol lists.

[cycript.git] / ObjectiveC / Output.mm

diff --git a/ObjectiveC/Output.mm b/ObjectiveC/Output.mm
index 1121121c33963e8806783f509e3b53bf4309f20b..d8f8345d1b822bbb8c064a09ffe9b52e38e812fa 100644 (file)
--- a/ObjectiveC/Output.mm
+++ b/ObjectiveC/Output.mm
@@ -70,6 +70,11 @@ void CYClass::Output(CYOutput &out, CYFlags flags) const {
         fields_->Output(out);
     if (messages_ != NULL)
         messages_->Output(out, false);
+    if (protocols_ != NULL) {
+        out << '<';
+        out << *protocols_;
+        out << '>';
+    }
     out << "objc_registerClassPair($cyc);";
     out << "return $cyc;";
     out << "}(";
@@ -104,6 +109,12 @@ void CYMessage::Output(CYOutput &out, bool replace) const {
     out << code_;
 }
 
+void CYProtocol::Output(CYOutput &out) const {
+    name_->Output(out, CYPA, CYNoFlags);
+    if (next_ != NULL)
+        out << ',' << ' ' << *next_;
+}
+
 void CYSelector::Output(CYOutput &out, CYFlags flags) const {
     out << "@selector" << '(' << name_ << ')';
 }