Add nil checks for placer
[oweals/minetest_game.git] / mods / screwdriver / init.lua
1 screwdriver = {}
2
3 screwdriver.ROTATE_FACE = 1
4 screwdriver.ROTATE_AXIS = 2
5 screwdriver.disallow = function(pos, node, user, mode, new_param2)
6         return false
7 end
8 screwdriver.rotate_simple = function(pos, node, user, mode, new_param2)
9         if mode ~= screwdriver.ROTATE_FACE then
10                 return false
11         end
12 end
13
14 -- For attached wallmounted nodes: returns true if rotation is valid
15 -- simplified version of minetest:builtin/game/falling.lua#L148.
16 local function check_attached_node(pos, rotation)
17         local d = minetest.wallmounted_to_dir(rotation)
18         local p2 = vector.add(pos, d)
19         local n = minetest.get_node(p2).name
20         local def2 = minetest.registered_nodes[n]
21         if def2 and not def2.walkable then
22                 return false
23         end
24         return true
25 end
26
27 screwdriver.rotate = {}
28
29 local facedir_tbl = {
30         [screwdriver.ROTATE_FACE] = {
31                 [0] = 1, [1] = 2, [2] = 3, [3] = 0,
32                 [4] = 5, [5] = 6, [6] = 7, [7] = 4,
33                 [8] = 9, [9] = 10, [10] = 11, [11] = 8,
34                 [12] = 13, [13] = 14, [14] = 15, [15] = 12,
35                 [16] = 17, [17] = 18, [18] = 19, [19] = 16,
36                 [20] = 21, [21] = 22, [22] = 23, [23] = 20,
37         },
38         [screwdriver.ROTATE_AXIS] = {
39                 [0] = 4, [1] = 4, [2] = 4, [3] = 4,
40                 [4] = 8, [5] = 8, [6] = 8, [7] = 8,
41                 [8] = 12, [9] = 12, [10] = 12, [11] = 12,
42                 [12] = 16, [13] = 16, [14] = 16, [15] = 16,
43                 [16] = 20, [17] = 20, [18] = 20, [19] = 20,
44                 [20] = 0, [21] = 0, [22] = 0, [23] = 0,
45         },
46 }
47
48 screwdriver.rotate.facedir = function(pos, node, mode)
49         local rotation = node.param2 % 32 -- get first 5 bits
50         local other = node.param2 - rotation
51         rotation = facedir_tbl[mode][rotation] or 0
52         return rotation + other
53 end
54
55 screwdriver.rotate.colorfacedir = screwdriver.rotate.facedir
56
57 local wallmounted_tbl = {
58         [screwdriver.ROTATE_FACE] = {[2] = 5, [3] = 4, [4] = 2, [5] = 3, [1] = 0, [0] = 1},
59         [screwdriver.ROTATE_AXIS] = {[2] = 5, [3] = 4, [4] = 2, [5] = 1, [1] = 0, [0] = 3}
60 }
61
62 screwdriver.rotate.wallmounted = function(pos, node, mode)
63         local rotation = node.param2 % 8 -- get first 3 bits
64         local other = node.param2 - rotation
65         rotation = wallmounted_tbl[mode][rotation] or 0
66         if minetest.get_item_group(node.name, "attached_node") ~= 0 then
67                 -- find an acceptable orientation
68                 for i = 1, 5 do
69                         if not check_attached_node(pos, rotation) then
70                                 rotation = wallmounted_tbl[mode][rotation] or 0
71                         else
72                                 break
73                         end
74                 end
75         end
76         return rotation + other
77 end
78
79 screwdriver.rotate.colorwallmounted = screwdriver.rotate.wallmounted
80
81 -- Handles rotation
82 screwdriver.handler = function(itemstack, user, pointed_thing, mode, uses)
83         if pointed_thing.type ~= "node" then
84                 return
85         end
86
87         local pos = pointed_thing.under
88         local player_name = user and user:get_player_name() or ""
89
90         if minetest.is_protected(pos, player_name) then
91                 minetest.record_protection_violation(pos, player_name)
92                 return
93         end
94
95         local node = minetest.get_node(pos)
96         local ndef = minetest.registered_nodes[node.name]
97         if not ndef then
98                 return itemstack
99         end
100         -- can we rotate this paramtype2?
101         local fn = screwdriver.rotate[ndef.paramtype2]
102         if not fn and not ndef.on_rotate then
103                 return itemstack
104         end
105
106         local should_rotate = true
107         local new_param2
108         if fn then
109                 new_param2 = fn(pos, node, mode)
110         else
111                 new_param2 = node.param2
112         end
113
114         -- Node provides a handler, so let the handler decide instead if the node can be rotated
115         if ndef.on_rotate then
116                 -- Copy pos and node because callback can modify it
117                 local result = ndef.on_rotate(vector.new(pos),
118                                 {name = node.name, param1 = node.param1, param2 = node.param2},
119                                 user, mode, new_param2)
120                 if result == false then -- Disallow rotation
121                         return itemstack
122                 elseif result == true then
123                         should_rotate = false
124                 end
125         elseif ndef.on_rotate == false then
126                 return itemstack
127         elseif ndef.can_dig and not ndef.can_dig(pos, user) then
128                 return itemstack
129         end
130
131         if should_rotate and new_param2 ~= node.param2 then
132                 node.param2 = new_param2
133                 minetest.swap_node(pos, node)
134                 minetest.check_for_falling(pos)
135         end
136
137         if not (creative and creative.is_enabled_for and
138                         creative.is_enabled_for(player_name)) then
139                 itemstack:add_wear(65535 / ((uses or 200) - 1))
140         end
141
142         return itemstack
143 end
144
145 -- Screwdriver
146 minetest.register_tool("screwdriver:screwdriver", {
147         description = "Screwdriver (left-click rotates face, right-click rotates axis)",
148         inventory_image = "screwdriver.png",
149         on_use = function(itemstack, user, pointed_thing)
150                 screwdriver.handler(itemstack, user, pointed_thing, screwdriver.ROTATE_FACE, 200)
151                 return itemstack
152         end,
153         on_place = function(itemstack, user, pointed_thing)
154                 screwdriver.handler(itemstack, user, pointed_thing, screwdriver.ROTATE_AXIS, 200)
155                 return itemstack
156         end,
157 })
158
159
160 minetest.register_craft({
161         output = "screwdriver:screwdriver",
162         recipe = {
163                 {"default:steel_ingot"},
164                 {"group:stick"}
165         }
166 })
167
168 minetest.register_alias("screwdriver:screwdriver1", "screwdriver:screwdriver")
169 minetest.register_alias("screwdriver:screwdriver2", "screwdriver:screwdriver")
170 minetest.register_alias("screwdriver:screwdriver3", "screwdriver:screwdriver")
171 minetest.register_alias("screwdriver:screwdriver4", "screwdriver:screwdriver")