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
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
local domain = require("clever_tee.domain")
local M = {}
local MotionPlanFactory = {}
M.MotionPlanFactory = MotionPlanFactory
local factory_records = setmetatable({}, { __mode = "k" })
local function fail(message, level)
error(message, (level or 1) + 1)
end
local function normalize_options(options)
if options == nil then
return {}
end
if type(options) == "table" and type(options.sample_search) == "function" then
return { policy = options }
end
if type(options) ~= "table" then
fail("MotionPlanFactory options must be a table", 3)
end
return options
end
local function require_policy(service)
if service ~= nil and (type(service) ~= "table"
or type(service.sample_search) ~= "function")
then
fail("MotionPlanFactory policy must provide sample_search", 3)
end
return service
end
local factory_metatable = {
__index = MotionPlanFactory,
__newindex = function()
fail("MotionPlanFactory values are immutable", 2)
end,
__tostring = function()
return "motion-plan-factory"
end,
__metatable = "clever_tee.motion_plan.MotionPlanFactory",
}
function MotionPlanFactory.new(options)
if MotionPlanFactory.is(options) then
return options
end
options = normalize_options(options)
local factory = setmetatable({}, factory_metatable)
factory_records[factory] = {
policy = require_policy(options.policy or options.policy_service),
}
return factory
end
function MotionPlanFactory.is(value)
return type(value) == "table" and factory_records[value] ~= nil
end
local function resolved_scope(factory, search_scope)
if search_scope == nil then
local policy = factory_records[factory].policy
if policy ~= nil then
search_scope = policy:sample_search().search_scope
else
search_scope = domain.SearchScope.BUFFER
end
elseif search_scope == "line" then
search_scope = domain.SearchScope.CURRENT_LINE
end
return domain.SearchScope.from_string(search_scope)
end
function MotionPlanFactory:build(
target_plan,
effective_descriptor,
search_scope,
endpoint_policy
)
if not domain.TargetPlan.is(target_plan) then
fail("motion target plan must be a TargetPlan", 2)
end
return domain.ResolvedMotionPlan.new({
target_plan = target_plan,
descriptor = effective_descriptor,
search_scope = resolved_scope(self, search_scope),
endpoint_policy = endpoint_policy or domain.EndpointPolicy.REGULAR,
})
end
local function selection_option(selection)
if selection == nil then
return domain.SelectionOption.INCLUSIVE
end
if domain.Selection.is(selection) then
return selection.option
end
if domain.SelectionOption.is(selection) then
return selection
end
if type(selection) == "table" and selection.option ~= nil then
return domain.SelectionOption.from_string(selection.option)
end
return domain.SelectionOption.from_string(selection)
end
function M.endpoint_policy(context, selection)
context = domain.ModeContext.from_full_mode(context)
local option = selection_option(selection)
local visual_kind = context.visual_kind
if option == domain.SelectionOption.EXCLUSIVE
and (visual_kind == domain.SelectionKind.CHARACTER
or visual_kind == domain.SelectionKind.LINE)
then
return domain.EndpointPolicy.VISUAL_EXCLUSIVE
end
return domain.EndpointPolicy.REGULAR
end
function MotionPlanFactory:endpoint_policy(context, selection)
return M.endpoint_policy(context, selection)
end
function MotionPlanFactory:build_for_context(
target_plan,
effective_descriptor,
context,
selection,
search_scope
)
return self:build(
target_plan,
effective_descriptor,
search_scope,
self:endpoint_policy(context, selection)
)
end
function M.new(options)
return MotionPlanFactory.new(options)
end
function M.build(
target_plan,
effective_descriptor,
search_scope,
endpoint_policy,
options
)
return MotionPlanFactory.new(options):build(
target_plan,
effective_descriptor,
search_scope,
endpoint_policy
)
end
function M.build_for_context(
target_plan,
effective_descriptor,
context,
selection,
search_scope,
options
)
return MotionPlanFactory.new(options):build_for_context(
target_plan,
effective_descriptor,
context,
selection,
search_scope
)
end
M.create = M.build
M.resolve = M.build
M.for_context = M.build_for_context
setmetatable(M, {
__call = function(_, options)
return MotionPlanFactory.new(options)
end,
})
return M
|
