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
|
--------------------------------------------------------------------------------
-- Module table
local M = {}
local lpeg = require('lpeg')
--------------------------------------------------------------------------------
-- lpeg rules
-- Base definitions
-- local tws = lpeg.S' ' ^ 1
local tnewline = lpeg.S'\n'
-- local tlowcasedword = lpeg.R'az' ^ 1
local tdigit = lpeg.locale()['digit']
local talphanum = lpeg.locale()['alnum']
local tanyprintable = lpeg.locale()['print']
-- local tcontrol = lpeg.locale()['cntrl']
local ttabtrigger = tanyprintable ^ 1
local ttag = lpeg.Cg(lpeg.Cp(), 'selstart')
* lpeg.P'${'
* lpeg.Cg(tdigit^1, 'tag-order')
* (
(lpeg.S':' * lpeg.Cg(talphanum^1, 'default-value') * lpeg.S'}')
+ lpeg.S'}'
)
* lpeg.Cg(lpeg.Cp(), 'selend')
local tsnippetdecl = lpeg.P'snippet' * lpeg.S' ' * lpeg.Cg(ttabtrigger, 'tabtrigger') * tnewline
local tsnippetcontent = lpeg.C(
lpeg.Cp() *
(lpeg.S'\t '^1
* (lpeg.Ct(ttag) + tanyprintable)^1
* tnewline
)^1
)
-- Constructs
local tsnippet = tsnippetdecl * tsnippetcontent
local tcomment = lpeg.S'#' * tanyprintable^0 * tnewline
-- The way grammar captures:
-- Every snippet gets its own table, and every table has:
-- 'tabtrigger' - the tabtrigger
-- [1] - full content
-- [2] - start of snippet content (need to subtract from selstart/selend
-- [3..n] - tags
local tsnippetsfile = lpeg.Ct((tcomment + lpeg.Ct(tsnippet) + tnewline) ^1)
--------------------------------------------------------------------------------
-- Functions
local function trim_tabs(content)
local trim = function (s)
return (string.gsub(s, "^\t(.-)$", "%1"))
end
local ret=''
for str in string.gmatch(content, '([^\n]+)') do
ret = ret .. trim(str) .. '\n'
end
return ret
end
-- Tags are on the top level of th table,
-- defined starting with index '3'
-- Index '2' is start of the content
-- Structure:
-- { tag-order: int
-- , selstart: int
-- , selend: int
-- , default-value: str
-- }
local function extract_tags(tableau)
local tags = {}
for k,v in ipairs(tableau) do
if k >= 3 then -- Only process starting with ix 2
tags[k - 2] = { selstart = v.selstart - tableau[2] - 1
, selend = v.selend - tableau[2] - 1
, default = v['default-value']
, order = v['tag-order']
}
-- vis:message('snippet ' .. tableau.tabtrigger .. ' tag ' ..
-- tostring(tags[k - 1].order) .. ' has start/end: ' ..
-- tostring(tags[k - 1].selstart) .. '/' ..
-- tostring(tags[k - 1].selend))
end
end
return tags
end
M.load_snippets = function(snippetfile)
local snippets = {}
local f = io.open(snippetfile, 'r')
if f then
local content = f:read("*all")
-- TODO hmmm, this'll make whole file unsuable, when it could
-- in fact have usable snippets
local m = tsnippetsfile:match(content)
if not m then
vis:info('Failed to parse SnipMate file: '.. snippetfile)
return nil
else
-- k is index of snippet definition, v is table of snippet def
for _,v in pairs(m) do
snippets[v.tabtrigger] = { description = nil
, options = {}
, content = { str = trim_tabs(v[1])
, tags = extract_tags(v)
}
}
end
end
f:close()
return snippets, true
else
return snippets, false
end
end
--------------------------------------------------------------------------------
-- End module
return M
|
