import AppKitimport CodexBarCoreimport CodexBarMacroSupportimport Foundationimport SwiftUI @ProviderImplementationRegistrationstruct AugmentProviderImplementation: ProviderImplementation { let id: UsageProvider = .augment @MainActor func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] { let cookieBinding = Binding( get: { context.settings.augmentCookieSource.rawValue }, set: { raw in context.settings.augmentCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto }) let cookieOptions: [ProviderSettingsPickerOption] = [ ProviderSettingsPickerOption( id: ProviderCookieSource.auto.rawValue, title: ProviderCookieSource.auto.displayName), ProviderSettingsPickerOption( id: ProviderCookieSource.manual.rawValue, title: ProviderCookieSource.manual.displayName), ] let cookieSubtitle: () -> String? = { switch context.settings.augmentCookieSource { case .auto: "Automatic imports browser cookies." case .manual: "Paste a Cookie header or cURL capture from the Augment dashboard." case .off: "Augment cookies are disabled." } } return [ ProviderSettingsPickerDescriptor( id: "augment-cookie-source", title: "Cookie source", subtitle: "Automatic imports browser cookies.", dynamicSubtitle: cookieSubtitle, binding: cookieBinding, options: cookieOptions, isVisible: nil, onChange: nil), ] } @MainActor func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] { // Actions for auto mode (browser cookies) let autoModeActions: [ProviderSettingsActionDescriptor] = [ ProviderSettingsActionDescriptor( id: "augment-refresh-browser", title: "Refresh Browser Cookies", style: .bordered, isVisible: nil, perform: { // Open Augment dashboard to refresh the session if let url = URL(string: "https://app.augmentcode.com") { NSWorkspace.shared.open(url) } // Show alert with instructions let alert = NSAlert() alert.messageText = "Refresh Browser Cookies" alert.informativeText = """ To refresh your Augment session cookies: 1. The Augment dashboard should now be open in your browser 2. If you're not logged in, log in now 3. If you are logged in, refresh the page (⌘R) 4. Wait a few seconds for cookies to update 5. Click "Force Refresh Session" below to reload Note: Browser cookies may take a few seconds to write to disk. If it still doesn't work, try closing and reopening your browser. """ alert.alertStyle = .informational alert.addButton(withTitle: "OK") alert.runModal() }), ProviderSettingsActionDescriptor( id: "augment-force-refresh", title: "Force Refresh Session", style: .bordered, isVisible: nil, perform: { print("[CodexBar] Force refresh requested by user") await context.store.forceRefreshAugmentSession() }), ProviderSettingsActionDescriptor( id: "augment-open-dashboard", title: "Open Augment Dashboard", style: .link, isVisible: nil, perform: { if let url = URL(string: "https://app.augmentcode.com") { NSWorkspace.shared.open(url) } }), ] // Actions for manual mode (cookie header) let manualModeActions: [ProviderSettingsActionDescriptor] = [ ProviderSettingsActionDescriptor( id: "augment-open-dashboard-manual", title: "Open Augment", style: .link, isVisible: nil, perform: { if let url = URL(string: "https://augmentcode.com") { NSWorkspace.shared.open(url) } }), ] return [ ProviderSettingsFieldDescriptor( id: "augment-cookie", title: "", subtitle: "", kind: .secure, placeholder: "Cookie: …", binding: context.stringBinding(\.augmentCookieHeader), actions: context.settings.augmentCookieSource == .auto ? autoModeActions : manualModeActions, isVisible: { context.settings.augmentCookieSource == .manual }, onActivate: { context.settings.ensureAugmentCookieLoaded() }), ] }}
import AppKit
import CodexBarCore
import CodexBarMacroSupport
import Foundation
import SwiftUI
@ProviderImplementationRegistration
struct AugmentProviderImplementation: ProviderImplementation {
let id: UsageProvider = .augment
@MainActor
func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {
let cookieBinding = Binding(
get: { context.settings.augmentCookieSource.rawValue },
set: { raw in
context.settings.augmentCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto
})
let cookieOptions: [ProviderSettingsPickerOption] = [
ProviderSettingsPickerOption(
id: ProviderCookieSource.auto.rawValue,
title: ProviderCookieSource.auto.displayName),
ProviderSettingsPickerOption(
id: ProviderCookieSource.manual.rawValue,
title: ProviderCookieSource.manual.displayName),
]
let cookieSubtitle: () -> String? = {
switch context.settings.augmentCookieSource {
case .auto:
"Automatic imports browser cookies."
case .manual:
"Paste a Cookie header or cURL capture from the Augment dashboard."
case .off:
"Augment cookies are disabled."
}
}
return [
ProviderSettingsPickerDescriptor(
id: "augment-cookie-source",
title: "Cookie source",
subtitle: "Automatic imports browser cookies.",
dynamicSubtitle: cookieSubtitle,
binding: cookieBinding,
options: cookieOptions,
isVisible: nil,
onChange: nil),
]
}
@MainActor
func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {
// Actions for auto mode (browser cookies)
let autoModeActions: [ProviderSettingsActionDescriptor] = [
ProviderSettingsActionDescriptor(
id: "augment-refresh-browser",
title: "Refresh Browser Cookies",
style: .bordered,
isVisible: nil,
perform: {
// Open Augment dashboard to refresh the session
if let url = URL(string: "https://app.augmentcode.com") {
NSWorkspace.shared.open(url)
}
// Show alert with instructions
let alert = NSAlert()
alert.messageText = "Refresh Browser Cookies"
alert.informativeText = """
To refresh your Augment session cookies:
1. The Augment dashboard should now be open in your browser
2. If you're not logged in, log in now
3. If you are logged in, refresh the page (⌘R)
4. Wait a few seconds for cookies to update
5. Click "Force Refresh Session" below to reload
Note: Browser cookies may take a few seconds to write to disk.
If it still doesn't work, try closing and reopening your browser.
"""
alert.alertStyle = .informational
alert.addButton(withTitle: "OK")
alert.runModal()
}),
ProviderSettingsActionDescriptor(
id: "augment-force-refresh",
title: "Force Refresh Session",
style: .bordered,
isVisible: nil,
perform: {
print("[CodexBar] Force refresh requested by user")
await context.store.forceRefreshAugmentSession()
}),
ProviderSettingsActionDescriptor(
id: "augment-open-dashboard",
title: "Open Augment Dashboard",
style: .link,
isVisible: nil,
perform: {
if let url = URL(string: "https://app.augmentcode.com") {
NSWorkspace.shared.open(url)
}
}),
]
// Actions for manual mode (cookie header)
let manualModeActions: [ProviderSettingsActionDescriptor] = [
ProviderSettingsActionDescriptor(
id: "augment-open-dashboard-manual",
title: "Open Augment",
style: .link,
isVisible: nil,
perform: {
if let url = URL(string: "https://augmentcode.com") {
NSWorkspace.shared.open(url)
}
}),
]
return [
ProviderSettingsFieldDescriptor(
id: "augment-cookie",
title: "",
subtitle: "",
kind: .secure,
placeholder: "Cookie: …",
binding: context.stringBinding(\.augmentCookieHeader),
actions: context.settings.augmentCookieSource == .auto ? autoModeActions : manualModeActions,
isVisible: { context.settings.augmentCookieSource == .manual },
onActivate: { context.settings.ensureAugmentCookieLoaded() }),
]
}
}