performSegue extension method on viewcontroller

Bluelich · Dec 1, 2015 · 841f4ed · 841f4ed
1 parent ae13050
commit 841f4ed
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/R.swift/main.swift b/R.swift/main.swift
@@ -30,6 +30,7 @@ do {
     ReuseIdentifierUITableViewExtension.description,
     ReuseIdentifierUICollectionViewExtension.description,
     NibUIViewControllerExtension.description,
+    SegueUIViewControllerExtension.description,
     ].joinWithSeparator("\n\n")
 
   // Write file if we have changes

diff --git a/R.swift/values.swift b/R.swift/values.swift
@@ -41,9 +41,10 @@ let ReuseIdentifier = Struct(
   functions: [],
   structs: [])
 
+let StoryboardSegueType = Type(name: "StoryboardSegue", genericArgs: ["Source", "Destination"])
 
 let StoryboardSegue = Struct(
-  type: Type(name: "StoryboardSegue", genericArgs: ["Source", "Destination"]),
+  type: StoryboardSegueType,
   implements: [Type(name: "CustomStringConvertible")],
   lets: [
     Let(
@@ -114,6 +115,23 @@ let NibUIViewControllerExtension = Extension(
   ]
 )
 
+let SegueUIViewControllerExtension = Extension(
+  type: Type._UIViewController,
+  functions: [
+    Function(
+      isStatic: false,
+      name: "performSegue",
+      generics: StoryboardSegueType.genericArgs.joinWithSeparator(","),
+      parameters: [
+        Function.Parameter(name: "segue", type: StoryboardSegueType),
+        Function.Parameter(name: "sender", type: Type._AnyObject.asOptional())
+      ],
+      returnType: Type._Void,
+      body: "performSegueWithIdentifier(segue.identifier, sender: sender)"
+    ),
+  ]
+)
+
 let ReuseIdentifierUITableViewExtension = Extension(
   type: Type._UITableView,
   functions: [