1 files changed, 322 insertions, 0 deletions
diff --git a/Data/BuiltIn/Libraries/lua-stdlib/lib/std/io.lua b/Data/BuiltIn/Libraries/lua-stdlib/lib/std/io.lua
new file mode 100644
index 0000000..1a2b79f
--- /dev/null
+++ b/Data/BuiltIn/Libraries/lua-stdlib/lib/std/io.lua
@@ -0,0 +1,322 @@
+--[[
+ General Lua Libraries for Lua 5.1, 5.2 & 5.3
+ Copyright (C) 2002-2018 stdlib authors
+]]
+--[[--
+ Additions to the core io module.
+
+ The module table returned by `std.io` also contains all of the entries from
+ the core `io` module table.   An hygienic way to import this module, then,
+ is simply to override core `io` locally:
+
+      local io = require 'std.io'
+
+ @corelibrary std.io
+]]
+
+
+local _ = require 'std._base'
+
+local argscheck = _.typecheck and _.typecheck.argscheck
+local catfile = _.io.catfile
+local leaves = _.tree.leaves
+local split = _.string.split
+
+_ = nil
+
+
+local _ENV = require 'std.normalize' {
+   'io',
+   _G = _G,  -- FIXME: don't use the host _G as an API!
+   concat = 'table.concat',
+   dirsep = 'package.dirsep',
+   format = 'string.format',
+   gsub = 'string.gsub',
+   input = 'io.input',
+   insert = 'table.insert',
+   io_type = 'io.type',
+   merge = 'table.merge',
+   open = 'io.open',
+   output = 'io.output',
+   popen = 'io.popen',
+   stderr = 'io.stderr',
+   stdin = 'io.stdin',
+   write = 'io.write',
+}
+
+
+--[[ =============== ]]--
+--[[ Implementation. ]]--
+--[[ =============== ]]--
+
+
+local M
+
+
+local function input_handle(h)
+   if h == nil then
+      return input()
+   elseif type(h) == 'string' then
+      return open(h)
+   end
+   return h
+end
+
+
+local function slurp(file)
+   local h, err = input_handle(file)
+   if h == nil then
+      argerror('std.io.slurp', 1, err, 2)
+   end
+
+   if h then
+      local s = h:read('*a')
+      h:close()
+      return s
+   end
+end
+
+
+local function readlines(file)
+   local h, err = input_handle(file)
+   if h == nil then
+      argerror('std.io.readlines', 1, err, 2)
+   end
+
+   local l = {}
+   for line in h:lines() do
+      l[#l + 1] = line
+   end
+   h:close()
+   return l
+end
+
+
+local function writelines(h, ...)
+   if io_type(h) ~= 'file' then
+      write(h, '\n')
+      h = output()
+   end
+   for v in leaves(ipairs, {...}) do
+      h:write(v, '\n')
+   end
+end
+
+
+local function process_files(fn)
+   -- N.B. 'arg' below refers to the global array of command-line args
+   if len(arg) == 0 then
+      insert(arg, '-')
+   end
+   for i, v in ipairs(arg) do
+      if v == '-' then
+         input(stdin)
+      else
+         input(v)
+      end
+      fn(v, i)
+   end
+end
+
+
+local function warnfmt(msg, ...)
+   local prefix = ''
+   local prog = rawget(_G, 'prog') or {}
+   local opts = rawget(_G, 'opts') or {}
+   if prog.name then
+      prefix = prog.name .. ':'
+      if prog.line then
+         prefix = prefix .. str(prog.line) .. ':'
+      end
+   elseif prog.file then
+      prefix = prog.file .. ':'
+      if prog.line then
+         prefix = prefix .. str(prog.line) .. ':'
+      end
+   elseif opts.program then
+      prefix = opts.program .. ':'
+      if opts.line then
+         prefix = prefix .. str(opts.line) .. ':'
+      end
+   end
+   if #prefix > 0 then
+      prefix = prefix .. ' '
+   end
+   return prefix .. format(msg, ...)
+end
+
+
+local function warn(msg, ...)
+   writelines(stderr, warnfmt(msg, ...))
+end
+
+
+
+--[[ ================= ]]--
+--[[ Public Interface. ]]--
+--[[ ================= ]]--
+
+
+local function X(decl, fn)
+   return argscheck and argscheck('std.io.' .. decl, fn) or fn
+end
+
+
+M = {
+   --- Diagnostic functions
+   -- @section diagnosticfuncs
+
+   --- Die with error.
+   -- This function uses the same rules to build a message prefix
+   -- as @{warn}.
+   -- @function die
+   -- @string msg format string
+   -- @param ... additional arguments to plug format string specifiers
+   -- @see warn
+   -- @usage
+   --    die('oh noes!(%s)', tostring(obj))
+   die = X('die(string, [any...])', function(...)
+      error(warnfmt(...), 0)
+   end),
+
+   --- Give warning with the name of program and file(if any).
+   -- If there is a global `prog` table, prefix the message with
+   -- `prog.name` or `prog.file`, and `prog.line` if any.   Otherwise
+   -- if there is a global `opts` table, prefix the message with
+   -- `opts.program` and `opts.line` if any.
+   -- @function warn
+   -- @string msg format string
+   -- @param ... additional arguments to plug format string specifiers
+   -- @see die
+   -- @usage
+   --    local OptionParser = require 'std.optparse'
+   --    local parser = OptionParser 'eg 0\nUsage: eg\n'
+   --    _G.arg, _G.opts = parser:parse(_G.arg)
+   --    if not _G.opts.keep_going then
+   --       require 'std.io'.warn 'oh noes!'
+   --    end
+   warn = X('warn(string, [any...])', warn),
+
+
+   --- Path Functions
+   -- @section pathfuncs
+
+   --- Concatenate directory names into a path.
+   -- @function catdir
+   -- @string ... path components
+   -- @return path without trailing separator
+   -- @see catfile
+   -- @usage
+   --    dirpath = catdir('', 'absolute', 'directory')
+   catdir = X('catdir(string...)', function(...)
+      return(gsub(concat({...}, dirsep), '^$', dirsep))
+   end),
+
+   --- Concatenate one or more directories and a filename into a path.
+   -- @function catfile
+   -- @string ... path components
+   -- @treturn string path
+   -- @see catdir
+   -- @see splitdir
+   -- @usage
+   --    filepath = catfile('relative', 'path', 'filename')
+   catfile = X('catfile(string...)', catfile),
+
+   --- Remove the last dirsep delimited element from a path.
+   -- @function dirname
+   -- @string path file path
+   -- @treturn string a new path with the last dirsep and following
+   --    truncated
+   -- @usage
+   --    dir = dirname '/base/subdir/filename'
+   dirname = X('dirname(string)', function(path)
+      return(gsub(path, catfile('', '[^', ']*$'), ''))
+   end),
+
+   --- Split a directory path into components.
+   -- Empty components are retained: the root directory becomes `{'', ''}`.
+   -- @function splitdir
+   -- @param path path
+   -- @return list of path components
+   -- @see catdir
+   -- @usage
+   --    dir_components = splitdir(filepath)
+   splitdir = X('splitdir(string)', function(path)
+      return split(path, dirsep)
+   end),
+
+
+   --- IO Functions
+   -- @section iofuncs
+
+   --- Process files specified on the command-line.
+   -- Each filename is made the default input source with `io.input`, and
+   -- then the filename and argument number are passed to the callback
+   -- function. In list of filenames, `-` means `io.stdin`.   If no
+   -- filenames were given, behave as if a single `-` was passed.
+   -- @todo Make the file list an argument to the function.
+   -- @function process_files
+   -- @tparam fileprocessor fn function called for each file argument
+   -- @usage
+   --    #! /usr/bin/env lua
+   --    -- minimal cat command
+   --    local io = require 'std.io'
+   --    io.process_files(function() io.write(io.slurp()) end)
+   process_files = X('process_files(function)', process_files),
+
+   --- Read a file or file handle into a list of lines.
+   -- The lines in the returned list are not `\n` terminated.
+   -- @function readlines
+   -- @tparam[opt=io.input()] file|string file file handle or name;
+   --    if file is a file handle, that file is closed after reading
+   -- @treturn list lines
+   -- @usage
+   --    list = readlines '/etc/passwd'
+   readlines = X('readlines(?file|string)', readlines),
+
+   --- Perform a shell command and return its output.
+   -- @function shell
+   -- @string c command
+   -- @treturn string output, or nil if error
+   -- @see os.execute
+   -- @usage
+   --    users = shell [[cat /etc/passwd | awk -F: '{print $1;}']]
+   shell = X('shell(string)', function(c) return slurp(popen(c)) end),
+
+   --- Slurp a file handle.
+   -- @function slurp
+   -- @tparam[opt=io.input()] file|string file file handle or name;
+   --    if file is a file handle, that file is closed after reading
+   -- @return contents of file or handle, or nil if error
+   -- @see process_files
+   -- @usage
+   --    contents = slurp(filename)
+   slurp = X('slurp(?file|string)', slurp),
+
+   --- Write values adding a newline after each.
+   -- @function writelines
+   -- @tparam[opt=io.output()] file h open writable file handle;
+   --    the file is **not** closed after writing
+   -- @tparam string|number ... values to write(as for write)
+   -- @usage
+   --    writelines(io.stdout, 'first line', 'next line')
+   writelines = X('writelines(?file|string|number, [string|number...])', writelines),
+}
+
+
+return merge(io, M)
+
+
+
+--- Types
+-- @section Types
+
+--- Signature of @{process_files} callback function.
+-- @function fileprocessor
+-- @string filename filename
+-- @int i argument number of *filename*
+-- @usage
+--    local fileprocessor = function(filename, i)
+--       io.write(tostring(i) .. ':\n===\n' .. io.slurp(filename) .. '\n')
+--    end
+--    io.process_files(fileprocessor)