PrivacyIconLogic.swift

//
//  PrivacyIconLogic.swift
//  DuckDuckGo
//
//  Copyright © 2022 DuckDuckGo. All rights reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

import Foundation
import Core
import PrivacyDashboard

final class PrivacyIconLogic {
    
    static func privacyIcon(for url: URL?) -> PrivacyIcon {
        if let url = url, url.isDuckDuckGoSearch {
            return .daxLogo
        } else {
            return .shield
        }
    }
    
    static func privacyIcon(for privacyInfo: PrivacyInfo) -> PrivacyIcon {
        if privacyInfo.url.isDuckDuckGoSearch {
            return .daxLogo
        } else {
            let config = ContentBlocking.shared.privacyConfigurationManager.privacyConfig
            let isUserUnprotected = config.isUserUnprotected(domain: privacyInfo.url.host)

            let isServerTrustInvalid = (privacyInfo.shouldCheckServerTrust ? privacyInfo.serverTrust == nil : false)
            let notFullyProtected = !privacyInfo.https || isUserUnprotected || isServerTrustInvalid

            return notFullyProtected ? .shieldWithDot : .shield
        }
    }
}