aerospace fork
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697import AppKitimport Common
@MainActorprivate var resizeWithMouseTask: Task<(), any Error>? = nil
func resizedObs(_: AXObserver, ax: AXUIElement, notif: CFString, _: UnsafeMutableRawPointer?) { let notif = notif as String let windowId = ax.containingWindowId() Task.startUnstructured { @MainActor in guard let token: RunSessionGuard = .isServerEnabled else { return } guard let windowId, let window = Window.get(byId: windowId), try await isManipulatedWithMouse(window) else { scheduleCancellableCompleteRefreshSession(.ax(notif)) return } resizeWithMouseTask?.cancel() resizeWithMouseTask = Task.startUnstructured { try checkCancellation() try await runLightSession(.ax(notif), token) { try await resizeWithMouse(window) } } }}
@MainActorfunc resetManipulatedWithMouseIfPossible() async throws { if currentlyManipulatedWithMouseWindowId != nil { currentlyManipulatedWithMouseWindowId = nil for workspace in Workspace.all { workspace.resetResizeWeightBeforeResizeRecursive() } scheduleCancellableCompleteRefreshSession(.resetManipulatedWithMouse, optimisticallyPreLayoutWorkspaces: true) }}
private let adaptiveWeightBeforeResizeWithMouseKey = TreeNodeUserDataKey<CGFloat>(key: "adaptiveWeightBeforeResizeWithMouseKey")
@MainActorprivate func resizeWithMouse(_ window: Window) async throws { // todo cover with tests resetClosedWindowsCache() switch window.windowParentCases { case .unbound: return case .floatingWindowsContainer, .macosMinimizedWindowsContainer, .macosFullscreenWindowsContainer, .macosPopupWindowsContainer, .macosHiddenAppsWindowsContainer: return // Nothing to do for floating, or unconventional windows case .tilingContainer: guard let rect = try await window.getAxRect(.cancellable) else { return } guard let lastAppliedLayoutRect = window.lastAppliedLayoutPhysicalRect else { return } let (lParent, lOwnIndex) = window.closestParent(hasChildrenInDirection: .left, withLayout: .tiles) ?? (nil, nil) let (dParent, dOwnIndex) = window.closestParent(hasChildrenInDirection: .down, withLayout: .tiles) ?? (nil, nil) let (uParent, uOwnIndex) = window.closestParent(hasChildrenInDirection: .up, withLayout: .tiles) ?? (nil, nil) let (rParent, rOwnIndex) = window.closestParent(hasChildrenInDirection: .right, withLayout: .tiles) ?? (nil, nil) let table: [(CGFloat, TilingContainer?, Int?, Int?)] = [ (lastAppliedLayoutRect.minX - rect.minX, lParent, 0, lOwnIndex), // Horizontal, to the left of the window (rect.maxY - lastAppliedLayoutRect.maxY, dParent, dOwnIndex.map { $0 + 1 }, dParent?.children.count), // Vertical, to the down of the window (lastAppliedLayoutRect.minY - rect.minY, uParent, 0, uOwnIndex), // Vertical, to the up of the window (rect.maxX - lastAppliedLayoutRect.maxX, rParent, rOwnIndex.map { $0 + 1 }, rParent?.children.count), // Horizontal, to the right of the window ] for (diff, parent, startIndex, pastTheEndIndex) in table { if let parent, let startIndex, let pastTheEndIndex, pastTheEndIndex - startIndex > 0 && abs(diff) > 5 { // 5 pixels should be enough to fight with accumulated floating precision error let siblingDiff = diff.div(pastTheEndIndex - startIndex).orDie() let orientation = parent.orientation
window.parentsWithSelf.lazy .prefix(while: { $0 != parent }) .filter { let parent = $0.parent as? TilingContainer return parent?.orientation == orientation && parent?.layout == .tiles } .forEach { $0.setWeight(orientation, $0.getWeightBeforeResize(orientation) + diff) } for sibling in parent.children[startIndex ..< pastTheEndIndex] { sibling.setWeight(orientation, sibling.getWeightBeforeResize(orientation) - siblingDiff) } } } currentlyManipulatedWithMouseWindowId = window.windowId }}
extension TreeNode { @MainActor fileprivate func getWeightBeforeResize(_ orientation: Orientation) -> CGFloat { let currentWeight = getWeight(orientation) // Check assertions return getUserData(key: adaptiveWeightBeforeResizeWithMouseKey) ?? (lastAppliedLayoutVirtualRect?.getDimension(orientation) ?? currentWeight) .also { putUserData(key: adaptiveWeightBeforeResizeWithMouseKey, data: $0) } }
fileprivate func resetResizeWeightBeforeResizeRecursive() { cleanUserData(key: adaptiveWeightBeforeResizeWithMouseKey) for child in children { child.resetResizeWeightBeforeResizeRecursive() } }}