Generating properties and closures for enums

CouchTrackerCore/CodeGeneration/EnumClosures.swift

@@ -0,0 +1,31 @@
+/*
+ Given the following enum
+
+ ```
+ enum ViewState<T>: EnumClosures {
+    case start(data: T)
+    case loading
+    case completed(count: Int, message: String)
+}
+ ```
+
+ will generate the following code
+
+ ```
+ extension ViewState {
+    internal func onStart(_ fn: (T) -> Void) {
+        guard case let .start(data) = self else { return }
+        fn(data)
+    }
+    internal func onLoading(_ fn: () -> Void) {
+        guard case .loading = self else { return }
+        fn()
+    }
+    internal func onCompleted(_ fn: (Int, String) -> Void) {
+        guard case let .completed(count, message) = self else { return }
+        fn(count, message)
+    }
+}
+ ```
+ */
+public protocol EnumClosures {}

CouchTrackerCore/CodeGeneration/EnumPoetry.swift

@@ -1,5 +1,2 @@
-public protocol EnumClosures {}
-
-public protocol EnumProperties {}
-
+// See `EnumClosures` and `EnumProperties`
 public protocol EnumPoetry: EnumClosures, EnumProperties {}

CouchTrackerCore/CodeGeneration/EnumProperties.swift

@@ -0,0 +1,40 @@
+/*
+ Given the following enum
+
+ ```
+ enum ViewState<T>: EnumProperties {
+ case start(T)
+ case loading
+ case completed(Int, String)
+ }
+ ```
+
+ will generate the following code
+
+ ```
+extension ViewState {
+    internal var isStart: Bool {
+        guard case .start = self else { return false }
+        return true
+    }
+    internal var isLoading: Bool {
+        guard case .loading = self else { return false }
+        return true
+    }
+    internal var isCompleted: Bool {
+        guard case .completed = self else { return false }
+        return true
+    }
+
+    internal var start: T? {
+        guard case let .start(data) = self else { return nil }
+        return (data)
+    }
+    internal var completed: (count: Int, message: String)? {
+        guard case let .completed(count, message) = self else { return nil }
+        return (count, message)
+    }
+}
+ ```
+ */
+public protocol EnumProperties {}

CouchTrackerCore/Movies/Manager/MoviesManagerViewState.swift

@@ -2,10 +2,3 @@ public enum MoviesManagerViewState: Hashable, EnumPoetry {
   case loading
   case showing(pages: [ModulePage], selectedIndex: Int)
 }
-
-extension MoviesManagerViewState {
-    var showing: (pages: [ModulePage], selectedIndex: Int)? {
-        guard case let .showing(pages, x) = self else { return nil }
-        return (pages: pages, selectedIndex: x)
-    }
-}

SourceryTemplates/EnumClosures.stencil

@@ -1,8 +1,6 @@
 {% macro associatedValueNamedType associatedValue %}{% if associatedValue.localName %}{{ associatedValue.localName }}{% else %}value{{ forloop.counter }}{% endif %}{% endmacro %}
-
 {% macro associatedValueTypes associatedValues %}{% for associatedValue in associatedValues %}{% if forloop.first %}{{ associatedValue.typeName }}{% else %}, {{ associatedValue.typeName }}{% endif %}{% endfor %}{% endmacro %}
 {% macro associatedValueNames associatedValues %}{% for associatedValue in case.associatedValues %}{% if forloop.first %}{% call associatedValueNamedType associatedValue %}{% else %}, {% call associatedValueNamedType associatedValue %}{% endif %}{% endfor %}{% endmacro %}
-
 // MARK: - EnumClosures
 
 {% for enum in types.enums where enum.implements.EnumClosures %}

SourceryTemplates/EnumProperties.stencil

@@ -1,8 +1,7 @@
 {% macro associatedValueNamedType associatedValue %}{% if associatedValue.localName %}{{ associatedValue.localName }}{% else %}value{{ forloop.counter }}{% endif %}{% endmacro %}
-
 {% macro associatedValueTypes associatedValues %}{% for associatedValue in associatedValues %}{% if forloop.first %}{{ associatedValue.typeName }}{% else %}, {{ associatedValue.typeName }}{% endif %}{% endfor %}{% endmacro %}
 {% macro associatedValueNames associatedValues %}{% for associatedValue in case.associatedValues %}{% if forloop.first %}{% call associatedValueNamedType associatedValue %}{% else %}, {% call associatedValueNamedType associatedValue %}{% endif %}{% endfor %}{% endmacro %}
-
+{% macro associatedTypeAsTuple associatedValues %}{% if associatedValues.count == 1 %}{{ associatedValues[0].typeName }}{% else %}({% for associatedValue in associatedValues %}{% if forloop.first %}{% call associatedValueNamedType associatedValue %}: {{ associatedValue.typeName }}{% else %}, {% call associatedValueNamedType associatedValue %}: {{ associatedValue.typeName }}{% endif %}{% endfor %}){% endif %}{% endmacro %}
 // MARK: - EnumProperties
 
 {% for enum in types.enums where enum.implements.EnumProperties %}
@@ -13,10 +12,9 @@ extension {{ enum.name }} {
         return true
     }
     {% endfor %}
-
     {% for case in enum.cases %}
     {% if case.hasAssociatedValue %}
-    {{ enum.accessLevel }} var {{ case.name }}: ({% call associatedValueTypes case.associatedValues %})? {
+    {{ enum.accessLevel }} var {{ case.name }}: {% call associatedTypeAsTuple case.associatedValues %}? {
         guard case let .{{ case.name }}({% call associatedValueNames case.associatedValues %}) = self else { return nil }
         return ({% call associatedValueNames case.associatedValues %})
     }