diff --git a/App/Script Policy UI/ScriptPolicyViewController.swift b/App/Script Policy UI/ScriptPolicyViewController.swift index c71a152..83286e1 100644 --- a/App/Script Policy UI/ScriptPolicyViewController.swift +++ b/App/Script Policy UI/ScriptPolicyViewController.swift @@ -98,9 +98,13 @@ class ScriptPolicyViewController: UIViewController, UICollectionViewDelegate let listLayout = UICollectionViewCompositionalLayout.list(using: listConfig) let collectionView = UICollectionView(frame: .zero, collectionViewLayout: listLayout) + // Allowed scripts go to the top + let allowedScripts = loadedScripts.filter { policyManager.allowedOriginsForScriptResources().contains($0) } + // Make sure host origin goes first in the list. - let otherOriginScripts = loadedScripts.subtracting([ hostOrigin ]) - let originItems = [ hostOrigin ] + otherOriginScripts + let otherOriginScripts = loadedScripts.subtracting([ hostOrigin ]).subtracting(allowedScripts) + + let originItems = [ hostOrigin ] + allowedScripts + otherOriginScripts let switchCellRegistry = UICollectionView.CellRegistration { [unowned self] (listCell, indexPath, item) in var config = listCell.defaultContentConfiguration()