passepartout-apple/PassepartoutLibrary/Sources/PassepartoutCore/Reusable/CoreLocationWifiObserver.swift

//
//  CoreLocationWifiObserver.swift
//  Passepartout
//
//  Created by Davide De Rosa on 5/21/23.
//  Copyright (c) 2023 Davide De Rosa. All rights reserved.
//
//  https://github.com/passepartoutvpn
//
//  This file is part of Passepartout.
//
//  Passepartout is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  Passepartout is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with Passepartout.  If not, see <http://www.gnu.org/licenses/>.
//

import CoreLocation
import Foundation

public final class CoreLocationWifiObserver: NSObject, WifiObserver {
    private let locationManager = CLLocationManager()

    private var continuation: CheckedContinuation<String, Error>?

    public func currentSSID() async throws -> String {
        switch locationManager.authorizationStatus {
        case .authorizedAlways, .authorizedWhenInUse, .denied:
            return await currentSSIDWithoutAuthorization()

        default:
            return try await withCheckedThrowingContinuation { continuation in
                self.continuation = continuation

                locationManager.delegate = self
                locationManager.requestWhenInUseAuthorization()
            }
        }
    }

    private func currentSSIDWithoutAuthorization() async -> String {
        await Utils.currentWifiSSID() ?? ""
    }
}

extension CoreLocationWifiObserver: CLLocationManagerDelegate {
    public func locationManagerDidChangeAuthorization(_ manager: CLLocationManager) {
        switch manager.authorizationStatus {
        case .authorizedWhenInUse, .authorizedAlways, .denied:
            Task {
                continuation?.resume(returning: await currentSSIDWithoutAuthorization())
                continuation = nil
            }

        default:
            continuation?.resume(with: .success(""))
            continuation = nil
        }
    }

    public func locationManager(_ manager: CLLocationManager, didFailWithError error: Error) {
        continuation?.resume(throwing: error)
        continuation = nil
    }
}