78 lines
2.3 KiB
Swift
78 lines
2.3 KiB
Swift
import Commander
|
|
import Foundation
|
|
|
|
@MainActor
|
|
struct ServiceRootCommand: ParsableCommand {
|
|
static var commandDescription: CommandDescription {
|
|
CommandDescription(
|
|
commandName: "service",
|
|
abstract: "Manage launchd agent",
|
|
subcommands: [ServiceInstall.self, ServiceUninstall.self, ServiceStatus.self])
|
|
}
|
|
}
|
|
|
|
private enum LaunchdHelper {
|
|
static let label = "com.swabble.agent"
|
|
|
|
static var plistURL: URL {
|
|
FileManager.default
|
|
.homeDirectoryForCurrentUser
|
|
.appendingPathComponent("Library/LaunchAgents/\(label).plist")
|
|
}
|
|
|
|
static func writePlist(executable: String) throws {
|
|
let plist: [String: Any] = [
|
|
"Label": label,
|
|
"ProgramArguments": [executable, "serve"],
|
|
"RunAtLoad": true,
|
|
"KeepAlive": true
|
|
]
|
|
let data = try PropertyListSerialization.data(fromPropertyList: plist, format: .xml, options: 0)
|
|
try data.write(to: plistURL)
|
|
}
|
|
|
|
static func removePlist() throws {
|
|
try? FileManager.default.removeItem(at: plistURL)
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
struct ServiceInstall: ParsableCommand {
|
|
static var commandDescription: CommandDescription {
|
|
CommandDescription(commandName: "install", abstract: "Install user launch agent")
|
|
}
|
|
|
|
mutating func run() async throws {
|
|
let exe = CommandLine.arguments.first ?? "/usr/local/bin/swabble"
|
|
try LaunchdHelper.writePlist(executable: exe)
|
|
print("launchctl load -w \(LaunchdHelper.plistURL.path)")
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
struct ServiceUninstall: ParsableCommand {
|
|
static var commandDescription: CommandDescription {
|
|
CommandDescription(commandName: "uninstall", abstract: "Remove launch agent")
|
|
}
|
|
|
|
mutating func run() async throws {
|
|
try LaunchdHelper.removePlist()
|
|
print("launchctl bootout gui/$(id -u)/\(LaunchdHelper.label)")
|
|
}
|
|
}
|
|
|
|
@MainActor
|
|
struct ServiceStatus: ParsableCommand {
|
|
static var commandDescription: CommandDescription {
|
|
CommandDescription(commandName: "status", abstract: "Show launch agent status")
|
|
}
|
|
|
|
mutating func run() async throws {
|
|
if FileManager.default.fileExists(atPath: LaunchdHelper.plistURL.path) {
|
|
print("plist present at \(LaunchdHelper.plistURL.path)")
|
|
} else {
|
|
print("launchd plist not installed")
|
|
}
|
|
}
|
|
}
|