summaryrefslogtreecommitdiffstatshomepage
path: root/plugins/pretty-print/init.lua
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/pretty-print/init.lua')
-rw-r--r--plugins/pretty-print/init.lua362
1 files changed, 362 insertions, 0 deletions
diff --git a/plugins/pretty-print/init.lua b/plugins/pretty-print/init.lua
new file mode 100644
index 00000000000..f1c112d6e4f
--- /dev/null
+++ b/plugins/pretty-print/init.lua
@@ -0,0 +1,362 @@
+--[[
+
+Copyright 2014-2015 The Luvit Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS-IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+--]]
+
+-- Luv port by Tim Caswell <tim@creationix.com>
+
+local exports = {}
+exports.name = "luvit/pretty-print"
+exports.version = "1.0.3"
+exports.homepage = "https://github.com/luvit/luvit/blob/master/deps/pretty-print.lua"
+exports.description = "A lua value pretty printer and colorizer for terminals."
+exports.tags = {"colors", "tty"}
+exports.license = "Apache 2"
+exports.author = { name = "Tim Caswell" }
+
+local uv = require('luv')
+
+local prettyPrint, dump, strip, color, colorize, loadColors
+local theme = {}
+local useColors = false
+local defaultTheme
+
+local stdout, stdin, stderr, width
+
+local quote, quote2, dquote, dquote2, obracket, cbracket, obrace, cbrace, comma, equals, controls
+
+local themes = {
+ -- nice color theme using 16 ansi colors
+ [16] = {
+ property = "0;37", -- white
+ sep = "1;30", -- bright-black
+ braces = "1;30", -- bright-black
+
+ ["nil"] = "1;30", -- bright-black
+ boolean = "0;33", -- yellow
+ number = "1;33", -- bright-yellow
+ string = "0;32", -- green
+ quotes = "1;32", -- bright-green
+ escape = "1;32", -- bright-green
+ ["function"] = "0;35", -- purple
+ thread = "1;35", -- bright-purple
+
+ table = "1;34", -- bright blue
+ userdata = "1;36", -- bright cyan
+ cdata = "0;36", -- cyan
+
+ err = "1;31", -- bright red
+ success = "1;33;42", -- bright-yellow on green
+ failure = "1;33;41", -- bright-yellow on red
+ highlight = "1;36;44", -- bright-cyan on blue
+ },
+ -- nice color theme using ansi 256-mode colors
+ [256] = {
+ property = "38;5;253",
+ braces = "38;5;247",
+ sep = "38;5;240",
+
+ ["nil"] = "38;5;244",
+ boolean = "38;5;220", -- yellow-orange
+ number = "38;5;202", -- orange
+ string = "38;5;34", -- darker green
+ quotes = "38;5;40", -- green
+ escape = "38;5;46", -- bright green
+ ["function"] = "38;5;129", -- purple
+ thread = "38;5;199", -- pink
+
+ table = "38;5;27", -- blue
+ userdata = "38;5;39", -- blue2
+ cdata = "38;5;69", -- teal
+
+ err = "38;5;196", -- bright red
+ success = "38;5;120;48;5;22", -- bright green on dark green
+ failure = "38;5;215;48;5;52", -- bright red on dark red
+ highlight = "38;5;45;48;5;236", -- bright teal on dark grey
+ },
+}
+
+local special = {
+ [7] = 'a',
+ [8] = 'b',
+ [9] = 't',
+ [10] = 'n',
+ [11] = 'v',
+ [12] = 'f',
+ [13] = 'r'
+}
+
+function strip(str)
+ return string.gsub(str, '\027%[[^m]*m', '')
+end
+
+
+function loadColors(index)
+ if index == nil then index = defaultTheme end
+
+ -- Remove the old theme
+ for key in pairs(theme) do
+ theme[key] = nil
+ end
+
+ if index then
+ local new = themes[index]
+ if not new then error("Invalid theme index: " .. tostring(index)) end
+ -- Add the new theme
+ for key in pairs(new) do
+ theme[key] = new[key]
+ end
+ useColors = true
+ else
+ useColors = false
+ end
+
+ quote = colorize('quotes', "'", 'string')
+ quote2 = colorize('quotes', "'")
+ dquote = colorize('quotes', '"', 'string')
+ dquote2 = colorize('quotes', '"')
+ obrace = colorize('braces', '{ ')
+ cbrace = colorize('braces', '}')
+ obracket = colorize('property', '[')
+ cbracket = colorize('property', ']')
+ comma = colorize('sep', ', ')
+ equals = colorize('sep', ' = ')
+
+ controls = {}
+ for i = 0, 31 do
+ local c = special[i]
+ if not c then
+ if i < 10 then
+ c = "00" .. tostring(i)
+ else
+ c = "0" .. tostring(i)
+ end
+ end
+ controls[i] = colorize('escape', '\\' .. c, 'string')
+ end
+ controls[92] = colorize('escape', '\\\\', 'string')
+ controls[34] = colorize('escape', '\\"', 'string')
+ controls[39] = colorize('escape', "\\'", 'string')
+ for i = 128, 255 do
+ local c
+ if i < 100 then
+ c = "0" .. tostring(i)
+ else
+ c = tostring(i)
+ end
+ controls[i] = colorize('escape', '\\' .. c, 'string')
+ end
+
+end
+
+function color(colorName)
+ return '\27[' .. (theme[colorName] or '0') .. 'm'
+end
+
+function colorize(colorName, string, resetName)
+ return useColors and
+ (color(colorName) .. tostring(string) .. color(resetName)) or
+ tostring(string)
+end
+
+local function stringEscape(c)
+ return controls[string.byte(c, 1)]
+end
+
+function dump(value, recurse, nocolor)
+ local seen = {}
+ local output = {}
+ local offset = 0
+ local stack = {}
+
+ local function recalcOffset(index)
+ for i = index + 1, #output do
+ local m = string.match(output[i], "\n([^\n]*)$")
+ if m then
+ offset = #(strip(m))
+ else
+ offset = offset + #(strip(output[i]))
+ end
+ end
+ end
+
+ local function write(text, length)
+ if not length then length = #(strip(text)) end
+ -- Create room for data by opening parent blocks
+ -- Start at the root and go down.
+ local i = 1
+ while offset + length > width and stack[i] do
+ local entry = stack[i]
+ if not entry.opened then
+ entry.opened = true
+ table.insert(output, entry.index + 1, "\n" .. string.rep(" ", i))
+ -- Recalculate the offset
+ recalcOffset(entry.index)
+ -- Bump the index of all deeper entries
+ for j = i + 1, #stack do
+ stack[j].index = stack[j].index + 1
+ end
+ end
+ i = i + 1
+ end
+ output[#output + 1] = text
+ offset = offset + length
+ if offset > width then
+ return dump(stack)
+ end
+ end
+
+ local function indent()
+ stack[#stack + 1] = {
+ index = #output,
+ opened = false,
+ }
+ end
+
+ local function unindent()
+ stack[#stack] = nil
+ end
+
+ local function process(value)
+ local typ = type(value)
+ if typ == 'string' then
+ if string.match(value, "'") and not string.match(value, '"') then
+ write(dquote .. string.gsub(value, '[%c\\\128-\255]', stringEscape) .. dquote2)
+ else
+ write(quote .. string.gsub(value, "[%c\\'\128-\255]", stringEscape) .. quote2)
+ end
+ elseif typ == 'table' and not seen[value] then
+ if not recurse then seen[value] = true end
+ write(obrace)
+ local i = 1
+ -- Count the number of keys so we know when to stop adding commas
+ local total = 0
+ for _ in pairs(value) do total = total + 1 end
+
+ local nextIndex = 1
+ for k, v in pairs(value) do
+ indent()
+ if k == nextIndex then
+ -- if the key matches the last numerical index + 1
+ -- This is how lists print without keys
+ nextIndex = k + 1
+ process(v)
+ else
+ if type(k) == "string" and string.find(k,"^[%a_][%a%d_]*$") then
+ write(colorize("property", k) .. equals)
+ else
+ write(obracket)
+ process(k)
+ write(cbracket .. equals)
+ end
+ if type(v) == "table" then
+ process(v)
+ else
+ indent()
+ process(v)
+ unindent()
+ end
+ end
+ if i < total then
+ write(comma)
+ else
+ write(" ")
+ end
+ i = i + 1
+ unindent()
+ end
+ write(cbrace)
+ else
+ write(colorize(typ, tostring(value)))
+ end
+ end
+
+ process(value)
+ local s = table.concat(output, "")
+ return nocolor and strip(s) or s
+end
+
+-- Print replacement that goes through libuv. This is useful on windows
+-- to use libuv's code to translate ansi escape codes to windows API calls.
+function _G.print(...)
+ local n = select('#', ...)
+ local arguments = {...}
+ for i = 1, n do
+ arguments[i] = tostring(arguments[i])
+ end
+ uv.write(stdout, table.concat(arguments, "\t") .. "\n")
+end
+
+function prettyPrint(...)
+ local n = select('#', ...)
+ local arguments = { ... }
+
+ for i = 1, n do
+ arguments[i] = dump(arguments[i])
+ end
+
+ print(table.concat(arguments, "\t"))
+end
+
+function strip(str)
+ return string.gsub(str, '\027%[[^m]*m', '')
+end
+
+if uv.guess_handle(0) == 'tty' then
+ stdin = assert(uv.new_tty(0, true))
+else
+ stdin = uv.new_pipe(false)
+ uv.pipe_open(stdin, 0)
+end
+
+if uv.guess_handle(1) == 'tty' then
+ stdout = assert(uv.new_tty(1, false))
+ width = uv.tty_get_winsize(stdout)
+ if width == 0 then width = 80 end
+ -- auto-detect when 16 color mode should be used
+ local term = os.getenv("TERM")
+ if term == 'xterm' or term == 'xterm-256color' then
+ defaultTheme = 256
+ else
+ defaultTheme = 16
+ end
+else
+ stdout = uv.new_pipe(false)
+ uv.pipe_open(stdout, 1)
+ width = 80
+end
+loadColors()
+
+if uv.guess_handle(2) == 'tty' then
+ stderr = assert(uv.new_tty(2, false))
+else
+ stderr = uv.new_pipe(false)
+ uv.pipe_open(stderr, 2)
+end
+
+exports.loadColors = loadColors
+exports.theme = theme
+exports.print = print
+exports.prettyPrint = prettyPrint
+exports.dump = dump
+exports.color = color
+exports.colorize = colorize
+exports.stdin = stdin
+exports.stdout = stdout
+exports.stderr = stderr
+exports.strip = strip
+
+return exports