Fix configuration setting for edgefn (#950)

Co-authored-by: Brandon Sneed <[email protected]>

Author: bsneed

Segment/Classes/SEGAnalytics.m

@@ -52,11 +52,6 @@ - (instancetype)initWithConfiguration:(SEGAnalyticsConfiguration *)configuration
         // TODO: Figure out if this is really the best way to do things here.
         self.integrationsManager = [[SEGIntegrationsManager alloc] initWithAnalytics:self];
 
-        if (configuration.edgeFunctionMiddleware) {
-            configuration.sourceMiddleware = @[[configuration.edgeFunctionMiddleware sourceMiddleware]];
-            configuration.destinationMiddleware = @[[configuration.edgeFunctionMiddleware destinationMiddleware]];
-        }
-
         self.runner = [[SEGMiddlewareRunner alloc] initWithMiddleware:
                                                        [configuration.sourceMiddleware ?: @[] arrayByAddingObject:self.integrationsManager]];
 

Segment/Classes/SEGAnalyticsConfiguration.m

@@ -8,6 +8,7 @@
 
 #import "SEGAnalyticsConfiguration.h"
 #import "SEGAnalytics.h"
+#import "SEGMiddleware.h"
 #import "SEGCrypto.h"
 #if TARGET_OS_IPHONE
 #import <UIKit/UIKit.h>
@@ -107,4 +108,10 @@ - (void)setMiddlewares:(NSArray<id<SEGMiddleware>> *)middlewares
     return self.sourceMiddleware;
 }
 
+- (void)setEdgeFunctionMiddleware:(id<SEGEdgeFunctionMiddleware>)edgeFunctionMiddleware
+{
+    self.sourceMiddleware = edgeFunctionMiddleware.sourceMiddleware;
+    self.destinationMiddleware = edgeFunctionMiddleware.destinationMiddleware;
+}
+
 @end