NSOpenPanelExtensions.swift

//
//  NSOpenPanelExtensions.swift
//
//  Copyright © 2021 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 AppKit
import UniformTypeIdentifiers

extension NSOpenPanel {

    static func downloadDirectoryPanel() -> NSOpenPanel {
        let downloadPreferences = DownloadsPreferences()
        let panel = NSOpenPanel()

        panel.directoryURL = downloadPreferences.effectiveDownloadLocation
        panel.canChooseFiles = false
        panel.canChooseDirectories = true
        panel.canCreateDirectories = true

        return panel
    }

    static func filePanel(allowedExtension: String) -> NSOpenPanel {
        let panel = NSOpenPanel()

        panel.directoryURL = FileManager.default.homeDirectoryForCurrentUser.appendingPathComponent("Desktop")
        panel.canChooseFiles = true
        if #available(macOS 11.0, *) {
            panel.allowedContentTypes = [UniformTypeIdentifiers.UTType.init(filenameExtension: allowedExtension)].compactMap { $0 }
        } else {
            panel.allowedFileTypes = [allowedExtension]
        }

        panel.canChooseDirectories = false

        return panel
    }
}