From 2a52011492e1dc7b237bba7ed6d755476d21b12d Mon Sep 17 00:00:00 2001
From: Myk Taylor <myk002@yahoo.com>
Date: Tue, 14 Nov 2023 09:28:57 -0800
Subject: [PATCH] label the mechanisms appropriately when linking a plate

only call it a lever when it's a lever
---
 plugins/lua/buildingplan/mechanisms.lua | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/plugins/lua/buildingplan/mechanisms.lua b/plugins/lua/buildingplan/mechanisms.lua
index 163255ca3..e6c482ae2 100644
--- a/plugins/lua/buildingplan/mechanisms.lua
+++ b/plugins/lua/buildingplan/mechanisms.lua
@@ -18,6 +18,16 @@ MechanismOverlay.ATTRS{
     frame={w=57, h=13},
 }
 
+local function get_label()
+    local bld = dfhack.gui.getSelectedBuilding(true)
+    if bld and bld:getType() == df.building_type.Trap and
+        bld:getSubtype() == df.trap_type.PressurePlate
+    then
+        return 'Plate:'
+    end
+    return 'Lever:'
+end
+
 function MechanismOverlay:init()
     self:addviews{
         widgets.BannerPanel{
@@ -31,7 +41,7 @@ function MechanismOverlay:init()
                     view_id='safety_lever',
                     frame={t=0, l=20, w=15},
                     key='CUSTOM_G',
-                    label='Lever:',
+                    label=get_label,
                     options={
                         {label='Any', value=0},
                         {label='Magma', value=2, pen=COLOR_RED},