CleanCommand.swift•6.02 kB
import ArgumentParser
import Foundation
import PeekabooCore
/// Clean up session cache and temporary files
@available(macOS 14.0, *)
struct CleanCommand: AsyncParsableCommand {
static let configuration = CommandConfiguration(
commandName: "clean",
abstract: "Clean up session cache and temporary files",
discussion: """
EXAMPLES:
peekaboo clean --all-sessions # Remove all session data
peekaboo clean --older-than 24 # Remove sessions older than 24 hours
peekaboo clean --session 12345 # Remove specific session
peekaboo clean --dry-run # Preview what would be deleted
SESSION CACHE:
Sessions are stored in ~/.peekaboo/session/<PID>/
Each session contains:
- raw.png: Original screenshot
- annotated.png: Screenshot with UI markers (if generated)
- map.json: UI element mapping data
"""
)
@Flag(help: "Remove all session data")
var allSessions = false
@Option(help: "Remove sessions older than specified hours (default: 24)")
var olderThan: Int?
@Option(help: "Remove specific session by ID")
var session: String?
@Flag(help: "Show what would be deleted without actually deleting")
var dryRun = false
@Flag(help: "Output in JSON format")
var jsonOutput = false
mutating func run() async throws {
let startTime = Date()
do {
// Validate options
let optionCount = [allSessions, olderThan != nil, self.session != nil].count { $0 }
guard optionCount == 1 else {
throw ValidationError("Specify exactly one of: --all-sessions, --older-than, or --session")
}
// Perform cleanup based on option using the FileService
let result: CleanResult
if self.allSessions {
result = try await PeekabooServices.shared.files.cleanAllSessions(dryRun: self.dryRun)
} else if let hours = olderThan {
result = try await PeekabooServices.shared.files.cleanOldSessions(hours: hours, dryRun: self.dryRun)
} else if let sessionId = session {
result = try await PeekabooServices.shared.files.cleanSpecificSession(
sessionId: sessionId,
dryRun: self.dryRun
)
} else {
throw ValidationError("No cleanup option specified")
}
// Calculate execution time
let executionTime = Date().timeIntervalSince(startTime)
// Output results
if self.jsonOutput {
// Create a wrapper for the clean result with execution time
struct CleanResultWithTime: Codable {
let sessionsRemoved: Int
let bytesFreed: Int64
let sessionDetails: [SessionDetail]
let dryRun: Bool
let executionTime: TimeInterval
}
let outputData = CleanResultWithTime(
sessionsRemoved: result.sessionsRemoved,
bytesFreed: result.bytesFreed,
sessionDetails: result.sessionDetails,
dryRun: result.dryRun,
executionTime: executionTime
)
outputSuccessCodable(data: outputData)
} else {
self.printResults(result, executionTime: executionTime)
}
} catch let error as FileServiceError {
handleFileServiceError(error, jsonOutput: self.jsonOutput)
throw ExitCode(1)
} catch {
if self.jsonOutput {
outputError(message: error.localizedDescription, code: .INTERNAL_SWIFT_ERROR)
} else {
var localStandardErrorStream = FileHandleTextOutputStream(FileHandle.standardError)
print("Error: \(error.localizedDescription)", to: &localStandardErrorStream)
}
throw ExitCode.failure
}
}
private func printResults(_ result: CleanResult, executionTime: TimeInterval) {
if result.dryRun {
print("🔍 Dry run mode - no files will be deleted")
print("")
}
if result.sessionsRemoved == 0 {
print("✅ No sessions to clean")
} else {
let action = result.dryRun ? "Would remove" : "Removed"
print("🗑️ \(action) \(result.sessionsRemoved) session\(result.sessionsRemoved == 1 ? "" : "s")")
print("💾 Space \(result.dryRun ? "to be freed" : "freed"): \(self.formatBytes(result.bytesFreed))")
if result.sessionDetails.count <= 5 {
print("\nSessions:")
for detail in result.sessionDetails {
print(" - \(detail.sessionId) (\(self.formatBytes(detail.size)))")
}
}
}
print("\n⏱️ Completed in \(String(format: "%.2f", executionTime))s")
}
private func formatBytes(_ bytes: Int64) -> String {
let formatter = ByteCountFormatter()
formatter.countStyle = .file
return formatter.string(fromByteCount: bytes)
}
}
// MARK: - Error Handling
private func handleFileServiceError(_ error: FileServiceError, jsonOutput: Bool) {
let errorCode: ErrorCode = switch error {
case .sessionNotFound:
.SESSION_NOT_FOUND
case .directoryNotFound:
.FILE_IO_ERROR
case .insufficientPermissions:
.PERMISSION_DENIED
case .fileSystemError:
.FILE_IO_ERROR
}
if jsonOutput {
outputError(message: error.localizedDescription, code: errorCode)
} else {
var localStandardErrorStream = FileHandleTextOutputStream(FileHandle.standardError)
print("❌ \(error.localizedDescription)", to: &localStandardErrorStream)
}
}