ControllerModel.swift

//
//  ControllerModel.swift
//  Controllers
//
//  Created by 강동영 on 5/10/25.
//

import SwiftUI
import StoreKit
import ControllerKit

final class ControllerModel: ObservableObject {
    @Published var isConnected: Bool = false {
        didSet {
            UserDefaults.shared.setValue(isConnected, forKey: StringKey.CONTROLLER_CONNECTED)
        }
    }
    @Published var batteryLevel: Float = 0.0
    @Published var vendorName: String = ""
    @Published var isCharging: Bool = false
    /// 상태 unknown·잔량 0이면 배터리 정보를 주지 않는 컨트롤러다. (시뮬레이터 가상 Gamepad, 일부 비공식 컨트롤러)
    @Published var hasBatteryInfo: Bool = true
    
    private let manager = GameControllerManager.shared
    
    init() {
        UserDefaults.shared.setValue(false, forKey: StringKey.CONTROLLER_CONNECTED)
        addControllerObservers()
        #if DEBUG
        applyScreenshotFixture()
        // -ACReviewTest YES: 누적 연결 횟수를 2로 맞추고 연결을 한 번 흉내 내 평가 요청 경로를 탄다.
        if UserDefaults.standard.bool(forKey: "ACReviewTest") {
            UserDefaults.standard.set(2, forKey: StringKey.CONTROLLER_CONNECT_COUNT)
            UserDefaults.standard.removeObject(forKey: StringKey.LAST_REVIEW_REQUEST_VERSION)
            countConnectionAndRequestReview()
        }
        #endif
    }
    
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    // MARK: - Life Cycle
    public func updateControllerInfo() {
        #if DEBUG
        // 스크린샷 모드에서는 실행 인자 상태를 실제 조회로 덮어쓰지 않는다.
        if ScreenshotFixture.isEnabled { return }
        #endif
        guard let info = manager.getControlelrInfo() else {
            WidgetSharedState.save(isConnected: false)
            batteryLevel = 0.0
            isCharging = false
            return
        }

        NSLog("batteryLevel: \(info.batteryLevel)")
        vendorName = info.vendorName
        batteryLevel = info.batteryLevel
        isCharging = info.batteryState == .charging
        hasBatteryInfo = !(info.batteryState == .unknown && info.batteryLevel == 0)
        WidgetSharedState.save(isConnected: true, batteryLevel: info.batteryLevel,
                               batteryState: info.batteryState.rawValue, vendorName: info.vendorName)

        // 활성화된 Live Activity 가 있으면 배터리 정보를 갱신한다.
        LiveActivityManager.shared.update(batteryLevel: info.batteryLevel,
                                          batteryState: info.batteryState.rawValue)
    }
}

// MARK: - Live Activity
extension ControllerModel {
    /// 컨트롤러 연결 시 배터리 Live Activity 를 시작한다. (구독자·광고 이용권 사용자 전용)
    private func startLiveActivity() {
        guard let info = manager.getControlelrInfo() else { return }
        LiveActivityManager.shared.start(vendorName: info.vendorName,
                                         batteryLevel: info.batteryLevel,
                                         batteryState: info.batteryState.rawValue)
    }
}

extension ControllerModel {
    private func addControllerObservers() {
        #if DEBUG
        // 스크린샷 모드에서는 시뮬레이터가 넘겨주는 가상 컨트롤러("Gamepad")를 무시하고 실행 인자 상태만 보여 준다.
        if ScreenshotFixture.isEnabled { return }
        #endif
        manager.addDelegate(self)
        // 컨트롤러가 연결된 채로 광고 이용권을 받으면 바로 Live Activity 를 띄운다.
        NotificationCenter.default.addObserver(forName: .proPassGranted, object: nil, queue: .main) { [weak self] _ in
            guard self?.isConnected == true else { return }
            self?.startLiveActivity()
        }
    }
}

// MARK: - GameControllerDelegate Method
extension ControllerModel: GameControllerDelegate {
    func didConnectedController() {
        #if DEBUG
        // 시뮬레이터가 알리는 가상 컨트롤러 대신 스크린샷용 모의 상태를 유지한다.
        if ScreenshotFixture.isEnabled { return }
        #endif
        isConnected = true
        updateControllerInfo()
        startLiveActivity()
        // 횟수를 먼저 세야 Pro 팝업이 이번 실행에 평가 요청 차례인지 알 수 있다.
        countConnectionAndRequestReview()
        // 유료 기능을 쓰지 않는 사용자에게 앱 실행 후 첫 연결 때 Pro 소개 팝업을 띄운다.
        Task { @MainActor in
            ProPromotionManager.shared.controllerDidConnect()
        }
    }

    func didDisConnectedController() {
        isConnected = false
        updateControllerInfo()
        LiveActivityManager.shared.end()
    }
}

// MARK: - Review Request
extension ControllerModel {
    /// 누적 연결 3회 이상이고 이번 버전에 아직 요청하지 않았으면 true.
    /// 이 값이 true인 실행은 Pro 팝업보다 평가 요청을 먼저 한다. (ProPromotionManager도 이 값을 본다)
    static var isReviewRequestDue: Bool {
        let defaults = UserDefaults.standard
        return defaults.integer(forKey: StringKey.CONTROLLER_CONNECT_COUNT) >= 3
            && defaults.string(forKey: StringKey.LAST_REVIEW_REQUEST_VERSION) != Bundle.main.versionString
    }

    /// 연결 횟수를 세고, 세 번째 연결부터 버전당 한 번 평가를 요청한다.
    private func countConnectionAndRequestReview() {
        let defaults = UserDefaults.standard
        let count = defaults.integer(forKey: StringKey.CONTROLLER_CONNECT_COUNT) + 1
        defaults.set(count, forKey: StringKey.CONTROLLER_CONNECT_COUNT)
        guard Self.isReviewRequestDue else { return }

        // 연결 직후 2초 기다린 뒤, 시작 흐름(알림·광고 동의·ATT)·광고·다른 화면이 끝나기를 최대 30초 더 기다린다.
        // 끝내 못 띄우면 횟수를 소모하지 않아 다음 연결에서 다시 시도한다.
        Task { @MainActor in
            try? await Task.sleep(nanoseconds: 2_000_000_000)
            for _ in 0..<60 {
                if let windowScene = Self.reviewPresentableScene() {
                    let version = Bundle.main.versionString
                    defaults.set(version, forKey: StringKey.LAST_REVIEW_REQUEST_VERSION)
                    NSLog("Review requested (connect count: \(count), version: \(version))")
                    SKStoreReviewController.requestReview(in: windowScene)
                    return
                }
                try? await Task.sleep(nanoseconds: 500_000_000)
            }
            NSLog("Review request skipped (connect count: \(count))")
        }
    }

    /// 앱이 활성 상태이고 시작 흐름·광고·모달이 없을 때만 평가를 띄울 화면을 돌려준다.
    @MainActor
    private static func reviewPresentableScene() -> UIWindowScene? {
        guard ProPromotionManager.shared.isLaunchFlowFinished,
              !AppOpenAdManager.shared.isShowingAd,
              !RewardedPassManager.shared.isShowingAd,
              UIApplication.shared.applicationState == .active,
              let topViewController = UIApplication.topViewController(),
              topViewController.presentingViewController == nil else { return nil }
        return topViewController.view.window?.windowScene
    }
}

#if DEBUG
// MARK: - Screenshot Fixture
extension ControllerModel {
    /// 스크린샷용 실행 인자가 있으면 모의 컨트롤러 상태를 반영한다.
    private func applyScreenshotFixture() {
        guard ScreenshotFixture.isEnabled else { return }
        if let mock = ScreenshotFixture.mockController {
            isConnected = true
            vendorName = mock.vendorName
            batteryLevel = mock.batteryLevel
            isCharging = mock.isCharging
            hasBatteryInfo = mock.hasBatteryInfo
        }
        ScreenshotFixture.writeWidgetState()
    }
}
#endif