1 files changed, 180 insertions, 0 deletions
diff --git a/lib/mods/theme/scpt/stores.lua b/lib/mods/theme/scpt/stores.lua
new file mode 100644
index 00000000..a504effb
--- /dev/null
+++ b/lib/mods/theme/scpt/stores.lua
@@ -0,0 +1,180 @@
+-- Whats shops can buy what
+store_buy_list
+{
+	["General Store"] =
+	{
+		TV_CORPSE,
+		TV_FOOD,
+		TV_LITE,
+		TV_FLASK,
+		TV_SPIKE,
+		TV_SHOT,
+		TV_ARROW,
+		TV_BOLT,
+		TV_DIGGING,
+		TV_CLOAK,
+		TV_BOTTLE,
+	},
+	["Armoury"] =
+	{
+		TV_BOOTS,
+		TV_GLOVES,
+		TV_CROWN,
+		TV_HELM,
+		TV_SHIELD,
+		TV_CLOAK,
+		TV_SOFT_ARMOR,
+		TV_HARD_ARMOR,
+		TV_DRAG_ARMOR,
+	},
+	["Weaponsmith"] =
+	{
+		TV_SHOT,
+		TV_BOLT,
+		TV_ARROW,
+		TV_BOOMERANG,
+		TV_BOW,
+		TV_DIGGING,
+		TV_HAFTED,
+		TV_POLEARM,
+		TV_SWORD,
+		TV_AXE,
+		TV_MSTAFF,
+	},
+	-- We use a function because we want to restrict to blessed weapons and god spells
+	["Temple"] = function (obj)
+		if obj.tval == TV_DRUID_BOOK then return TRUE
+		elseif obj.tval == TV_BOOK and obj.sval == 255 and (can_spell_random(obj.pval) == SKILL_SPIRITUALITY) then return TRUE
+		elseif obj.tval == TV_SCROLL then return TRUE
+		elseif obj.tval == TV_POTION2 then return TRUE
+		elseif obj.tval == TV_POTION then return TRUE
+		elseif obj.tval == TV_HAFTED then return TRUE
+		elseif obj.tval == TV_POLEARM and is_blessed(obj) == TRUE then return TRUE
+		elseif obj.tval == TV_SWORD and is_blessed(obj) == TRUE then return TRUE
+		elseif obj.tval == TV_AXE and is_blessed(obj) == TRUE then return TRUE
+		elseif obj.tval == TV_BOOMERANG and is_blessed(obj) == TRUE then return TRUE
+		end
+	end,
+	["Alchemy shop"] =
+	{
+		TV_SCROLL,
+		TV_POTION2,
+		TV_POTION,
+		TV_BATERIE,
+		TV_BOTTLE,
+	},
+	-- We use a function because we dont want god spells
+	["Magic shop"] = function (obj)
+		local buy =
+		{
+			[TV_SYMBIOTIC_BOOK] = TRUE,
+			[TV_AMULET] = TRUE,
+			[TV_RING] = TRUE,
+			[TV_STAFF] = TRUE,
+			[TV_WAND] = TRUE,
+			[TV_ROD] = TRUE,
+			[TV_ROD_MAIN] = TRUE,
+			[TV_SCROLL] = TRUE,
+			[TV_POTION2] = TRUE,
+			[TV_POTION] = TRUE,
+			[TV_MSTAFF] = TRUE,
+			[TV_RANDART] = TRUE,
+		}
+
+		if obj.tval == TV_BOOK and obj.sval == 255 and (can_spell_random(obj.pval) == SKILL_MAGIC) then return TRUE
+		elseif obj.tval == TV_BOOK and obj.sval ~= 255 then return TRUE
+		elseif buy[obj.tval] == TRUE then return TRUE
+		end
+	end,
+	-- Black markets wants ALL!
+	["Black Market"] = function (obj)
+		return TRUE
+	end,
+	["Book Store"] =
+	{
+		TV_BOOK,
+		TV_SYMBIOTIC_BOOK,
+		TV_MUSIC_BOOK,
+		TV_DAEMON_BOOK,
+		TV_DRUID_BOOK,
+	},
+	["Pet Shop"] =
+	{
+		TV_EGG,
+	},
+-- Theme stores
+
+	["Hunting Supply Store"] =
+	{
+		TV_TRAPKIT,
+		TV_BOOMERANG,
+		TV_SHOT,
+		TV_BOLT,
+		TV_ARROW,
+		TV_BOW,
+		TV_POTION2,
+	},
+
+	["Runic Magic Shop"] = 
+	{
+		TV_RUNE1,
+		TV_RUNE2,
+	},
+
+	["Construction Supply Store"] = 
+	{
+		TV_LITE,
+		TV_DIGGING,
+	},
+
+	["Music Store"] = 
+	{
+		TV_INSTRUMENT,
+	},
+}
+
+-- Test only
+function out_sticks()
+	local i
+	for i = 0, __tmp_spells_num - 1 do
+		if __tmp_spells[i].stick then
+			if __tmp_spells[i].stick[TV_WAND] then
+				print("Wand: " .. __tmp_spells[i].name)
+			end
+		end
+	end
+	for i = 0, __tmp_spells_num - 1 do
+		if __tmp_spells[i].stick then
+			if __tmp_spells[i].stick[TV_STAFF] then
+				print("Staff: " .. __tmp_spells[i].name)
+			end
+		end
+	end
+end
+
+-- Take care to have Magic shop/Temple have specific spells only
+add_hooks
+{
+	[HOOK_STORE_STOCK] = function (index, name, level)
+			if name == "Magic shop" then
+				-- Books
+				if magik(20) == TRUE then
+					object_prep(obj_forge, lookup_kind(TV_BOOK, 255))
+					local spell = get_random_spell(SKILL_MAGIC, 20)
+					if spell > -1 then
+						obj_forge.pval = spell
+						return TRUE, obj_forge
+					end
+				end
+			elseif name == "Temple" then
+				if magik(20) == TRUE then
+					object_prep(obj_forge, lookup_kind(TV_BOOK, 255))
+					local spell = get_random_spell(SKILL_SPIRITUALITY, 20)
+					if spell > -1 then
+						obj_forge.pval = spell
+						return TRUE, obj_forge
+					end
+				end
+			end
+	end,
+}