From 9560cef37f664929ea1fc5f486e78f16b1bd4311 Mon Sep 17 00:00:00 2001
From: plule <630159+plule@users.noreply.github.com>
Date: Mon, 5 Jan 2026 11:38:55 +0100
Subject: [PATCH] Fix the suspendmanager overlay appearing for followed units

---
 docs/changelog.txt             |  1 +
 plugins/lua/suspendmanager.lua | 32 +++++++++++++++++++++++++++-----
 2 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/docs/changelog.txt b/docs/changelog.txt
index 2eb584644f..cc0acf0217 100644
--- a/docs/changelog.txt
+++ b/docs/changelog.txt
@@ -59,6 +59,7 @@ Template for new versions:
 ## New Features
 
 ## Fixes
+- `suspendmanager`: Fix the overlay appearing where it should not when following a unit
 
 ## Misc Improvements
 
diff --git a/plugins/lua/suspendmanager.lua b/plugins/lua/suspendmanager.lua
index c15be19402..a7de42e5a0 100644
--- a/plugins/lua/suspendmanager.lua
+++ b/plugins/lua/suspendmanager.lua
@@ -26,6 +26,25 @@ function isBuildingPlanJob(job)
     return suspendmanager_isBuildingPlanJob(job)
 end
 
+--- Return the selected construction job
+local function getSelectedBuildingJob()
+    -- This is not relying on dfhack.gui.getSelectedJob() because we don't want
+    -- the job of a selected or followed unit, only of a selected building
+    local building = dfhack.gui.getSelectedBuilding(true)
+    if not building then
+        return nil
+    end
+
+    -- Find if the building is being constructed
+    for _, job in ipairs(building.jobs) do
+        if job.job_type == df.job_type.ConstructBuilding then
+            return job
+        end
+    end
+
+    return nil
+end
+
 function runOnce(prevent_blocking, quiet, unsuspend_everything)
     suspendmanager_runOnce(prevent_blocking, unsuspend_everything)
     if (not quiet) then
@@ -69,7 +88,7 @@ function StatusOverlay:init()
 end
 
 function StatusOverlay:get_status_string()
-    local job = dfhack.gui.getSelectedJob(true)
+    local job = getSelectedBuildingJob()
     if job and job.flags.suspend then
         return "Suspended because: " .. suspendmanager_suspensionDescription(job) .. "."
     end
@@ -77,8 +96,11 @@ function StatusOverlay:get_status_string()
 end
 
 function StatusOverlay:render(dc)
-    local job = dfhack.gui.getSelectedJob(true)
-    if not job or job.job_type ~= df.job_type.ConstructBuilding or not isEnabled() or isBuildingPlanJob(job) then
+    if not isEnabled() then
+        return
+    end
+    local job = getSelectedBuildingJob()
+    if not job or isBuildingPlanJob(job) then
         return
     end
     StatusOverlay.super.render(self, dc)
@@ -110,8 +132,8 @@ function ToggleOverlay:init()
 end
 
 function ToggleOverlay:shouldRender()
-    local job = dfhack.gui.getSelectedJob(true)
-    return job and job.job_type == df.job_type.ConstructBuilding and not isBuildingPlanJob(job)
+    local job = getSelectedBuildingJob()
+    return job and not isBuildingPlanJob(job)
 end
 
 function ToggleOverlay:render(dc)