code / nomsu

   1 require("containers")
   2 local Text = require('text')
   3 local C = require('colors')
   4 local box
   5 box = function(text)
   6   local max_line = 0
   7   for line in text:gmatch("[^\n]+") do
   8     max_line = math.max(max_line, #(line:gsub("\027%[[0-9;]*m", "")))
   9   end
  10   local ret = ("\n" .. text):gsub("\n([^\n]*)", function(line)
  11     local line_len = #(line:gsub("\027%[[0-9;]*m", ""))
  12     return "\n" .. tostring(line) .. tostring((" "):rep(max_line - line_len)) .. " " .. C('reset')
  13   end)
  14   return ret:sub(2, -1), max_line
  15 end
  16 local format_error
  17 format_error = function(err)
  18   local context = err.context or 2
  19   local err_line, err_linenum, err_linepos = err.source:line_info_at(err.start)
  20   local err_size = math.min((err.stop - err.start), (#err_line - err_linepos) + 1)
  21   local nl_indicator = (err_linepos > #err_line) and " " or ""
  22   local fmt_str = " %" .. tostring(#tostring(err_linenum + context)) .. "d|"
  23   local pointer = (" "):rep(err_linepos + #fmt_str:format(0) - 1) .. ("^"):rep(err_size)
  24   local err_msg = C('bold red', err.title or "Error") .. C('red', " at " .. tostring(err.filename or '???') .. ":" .. tostring(err_linenum) .. "," .. tostring(err_linepos))
  25   if not (COLOR_ENABLED) then
  26     err_msg = err_msg .. "\n"
  27   end
  28   local lines = err.source:lines()
  29   for i = err_linenum - context, err_linenum - 1 do
  30     do
  31       local line = lines[i]
  32       if line then
  33         err_msg = err_msg .. ("\n" .. C('dim', fmt_str:format(i)) .. line)
  34       end
  35     end
  36   end
  37   if err_line then
  38     local before = err_line:sub(1, err_linepos - 1)
  39     local during = err_line:sub(err_linepos, err_linepos + err_size - 1)
  40     local after = err_line:sub(err_linepos + err_size, -1)
  41     err_line = before .. C('black on red', during .. nl_indicator) .. after
  42     err_msg = err_msg .. ("\n" .. C('dim', fmt_str:format(err_linenum)) .. err_line)
  43   end
  44   local _, err_linenum_end, err_linepos_end = err.source:line_info_at(err.stop)
  45   err_linenum_end = err_linenum_end or err_linenum
  46   if err_linenum_end == err_linenum then
  47     err_msg = err_msg .. "\n" .. tostring(pointer)
  48   else
  49     for i = err_linenum + 1, err_linenum_end do
  50       do
  51         local line = lines[i]
  52         if line then
  53           if i == err_linenum_end then
  54             local during, after = line:sub(1, err_linepos_end - 1), line:sub(err_linepos_end, -1)
  55             err_msg = err_msg .. ("\n" .. C('dim', fmt_str:format(i)) .. C('black on red', during) .. after)
  56           else
  57             err_msg = err_msg .. ("\n" .. C('dim', fmt_str:format(i)) .. C('black on red', line))
  58           end
  59         end
  60       end
  61       if i > err_linenum + 1 + 5 then
  62         err_msg = err_msg .. "\n       ...\n"
  63         break
  64       end
  65     end
  66   end
  67   local box_width = 70
  68   local err_text = C('black on white', " " .. err.error:wrapped_to(box_width, 16):gsub("\n", "\n" .. C('black on white') .. " "))
  69   if err.hint then
  70     err_text = err_text .. ("\n" .. C('italic black on white', (" Suggestion: " .. tostring(err.hint)):wrapped_to(box_width, 16):gsub("\n", "\n" .. C('italic black on white') .. " ")))
  71   end
  72   err_msg = err_msg .. ("\n " .. box(err_text):gsub("\n", "\n "))
  73   if not (COLOR_ENABLED) then
  74     err_msg = err_msg .. "\n"
  75   end
  76   for i = err_linenum_end + 1, err_linenum_end + context do
  77     do
  78       local line = lines[i]
  79       if line then
  80         err_msg = err_msg .. ("\n" .. C('dim', fmt_str:format(i)) .. line)
  81       end
  82     end
  83   end
  84   if not (COLOR_ENABLED) then
  85     err_msg = err_msg .. "\n"
  86   end
  87   return err_msg
  88 end
  89 return format_error