From 465c22f769eab09bc5404dddb301a48dfcb66f8e Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Sat, 8 Dec 2018 01:54:18 +0530
Subject: On-Demand: Move detail text to TunnelViewModel

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 WireGuard/WireGuard/UI/TunnelViewModel.swift        | 12 ++++++++++++
 .../UI/iOS/TunnelDetailTableViewController.swift    | 21 +--------------------
 2 files changed, 13 insertions(+), 20 deletions(-)

(limited to 'WireGuard/WireGuard')

diff --git a/WireGuard/WireGuard/UI/TunnelViewModel.swift b/WireGuard/WireGuard/UI/TunnelViewModel.swift
index f237660..2a3f36b 100644
--- a/WireGuard/WireGuard/UI/TunnelViewModel.swift
+++ b/WireGuard/WireGuard/UI/TunnelViewModel.swift
@@ -464,6 +464,18 @@ extension TunnelViewModel {
         }
     }
 
+    static func activateOnDemandDetailText(for activateOnDemandSetting: ActivateOnDemandSetting?) -> String {
+        if let activateOnDemandSetting = activateOnDemandSetting {
+            if (activateOnDemandSetting.isActivateOnDemandEnabled) {
+                return TunnelViewModel.activateOnDemandOptionText(for: activateOnDemandSetting.activateOnDemandOption)
+            } else {
+                return TunnelViewModel.activateOnDemandOptionText(for: .none)
+            }
+        } else {
+            return TunnelViewModel.activateOnDemandOptionText(for: .none)
+        }
+    }
+
     func defaultActivateOnDemandOption() -> ActivateOnDemandOption {
         return .useOnDemandOverWiFiOrCellular
     }
diff --git a/WireGuard/WireGuard/UI/iOS/TunnelDetailTableViewController.swift b/WireGuard/WireGuard/UI/iOS/TunnelDetailTableViewController.swift
index 5873f1d..a888924 100644
--- a/WireGuard/WireGuard/UI/iOS/TunnelDetailTableViewController.swift
+++ b/WireGuard/WireGuard/UI/iOS/TunnelDetailTableViewController.swift
@@ -426,26 +426,7 @@ class TunnelDetailTableViewActivateOnDemandCell: UITableViewCell {
     }
 
     func update(from activateOnDemandSetting: ActivateOnDemandSetting?) {
-        let detailText: String
-        if let activateOnDemandSetting = activateOnDemandSetting {
-            if (activateOnDemandSetting.isActivateOnDemandEnabled) {
-                switch (activateOnDemandSetting.activateOnDemandOption) {
-                case .none:
-                    detailText = "Off"
-                case .useOnDemandOverWiFiOrCellular:
-                    detailText = "Wi-Fi or cellular"
-                case .useOnDemandOverWiFiOnly:
-                    detailText = "Wi-Fi only"
-                case .useOnDemandOverCellularOnly:
-                    detailText = "Cellular only"
-                }
-            } else {
-                detailText = "Off"
-            }
-        } else {
-            detailText = "Off"
-        }
-        detailTextLabel?.text = detailText
+        detailTextLabel?.text = TunnelViewModel.activateOnDemandDetailText(for: activateOnDemandSetting)
     }
 
     override func prepareForReuse() {
-- 
cgit v1.2.3-59-g8ed1b