fx_cast/app/NativeMacReceiverSelector/ViewController.swift

import Cocoa


class ViewController : NSViewController {
    var mediaTypePopUpButton: NSPopUpButton!
    var receiverViews = [ReceiverView]()

    var filePath: String?


    override func loadView () {
        /*let visualEffectView = NSVisualEffectView()
        visualEffectView.blendingMode = .behindWindow
        visualEffectView.state = .active*/

        self.view = NSView()
    }

    override func viewDidLoad () {
        super.viewDidLoad()

        let initData = InitDataProvider.shared.data

        /**
         * View Hierarchy
         * --------------
         *
         * stackView                  \ (NSStackView)
         *  ├ mediaTypeStackView      \  (NSStackView)
         *  │  ├ mediaSelectCastLabel \    (NSTextField)
         *  │  ├ mediaTypePopUpButton \    (NSPopUpButton)
         *  │  └ mediaSelectToLabel   \    (NSTextField)
         *  │
         *  ├ receiverSeparator       \  (NSBox)                    ┐
         *  └ receiverView            \  (ReceiverView:NSStackView) │
         *     ├ metaStackView        \    (NSStackView)            │
         *     │  ├ receiver name     \      (NSTextField)          ├ Repeats
         *     │  └ receiver host     \      (NSTextField)          │
         *     ├ spinner              \    (NSProgressIndicator)    │
         *     └ button               \    (NSButton)               ┘
         */


        let stackView = NSStackView()
        stackView.orientation = .vertical
        stackView.alignment = .leading
        stackView.autoresizingMask = [ .width, .height ]
        stackView.edgeInsets = NSEdgeInsetsMake(8, 8, 8, 8)


        self.mediaTypePopUpButton = NSPopUpButton()
        self.mediaTypePopUpButton.autoenablesItems = false
        self.mediaTypePopUpButton.addItems(withTitles: [
            initData.i18n_mediaTypeApp
          , initData.i18n_mediaTypeTab
          , initData.i18n_mediaTypeScreen
          , initData.i18n_mediaTypeFile
        ])

        if let appItem = self.mediaTypePopUpButton
                .item(withTitle: initData.i18n_mediaTypeApp)
         , let tabItem = self.mediaTypePopUpButton
                .item(withTitle: initData.i18n_mediaTypeTab)
         , let screenItem = self.mediaTypePopUpButton
                .item(withTitle: initData.i18n_mediaTypeScreen)
         , let fileItem = self.mediaTypePopUpButton
                .item(withTitle: initData.i18n_mediaTypeFile) {

            if let mediaTypePopUpButtonMenu = self.mediaTypePopUpButton.menu {
                mediaTypePopUpButtonMenu.delegate = self
                mediaTypePopUpButtonMenu.insertItem(NSMenuItem.separator()
                      , at: mediaTypePopUpButtonMenu.index(of: fileItem))
            }

            // Set tags to enum value
            appItem.tag = MediaType.app.rawValue
            tabItem.tag = MediaType.tab.rawValue
            screenItem.tag = MediaType.screen.rawValue
            fileItem.tag = MediaType.file.rawValue
        }

        for item in self.mediaTypePopUpButton.itemArray {
            if (initData.availableMediaTypes & item.tag) == 0 {
                item.isEnabled = false
            }
        }

        self.mediaTypePopUpButton.selectItem(
                withTag: initData.defaultMediaType.rawValue)


        let mediaTypeStackView = NSStackView()

        if initData.i18n_mediaSelectCastLabel != "" {
            mediaTypeStackView.addView(
                    makeLabel(initData.i18n_mediaSelectCastLabel), in: .leading)
        }

        mediaTypeStackView.addView(self.mediaTypePopUpButton, in: .leading)

        if initData.i18n_mediaSelectToLabel != "" {
            mediaTypeStackView.addView(
                    makeLabel(initData.i18n_mediaSelectToLabel), in: .leading)
        }

        stackView.addArrangedSubview(mediaTypeStackView)


        /**
         * For each receiver in the initData list, create a new
         * ReceiverView, set self as a ReceiverViewDelegate and
         * appends to main stack view.
         *
         * Keeps a reference to the receiver view to call disable()
         * later.
         */
        for receiver in initData.receivers {
            // Create separator between last receiver / media type
            let receiverSeparator = NSBox()
            receiverSeparator.boxType = .separator

            let receiverView = ReceiverView(receiver: receiver)
            receiverView.receiverViewDelegate = self

            if UInt(initData.availableMediaTypes) == 0
                    || (initData.availableMediaTypes
                            & initData.defaultMediaType.rawValue) == 0 {
                receiverView.isEnabled = false
            }


            self.receiverViews.append(receiverView)

            stackView.addArrangedSubview(receiverSeparator)
            stackView.addArrangedSubview(receiverView)
        }


        // Add to main view and set width to resize window
        self.view.addSubview(stackView)
        self.view.autoresizesSubviews = true
        self.view.frame.size.width = 350
    }

    override func viewDidAppear () {
        // Set window title and update visibility
        if let window = self.view.window {
            window.title = InitDataProvider.shared.data.i18n_extensionName
            window.titleVisibility = .visible
        }
    }
}

extension ViewController : NSMenuDelegate {
    func menuDidClose (_ menu: NSMenu) {
        let initData = InitDataProvider.shared.data

        guard let selectedItem = self.mediaTypePopUpButton.selectedItem
            , let mediaType = MediaType(rawValue: selectedItem.tag) else {
            return
        }

        if initData.availableMediaTypes & mediaType.rawValue != 0 {
            for receiverView in self.receiverViews {
                receiverView.isEnabled = true
            }
        }

        guard let fileItem = self.mediaTypePopUpButton
                .item(at: self.mediaTypePopUpButton.indexOfItem(
                        withTag: MediaType.file.rawValue)) else {
            return
        }

        if (mediaType == .file) {
            let panel = NSOpenPanel()
            panel.allowsMultipleSelection = false
            panel.allowedFileTypes = [ "aac", "mp3", "mp4", "wav", "webm" ]
            panel.canChooseDirectories = false
            panel.canCreateDirectories = false
            panel.canChooseFiles = true

            guard let window = self.view.window else { return }

            panel.beginSheetModal(for: window) { (result) in
                if (result == .OK) {
                    let url = panel.urls[0]
                    let fileName = url.lastPathComponent

                    // Truncate file name and set as title
                    fileItem.title = fileName.count > 12
                        ? "\(fileName.prefix(12))..."
                        : fileName

                    self.filePath = url.path

                    return
                } else {
                    // Re-select the default media type item
                    self.mediaTypePopUpButton.selectItem(
                            withTag: initData.defaultMediaType.rawValue)

                    let defaultMediaTypeAvailable = initData.availableMediaTypes
                            & initData.defaultMediaType.rawValue != 0

                    for receiverView in self.receiverViews {
                        receiverView.isEnabled = defaultMediaTypeAvailable
                    }
                }
            }
        }


        // Reset file item
        fileItem.title = initData.i18n_mediaTypeFile
        self.filePath = nil
    }
}

extension ViewController : ReceiverViewDelegate {
    func didCast (_ receiver: Receiver) {

        // Disable media type UI
        self.mediaTypePopUpButton.isEnabled = false

        // Disable cast buttons
        for receiverView in self.receiverViews {
            receiverView.isEnabled = false
        }


        guard let selectedItem = self.mediaTypePopUpButton.selectedItem
            , let mediaType = MediaType(rawValue: selectedItem.tag) else {
            return
        }

        let selection = ReceiverSelection(
                receiver: receiver
              , mediaType: mediaType
              , filePath: self.filePath ?? nil)

        if let jsonData = try? JSONEncoder().encode(selection)
         , let jsonString = String(data: jsonData, encoding: .utf8) {
            print(jsonString)
            fflush(stdout)
        } else {
            fatalError("Error: Failed to encode output data")
        }
    }
}