From 06bbe907b9d1f54e7c531fd7acc0f4731caeb524 Mon Sep 17 00:00:00 2001
From: Michael Kirk <michael@signal.org>
Date: Wed, 25 Jul 2018 13:54:47 -0600
Subject: [PATCH] builder pattern for proto construction

---
 .../Protos/ProtoWrappers/SSKEnvelope.swift    | 52 +++++++++----------
 1 file changed, 26 insertions(+), 26 deletions(-)

diff --git a/SignalServiceKit/src/Protos/ProtoWrappers/SSKEnvelope.swift b/SignalServiceKit/src/Protos/ProtoWrappers/SSKEnvelope.swift
index be35d102b..c43949734 100644
--- a/SignalServiceKit/src/Protos/ProtoWrappers/SSKEnvelope.swift
+++ b/SignalServiceKit/src/Protos/ProtoWrappers/SSKEnvelope.swift
@@ -114,34 +114,34 @@ public class SSKEnvelope: NSObject {
     }
 
     private var asProtobuf: SignalServiceProtos_Envelope {
-        var proto = SignalServiceProtos_Envelope()
-
-        proto.source = self.source
-
-        proto.type = {
-            switch self.type {
-            case .unknown:
-                return .unknown
-            case .ciphertext:
-                return .ciphertext
-            case .keyExchange:
-                return .keyExchange
-            case .prekeyBundle:
-                return .prekeyBundle
-            case .receipt:
-                return .receipt
+        let proto = SignalServiceProtos_Envelope.with { (builder) in
+            builder.source = self.source
+
+            builder.type = {
+                switch self.type {
+                case .unknown:
+                    return .unknown
+                case .ciphertext:
+                    return .ciphertext
+                case .keyExchange:
+                    return .keyExchange
+                case .prekeyBundle:
+                    return .prekeyBundle
+                case .receipt:
+                    return .receipt
+                }
+            }()
+
+            builder.timestamp = self.timestamp
+            builder.sourceDevice = self.sourceDevice
+
+            if let relay = self.relay {
+                builder.relay = relay
             }
-        }()
 
-        proto.timestamp = self.timestamp
-        proto.sourceDevice = self.sourceDevice
-
-        if let relay = self.relay {
-            proto.relay = relay
-        }
-
-        if let content = self.content {
-            proto.content = content
+            if let content = self.content {
+                builder.content = content
+            }
         }
 
         return proto