Merge pull request #485 from sopt-makers/hotfix/#484-web_download_image_save_library

meltsplit · web-flow · commit 6fe0b7c3fc0e · 2025-01-25T13:33:33.000+09:00
[Hotfix] #484 - WKWebView에서 이미지 다운로드 발생 시 갤러리에 저장하는 기능을 구현한다.
diff --git a/SOPT-iOS/Projects/Core/Sources/Utils/makeAlert.swift b/SOPT-iOS/Projects/Core/Sources/Utils/makeAlert.swift
@@ -34,4 +34,20 @@ public extension UIViewController {
         alertViewController.addAction(okAction)
         self.present(alertViewController, animated: true, completion: completion)
     }
+    
+    func makeAlert(
+        title: String,
+        message: String,
+        actions: UIAlertAction...,
+        completion: (() -> Void)? = nil
+    ) {
+        makeVibrate()
+        let alertVC = UIAlertController(
+            title: title,
+            message: message,
+            preferredStyle: .alert
+        )
+        actions.forEach { alertVC.addAction($0) }
+        self.present(alertVC, animated: true, completion: completion)
+    }
 }
diff --git a/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKDownloadExecutable.swift b/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKDownloadExecutable.swift
@@ -0,0 +1,24 @@
+//
+//  WKDownloadExecutable.swift
+//  BaseFeatureDependency
+//
+//  Created by 장석우 on 1/24/25.
+//  Copyright © 2025 SOPT-iOS. All rights reserved.
+//
+
+import Foundation
+import UniformTypeIdentifiers
+import WebKit
+
+public protocol WKDownloadExecutable {
+    
+    static var key: UTType { get }
+    
+    func execute(
+        _ download: WKDownload,
+        _ response: URLResponse,
+        _ suggestedFilename: String,
+        _ webVC: SOPTWebViewControllable?
+    ) async -> URL?
+}
+
diff --git a/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKDownloadManager.swift b/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKDownloadManager.swift
@@ -0,0 +1,52 @@
+//
+//  WKDownloadManager.swift
+//  BaseFeatureDependency
+//
+//  Created by 장석우 on 1/24/25.
+//  Copyright © 2025 SOPT-iOS. All rights reserved.
+//
+
+import Foundation
+import UniformTypeIdentifiers
+import WebKit
+
+public final class WKDownloadManager {
+    
+    private var downloadHandlers: [UTType: WKDownloadExecutable]
+    var webVC: SOPTWebViewControllable?
+    
+    init(
+        downloadHandlers: [UTType : WKDownloadExecutable] = [:],
+        webVC: SOPTWebViewControllable? = nil
+    ) {
+        self.downloadHandlers = downloadHandlers
+        self.webVC = webVC
+    }
+    
+    public func register<T: WKDownloadExecutable>(_ object: T) {
+        self.downloadHandlers[T.key] = object
+    }
+    
+    func download(
+        _ download: WKDownload,
+        decideDestinationUsing response: URLResponse,
+        suggestedFilename: String
+    ) async -> URL? {
+        guard let mimeType = response.mimeType,
+              let utType = UTType(mimeType: mimeType)
+        else { return nil }
+        
+        guard let handler = downloadHandlers.first(where: { utType.conforms(to: $0.key) })?.value
+        else { return nil }
+        
+        return await handler.execute(download, response, suggestedFilename, webVC)
+    }
+    
+    public static let `default`: WKDownloadManager = {
+        let manager = WKDownloadManager()
+        manager.register(WKImageDownloadHandler())
+        return manager
+    }()
+}
+
+
diff --git a/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKImageDownloadHandler.swift b/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/Download/WKImageDownloadHandler.swift
@@ -0,0 +1,105 @@
+//
+//  WKImageDownloadHandler.swift
+//  BaseFeatureDependency
+//
+//  Created by 장석우 on 1/24/25.
+//  Copyright © 2025 SOPT-iOS. All rights reserved.
+//
+
+import Foundation
+import WebKit
+import Photos
+
+struct WKImageDownloadHandler: WKDownloadExecutable {
+    
+    static let key = UTType.image
+    
+    func execute(
+        _ download: WKDownload,
+        _ response: URLResponse,
+        _ suggestedFilename: String,
+        _ webVC: SOPTWebViewControllable?
+    ) async -> URL? {
+        
+        guard let webVC else { return nil }
+        
+        guard await requestAuthorization() else {
+            await presentGoToSettingAlert(from: webVC)
+            return nil
+        }
+        
+        guard let urlString = response.url?.absoluteString,
+              let range = urlString.range(of: "base64,"),
+              let encodedData = Data(base64Encoded: String(urlString[range.upperBound...])),
+              let image = UIImage(data: encodedData),
+              let pngData = image.pngData()
+        else {
+            return nil
+        }
+        
+        guard let fileURL = try? saveToTemporaryDirectory(pngData, suggestedFilename)
+        else { return nil }
+        
+        await presentActivityVC(fileURL, from: webVC)
+        
+        return nil
+    }
+    
+    private func requestAuthorization() async -> Bool {
+        let status = await PHPhotoLibrary.requestAuthorization(for: .addOnly)
+        return status == .authorized || status == .limited
+    }
+    
+    private func saveToTemporaryDirectory(_ data: Data, _ suggestedFilename: String) throws -> URL {
+        let temporaryURL = FileManager.default.temporaryDirectory
+            .appendingPathComponent(suggestedFilename)
+            .appendingPathExtension("png")
+        
+        try data.write(to: temporaryURL, options: [])
+        return temporaryURL
+    }
+    
+    @MainActor
+    private func presentActivityVC(_ fileURL: URL, from webVC: SOPTWebViewControllable) {
+        
+        let activityVC = UIActivityViewController(
+            activityItems: [fileURL],
+            applicationActivities: nil
+        )
+        
+        activityVC.completionWithItemsHandler = {
+            activityType, completed, returnedItems, activityError in
+            if activityType == .saveToCameraRoll {
+                guard activityError == nil else {
+                    ToastUtils.showMDSToast(type: .error, text: "이미지 저장에 실패했습니다.")
+                    return
+                }
+                
+                if completed {
+                    ToastUtils.showMDSToast(type: .success, text: "이미지가 저장되었습니다.")
+                }
+            }
+        }
+        // 공유 화면 표시
+        webVC.vc.present(activityVC, animated: true, completion: nil)
+        
+    }
+    
+    @MainActor
+    private func presentGoToSettingAlert(from webVC: SOPTWebViewControllable) {
+        webVC.vc.makeAlert(
+            title: "갤러리 접근 권한 설정",
+            message: "이미지를 저장하시려면 갤러리 접근 권한이 필요합니다.",
+            actions: .init(title: "나중에", style: .destructive),
+            .init(title: "설정", style: .default, handler: { _ in
+                guard let url = URL(string: UIApplication.openSettingsURLString),
+                      UIApplication.shared.canOpenURL(url) else { return }
+                
+                UIApplication.shared.open(url, completionHandler: nil)
+            })
+        )
+    }
+}
+
+
+
diff --git a/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/SOPTWebView.swift b/SOPT-iOS/Projects/Features/BaseFeatureDependency/Sources/WebView/SOPTWebView.swift
@@ -14,36 +14,44 @@ import WebKit
 
 import SnapKit
 
-public final class SOPTWebView: UIViewController {
+public protocol SOPTWebViewControllable {
+    var vc: UIViewController { get }
+    var webView: WKWebView { get }
+}
+
+public final class SOPTWebView: UIViewController, SOPTWebViewControllable {
     private enum Metric {
         static let navigationBarHeight = 44.f
     }
     
     private lazy var navigationBar = WebViewNavigationBar(frame: self.view.frame)
-    private let wkwebView: WKWebView
+    public let webView: WKWebView
+    public var vc: UIViewController { self }
+    private let downloadManager: WKDownloadManager
     
     // MARK: Variables
     private let cancelbag = CancelBag()
     private var barrier = false
     
     public init(
         config: WebViewConfig = WebViewConfig(),
-        startWith url: URL
+        startWith url: URL,
+        downloadManager: WKDownloadManager = .default
     ) {
         let configuration = WKWebViewConfiguration().then {
             $0.allowsInlineMediaPlayback = config.allowsInlineMediaPlayback
             $0.mediaTypesRequiringUserActionForPlayback = config.mediaTypesRequiringUserActionForPlayback
         }
         
-        self.wkwebView = WKWebView(frame: .zero, configuration: configuration).then {
+        self.webView = WKWebView(frame: .zero, configuration: configuration).then {
             $0.allowsBackForwardNavigationGestures = config.allowsBackForwardNavigationGestures
         }
-        
+        self.downloadManager = downloadManager
         super.init(nibName: nil, bundle: nil)
         
         DispatchQueue.main.async {
             let request = URLRequest(url: url)
-            self.wkwebView.load(request)
+            self.webView.load(request)
         }
     }
     
@@ -56,10 +64,10 @@ public final class SOPTWebView: UIViewController {
         
         self.view.backgroundColor = DSKitAsset.Colors.black100.color
         
-        self.wkwebView.scrollView.delegate = self
-        self.wkwebView.navigationDelegate = self
-        self.wkwebView.uiDelegate = self
-        
+        self.webView.scrollView.delegate = self
+        self.webView.navigationDelegate = self
+        self.webView.uiDelegate = self
+        downloadManager.webVC = self
         self.initializeViews()
         self.setupConstraints()
         self.setupNavigationButtonActions()
@@ -68,7 +76,7 @@ public final class SOPTWebView: UIViewController {
 
 extension SOPTWebView {
     private func initializeViews() {
-        self.view.addSubviews(self.navigationBar, self.wkwebView)
+        self.view.addSubviews(self.navigationBar, self.webView)
     }
     
     private func setupConstraints() {
@@ -78,7 +86,7 @@ extension SOPTWebView {
             $0.height.equalTo(Metric.navigationBarHeight)
         }
         
-        self.wkwebView.snp.makeConstraints {
+        self.webView.snp.makeConstraints {
             $0.top.equalTo(self.navigationBar.snp.bottom)
             $0.leading.trailing.bottom.equalToSuperview()
         }
@@ -88,12 +96,12 @@ extension SOPTWebView {
         self.navigationBar
             .signalForClickLeftButton()
             .sink { [weak self] _ in
-                guard self?.wkwebView.canGoBack == true else {
+                guard self?.webView.canGoBack == true else {
                     self?.navigationController?.popViewController(animated: true)
                     return
                 }
                     
-                self?.wkwebView.goBack()
+                self?.webView.goBack()
             }.store(in: self.cancelbag)
         
         self.navigationBar
@@ -111,10 +119,10 @@ extension SOPTWebView: WKNavigationDelegate {
         }
         
         self.barrier = true
-        self.wkwebView.evaluateJavaScript(
+        self.webView.evaluateJavaScript(
             "localStorage.setItem(\"serviceAccessToken\", \"\(playgroundToken)\")"
         )
-        self.wkwebView.reload()
+        self.webView.reload()
     }
 }
 
@@ -138,3 +146,22 @@ extension SOPTWebView: UIScrollViewDelegate {
         scrollView.pinchGestureRecognizer?.isEnabled = false
     }
 }
+
+extension SOPTWebView: WKDownloadDelegate {
+    
+    public func download(_ download: WKDownload, decideDestinationUsing response: URLResponse, suggestedFilename: String) async -> URL? {
+        return await downloadManager.download(
+            download,
+            decideDestinationUsing: response,
+            suggestedFilename: suggestedFilename
+        )
+    }
+    
+    public func webView(_ webView: WKWebView, navigationAction: WKNavigationAction, didBecome download: WKDownload) {
+        download.delegate = self
+    }
+    
+    public func webView(_ webView: WKWebView, navigationResponse: WKNavigationResponse, didBecome download: WKDownload) {
+        download.delegate = self
+    }
+}
