passepartout-apple/Passepartout/Library/Sources/AppUI/Mock/Mock.swift

//
//  Mock.swift
//  Passepartout
//
//  Created by Davide De Rosa on 6/22/24.
//  Copyright (c) 2024 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 AppLibrary
import Combine
import Foundation
import PassepartoutKit
import UtilsLibrary

extension AppContext {
    public static let mock: AppContext = .mock(withRegistry: Registry())

    public static func mock(withRegistry registry: Registry) -> AppContext {
        let env = InMemoryEnvironment()
        return AppContext(
            iapManager: IAPManager(
                customUserLevel: nil,
                receiptReader: MockReceiptReader(),
                unrestrictedFeatures: [
                    .interactiveLogin,
                    .onDemand,
                    .sharing
                ],
                productsAtBuild: { _ in
                    []
                }
            ),
            profileManager: {
                let profiles: [Profile] = (0..<20)
                    .reduce(into: []) { list, _ in
                        list.append(.newMockProfile())
                    }
                return ProfileManager(profiles: profiles)
            }(),
            profileProcessor: ProfileProcessor {
                "Passepartout.Mock: \($0.name)"
            } processed: {
                try $0.withProviderModules()
            },
            tunnel: Tunnel(strategy: FakeTunnelStrategy(environment: env)),
            tunnelEnvironment: env,
            registry: registry,
            providerManager: ProviderManager(
                repository: InMemoryProviderRepository()
            ),
            constants: .shared
        )
    }
}

extension IAPManager {
    public static var mock: IAPManager {
        AppContext.mock.iapManager
    }
}

extension ProfileManager {
    public static var mock: ProfileManager {
        AppContext.mock.profileManager
    }
}

extension ProfileProcessor {
    public static var mock: ProfileProcessor {
        AppContext.mock.profileProcessor
    }
}

extension Tunnel {
    public static var mock: Tunnel {
        AppContext.mock.tunnel
    }
}

extension ConnectionObserver {
    public static var mock: ConnectionObserver {
        AppContext.mock.connectionObserver
    }
}

extension ProviderManager {
    public static var mock: ProviderManager {
        AppContext.mock.providerManager
    }
}

// MARK: - Profile

extension Profile {
    public static let mock: Profile = {
        var profile = Profile.Builder()
        profile.name = "Mock profile"

        do {
            var ovpn = OpenVPNModule.Builder()
            ovpn.configurationBuilder = OpenVPN.Configuration.Builder(withFallbacks: true)
            ovpn.configurationBuilder?.ca = .init(pem: "some CA")
            ovpn.configurationBuilder?.remotes = [
                try .init("1.2.3.4", .init(.udp, 80))
            ]
            profile.modules.append(try ovpn.tryBuild())

            var dns = DNSModule.Builder()
            dns.protocolType = .https
            dns.servers = ["1.1.1.1"]
            dns.dohURL = "https://1.1.1.1/dns-query"
            profile.modules.append(try dns.tryBuild())

            var proxy = HTTPProxyModule.Builder()
            proxy.address = "1.1.1.1"
            proxy.port = 1080
            proxy.secureAddress = "2.2.2.2"
            proxy.securePort = 8080
            proxy.bypassDomains = ["bypass.com"]
            profile.modules.append(try proxy.tryBuild())

            profile.activeModulesIds = [ovpn.id, dns.id]

            return try profile.tryBuild()
        } catch {
            fatalError("Unable to build: \(error)")
        }
    }()

    public static func newMockProfile() -> Profile {
        do {
            var copy = mock.builder(withNewId: true)
            copy.name = String(copy.id.uuidString.prefix(8))
            return try copy.tryBuild()
        } catch {
            fatalError("Unable to build: \(error)")
        }
    }
}