From cc20211a7dc044c35bd806d99f6d2200a2065e2e Mon Sep 17 00:00:00 2001
From: Renan Mav <renan.mav@hotmail.com>
Date: Tue, 22 Oct 2024 20:24:58 -0300
Subject: [PATCH] feat: start implementing ASWebAuthenticationSession

---
 example/App.tsx                               | 12 +++++
 ios/HybridASWebAuthenticationSession.swift    | 50 +++++++++++++++++++
 nitro.json                                    |  3 ++
 .../ios/NitroInAppBrowserAutolinking.mm       |  8 +++
 .../ios/NitroInAppBrowserAutolinking.swift    | 12 +++++
 src/ASWebAuthenticationSession.ts             | 31 ++++++++++++
 src/index.ts                                  |  1 +
 src/specs/ASWebAuthenticationSession.nitro.ts |  2 +-
 8 files changed, 118 insertions(+), 1 deletion(-)
 create mode 100644 ios/HybridASWebAuthenticationSession.swift
 create mode 100644 src/ASWebAuthenticationSession.ts

diff --git a/example/App.tsx b/example/App.tsx
index 9df1fd4..ff54a88 100644
--- a/example/App.tsx
+++ b/example/App.tsx
@@ -24,6 +24,7 @@ import {
   InAppBrowser,
   SFSafariViewController,
   ModalPresentationStyle,
+  ASWebAuthenticationSession,
   ChromeCustomTabs,
 } from 'react-native-in-app-browser'
 
@@ -60,6 +61,7 @@ function Section({ children, title }: SectionProps): React.JSX.Element {
 }
 
 const URL = 'https://reactnative.dev'
+const AUTH_URL = 'https://github.com/login'
 
 function App(): React.JSX.Element {
   const isDarkMode = useColorScheme() === 'dark'
@@ -114,6 +116,16 @@ function App(): React.JSX.Element {
               }}
             />
           </Section>
+          <Section title="🍎 ASWebAuthenticationSession">
+            <Button
+              title={`start(${AUTH_URL})`}
+              onPress={() => {
+                ASWebAuthenticationSession.start({
+                  url: AUTH_URL,
+                })
+              }}
+            />
+          </Section>
           <Section title="🤖 ChromeCustomTabs">
             <Button
               title={`launch(${URL})`}
diff --git a/ios/HybridASWebAuthenticationSession.swift b/ios/HybridASWebAuthenticationSession.swift
new file mode 100644
index 0000000..6ec09fb
--- /dev/null
+++ b/ios/HybridASWebAuthenticationSession.swift
@@ -0,0 +1,50 @@
+import Foundation
+import AuthenticationServices
+
+class HybridASWebAuthenticationSession: HybridASWebAuthenticationSessionSpec {
+  var hybridContext = margelo.nitro.HybridContext()
+  
+  var memorySize: Int {
+    return getSizeOf(self)
+  }
+  
+  private var authSession: ASWebAuthenticationSession?
+  
+  var prefersEphemeralWebBrowserSession: Bool = false
+  
+  func start(params: ASWebAuthenticationSessionStartParams) throws -> Void {
+    NSLog("HybridASWebAuthenticationSession.start(url:%@) is being called", params.url)
+    
+    guard let nativeUrl = URL(string: params.url) else {
+      throw NSError(domain: "HybridASWebAuthenticationSession", code: 0, userInfo: [NSLocalizedDescriptionKey: "Invalid URL"])
+    }
+    
+    authSession = ASWebAuthenticationSession(url: nativeUrl, callbackURLScheme: nil) { session, error in
+      if let error = error {
+        NSLog("ASWebAuthenticationSession failed with error: %@", error.localizedDescription)
+        // TODO: Implement error handling and callback to JavaScript
+      } else if let callbackURL = session {
+        NSLog("ASWebAuthenticationSession succeeded with URL: %@", callbackURL.absoluteString)
+        // TODO: Implement success callback to JavaScript with the callbackURL
+      } else {
+        NSLog("ASWebAuthenticationSession completed without error or callback URL")
+        // TODO: Implement callback to JavaScript for completion without result
+      }
+      
+      self.authSession = nil
+    }
+    
+    if !(authSession?.start() ?? false) {
+      throw NSError(domain: "HybridASWebAuthenticationSession", code: 1, userInfo: [NSLocalizedDescriptionKey: "ASWebAuthenticationSession failed to start"])
+    }
+  }
+  
+  func cancel() throws -> Void {
+    guard let authSession = authSession else {
+      throw NSError(domain: "HybridASWebAuthenticationSession", code: 2, userInfo: [NSLocalizedDescriptionKey: "ASWebAuthenticationSession is not initialized"])
+    }
+    
+    authSession.cancel()
+    self.authSession = nil
+  }
+}
diff --git a/nitro.json b/nitro.json
index 66a8b72..97cfd50 100644
--- a/nitro.json
+++ b/nitro.json
@@ -11,6 +11,9 @@
     "SFSafariViewController": {
       "swift": "HybridSFSafariViewController"
     },
+    "ASWebAuthenticationSession": {
+      "swift": "HybridASWebAuthenticationSession"
+    },
     "ChromeCustomTabs": {
       "kotlin": "HybridChromeCustomTabs"
     }
diff --git a/nitrogen/generated/ios/NitroInAppBrowserAutolinking.mm b/nitrogen/generated/ios/NitroInAppBrowserAutolinking.mm
index 508930a..e09c36a 100644
--- a/nitrogen/generated/ios/NitroInAppBrowserAutolinking.mm
+++ b/nitrogen/generated/ios/NitroInAppBrowserAutolinking.mm
@@ -11,6 +11,7 @@
 #import <type_traits>
 
 #include "HybridSFSafariViewControllerSpecSwift.hpp"
+#include "HybridASWebAuthenticationSessionSpecSwift.hpp"
 
 @interface NitroInAppBrowserAutolinking : NSObject
 @end
@@ -28,6 +29,13 @@ + (void) load {
       return std::make_shared<HybridSFSafariViewControllerSpecSwift>(swiftPart);
     }
   );
+  HybridObjectRegistry::registerHybridObjectConstructor(
+    "ASWebAuthenticationSession",
+    []() -> std::shared_ptr<HybridObject> {
+      auto swiftPart = NitroInAppBrowser::NitroInAppBrowserAutolinking::createASWebAuthenticationSession();
+      return std::make_shared<HybridASWebAuthenticationSessionSpecSwift>(swiftPart);
+    }
+  );
 }
 
 @end
diff --git a/nitrogen/generated/ios/NitroInAppBrowserAutolinking.swift b/nitrogen/generated/ios/NitroInAppBrowserAutolinking.swift
index daad6fe..92f4709 100644
--- a/nitrogen/generated/ios/NitroInAppBrowserAutolinking.swift
+++ b/nitrogen/generated/ios/NitroInAppBrowserAutolinking.swift
@@ -17,4 +17,16 @@ public final class NitroInAppBrowserAutolinking {
     let hybridObject = HybridSFSafariViewController()
     return hybridObject.createCxxBridge()
   }
+  
+  /**
+   * Creates an instance of a Swift class that implements `HybridASWebAuthenticationSessionSpec`,
+   * and wraps it in a Swift class that can directly interop with C++ (`HybridASWebAuthenticationSessionSpecCxx`)
+   *
+   * This is generated by Nitrogen and will initialize the class specified
+   * in the `"autolinking"` property of `nitro.json` (in this case, `HybridASWebAuthenticationSession`).
+   */
+  public static func createASWebAuthenticationSession() -> HybridASWebAuthenticationSessionSpecCxx {
+    let hybridObject = HybridASWebAuthenticationSession()
+    return hybridObject.createCxxBridge()
+  }
 }
diff --git a/src/ASWebAuthenticationSession.ts b/src/ASWebAuthenticationSession.ts
new file mode 100644
index 0000000..15f1463
--- /dev/null
+++ b/src/ASWebAuthenticationSession.ts
@@ -0,0 +1,31 @@
+import { Platform } from 'react-native'
+import { NitroModules } from 'react-native-nitro-modules'
+import type { ASWebAuthenticationSession as ASWebAuthenticationSessionType } from './specs/ASWebAuthenticationSession.nitro'
+
+export * from './specs/ASWebAuthenticationSession.nitro'
+
+let ASWebAuthenticationSession: ASWebAuthenticationSessionType
+
+const ASWebAuthenticationSessionStub: ASWebAuthenticationSessionType = {
+  name: 'ASWebAuthenticationSession',
+  equals: () => false,
+  dispose: () => {},
+  prefersEphemeralWebBrowserSession: false,
+  start: () => {
+    console.warn('ASWebAuthenticationSession.start is only supported on iOS.')
+  },
+  cancel: () => {
+    console.warn('ASWebAuthenticationSession.cancel is only supported on iOS.')
+  },
+}
+
+if (Platform.OS === 'ios') {
+  ASWebAuthenticationSession =
+    NitroModules.createHybridObject<ASWebAuthenticationSessionType>(
+      'ASWebAuthenticationSession'
+    )
+} else {
+  ASWebAuthenticationSession = ASWebAuthenticationSessionStub
+}
+
+export { ASWebAuthenticationSession }
diff --git a/src/index.ts b/src/index.ts
index 4605bec..3ab5ede 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -3,4 +3,5 @@ export * from './InAppBrowser'
 
 // Platform specific interfaces
 export * from './SFSafariViewController'
+export * from './ASWebAuthenticationSession'
 export * from './ChromeCustomTabs'
diff --git a/src/specs/ASWebAuthenticationSession.nitro.ts b/src/specs/ASWebAuthenticationSession.nitro.ts
index a094e80..b281582 100644
--- a/src/specs/ASWebAuthenticationSession.nitro.ts
+++ b/src/specs/ASWebAuthenticationSession.nitro.ts
@@ -2,9 +2,9 @@ import type { HybridObject } from 'react-native-nitro-modules'
 
 export interface ASWebAuthenticationSession
   extends HybridObject<{ ios: 'swift' }> {
+  prefersEphemeralWebBrowserSession: boolean
   start(params: ASWebAuthenticationSessionStartParams): void
   cancel(): void
-  prefersEphemeralWebBrowserSession: boolean
 }
 
 export interface ASWebAuthenticationSessionStartParams {