Attractor/App/Backend/ResourcePolicyManager.swift

//
//  ResourcePolicyManager.swift
//  SBrowser
//
//  Created by James Magahern on 7/22/20.
//

import Foundation

class ResourcePolicyManager: NSObject, SBRResourceOriginPolicyDataSource
{
    static let AllowedOriginsDefaultsKey = "allowedOrigins"
    static let EnabledOriginsDefaultsKey = "enabledOrigins"
    
    private static func stringSetForKey(_ key: String) -> Set<String> {
        if let set = UserDefaults.standard.array(forKey: key) as? [String] {
            return Set<String>(set)
        }

        return Set<String>()
    }
    
    private static func saveStringSet(_ set: Set<String>, forKey key: String) {
        UserDefaults.standard.set(Array(set), forKey: key)
    }
    
    private var allowedOriginSet: Set<String> = stringSetForKey(AllowedOriginsDefaultsKey) {
        didSet { Self.saveStringSet(allowedOriginSet, forKey: Self.AllowedOriginsDefaultsKey) }
    }
    
    func allowedOriginsForScriptResources() -> Set<String> { allowedOriginSet }
    
    func allowOriginToLoadScriptResources(_ origin: String) {
        allowedOriginSet.formUnion([ origin ])
    }
    
    func disallowOriginToLoadScriptResources(_ origin: String) {
        allowedOriginSet.remove(origin)
    }
}