summaryrefslogtreecommitdiff
path: root/lua/clever_f/motion_plan.lua
blob: 362b97814df49982d6a263b31f74adfa57581c6b (plain)
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
local domain = require("clever_f.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_f.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

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

M.create = M.build
M.resolve = M.build

setmetatable(M, {
  __call = function(_, options)
    return MotionPlanFactory.new(options)
  end,
})

return M