FilesExpand file tree

 develop

/

script.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

164 lines (138 loc) · 3.8 KB

 develop

/

script.lua

Copy path

Top

File metadata and controls

• Code
• Blame

164 lines (138 loc) · 3.8 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

-- Support for scripted interaction sequences via coroutines.

local _ENV = mkmodule('gui.script')

local dlg = require('gui.dialogs')

--[[

Example:

start(function()

sleep(100, 'frames')

print(showYesNoPrompt('test', 'print true?'))

end)

]]

-- Table of running background scripts.

if not scripts then

scripts = {}

setmetatable(scripts, { __mode = 'k' })

end

local function do_resume(inst, ...)

inst.gen = inst.gen + 1

return (dfhack.saferesume(inst.coro, ...))

end

-- Starts a new background script by calling the function.

function start(fn,...)

local coro = coroutine.create(fn)

local inst = {

coro = coro,

gen = 0,

}

scripts[coro] = inst

return do_resume(inst, ...)

end

-- Checks if called from a background script

function in_script()

return scripts[coroutine.running()] ~= nil

end

local function getinst()

local inst = scripts[coroutine.running()]

if not inst then

error('Not in a gui script coroutine.')

end

return inst

end

local function invoke_resume(inst,gen,quiet,...)

local state = coroutine.status(inst.coro)

if state ~= 'suspended' then

if state ~= 'dead' then

dfhack.printerr(debug.traceback('resuming a non-waiting continuation'))

end

elseif inst.gen > gen then

if not quiet then

dfhack.printerr(debug.traceback('resuming an expired continuation'))

end

else

do_resume(inst, ...)

end

end

-- Returns a callback that resumes the script from wait with given return values

function mkresume(...)

local inst = getinst()

return curry(invoke_resume, inst, inst.gen, false, ...)

end

-- Like mkresume, but does not complain if already resumed from this wait

function qresume(...)

local inst = getinst()

return curry(invoke_resume, inst, inst.gen, true, ...)

end

-- Wait until a mkresume callback is called, then return its arguments.

-- Once it returns, all mkresume callbacks created before are invalidated.

function wait()

getinst() -- check that the context is right

return coroutine.yield()

end

-- Wraps dfhack.timeout for coroutines.

function sleep(time, quantity)

if dfhack.timeout(time, quantity, mkresume()) then

wait()

return true

else

return false

end

end

-- Some dialog wrappers:

function showMessage(title, text, tcolor)

dlg.MessageBox{

frame_title = title,

text = text,

text_pen = tcolor,

on_close = qresume(nil)

}:show()

return wait()

end

function showYesNoPrompt(title, text, tcolor)

dlg.MessageBox{

frame_title = title,

text = text,

text_pen = tcolor,

on_accept = mkresume(true),

on_cancel = mkresume(false),

on_close = qresume(nil)

}:show()

return wait()

end

function showInputPrompt(title, text, tcolor, input, min_width)

dlg.InputBox{

frame_title = title,

text = text,

text_pen = tcolor,

input = input,

frame_width = min_width,

on_input = mkresume(true),

on_cancel = mkresume(false),

on_close = qresume(nil)

}:show()

return wait()

end

function showListPrompt(title, text, tcolor, choices, min_width, filter)

dlg.ListBox{

frame_title = title,

text = text,

text_pen = tcolor,

choices = choices,

frame_width = min_width,

with_filter = filter,

on_select = mkresume(true),

on_cancel = mkresume(false),

on_close = qresume(nil)

}:show()

return wait()

end

function showMaterialPrompt(title, prompt)

require('gui.materials').MaterialDialog{

frame_title = title,

prompt = prompt,

on_select = mkresume(true),

on_cancel = mkresume(false),

on_close = qresume(nil)

}:show()

return wait()

end

return _ENV