aboutsummaryrefslogtreecommitdiff
path: root/src/nvim/generators/gen_char_blob.lua
blob: c40e0d6e82fc6d857dc316b6c85802ffee534560 (plain) (blame)
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
if arg[1] == '--help' then
  print('Usage:')
  print('  ' .. arg[0] .. ' [-c] target source varname [source varname]...')
  print('')
  print('Generates C file with big uint8_t blob.')
  print('Blob will be stored in a static const array named varname.')
  os.exit()
end

-- Recognized options:
--   -c   compile Lua bytecode
local options = {}

while true do
  local opt = string.match(arg[1], '^-(%w)')
  if not opt then
    break
  end

  options[opt] = true
  table.remove(arg, 1)
end

assert(#arg >= 3 and (#arg - 1) % 2 == 0)

local target_file = arg[1] or error('Need a target file')
local target = io.open(target_file, 'w')

target:write('#include <stdint.h>\n\n')

local index_items = {}

local warn_on_missing_compiler = true
local modnames = {}
for argi = 2, #arg, 2 do
  local source_file = arg[argi]
  local modname = arg[argi + 1]
  if modnames[modname] then
    error(string.format('modname %q is already specified for file %q', modname, modnames[modname]))
  end
  modnames[modname] = source_file

  local varname = string.gsub(modname, '%.', '_dot_') .. '_module'
  target:write(('static const uint8_t %s[] = {\n'):format(varname))

  local output
  if options.c then
    local luac = os.getenv('LUAC_PRG')
    if luac and luac ~= '' then
      output = io.popen(luac:format(source_file), 'r'):read('*a')
    elseif warn_on_missing_compiler then
      print('LUAC_PRG is missing, embedding raw source')
      warn_on_missing_compiler = false
    end
  end

  if not output then
    local source = io.open(source_file, 'r')
      or error(string.format("source_file %q doesn't exist", source_file))
    output = source:read('*a')
    source:close()
  end

  local num_bytes = 0
  local MAX_NUM_BYTES = 15 -- 78 / 5: maximum number of bytes on one line
  target:write(' ')

  local increase_num_bytes
  increase_num_bytes = function()
    num_bytes = num_bytes + 1
    if num_bytes == MAX_NUM_BYTES then
      num_bytes = 0
      target:write('\n ')
    end
  end

  for i = 1, string.len(output) do
    local byte = output:byte(i)
    target:write(string.format(' %3u,', byte))
    increase_num_bytes()
  end

  target:write('  0};\n')
  if modname ~= '_' then
    table.insert(
      index_items,
      '  { "' .. modname .. '", ' .. varname .. ', sizeof ' .. varname .. ' },\n\n'
    )
  end
end

target:write('static ModuleDef builtin_modules[] = {\n')
target:write(table.concat(index_items))
target:write('};\n')

target:close()