[FancyZones]Fix snap by hotkey for single-zone layouts (#22373)

* unsnap window from previous work area when snapping by hotkey

.github/actions/spell-check/expect.txt

@@ -1859,6 +1859,7 @@ UIs
 Ulaanbaatar
 ULARGE
 ULONGLONG
+unassign
 uncompilable
 UNCPRIORITY
 UNDNAME

src/modules/fancyzones/FancyZonesLib/FancyZones.cpp

@@ -808,6 +808,15 @@ bool FancyZones::OnSnapHotkeyBasedOnZoneNumber(HWND window, DWORD vkCode) noexce
             auto workArea = m_workAreaHandler.GetWorkArea(VirtualDesktop::instance().GetCurrentVirtualDesktopId(), *currMonitorInfo);
             if (m_windowMoveHandler.MoveWindowIntoZoneByDirectionAndIndex(window, vkCode, false /* cycle through zones */, workArea))
             {
+                // unassign from previous work area
+                for (auto& prevWorkArea : m_workAreaHandler.GetAllWorkAreas())
+                {
+                    if (workArea != prevWorkArea)
+                    {
+                        prevWorkArea->UnsnapWindow(window);
+                    }
+                }
+                
                 Trace::FancyZones::KeyboardSnapWindowToZone(workArea->GetLayout().get(), workArea->GetLayoutWindows().get());
                 return true;
             }

src/modules/fancyzones/FancyZonesLib/WindowMoveHandler.cpp

@@ -134,11 +134,7 @@ void WindowMoveHandler::MoveSizeStart(HWND window, HMONITOR monitor, POINT const
     auto workArea = workAreaMap.find(monitor);
     if (workArea != workAreaMap.end())
     {
-        const auto& layoutWindows = workArea->second->GetLayoutWindows();
+        workArea->second->UnsnapWindow(window);
-        if (layoutWindows)
-        {
-            layoutWindows->Dismiss(window);
-        }
     }
 }
 
@@ -289,6 +285,8 @@ void WindowMoveHandler::MoveSizeEnd(HWND window, const std::unordered_map<HMONIT
                         AppZoneHistory::instance().RemoveAppLastZone(window, workAreaPtr->UniqueId(), guidStr.value());
                     }
                 }
+
+                workAreaPtr->UnsnapWindow(window);
             }
         }
 

src/modules/fancyzones/FancyZonesLib/WorkArea.cpp

@@ -462,6 +462,22 @@ void WorkArea::SaveWindowProcessToZoneIndex(HWND window) noexcept
     }
 }
 
+bool WorkArea::UnsnapWindow(HWND window) noexcept
+{
+    if (!m_layoutWindows)
+    {
+        return false;    
+    }
+    
+    if (!m_layoutWindows->GetZoneIndexSetFromWindow(window).empty())
+    {
+        m_layoutWindows->Dismiss(window);
+        return true;
+    }
+
+    return false;
+}
+
 ZoneIndexSet WorkArea::GetWindowZoneIndexes(HWND window) const noexcept
 {
     if (m_layout)

src/modules/fancyzones/FancyZonesLib/WorkArea.h

@@ -69,6 +69,7 @@ public:
     bool MoveWindowIntoZoneByDirectionAndPosition(HWND window, DWORD vkCode, bool cycle) noexcept;
     bool ExtendWindowByDirectionAndPosition(HWND window, DWORD vkCode) noexcept;
     void SaveWindowProcessToZoneIndex(HWND window) noexcept;
+    bool UnsnapWindow(HWND window) noexcept;
 
     void UpdateActiveZoneSet() noexcept;