[P1] Country support for conversation sharing

OLMoE.swift/Constants/FeatureFlags.swift

@@ -5,12 +5,20 @@
 //  Created by Thomas Jones on 11/15/24.
 //
 
+import DeviceCheck
 
-enum FeatureFlags {
+public enum FeatureFlags {
 
     static let allowDeviceBypass = false
 
+    static let serverSideSharing: Bool = {
+        let appAttestService = DCAppAttestService()
+        return Locale.isCountrySupported() && appAttestService.isSupported
+    }()
+
     static let allowMockedModel = false
 
     static let useLLMCaching = true
+
+
 }

OLMoE.swift/Extensions/Locale+CountrySupport.swift

@@ -0,0 +1,24 @@
+//
+//  Locale+CountrySupport.swift
+//  OLMoE.swift
+//
+//  Created by Stanley Jovel on 3/5/25.
+//
+
+import Foundation
+
+extension Locale {
+    static let supportedCountries = ["US"]
+
+    /// Check if the user's country supports server-side sharing
+    static func isCountrySupported() -> Bool {
+        let userCountry: String?
+        if #available(iOS 16, *) {
+            userCountry = Locale.current.region?.identifier
+        } else {
+            userCountry = Locale.current.regionCode
+        }
+
+        return userCountry.map { supportedCountries.contains($0) } ?? false
+    }
+}

OLMoE.swift/Views/ContentView.swift

@@ -232,12 +232,18 @@ struct BotView: View {
         } else {
             ToolbarButton(action: {
                 isTextEditorFocused = false
-                // disclaimerHandlers.setActiveDisclaimer(Disclaimers.ShareDisclaimer())
-                // disclaimerHandlers.setCancelAction({ disclaimerHandlers.setShowDisclaimerPage(false) })
-                // disclaimerHandlers.setAllowOutsideTapDismiss(true)
-                // disclaimerHandlers.setConfirmAction({ shareConversation() })
-                // disclaimerHandlers.setShowDisclaimerPage(true)
-                showTextShareSheet = true
+
+                if FeatureFlags.serverSideSharing {
+                    // For supported countries, use the disclaimer and shareConversation flow
+                    disclaimerHandlers.setActiveDisclaimer(Disclaimers.ShareDisclaimer())
+                    disclaimerHandlers.setCancelAction({ disclaimerHandlers.setShowDisclaimerPage(false) })
+                    disclaimerHandlers.setAllowOutsideTapDismiss(true)
+                    disclaimerHandlers.setConfirmAction({ shareConversation() })
+                    disclaimerHandlers.setShowDisclaimerPage(true)
+                } else {
+                    // For unsupported countries, use the text share sheet
+                    showTextShareSheet = true
+                }
             }, imageName: "square.and.arrow.up")
              .disabled(isSharing || bot.history.isEmpty || isGenerating)
              .opacity(isSharing || bot.history.isEmpty || isGenerating ? 0.5 : 1)