From 181d64338d0cc88ea700725ec7ec8d3708fa1121 Mon Sep 17 00:00:00 2001
From: Qwerasd <qwerasd205@users.noreply.github.com>
Date: Sat, 3 Feb 2024 20:03:34 -0500
Subject: [PATCH] fix(macOS): Restore titlebar tabs when exiting non-native
 fullscreen

---
 .../Features/Terminal/TerminalWindow.swift    | 26 ++++++++++++++++---
 macos/Sources/Helpers/FullScreenHandler.swift |  5 ++++
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/macos/Sources/Features/Terminal/TerminalWindow.swift b/macos/Sources/Features/Terminal/TerminalWindow.swift
index 7db4db42e..ce94c5317 100644
--- a/macos/Sources/Features/Terminal/TerminalWindow.swift
+++ b/macos/Sources/Features/Terminal/TerminalWindow.swift
@@ -136,10 +136,8 @@ class TerminalWindow: NSWindow {
         
         addWindowButtonsBackdrop(titlebarView: titlebarView, toolbarView: toolbarView)
         guard let windowButtonsBackdrop = windowButtonsBackdrop else { return }
-        windowButtonsBackdrop.isHidden = false
         
         addWindowDragHandle(titlebarView: titlebarView, toolbarView: toolbarView)
-        windowDragHandle?.isHidden = false
         
         accessoryClipView.translatesAutoresizingMaskIntoConstraints = false
         accessoryClipView.leftAnchor.constraint(equalTo: windowButtonsBackdrop.rightAnchor).isActive = true
@@ -165,7 +163,17 @@ class TerminalWindow: NSWindow {
     }
     
     private func addWindowButtonsBackdrop(titlebarView: NSView, toolbarView: NSView) {
-        guard windowButtonsBackdrop == nil else { return }
+        // If we already made the view, just make sure it's unhidden and correctly placed as a subview.
+        if let view = windowButtonsBackdrop {
+            view.removeFromSuperview()
+            view.isHidden = false
+            titlebarView.addSubview(view)
+            view.leftAnchor.constraint(equalTo: toolbarView.leftAnchor).isActive = true
+            view.rightAnchor.constraint(equalTo: toolbarView.leftAnchor, constant: 80).isActive = true
+            view.topAnchor.constraint(equalTo: toolbarView.topAnchor).isActive = true
+            view.heightAnchor.constraint(equalTo: toolbarView.heightAnchor).isActive = true
+            return
+        }
         
         let view = NSView()
         view.identifier = NSUserInterfaceItemIdentifier("_windowButtonsBackdrop")
@@ -202,7 +210,17 @@ class TerminalWindow: NSWindow {
     }
     
     private func addWindowDragHandle(titlebarView: NSView, toolbarView: NSView) {
-        guard windowDragHandle == nil else { return }
+        // If we already made the view, just make sure it's unhidden and correctly placed as a subview.
+        if let view = windowDragHandle {
+            view.removeFromSuperview()
+            view.isHidden = false
+            titlebarView.superview?.addSubview(view)
+            view.leftAnchor.constraint(equalTo: toolbarView.leftAnchor).isActive = true
+            view.rightAnchor.constraint(equalTo: toolbarView.rightAnchor).isActive = true
+            view.topAnchor.constraint(equalTo: toolbarView.topAnchor).isActive = true
+            view.bottomAnchor.constraint(equalTo: toolbarView.topAnchor, constant: 12).isActive = true
+            return
+        }
         
         let view = WindowDragView()
         view.identifier = NSUserInterfaceItemIdentifier("_windowDragHandle")
diff --git a/macos/Sources/Helpers/FullScreenHandler.swift b/macos/Sources/Helpers/FullScreenHandler.swift
index 2bb716c00..03cc02c32 100644
--- a/macos/Sources/Helpers/FullScreenHandler.swift
+++ b/macos/Sources/Helpers/FullScreenHandler.swift
@@ -164,6 +164,11 @@ class FullScreenHandler {
         // Restore frame
         window.setFrame(window.frameRect(forContentRect: previousFrame), display: true)
         
+        // Have titlebar tabs set itself up again, since removing the titlebar when fullscreen breaks its constraints.
+        if let window = window as? TerminalWindow, window.titlebarTabs {
+            window.titlebarTabs = true
+        }
+        
         // If the window was previously in a tab group that isn't empty now, we re-add it
         if let group = previousTabGroup, let tabIndex = previousTabGroupIndex, !group.windows.isEmpty {
             var tabWindow: NSWindow?