feat(companion): add runnable macOS menu-bar reference app scaffold

This commit is contained in:
William Valentin
2026-02-26 20:51:00 -08:00
parent 3f6232d1fb
commit d303869866
21 changed files with 695 additions and 13 deletions
+5
View File
@@ -15,6 +15,7 @@ export { writeCompanionShellTemplate } from './shellTemplate.js';
export { verifyCompanionReleaseBundle } from './releaseVerify.js';
export { buildAndVerifyCompanionReleaseBundle } from './releasePipeline.js';
export { generateReferenceCompanionApps } from './referenceApps.js';
export { generateMacOSMenuBarApp } from './macosMenuBarApp.js';
export type {
CompanionRuntimeClientOptions,
@@ -104,3 +105,7 @@ export type {
GenerateReferenceCompanionAppsResult,
GeneratedReferenceCompanionApp,
} from './referenceApps.js';
export type {
GenerateMacOSMenuBarAppInput,
GenerateMacOSMenuBarAppResult,
} from './macosMenuBarApp.js';
+50
View File
@@ -0,0 +1,50 @@
import { mkdtemp, readFile, rm } from 'node:fs/promises';
import { join } from 'node:path';
import { tmpdir } from 'node:os';
import { describe, expect, it } from 'vitest';
import { generateMacOSMenuBarApp } from './macosMenuBarApp.js';
describe('generateMacOSMenuBarApp', () => {
it('writes a runnable Swift package scaffold for menu bar companion', async () => {
const tempDir = await mkdtemp(join(tmpdir(), 'flynn-macos-menubar-app-'));
const outputDir = join(tempDir, 'macos-app');
const result = await generateMacOSMenuBarApp({
outputDir,
manifest: {
schemaVersion: 1,
generatedAt: '2026-02-27T00:00:00.000Z',
gateway: { url: 'ws://127.0.0.1:18800' },
node: {
nodeId: 'macos-menu-node',
role: 'companion',
platform: 'macos',
capabilities: ['ui.canvas'],
},
runtime: {
heartbeatSeconds: 30,
handoffTimeoutMs: 120000,
autoReconnect: true,
},
},
});
const packageSwift = await readFile(`${outputDir}/Package.swift`, 'utf8');
const mainSwift = await readFile(`${outputDir}/Sources/FlynnCompanionMenuBar/main.swift`, 'utf8');
const bootstrap = await readFile(`${outputDir}/Sources/FlynnCompanionMenuBar/Resources/companion.bootstrap.json`, 'utf8');
const readme = await readFile(`${outputDir}/README.md`, 'utf8');
expect(result.files).toEqual([
`${outputDir}/Package.swift`,
`${outputDir}/Sources/FlynnCompanionMenuBar/main.swift`,
`${outputDir}/Sources/FlynnCompanionMenuBar/Resources/companion.bootstrap.json`,
`${outputDir}/README.md`,
]);
expect(packageSwift).toContain('FlynnCompanionMenuBar');
expect(mainSwift).toContain('NSStatusBar.system.statusItem');
expect(mainSwift).toContain('flynn');
expect(JSON.parse(bootstrap)).toMatchObject({ node: { nodeId: 'macos-menu-node' } });
expect(readme).toContain('menu bar companion');
await rm(tempDir, { recursive: true, force: true });
});
});
+304
View File
@@ -0,0 +1,304 @@
import { mkdir, writeFile } from 'node:fs/promises';
import type { CompanionBootstrapManifest } from './bootstrapManifest.js';
export interface GenerateMacOSMenuBarAppInput {
outputDir: string;
manifest: CompanionBootstrapManifest;
}
export interface GenerateMacOSMenuBarAppResult {
outputDir: string;
files: string[];
}
function packageSwift(): string {
return `// swift-tools-version: 5.9
import PackageDescription
let package = Package(
name: "FlynnCompanionMenuBar",
platforms: [
.macOS(.v13)
],
products: [
.executable(name: "FlynnCompanionMenuBar", targets: ["FlynnCompanionMenuBar"])
],
targets: [
.executableTarget(
name: "FlynnCompanionMenuBar",
resources: [.copy("Resources/companion.bootstrap.json")]
)
]
)
`;
}
function readmeBody(): string {
return `# Flynn macOS Companion Menu Bar App (Reference MVP)
This is a runnable Swift Package reference app that creates a macOS menu bar companion controller.
## Capabilities
- Loads companion.bootstrap.json from package resources.
- Starts/stops flynn companion as a child process from menu actions.
- Runs one-shot handoff prompts (flynn companion --once --handoff ...).
- Shows process state in the menu bar title (F, F* running, F! error).
## Build + Run
Run:
cd <this-directory>
swift run FlynnCompanionMenuBar
Prerequisites:
- macOS with Swift toolchain.
- flynn executable available in PATH.
`;
}
function mainSwift(manifest: CompanionBootstrapManifest): string {
return `import AppKit
import Foundation
struct CompanionBootstrap: Codable {
struct Gateway: Codable {
let url: String
let token: String?
}
struct Node: Codable {
let nodeId: String
let role: String
let platform: String
let capabilities: [String]
}
struct Runtime: Codable {
let heartbeatSeconds: Int
let handoffTimeoutMs: Int
let autoReconnect: Bool
}
let schemaVersion: Int
let generatedAt: String
let gateway: Gateway
let node: Node
let runtime: Runtime
}
final class CompanionProcessController {
private var process: Process?
private var isRunning = false
private(set) var lastError: String?
private let bootstrap: CompanionBootstrap
init(bootstrap: CompanionBootstrap) {
self.bootstrap = bootstrap
}
func start() {
guard process == nil else { return }
let proc = Process()
proc.executableURL = URL(fileURLWithPath: "/usr/bin/env")
var args: [String] = [
"flynn",
"companion",
"--url", bootstrap.gateway.url,
"--node-id", bootstrap.node.nodeId,
"--role", bootstrap.node.role,
"--platform", bootstrap.node.platform,
"--heartbeat", String(bootstrap.runtime.heartbeatSeconds),
"--handoff-timeout", String(bootstrap.runtime.handoffTimeoutMs),
]
if let token = bootstrap.gateway.token, !token.isEmpty {
args += ["--token", token]
}
for capability in bootstrap.node.capabilities {
args += ["--capability", capability]
}
if !bootstrap.runtime.autoReconnect {
args.append("--once")
}
proc.arguments = args
proc.terminationHandler = { [weak self] p in
DispatchQueue.main.async {
self?.process = nil
self?.isRunning = false
if p.terminationStatus != 0 {
self?.lastError = "companion exited with status \\(p.terminationStatus)"
}
}
}
do {
try proc.run()
process = proc
isRunning = true
lastError = nil
} catch {
lastError = error.localizedDescription
process = nil
isRunning = false
}
}
func stop() {
process?.terminate()
process = nil
isRunning = false
}
func runHandoff(message: String) {
guard !message.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty else { return }
let proc = Process()
proc.executableURL = URL(fileURLWithPath: "/usr/bin/env")
var args: [String] = [
"flynn",
"companion",
"--once",
"--url", bootstrap.gateway.url,
"--node-id", bootstrap.node.nodeId,
"--role", bootstrap.node.role,
"--platform", bootstrap.node.platform,
"--handoff-timeout", String(bootstrap.runtime.handoffTimeoutMs),
"--handoff", message,
]
if let token = bootstrap.gateway.token, !token.isEmpty {
args += ["--token", token]
}
proc.arguments = args
do {
try proc.run()
} catch {
lastError = "handoff failed: \\(error.localizedDescription)"
}
}
var running: Bool { isRunning }
}
final class AppDelegate: NSObject, NSApplicationDelegate {
private var statusItem: NSStatusItem!
private var startItem: NSMenuItem!
private var stopItem: NSMenuItem!
private var handoffItem: NSMenuItem!
private var errorItem: NSMenuItem!
private var controller: CompanionProcessController!
private var refreshTimer: Timer?
func applicationDidFinishLaunching(_ notification: Notification) {
let bootstrap = Self.loadBootstrap()
controller = CompanionProcessController(bootstrap: bootstrap)
statusItem = NSStatusBar.system.statusItem(withLength: NSStatusItem.variableLength)
let menu = NSMenu()
startItem = NSMenuItem(title: "Start Companion", action: #selector(startCompanion), keyEquivalent: "s")
stopItem = NSMenuItem(title: "Stop Companion", action: #selector(stopCompanion), keyEquivalent: "x")
handoffItem = NSMenuItem(title: "Send Handoff...", action: #selector(sendHandoff), keyEquivalent: "h")
errorItem = NSMenuItem(title: "Last error: (none)", action: nil, keyEquivalent: "")
errorItem.isEnabled = false
menu.addItem(startItem)
menu.addItem(stopItem)
menu.addItem(handoffItem)
menu.addItem(NSMenuItem.separator())
menu.addItem(errorItem)
menu.addItem(NSMenuItem.separator())
menu.addItem(NSMenuItem(title: "Quit", action: #selector(quitApp), keyEquivalent: "q"))
startItem.target = self
stopItem.target = self
handoffItem.target = self
statusItem.menu = menu
refreshUI()
refreshTimer = Timer.scheduledTimer(withTimeInterval: 1.0, repeats: true) { [weak self] _ in
self?.refreshUI()
}
}
func applicationWillTerminate(_ notification: Notification) {
refreshTimer?.invalidate()
controller.stop()
}
@objc private func startCompanion() {
controller.start()
refreshUI()
}
@objc private func stopCompanion() {
controller.stop()
refreshUI()
}
@objc private func sendHandoff() {
let alert = NSAlert()
alert.messageText = "Companion Handoff"
alert.informativeText = "Send a one-shot handoff message through Flynn companion."
let input = NSTextField(frame: NSRect(x: 0, y: 0, width: 280, height: 24))
input.placeholderString = "message"
alert.accessoryView = input
alert.addButton(withTitle: "Send")
alert.addButton(withTitle: "Cancel")
let response = alert.runModal()
if response == .alertFirstButtonReturn {
controller.runHandoff(message: input.stringValue)
refreshUI()
}
}
@objc private func quitApp() {
NSApplication.shared.terminate(nil)
}
private func refreshUI() {
if controller.running {
statusItem.button?.title = "F*"
startItem.isEnabled = false
stopItem.isEnabled = true
} else {
statusItem.button?.title = controller.lastError == nil ? "F" : "F!"
startItem.isEnabled = true
stopItem.isEnabled = false
}
let errorText = controller.lastError ?? "(none)"
errorItem.title = "Last error: \\(errorText)"
}
static func loadBootstrap() -> CompanionBootstrap {
guard let url = Bundle.module.url(forResource: "companion.bootstrap", withExtension: "json"),
let data = try? Data(contentsOf: url),
let decoded = try? JSONDecoder().decode(CompanionBootstrap.self, from: data)
else {
fatalError("Failed to load companion.bootstrap.json resource")
}
return decoded
}
}
let app = NSApplication.shared
let delegate = AppDelegate()
app.setActivationPolicy(.accessory)
app.delegate = delegate
app.run()
`;
}
export async function generateMacOSMenuBarApp(
input: GenerateMacOSMenuBarAppInput,
): Promise<GenerateMacOSMenuBarAppResult> {
const packagePath = `${input.outputDir}/Package.swift`;
const sourceDir = `${input.outputDir}/Sources/FlynnCompanionMenuBar`;
const sourcePath = `${sourceDir}/main.swift`;
const resourcesDir = `${sourceDir}/Resources`;
const resourceBootstrapPath = `${resourcesDir}/companion.bootstrap.json`;
const readmePath = `${input.outputDir}/README.md`;
await mkdir(sourceDir, { recursive: true });
await mkdir(resourcesDir, { recursive: true });
await writeFile(packagePath, packageSwift(), 'utf8');
await writeFile(sourcePath, mainSwift(input.manifest), 'utf8');
await writeFile(resourceBootstrapPath, `${JSON.stringify(input.manifest, null, 2)}\n`, 'utf8');
await writeFile(readmePath, readmeBody(), 'utf8');
return {
outputDir: input.outputDir,
files: [packagePath, sourcePath, resourceBootstrapPath, readmePath],
};
}
+4
View File
@@ -16,14 +16,18 @@ describe('generateReferenceCompanionApps', () => {
expect(result.generated.map((entry) => entry.platform)).toEqual(['macos', 'ios', 'android']);
const macosTemplate = await readFile(`${outputDir}/macos/MenuBarCompanion.swift`, 'utf8');
const macosAppMain = await readFile(`${outputDir}/macos-app/Sources/FlynnCompanionMenuBar/main.swift`, 'utf8');
const iosTemplate = await readFile(`${outputDir}/ios/CompanionBootstrap.swift`, 'utf8');
const androidTemplate = await readFile(`${outputDir}/android/CompanionBootstrap.kt`, 'utf8');
const rootReadme = await readFile(`${outputDir}/README.md`, 'utf8');
expect(macosTemplate).toContain('launchFlynnCompanion');
expect(macosAppMain).toContain('NSStatusBar.system.statusItem');
expect(iosTemplate).toContain('CompanionBootstrap');
expect(androidTemplate).toContain('data class CompanionBootstrap');
expect(rootReadme).toContain('Companion Reference Apps');
expect(result.macosMenuBarAppDir).toBe(`${outputDir}/macos-app`);
expect(result.macosMenuBarAppFiles.length).toBeGreaterThan(0);
await rm(tempDir, { recursive: true, force: true });
});
+28
View File
@@ -1,6 +1,7 @@
import { mkdir, writeFile } from 'node:fs/promises';
import { writeCompanionShellTemplate } from './shellTemplate.js';
import type { CompanionBootstrapPlatform } from './bootstrapManifest.js';
import { generateMacOSMenuBarApp } from './macosMenuBarApp.js';
export interface GenerateReferenceCompanionAppsInput {
outputDir: string;
@@ -17,6 +18,8 @@ export interface GeneratedReferenceCompanionApp {
export interface GenerateReferenceCompanionAppsResult {
rootDir: string;
generated: GeneratedReferenceCompanionApp[];
macosMenuBarAppDir: string;
macosMenuBarAppFiles: string[];
readmePath: string;
}
@@ -33,6 +36,7 @@ function rootReadmeBody(): string {
This directory contains generated companion starter shells for:
- macOS menu-bar style wrapper
- macOS runnable menu-bar app scaffold (Swift Package)
- iOS shell
- Android shell
@@ -76,11 +80,35 @@ export async function generateReferenceCompanionApps(
files: result.files,
});
}
const macosManifest = {
schemaVersion: 1 as const,
generatedAt: generatedAt.toISOString(),
gateway: {
url: input.gatewayUrl,
},
node: {
nodeId: 'macos-menu-bar-reference',
role: 'companion',
platform: 'macos' as const,
capabilities: defaultCapabilities('macos'),
},
runtime: {
heartbeatSeconds: 30,
handoffTimeoutMs: 120000,
autoReconnect: true,
},
};
const macosMenuBar = await generateMacOSMenuBarApp({
outputDir: `${input.outputDir}/macos-app`,
manifest: macosManifest,
});
const readmePath = `${input.outputDir}/README.md`;
await writeFile(readmePath, rootReadmeBody(), 'utf8');
return {
rootDir: input.outputDir,
generated,
macosMenuBarAppDir: macosMenuBar.outputDir,
macosMenuBarAppFiles: macosMenuBar.files,
readmePath,
};
}