Implement ViewCodeProtocol

emmendesf · May 23, 2021 · b7a76a1 · b7a76a1
1 parent f43956a
commit b7a76a1
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/ViewCodeGuide/ViewCodeGuide/Source/MyView.swift b/ViewCodeGuide/ViewCodeGuide/Source/MyView.swift
@@ -13,8 +13,7 @@ final class MyView: UIView {
         super.init(frame: .zero)
         self.backgroundColor = .orange
 
-        setupHierarchy()
-        setupConstraints()
+        build()
     }
 
     @available(*, unavailable)
@@ -28,6 +27,9 @@ final class MyView: UIView {
 
         return view
     }()
+}
+
+extension MyView: ViewCodeProtocol {
 
     func setupHierarchy() {
         addSubview(titleSubtitleView)
@@ -38,4 +40,8 @@ final class MyView: UIView {
         titleSubtitleView.leadingAnchor.constraint(equalTo: leadingAnchor, constant: 16).isActive = true
         titleSubtitleView.trailingAnchor.constraint(equalTo: trailingAnchor, constant: -16).isActive = true
     }
+
+    func additionalSetup() {
+        backgroundColor = .orange
+    }
 }
diff --git a/ViewCodeGuide/ViewCodeGuide/Source/TitleSubtitleView.swift b/ViewCodeGuide/ViewCodeGuide/Source/TitleSubtitleView.swift
@@ -19,8 +19,7 @@ final class TitleSubtitleView: UIView {
 
         super.init(frame: .zero)
 
-        setupHierarchy()
-        setupConstraints()
+        build()
     }
 
     @available(*, unavailable)
@@ -45,6 +44,9 @@ final class TitleSubtitleView: UIView {
 
         return subtitleLabel
     }()
+}
+
+extension TitleSubtitleView: ViewCodeProtocol {
 
     func setupHierarchy() {
         addSubview(titleLabel)

diff --git a/ViewCodeGuide/ViewCodeGuide/Source/ViewCodeProtocol.swift b/ViewCodeGuide/ViewCodeGuide/Source/ViewCodeProtocol.swift
@@ -9,12 +9,16 @@ protocol ViewCodeProtocol {
 
     func setupHierarchy()
     func setupConstraints()
+    func additionalSetup()
 }
 
 extension ViewCodeProtocol {
 
     func build() {
         setupHierarchy()
         setupConstraints()
+        additionalSetup()
     }
+
+    func additionalSetup() { }
 }