1 files changed, 29 insertions, 33 deletions
diff --git a/mesecons_walllever/init.lua b/mesecons_walllever/init.lua
index 4e3401e..6f1c310 100644
--- a/mesecons_walllever/init.lua
+++ b/mesecons_walllever/init.lua
@@ -1,42 +1,38 @@
 -- WALL LEVER
 minetest.register_node("mesecons_walllever:wall_lever_off", {
-    drawtype = "signlike",
-    tile_images = {"jeija_wall_lever_off.png"},
-    inventory_image = "jeija_wall_lever_off.png",
-    wield_image = "jeija_wall_lever_off.png",
-    paramtype = "light",
-    paramtype2 = "wallmounted",
-    legacy_wallmounted = true,
-    walkable = false,
-    selection_box = {
-        type = "wallmounted",
-    },
-    groups = {dig_immediate=2},
-    description="Lever",
+	drawtype = "signlike",
+	tile_images = {"jeija_wall_lever_off.png"},
+	inventory_image = "jeija_wall_lever_off.png",
+	wield_image = "jeija_wall_lever_off.png",
+	paramtype = "light",
+	paramtype2 = "wallmounted",
+	legacy_wallmounted = true,
+	walkable = false,
+	selection_box = {
+		type = "wallmounted",
+	},
+	groups = {dig_immediate=2},
+	description="Lever",
 })
 minetest.register_node("mesecons_walllever:wall_lever_on", {
-    drawtype = "signlike",
-    tile_images = {"jeija_wall_lever_on.png"},
-    inventory_image = "jeija_wall_lever_on.png",
-    paramtype = "light",
-    paramtype2 = "wallmounted",
-    legacy_wallmounted = true,
-    walkable = false,
-    selection_box = {
-        type = "wallmounted",
-    },
-    groups = {dig_immediate=2},
-    drop = '"mesecons_walllever:wall_lever_off" 1',
-    description="Lever",
+	drawtype = "signlike",
+	tile_images = {"jeija_wall_lever_on.png"},
+	inventory_image = "jeija_wall_lever_on.png",
+	paramtype = "light",
+	paramtype2 = "wallmounted",
+	legacy_wallmounted = true,
+	walkable = false,
+	selection_box = {
+		type = "wallmounted",
+	},
+	groups = {dig_immediate=2},
+	drop = '"mesecons_walllever:wall_lever_off" 1',
+	description="Lever",
+	after_dig_node = function(pos)
+		mesecon:receptor_off(pos, mesecon.button_get_rules(minetest.env:get_node(pos).param2))
+	end
 })
 
-minetest.register_on_dignode(
-    function(pos, oldnode, digger)
-        if oldnode.name == "mesecons_walllever:wall_lever_on" then
-            mesecon:receptor_off(pos, mesecon.button_get_rules(oldnode.param2))
-        end    
-    end
-)
 minetest.register_on_punchnode(function(pos, node, puncher)
 	if node.name == "mesecons_walllever:wall_lever_off" then
 		minetest.env:add_node(pos, {name="mesecons_walllever:wall_lever_on",param2=node.param2})