CopilotForXcode/Core/Sources/ConversationTab/ModeAndModelPicker/ModePicker/AgentModeMenu.swift at 0.44.150 · github/CopilotForXcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import AppKit
import ConversationServiceProvider

// MARK: - Agent Mode Menu Builder

struct AgentModeMenu {
    let builtInAgentModes: [ConversationMode]
    let customAgents: [ConversationMode]
    let selectedAgent: ConversationMode
    let fontScale: Double
    let onSelectAgent: (ConversationMode) -> Void
    let onEditAgent: (ConversationMode) -> Void
    let onDeleteAgent: (ConversationMode) -> Void
    let onCreateAgent: () -> Void

    func createMenu() -> NSMenu {
        let menu = NSMenu()

        let menuHasSelection = true // Always show checkmarks for clarity

        // Add built-in agent modes
        addBuiltInModes(to: menu, menuHasSelection: menuHasSelection)

        // Add custom agents if any
        if !customAgents.isEmpty {
            menu.addItem(.separator())
            addCustomAgents(to: menu, menuHasSelection: menuHasSelection)
        }

        // Add create option
        menu.addItem(.separator())
        addCreateOption(to: menu, menuHasSelection: menuHasSelection)

        return menu
    }

    private func addBuiltInModes(to menu: NSMenu, menuHasSelection: Bool) {
        for mode in builtInAgentModes {
            let agentItem = NSMenuItem()
            // Determine icon: use checklist for Plan, Agent icon for others
            let iconName = AgentModeIcon.icon(for: mode.name)
            let agentView = AgentModeButtonMenuItem(
                name: mode.name,
                iconName: iconName,
                isSelected: selectedAgent.id == mode.id,
                menuHasSelection: menuHasSelection,
                fontScale: fontScale,
                onSelect: { [onSelectAgent] in
                    onSelectAgent(mode)
                    menu.cancelTracking()
                }
            )
            agentView.toolTip = mode.description
            agentItem.view = agentView
            menu.addItem(agentItem)
        }
    }

    private func addCustomAgents(to menu: NSMenu, menuHasSelection: Bool) {
        for agent in customAgents {
            let agentItem = NSMenuItem()
            agentItem.representedObject = agent

            // Create custom view for the menu item
            let customView = AgentModeButtonMenuItem(
                name: agent.name,
                iconName: nil,
                isSelected: selectedAgent.id == agent.id,
                menuHasSelection: menuHasSelection,
                fontScale: fontScale,
                onSelect: { [onSelectAgent] in
                    onSelectAgent(agent)
                    menu.cancelTracking()
                },
                onEdit: { [onEditAgent] in
                    onEditAgent(agent)
                    menu.cancelTracking()
                },
                onDelete: { [onDeleteAgent] in
                    onDeleteAgent(agent)
                    menu.cancelTracking()
                }
            )

            customView.toolTip = agent.description
            agentItem.view = customView
            menu.addItem(agentItem)
        }
    }

    private func addCreateOption(to menu: NSMenu, menuHasSelection: Bool) {
        let createItem = NSMenuItem()
        let createView = AgentModeButtonMenuItem(
            name: "Create an agent",
            iconName: AgentModeIcon.plus,
            isSelected: false,
            menuHasSelection: menuHasSelection,
            fontScale: fontScale,
            onSelect: { [onCreateAgent] in
                onCreateAgent()
                menu.cancelTracking()
            }
        )
        createItem.view = createView
        menu.addItem(createItem)
    }

    func showMenu(relativeTo button: NSButton) {
        let menu = createMenu()

        // Show menu aligned to the button's edge, positioned below the button
        let buttonFrame = button.frame
        let menuOrigin = NSPoint(x: buttonFrame.minX, y: buttonFrame.maxY)
        menu.popUp(positioning: menu.items.first, at: menuOrigin, in: button.superview)
    }
}