native macOS codings agent orchestrator prowl.onev.cat
Something went wrong. Try again.
11 kB · 347 lines
Swift
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348import Foundationimport IdentifiedCollections
struct FailedWorktreeCleanup { let didRemoveWorktree: Bool let didUpdatePinned: Bool let didUpdateOrder: Bool let worktree: Worktree?}
func removePendingWorktree(_ id: String, state: inout RepositoriesFeature.State) { state.pendingWorktrees.removeAll { $0.id == id }}
func requestCanvasFocus( _ target: CanvasFocusRequest.Target, openedWorktreeID: Worktree.ID, state: inout RepositoriesFeature.State) { state.nextCanvasFocusRequestID += 1 state.pendingCanvasFocusRequest = CanvasFocusRequest( id: state.nextCanvasFocusRequestID, target: target ) state.openedWorktreeIDs.insert(openedWorktreeID)}
func updatePendingWorktreeProgress( _ id: String, progress: WorktreeCreationProgress, state: inout RepositoriesFeature.State) { guard let index = state.pendingWorktrees.firstIndex(where: { $0.id == id }) else { return } state.pendingWorktrees[index].progress = progress}
func insertWorktree( _ worktree: Worktree, repositoryID: Repository.ID, state: inout RepositoriesFeature.State) { guard let index = state.repositories.index(id: repositoryID) else { return } let repository = state.repositories[index] if repository.worktrees[id: worktree.id] != nil { return } var worktrees = repository.worktrees worktrees.insert(worktree, at: 0) state.repositories[index] = Repository( id: repository.id, rootURL: repository.rootURL, name: repository.name, worktrees: worktrees )}
@discardableResultfunc removeWorktree( _ worktreeID: Worktree.ID, repositoryID: Repository.ID, state: inout RepositoriesFeature.State) -> Bool { guard let index = state.repositories.index(id: repositoryID) else { return false } let repository = state.repositories[index] guard repository.worktrees[id: worktreeID] != nil else { return false } var worktrees = repository.worktrees worktrees.remove(id: worktreeID) state.repositories[index] = Repository( id: repository.id, rootURL: repository.rootURL, name: repository.name, worktrees: worktrees ) return true}
func cleanupFailedWorktree( repositoryID: Repository.ID, name: String?, baseDirectory: URL, state: inout RepositoriesFeature.State) -> FailedWorktreeCleanup { guard let name, !name.isEmpty else { return FailedWorktreeCleanup( didRemoveWorktree: false, didUpdatePinned: false, didUpdateOrder: false, worktree: nil ) } let repositoryRootURL = URL(fileURLWithPath: repositoryID).standardizedFileURL let normalizedBaseDirectory = baseDirectory.standardizedFileURL let worktreeURL = normalizedBaseDirectory .appending(path: name, directoryHint: .isDirectory) .standardizedFileURL guard isPathInsideBaseDirectory(worktreeURL, baseDirectory: normalizedBaseDirectory) else { return FailedWorktreeCleanup( didRemoveWorktree: false, didUpdatePinned: false, didUpdateOrder: false, worktree: nil ) } let worktreeID = worktreeURL.path(percentEncoded: false) let worktree = state.repositories[id: repositoryID]?.worktrees[id: worktreeID] ?? Worktree( id: worktreeID, name: name, detail: "", workingDirectory: worktreeURL, repositoryRootURL: repositoryRootURL ) let cleanup = cleanupWorktreeState( worktreeID, repositoryID: repositoryID, state: &state ) return FailedWorktreeCleanup( didRemoveWorktree: cleanup.didRemoveWorktree, didUpdatePinned: cleanup.didUpdatePinned, didUpdateOrder: cleanup.didUpdateOrder, worktree: worktree )}
func isPathInsideBaseDirectory(_ path: URL, baseDirectory: URL) -> Bool { PathPolicy.contains(path, in: baseDirectory)}
struct WorktreeCleanupStateResult { let didRemoveWorktree: Bool let didUpdatePinned: Bool let didUpdateOrder: Bool}
func cleanupWorktreeState( _ worktreeID: Worktree.ID, repositoryID: Repository.ID, state: inout RepositoriesFeature.State) -> WorktreeCleanupStateResult { let didRemoveWorktree = removeWorktree(worktreeID, repositoryID: repositoryID, state: &state) state.pendingWorktrees.removeAll { $0.id == worktreeID } state.pendingSetupScriptWorktreeIDs.remove(worktreeID) state.pendingTerminalFocusWorktreeIDs.remove(worktreeID) state.archivingWorktreeIDs.remove(worktreeID) state.archiveScriptProgressByWorktreeID.removeValue(forKey: worktreeID) state.deletingWorktreeIDs.remove(worktreeID) state.worktreeInfoByID.removeValue(forKey: worktreeID) state.forgetPullRequestRefreshHistory(of: worktreeID) let didUpdatePinned = state.pinnedWorktreeIDs.contains(worktreeID) if didUpdatePinned { state.pinnedWorktreeIDs.removeAll { $0 == worktreeID } } var didUpdateOrder = false if var order = state.worktreeOrderByRepository[repositoryID] { let countBefore = order.count order.removeAll { $0 == worktreeID } if order.count != countBefore { didUpdateOrder = true if order.isEmpty { state.worktreeOrderByRepository.removeValue(forKey: repositoryID) } else { state.worktreeOrderByRepository[repositoryID] = order } } } return WorktreeCleanupStateResult( didRemoveWorktree: didRemoveWorktree, didUpdatePinned: didUpdatePinned, didUpdateOrder: didUpdateOrder )}
nonisolated func archiveScriptCommand(_ script: String) -> String { let normalized = script.replacing("\n", with: "\\n") return "bash -lc \(shellQuote(normalized))"}
nonisolated func worktreeCreateCommand( baseDirectoryURL: URL, name: String, copyIgnored: Bool, copyUntracked: Bool, baseRef: String, directoryOverride: URL? = nil) -> String { let baseDir = baseDirectoryURL.path(percentEncoded: false) var parts = ["wt", "--base-dir", baseDir, "sw"] if copyIgnored { parts.append("--copy-ignored") } if copyUntracked { parts.append("--copy-untracked") } if !baseRef.isEmpty { parts.append("--from") parts.append(baseRef) } if let directoryOverride { parts.append("--path") parts.append(directoryOverride.path(percentEncoded: false)) } if copyIgnored || copyUntracked { parts.append("--verbose") } parts.append(name) return parts.map(shellQuote).joined(separator: " ")}
nonisolated func shellQuote(_ value: String) -> String { "'\(value.replacing("'", with: "'\"'\"'"))'"}
func updateWorktreeName( _ worktreeID: Worktree.ID, name: String, state: inout RepositoriesFeature.State) { for index in state.repositories.indices { var repository = state.repositories[index] guard let worktreeIndex = repository.worktrees.index(id: worktreeID) else { continue } let worktree = repository.worktrees[worktreeIndex] guard worktree.name != name else { return } var worktrees = repository.worktrees worktrees[id: worktreeID] = Worktree( id: worktree.id, name: name, detail: worktree.detail, workingDirectory: worktree.workingDirectory, repositoryRootURL: worktree.repositoryRootURL, createdAt: worktree.createdAt ) repository = Repository( id: repository.id, rootURL: repository.rootURL, name: repository.name, worktrees: worktrees ) state.repositories[index] = repository return }}
@discardableResultfunc updateWorktreeLineChanges( worktreeID: Worktree.ID, changes: GitLineChanges, state: inout RepositoriesFeature.State) -> Bool { var entry = state.worktreeInfoByID[worktreeID] ?? WorktreeInfoEntry() if changes.isEmpty { entry.addedLines = nil entry.removedLines = nil } else { entry.addedLines = changes.added entry.removedLines = changes.removed } entry.skippedUntrackedFileCount = changes.skippedUntrackedFileCount let previousEntry = state.worktreeInfoByID[worktreeID] if entry.isEmpty { guard previousEntry != nil else { return false } state.worktreeInfoByID.removeValue(forKey: worktreeID) return true } guard previousEntry != entry else { return false } state.worktreeInfoByID[worktreeID] = entry return true}
func updateWorktreePullRequest( worktreeID: Worktree.ID, pullRequest: GithubPullRequest?, state: inout RepositoriesFeature.State) { var entry = state.worktreeInfoByID[worktreeID] ?? WorktreeInfoEntry() entry.pullRequest = pullRequest if entry.isEmpty { state.worktreeInfoByID.removeValue(forKey: worktreeID) } else { state.worktreeInfoByID[worktreeID] = entry }}
nonisolated func normalizedLineChanges(_ entry: WorktreeInfoEntry?) -> GitLineChanges? { guard let added = entry?.addedLines, let removed = entry?.removedLines else { return nil } return normalizedLineChanges( GitLineChanges( added: added, removed: removed, skippedUntrackedFileCount: entry?.skippedUntrackedFileCount ?? 0 ))}
nonisolated func normalizedLineChanges(_ changes: GitLineChanges) -> GitLineChanges? { changes.isEmpty ? nil : changes}
extension RepositoriesFeature.State { // A worktree ID is its path, so a worktree created again at a removed one's path must start // without the removed worktree's refresh history. mutating func forgetPullRequestRefreshHistory(of worktreeID: Worktree.ID) { pullRequestRefreshCheckpointByWorktreeID.removeValue(forKey: worktreeID) pullRequestRefreshForcedWorktreeIDs.remove(worktreeID) for repositoryID in Array(sentPullRequestRefreshMarks.keys) { sentPullRequestRefreshMarks[repositoryID]?.remove(worktreeID) } }
/// Marks the refresh in flight for `repositoryID`, or for every repository, still carries go /// back on the next refresh: GitHub did not answer for those branches. mutating func restoreUnansweredPullRequestRefreshMarks(of repositoryID: Repository.ID? = nil) { if let repositoryID { if let unanswered = sentPullRequestRefreshMarks.removeValue(forKey: repositoryID) { pullRequestRefreshForcedWorktreeIDs.formUnion(unanswered) } return } for unanswered in sentPullRequestRefreshMarks.values { pullRequestRefreshForcedWorktreeIDs.formUnion(unanswered) } sentPullRequestRefreshMarks.removeAll() }
mutating func forgetPullRequestRefreshHistory(keeping worktreeIDs: Set<Worktree.ID>) { pullRequestRefreshCheckpointByWorktreeID = pullRequestRefreshCheckpointByWorktreeID.filter { worktreeIDs.contains($0.key) } pullRequestRefreshForcedWorktreeIDs.formIntersection(worktreeIDs) sentPullRequestRefreshMarks = sentPullRequestRefreshMarks.mapValues { $0.intersection(worktreeIDs) } }}