//
//  LDFirewallViewController.swift
//  Lockdown
//
//  Created by Aliaksandr Dvoineu on 17.04.23.
//  Copyright © 2023 Confirmed Inc. All rights reserved.
//

import UIKit
import CocoaLumberjackSwift
import PromiseKit
import NetworkExtension
import PopupDialog

final class LDFirewallViewController: BaseViewController {
    
    // MARK: Properties
    let kHasViewedTutorial = "hasViewedTutorial"
    let kHasSeenInitialFirewallConnectedDialog = "hasSeenInitialFirewallConnectedDialog11"
    let kHasSeenShare = "hasSeenShareDialog4"
    
    let ratingCountKey = "ratingCount" + lastVersionToAskForRating
    let ratingTriggeredKey = "ratingTriggered" + lastVersionToAskForRating
    
    var lastFirewallStatus: NEVPNStatus?
    var activePlans: [Subscription.PlanType] = []
    let vc = FirewallPaywallViewController()
    
    enum Mode {
        case newSubscription
        case upgrade(active: [Subscription.PlanType])
    }
    
    var mode = Mode.newSubscription
    
    var metricsTimer : Timer?
    
    private lazy var scrollView: UIScrollView = {
        let view = UIScrollView()
        view.isScrollEnabled = true
        return view
    }()
    
    private lazy var contentView: UIView = {
        let view = UIView()
        view.anchors.height.equal(800)
        return view
    }()
    
    lazy var yourCurrentPlanLabel: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Your current plan is", comment: "")
        label.font = fontRegular14
        label.numberOfLines = 0
        label.textColor = .label
        return label
    }()
    
    lazy var upgradeLabel: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Upgrade?", comment: "")
        label.font = fontBold13
        label.textColor = .tunnelsBlue
        label.isUserInteractionEnabled = true
        label.setOnClickListener {
            let vc = VPNPaywallViewController()
            self.present(vc, animated: true)
        }
        return label
    }()
    
    lazy var protectionPlanLabel: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Basic Protection", comment: "")
        label.font = fontBold22
        label.textColor = .label
        return label
    }()
    
    lazy var firewallTitle: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Get complete protection", comment: "")
        label.font = fontBold24
        label.numberOfLines = 0
        label.textColor = .label
        return label
    }()
    
    private lazy var firewallDescriptionLabel1: DescriptionLabel = {
        let label = DescriptionLabel()
        label.configure(with: DescriptionLabelViewModel(text: NSLocalizedString("Block as many trackers as you want", comment: "")))
        return label
    }()
    
    private lazy var firewallDescriptionLabel2: DescriptionLabel = {
        let label = DescriptionLabel()
        label.configure(with: DescriptionLabelViewModel(text: NSLocalizedString("Import and export your own block lists", comment: "")))
        return label
    }()
    
    private lazy var firewallDescriptionLabel3: DescriptionLabel = {
        let label = DescriptionLabel()
        label.configure(with: DescriptionLabelViewModel(text: NSLocalizedString("Access to new curated lists of trackers", comment: "")))
        return label
    }()
    
    private lazy var stackView: UIStackView = {
        let stackView  = UIStackView()
        stackView.addArrangedSubview(firewallTitle)
        stackView.addArrangedSubview(firewallDescriptionLabel1)
        stackView.addArrangedSubview(firewallDescriptionLabel2)
        stackView.addArrangedSubview(firewallDescriptionLabel3)
        stackView.axis = .vertical
        stackView.distribution = .fillProportionally
        stackView.spacing = 10
        return stackView
    }()
    
    private lazy var cpTitle: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Don't let those trackers know your every move – Upgrade to Advanced now!", comment: "")
        label.textColor = .black
        label.font = fontBold15
        label.numberOfLines = 0
        label.textAlignment = .center
        return label
    }()
    
    private lazy var cpTrackersGroupView1: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#1"
        view.placeNumber.textColor = .black
        view.titleLabel.textColor = .black
        view.number.textColor = .black
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_game_marketing")!, title: "Game Marketing", number: 4678))
        view.number.isHidden = true
        return view
    }()
    
    private lazy var cpTrackersGroupView2: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#2"
        view.placeNumber.textColor = .black
        view.titleLabel.textColor = .black
        view.number.textColor = .black
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_marketing_trackers")!, title: "Marketing Trackers", number: 3432))
        view.number.isHidden = true
        return view
    }()
    
    private lazy var cpTrackersGroupView3: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#3"
        view.placeNumber.textColor = .black
        view.titleLabel.textColor = .black
        view.number.textColor = .black
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_email_trackers")!, title: "Email Trackers", number: 2756))
        view.number.isHidden = true
        return view
    }()
    
    private lazy var cpStackView: UIStackView = {
        let stackView  = UIStackView()
        stackView.addArrangedSubview(cpTitle)
        stackView.addArrangedSubview(cpTrackersGroupView1)
        stackView.addArrangedSubview(cpTrackersGroupView2)
        stackView.addArrangedSubview(cpTrackersGroupView3)
        stackView.layer.cornerRadius = 8
        stackView.layer.borderWidth = 2
        stackView.layer.borderColor = UIColor.gray.cgColor
        stackView.axis = .vertical
        stackView.distribution = .fillEqually
        stackView.spacing = 0
        stackView.backgroundColor = .extraLightGray
        return stackView
    }()
    
    private lazy var upgradeButton: UIButton = {
        let button = UIButton(type: .system)
        button.tintColor = .white
        button.setTitle(NSLocalizedString("Upgrade", comment: ""), for: .normal)
        button.titleLabel?.font = fontBold18
        button.backgroundColor = .tunnelsBlue
        button.layer.cornerRadius = 28
        button.anchors.height.equal(56)
        button.addTarget(self, action: #selector(upgrade), for: .touchUpInside)
        return button
    }()
    
    private lazy var maTitle: UILabel = {
        let label = UILabel()
        label.text = NSLocalizedString("Most active this week", comment: "")
        label.textColor = .label
        label.font = fontBold15
        label.textAlignment = .center
        return label
    }()
    
    private lazy var maTrackersGroupView1: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#1"
        view.number.textColor = .label
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_facebook_trackers")!, title: "Facebook Trackers", number: 89))
        view.lockImage.isHidden = true
        return view
    }()
    
    private lazy var maTrackersGroupView2: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#2"
        view.number.textColor = .label
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_data_trackers")!, title: "Data Trackers", number: 32))
        view.lockImage.isHidden = true
        return view
    }()
    
    private lazy var maTrackersGroupView3: TrackersGroupView = {
        let view = TrackersGroupView()
        view.placeNumber.text = "#3"
        view.number.textColor = .label
        view.configure(with: TrackersGroupViewModel(image: UIImage(named: "icn_clickbait_trackers")!, title: "Clickbait", number: 21))
        view.lockImage.isHidden = true
        return view
    }()
    
    private lazy var maStackView: UIStackView = {
        let stackView  = UIStackView()
        stackView.addArrangedSubview(maTitle)
        stackView.addArrangedSubview(maTrackersGroupView1)
        stackView.addArrangedSubview(maTrackersGroupView2)
        stackView.addArrangedSubview(maTrackersGroupView3)
        stackView.layer.cornerRadius = 8
        stackView.layer.borderWidth = 1
        stackView.layer.borderColor = UIColor.lightGray.cgColor
        stackView.axis = .vertical
        stackView.distribution = .fillEqually
        stackView.spacing = 0
        return stackView
    }()
    
    lazy var statisitcsView: OverallStatiscticView = {
        let view = OverallStatiscticView()
        return view
    }()
    
    private lazy var firewallSwitchControl: CustomUISwitch = {
        let uiSwitch = CustomUISwitch(onImage: UIImage(named: "firewall-on-image")!, offImage: UIImage(named: "firewall-off-image")!)
        uiSwitch.setOnClickListener {
            self.toggleFirewall()
        }
        return uiSwitch
    }()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        VPNSubscription.cacheLocalizedPrices()
        
        updateFirewallButtonWithStatus(status: FirewallController.shared.status())
        updateMetrics()
        if metricsTimer == nil {
            metricsTimer = Timer.scheduledTimer(timeInterval: 2.0, target: self, selector: #selector(updateMetrics), userInfo: nil, repeats: true)
            metricsTimer?.fire()
        }
        
        view.backgroundColor = .systemBackground
        
        view.addSubview(firewallSwitchControl)
        firewallSwitchControl.anchors.bottom.safeAreaPin()
        firewallSwitchControl.anchors.leading.marginsPin()
        firewallSwitchControl.anchors.trailing.marginsPin()
        firewallSwitchControl.anchors.height.equal(56)
        
        view.addSubview(yourCurrentPlanLabel)
        yourCurrentPlanLabel.anchors.leading.marginsPin()
        yourCurrentPlanLabel.anchors.top.safeAreaPin()
        
        view.addSubview(upgradeLabel)
        upgradeLabel.anchors.trailing.marginsPin()
        upgradeLabel.anchors.centerY.equal(yourCurrentPlanLabel.anchors.centerY)
        
        view.addSubview(protectionPlanLabel)
        protectionPlanLabel.anchors.top.spacing(8, to: yourCurrentPlanLabel.anchors.bottom)
        protectionPlanLabel.anchors.leading.marginsPin()
        
        view.addSubview(scrollView)
        scrollView.anchors.top.spacing(12, to: protectionPlanLabel.anchors.bottom)
        scrollView.anchors.leading.pin()
        scrollView.anchors.trailing.pin()
        scrollView.anchors.bottom.spacing(8, to: firewallSwitchControl.anchors.top)
        
        scrollView.addSubview(contentView)
        contentView.anchors.top.pin()
        contentView.anchors.centerX.align()
        contentView.anchors.width.equal(scrollView.anchors.width)
        contentView.anchors.bottom.pin()

        contentView.addSubview(stackView)
        stackView.anchors.top.marginsPin()
        stackView.anchors.leading.marginsPin()
        stackView.anchors.trailing.marginsPin()
        
        contentView.addSubview(upgradeButton)
        upgradeButton.anchors.top.spacing(18, to: stackView.anchors.bottom)
        upgradeButton.anchors.leading.marginsPin()
        upgradeButton.anchors.trailing.marginsPin()
        
        contentView.addSubview(maStackView)
        maStackView.anchors.top.spacing(18, to: upgradeButton.anchors.bottom)
        maStackView.anchors.leading.marginsPin()
        maStackView.anchors.trailing.marginsPin()
        
        contentView.addSubview(statisitcsView)
        statisitcsView.anchors.top.spacing(18, to: maStackView.anchors.bottom)
        statisitcsView.anchors.leading.marginsPin()
        statisitcsView.anchors.trailing.marginsPin()
        
        updateProtectionPlanUI()

//        accountStateDidChange()
        
        NotificationCenter.default.addObserver(self, selector: #selector(tunnelStatusDidChange(_:)), name: .NEVPNStatusDidChange, object: nil)
    }
    
    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
        updateMetrics()
    }
}

extension LDFirewallViewController: Loadable {
    
    @objc func accountStateDidChange() {
        updateActiveSubscription()
    }
    
    func updateProtectionPlanUI() {
        if UserDefaults.hasSeenUniversalPaywall {
            updateUI()
            protectionPlanLabel.text = "Universal protection"
        } else if UserDefaults.hasSeenAnonymousPaywall {
            updateUI()
            protectionPlanLabel.text = "Anonymous protection"
        } else if UserDefaults.hasSeenAdvancedPaywall {
            updateUI()
            protectionPlanLabel.text = "Advanced protection"
        } else {
            protectionPlanLabel.text = "Basic protection"
        }
    }
    
    func updateUI() {
        firewallTitle.isHidden = true
        firewallDescriptionLabel1.isHidden = true
        firewallDescriptionLabel2.isHidden = true
        firewallDescriptionLabel3.isHidden = true
        upgradeButton.isHidden = true
        upgradeButton.anchors.height.equal(0)
        cpTrackersGroupView1.lockImage.isHidden = true
        cpTrackersGroupView1.number.isHidden = false
        cpTrackersGroupView2.lockImage.isHidden = true
        cpTrackersGroupView2.number.isHidden = false
        cpTrackersGroupView3.lockImage.isHidden = true
        cpTrackersGroupView3.number.isHidden = false
    }
    
    func updateActiveSubscription() {
        showLoadingView()
        // not logged in via email, use receipt
        firstly {
            try Client.signIn()
        }.then { _ in
            try Client.activeSubscriptions()
        }.ensure {
            self.hideLoadingView()
        }.done { [self] subscriptions in
            self.activePlans = subscriptions.map({ $0.planType })
            if let active = subscriptions.first {
                if active.planType == .universalAnnual || active.planType == .universalMonthly {
                    protectionPlanLabel.text = "Universal protection"
                    updateUI()
                } else if active.planType == .anonymousMonthly || active.planType == .anonymousAnnual {
                    updateUI()
                    protectionPlanLabel.text = "Anonymous protection"
                } else if active.planType == .advancedMonthly || active.planType == .advancedAnnual {
                    updateUI()
                    protectionPlanLabel.text = "Advanced protection"
                } else {
                    firewallTitle.textColor = .red
                }
            } else {
                firewallTitle.textColor = .red
            }
        }.catch { [self] error in
            DDLogError("Error reloading subscription: \(error.localizedDescription)")
            hideLoadingView()
            if let apiError = error as? ApiError {
                switch apiError.code {
                case kApiCodeNoSubscriptionInReceipt, kApiCodeNoActiveSubscription:
                    UserDefaults.hasSeenAdvancedPaywall = false
                case kApiCodeSandboxReceiptNotAllowed:
                    UserDefaults.hasSeenAdvancedPaywall = false
                default:
                    DDLogError("Error loading plan: API error code - \(apiError.code)")
                    UserDefaults.hasSeenAdvancedPaywall = false
                }
            } else {
                DDLogError("Error loading plan: Non-API Error - \(error.localizedDescription)")
                UserDefaults.hasSeenAdvancedPaywall = false
            }
        }
    }
    
    @objc func upgrade() {
        let vc = FirewallPaywallViewController()
        present(vc, animated: true)
    }
    
    @objc func tunnelStatusDidChange(_ notification: Notification) {
        // Firewall
        if let tunnelProviderSession = notification.object as? NETunnelProviderSession {
            DDLogInfo("VPNStatusDidChange as NETunnelProviderSession with status: \(tunnelProviderSession.status.description)");
            if (!getUserWantsFirewallEnabled()) {
                updateFirewallButtonWithStatus(status: .disconnected)
            }
            else {
                updateFirewallButtonWithStatus(status: tunnelProviderSession.status)
                if (tunnelProviderSession.status == .connected && defaults.bool(forKey: kHasSeenInitialFirewallConnectedDialog) == false) {
                    defaults.set(true, forKey: kHasSeenInitialFirewallConnectedDialog)
//                    self.tapToActivateFirewallLabel.isHidden = true
//                    if (VPNController.shared.status() == .invalid) {
//                        self.showVPNSubscriptionDialog(title: NSLocalizedString("🔥🧱 Firewall Activated 🎊🎉", comment: ""), message: NSLocalizedString("Trackers, ads, and other malicious scripts are now blocked in all your apps, even outside of Safari.\n\nGet maximum privacy with a Secure Tunnel that protects connections, anonymizes your browsing, and hides your location.", comment: ""))
//                    }
                }
            }
        }
    }
    
    @objc func updateMetrics() {
        DispatchQueue.main.async { [unowned self] in
            self.statisitcsView.enabledBoxView.numberLabel.text = String(getTotalEnabled().count)
            self.statisitcsView.disabledBoxView.numberLabel.text = String(getTotalDisabled().count)
            self.statisitcsView.blockedBoxView.numberLabel.text = String(getAllBlockedDomains().count)
        }
    }
    
    func toggleFirewall() {
        if (defaults.bool(forKey: kHasAgreedToFirewallPrivacyPolicy) == false) {
            let storyboard = UIStoryboard(name: "Main", bundle: nil)
            let viewController = storyboard.instantiateViewController(withIdentifier: "firewallPrivacyPolicyViewController") as! PrivacyPolicyViewController
            viewController.privacyPolicyKey = kHasAgreedToFirewallPrivacyPolicy
            viewController.parentVC1 = self
            self.present(viewController, animated: true, completion: nil)
            return
        }
        
        if getIsCombinedBlockListEmpty() {
            FirewallController.shared.setEnabled(false, isUserExplicitToggle: true)
            self.showPopupDialog(title: NSLocalizedString("No Block Lists Enabled", comment: ""), message: NSLocalizedString("Please tap Block List and enable at least one block list to activate Firewall.", comment: ""), acceptButton: NSLocalizedString("Okay", comment: ""))
            return
        }
        
        switch FirewallController.shared.status() {
        case .invalid:
            FirewallController.shared.setEnabled(true, isUserExplicitToggle: true)
            //ensureFirewallWorkingAfterEnabling(waitingSeconds: 5.0)
        case .disconnected:
            updateFirewallButtonWithStatus(status: .connecting)
            FirewallController.shared.setEnabled(true, isUserExplicitToggle: true)
            //ensureFirewallWorkingAfterEnabling(waitingSeconds: 5.0)
            
//            checkForAskRating()
        case .connected:
            updateFirewallButtonWithStatus(status: .disconnecting)
            FirewallController.shared.setEnabled(false, isUserExplicitToggle: true)
        case .connecting, .disconnecting, .reasserting:
            break;
        }
    }
    
    func ensureFirewallWorkingAfterEnabling(waitingSeconds: TimeInterval) {
        FirewallController.shared.existingManagerCount { (count) in
            if let count = count, count > 0 {
                DispatchQueue.main.asyncAfter(deadline: .now() + waitingSeconds) {
                    DDLogInfo("\(waitingSeconds) seconds passed, checking if Firewall is enabled")
                    guard getUserWantsFirewallEnabled() else {
                        // firewall shouldn't be enabled, no need to act
                        DDLogInfo("User doesn't want Firewall enabled, no action")
                        return
                    }
                    
                    let status = FirewallController.shared.status()
                    switch status {
                    case .connecting, .disconnecting, .reasserting:
                        // check again in three seconds
                        DDLogInfo("Firewall is in transient state, will check again in 3 seconds")
                        self.ensureFirewallWorkingAfterEnabling(waitingSeconds: 3.0)
                    case .connected:
                        // all good
                        DDLogInfo("Firewall is connected, no action")
                        break
                    case .disconnected, .invalid:
                        // we suppose that the connection is somehow broken, trying to fix
                        DDLogInfo("Firewall is not connected even though it should be, attempting to fix")
                        self.showFixFirewallConnectionDialog {
                            FirewallController.shared.deleteConfigurationAndAddAgain()
                        }
                    }
                }
            } else {
                DDLogInfo("No Firewall configurations in settings (likely fresh install): not checking")
                return
            }
        }
    }
    
    func updateFirewallButtonWithStatus(status: NEVPNStatus) {
        DDLogInfo("UpdateFirewallButton")
        switch status {
        case .connected:
            LatestKnowledge.isFirewallEnabled = true
        case .disconnected:
            LatestKnowledge.isFirewallEnabled = false
        default:
            break
        }
        updateToggleButtonWithStatus(lastStatus: lastFirewallStatus,
                                     newStatus: status,
                                     switchControl: firewallSwitchControl)
    }
    
    func updateToggleButtonWithStatus(lastStatus: NEVPNStatus?,
                                      newStatus: NEVPNStatus,
                                      switchControl: CustomUISwitch) {
        DDLogInfo("UpdateToggleButton")
        if (newStatus == lastStatus) {
            DDLogInfo("No status change from last time, ignoring.");
        }
        else {
            DispatchQueue.main.async() {
                switch newStatus {
                case .connected:
                    switchControl.status = true
                case .connecting:
                    switchControl.status = true
                case .disconnected, .invalid:
                    switchControl.status = false
                case .disconnecting:
                    switchControl.status = false
                case .reasserting:
                    break;
                }
            }
        }
    }
}