FilesExpand file tree

 develop

/

eventful.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

169 lines (163 loc) · 7.02 KB

 develop

/

eventful.lua

Copy path

Top

File metadata and controls

• Code
• Blame

169 lines (163 loc) · 7.02 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

local _ENV = mkmodule('plugins.eventful')

--[[

--]]

local function getShopName(btype,bsubtype,bcustom)

local typenames_shop={[df.workshop_type.Carpenters]="CARPENTERS",[df.workshop_type.Farmers]="FARMERS",

[df.workshop_type.Masons]="MASONS",[df.workshop_type.Craftsdwarfs]="CRAFTSDWARFS",

[df.workshop_type.Jewelers]="JEWELERS",[df.workshop_type.MetalsmithsForge]="METALSMITHSFORGE",

[df.workshop_type.MagmaForge]="MAGMAFORGE",[df.workshop_type.Bowyers]="BOWYERS",

[df.workshop_type.Mechanics]="MECHANICS",[df.workshop_type.Siege]="SIEGE",

[df.workshop_type.Butchers]="BUTCHERS",[df.workshop_type.Leatherworks]="LEATHERWORKS",

[df.workshop_type.Tanners]="TANNERS",[df.workshop_type.Clothiers]="CLOTHIERS",

[df.workshop_type.Fishery]="FISHERY",[df.workshop_type.Still]="STILL",

[df.workshop_type.Loom]="LOOM",[df.workshop_type.Quern]="QUERN",

[df.workshop_type.Kennels]="KENNELS",[df.workshop_type.Ashery]="ASHERY",

[df.workshop_type.Kitchen]="KITCHEN",[df.workshop_type.Dyers]="DYERS",

[df.workshop_type.Tool]="TOOL",[df.workshop_type.Millstone]="MILLSTONE",

}

local typenames_furnace={[df.furnace_type.WoodFurnace]="WOOD_FURNACE",[df.furnace_type.Smelter]="SMELTER",

[df.furnace_type.GlassFurnace]="GLASS_FURNACE",[df.furnace_type.MagmaSmelter]="MAGMA_SMELTER",

[df.furnace_type.MagmaGlassFurnace]="MAGMA_GLASS_FURNACE",[df.furnace_type.MagmaKiln]="MAGMA_KILN",

[df.furnace_type.Kiln]="KILN"}

if btype==df.building_type.Workshop then

if typenames_shop[bsubtype]~=nil then

return typenames_shop[bsubtype]

else

return df.building_def_workshopst.find(bcustom).code

end

elseif btype==df.building_type.Furnace then

if typenames_furnace[bsubtype]~=nil then

return typenames_furnace[bsubtype]

else

return df.building_def_furnacest.find(bcustom).code

end

end

end

_registeredStuff={}

local function unregall(state)

if state==SC_WORLD_UNLOADED then

onReactionComplete._library=nil

postWorkshopFillSidebarMenu._library=nil

dfhack.onStateChange.eventful= nil

_registeredStuff={}

end

end

local function onReact(reaction,reaction_product,unit,input_items,input_reagents,output_items,call_native)

if _registeredStuff.reactionCallbacks and _registeredStuff.reactionCallbacks[reaction.code] then

_registeredStuff.reactionCallbacks[reaction.code](reaction,reaction_product,unit,input_items,input_reagents,output_items,call_native)

end

end

local function onPostSidebar(workshop)

local shop_id=getShopName(workshop:getType(),workshop:getSubtype(),workshop:getCustomType())

if shop_id then

if _registeredStuff.shopNonNative and _registeredStuff.shopNonNative[shop_id] then

if _registeredStuff.shopNonNative[shop_id].all then

--[[for _,button in ipairs(df.global.ui_sidebar_menus.workshop_job.choices_all) do

button.is_hidden=true

end]]

df.global.ui_sidebar_menus.workshop_job.choices_visible:resize(0)

else

--todo by name

end

end

if _registeredStuff.reactionToShop and _registeredStuff.reactionToShop[shop_id] then

for _,reaction_name in ipairs(_registeredStuff.reactionToShop[shop_id]) do

local new_button=df.interface_button_building_new_jobst:new()

--new_button.hotkey_id=--todo get hotkey

new_button.is_hidden=false

new_button.building=workshop

new_button.job_type=df.job_type.CustomReaction --could be used for other stuff too i guess...

new_button.reaction_name=reaction_name

new_button.is_custom=true

local wjob=df.global.ui_sidebar_menus.workshop_job

wjob.choices_all:insert("#",new_button)

wjob.choices_visible:insert("#",new_button)

end

end

if _registeredStuff.customSidebar and _registeredStuff.customSidebar[shop_id] then

_registeredStuff.customSidebar[shop_id](workshop)

end

end

end

local function customSidebarsCallback(workshop)

local shop_id=getShopName(workshop:getType(),workshop:getSubtype(),workshop:getCustomType())

if shop_id then

if _registeredStuff.customSidebar and _registeredStuff.customSidebar[shop_id] then

_registeredStuff.customSidebar[shop_id](workshop)

end

end

end

function registerReaction(reaction_name,callback)

_registeredStuff.reactionCallbacks=_registeredStuff.reactionCallbacks or {}

_registeredStuff.reactionCallbacks[reaction_name]=callback

onReactionComplete._library=onReact

dfhack.onStateChange.eventful=unregall

end

function registerSidebar(shop_name,callback)

if type(callback)=="function" then

_registeredStuff.customSidebar=_registeredStuff.customSidebar or {}

_registeredStuff.customSidebar[shop_name]=callback

onWorkshopFillSidebarMenu._library=customSidebarsCallback

dfhack.onStateChange.eventful=unregall

else

local function drawSidebar( wshop )

local valid_focus="dwarfmode/QueryBuilding"

local another_overlay="dfhack/lua/WorkshopOverlay"

if wshop:getMaxBuildStage()==wshop:getBuildStage() then

local sidebar=callback{workshop=wshop}

if string.sub(dfhack.gui.getCurFocus(true),1,#another_overlay)==another_overlay then

dfhack.screen.dismiss(dfhack.gui.getCurViewscreen(true))

end

if string.sub(dfhack.gui.getCurFocus(true),1,#valid_focus)==valid_focus then

sidebar:show(dfhack.gui.getCurViewscreen(true))

end

end

end

registerSidebar(shop_name,drawSidebar)

end

end

function removeNative(shop_name,name)

_registeredStuff.shopNonNative=_registeredStuff.shopNonNative or {}

local shops=_registeredStuff.shopNonNative

shops[shop_name]=shops[shop_name] or {}

if name~=nil then

table.insert(shops[shop_name],name)

else

shops[shop_name].all=true

end

postWorkshopFillSidebarMenu._library=onPostSidebar

dfhack.onStateChange.eventful=unregall

end

function addReactionToShop(reaction_name,shop_name)

_registeredStuff.reactionToShop=_registeredStuff.reactionToShop or {}

local shops=_registeredStuff.reactionToShop

shops[shop_name]=shops[shop_name] or {}

table.insert(shops[shop_name],reaction_name)

postWorkshopFillSidebarMenu._library=onPostSidebar

dfhack.onStateChange.eventful=unregall

end

local function invertTable(tbl)

local ret={}

for k,v in pairs(tbl) do

ret[v]=k

end

return ret

end

eventType=invertTable{

[0]="TICK",

"JOB_INITIATED",

"JOB_COMPLETED",

"UNIT_DEATH",

"ITEM_CREATED",

"BUILDING",

"CONSTRUCTION",

"SYNDROME",

"INVASION",

"INVENTORY_CHANGE",

"REPORT",

"UNIT_ATTACK",

"UNLOAD",

"INTERACTION",

"EVENT_MAX"

}

return _ENV