Fix crash when using the screwdriver on an unknown node
[oweals/minetest_game.git] / mods / screwdriver / init.lua
1
2 local mode_text = {
3         {"Change rotation, Don't change axisdir."},
4         {"Keep choosen face in front then rotate it."},
5         {"Change axis dir, Reset rotation."},
6         {"Bring top in front then rotate it."},
7 }
8
9 local opposite_faces = {
10         [0] = 5,
11         [1] = 2,
12         [2] = 1,
13         [3] = 4,
14         [4] = 3,
15         [5] = 0,
16 }
17
18 local function screwdriver_setmode(user,itemstack)
19         local player_name = user:get_player_name()
20         local item = itemstack:to_table()
21         local mode = tonumber(itemstack:get_metadata())
22         if not mode then
23                 minetest.chat_send_player(player_name, "Hold shift and use to change screwdriwer modes.")
24                 mode = 0
25         end
26         mode = mode + 1
27         if mode == 5 then
28                 mode = 1
29         end
30         minetest.chat_send_player(player_name, "Screwdriver mode : "..mode.." - "..mode_text[mode][1] )
31         itemstack:set_name("screwdriver:screwdriver"..mode)
32         itemstack:set_metadata(mode)
33         return itemstack
34 end
35
36 local function get_node_face(pointed_thing)
37         local ax, ay, az = pointed_thing.above.x, pointed_thing.above.y, pointed_thing.above.z
38         local ux, uy, uz = pointed_thing.under.x, pointed_thing.under.y, pointed_thing.under.z
39         if     ay > uy then return 0 -- Top
40         elseif az > uz then return 1 -- Z+ side
41         elseif az < uz then return 2 -- Z- side
42         elseif ax > ux then return 3 -- X+ side
43         elseif ax < ux then return 4 -- X- side
44         elseif ay < uy then return 5 -- Bottom
45         else
46                 error("pointed_thing.above and under are the same!")
47         end
48 end
49
50 local function nextrange(x, max)
51         x = x + 1
52         if x > max then
53                 x = 0
54         end
55         return x
56 end
57
58 local function screwdriver_handler(itemstack, user, pointed_thing)
59         if pointed_thing.type ~= "node" then
60                 return
61         end
62         local pos = pointed_thing.under
63         local keys = user:get_player_control()
64         local player_name = user:get_player_name()
65         local mode = tonumber(itemstack:get_metadata())
66         if not mode or keys["sneak"] == true then
67                 return screwdriver_setmode(user, itemstack)
68         end
69         if minetest.is_protected(pos, user:get_player_name()) then
70                 minetest.record_protection_violation(pos, user:get_player_name())
71                 return
72         end
73         local node = minetest.get_node(pos)
74         local ndef = minetest.registered_nodes[node.name]
75         if ndef and ndef.paramtype2 == "facedir" then
76                 if ndef.drawtype == "nodebox" and ndef.node_box.type ~= "fixed" then
77                         return
78                 end
79                 if node.param2 == nil then
80                         return
81                 end
82                 -- Get ready to set the param2
83                 local n = node.param2
84                 local axisdir = math.floor(n / 4)
85                 local rotation = n - axisdir * 4
86                 if mode == 1 then
87                         n = axisdir * 4 + nextrange(rotation, 3)
88                 elseif mode == 2 then
89                         -- If you are pointing at the axisdir face or the
90                         -- opposite one then you can just rotate the node.
91                         -- Otherwise change the axisdir, avoiding the facing
92                         -- and opposite axes.
93                         local face = get_node_face(pointed_thing)
94                         if axisdir == face or axisdir == opposite_faces[face] then
95                                 n = axisdir * 4 + nextrange(rotation, 3)
96                         else
97                                 axisdir = nextrange(axisdir, 5)
98                                 -- This is repeated because switching from the face
99                                 -- can move to to the opposite and vice-versa
100                                 if axisdir == face or axisdir == opposite_faces[face] then
101                                         axisdir = nextrange(axisdir, 5)
102                                 end
103                                 if axisdir == face or axisdir == opposite_faces[face] then
104                                         axisdir = nextrange(axisdir, 5)
105                                 end
106                                 n = axisdir * 4
107                         end
108                 elseif mode == 3 then
109                         n = nextrange(axisdir, 5) * 4
110                 elseif mode == 4 then
111                         local face = get_node_face(pointed_thing)
112                         if axisdir == face then
113                                 n = axisdir * 4 + nextrange(rotation, 3)
114                         else
115                                 n = face * 4
116                         end
117                 end
118                 --print (dump(axisdir..", "..rotation))
119                 node.param2 = n
120                 minetest.swap_node(pos, node)
121                 local item_wear = tonumber(itemstack:get_wear())
122                 item_wear = item_wear + 327
123                 if item_wear > 65535 then
124                         itemstack:clear()
125                         return itemstack
126                 end
127                 itemstack:set_wear(item_wear)
128                 return itemstack
129         end
130 end
131
132 minetest.register_craft({
133         output = "screwdriver:screwdriver",
134         recipe = {
135                 {"default:steel_ingot"},
136                 {"group:stick"}
137         }
138 })
139
140 minetest.register_tool("screwdriver:screwdriver", {
141         description = "Screwdriver",
142         inventory_image = "screwdriver.png",
143         on_use = function(itemstack, user, pointed_thing)
144                 screwdriver_handler(itemstack, user, pointed_thing)
145                 return itemstack
146         end,
147 })
148
149 for i = 1, 4 do
150         minetest.register_tool("screwdriver:screwdriver"..i, {
151                 description = "Screwdriver in Mode "..i,
152                 inventory_image = "screwdriver.png^tool_mode"..i..".png",
153                 wield_image = "screwdriver.png",
154                 groups = {not_in_creative_inventory=1},
155                 on_use = function(itemstack, user, pointed_thing)
156                         screwdriver_handler(itemstack, user, pointed_thing)
157                         return itemstack
158                 end,
159         })
160 end
161