.nyagos

-- NYAGOS settings
--
-- Maintainer:   DeaR <nayuri@kuonn.mydns.jp>
-- Last Change:  01-Sep-2017.
-- License:      MIT License {{{
--     Copyright (c) 2013 DeaR <nayuri@kuonn.mydns.jp>
--
--     Permission is hereby granted, free of charge, to any person obtaining a
--     copy of this software and associated documentation files (the
--     "Software"), to deal in the Software without restriction, including
--     without limitation the rights to use, copy, modify, merge, publish,
--     distribute, sublicense, and/or sell copies of the Software, and to
--     permit persons to whom the Software is furnished to do so, subject to
--     the following conditions:
--
--     The above copyright notice and this permission notice shall be included
--     in all copies or substantial portions of the Software.
--
--     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
--     OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
--     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
--     IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
--     CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
--     TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
--     SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
-- }}}

-------------------------------------------------------------------------------
-- Pre Init: {{{
local home = nyagos.getenv('HOME') or nyagos.getenv('USERPROFILE')
local prof = nyagos.getenv('USERPROFILE')
local arch = nyagos.getenv('PROGRAMFILES(X86)') and true or false
local pf64 =
  nyagos.getenv('PROGRAMW6432')      or nyagos.getenv('PROGRAMFILES')
local pf32 =
  nyagos.getenv('PROGRAMFILES(X86)') or nyagos.getenv('PROGRAMFILES')

set {
  HOME = home,

  XDG_DATA_HOME   =
    nyagos.getenv('XDG_DATA_HOME')   or (home .. '\\.local\\share'),
  XDG_CACHE_HOME  =
    nyagos.getenv('XDG_CACHE_HOME')  or (home .. '\\.cache'),
  XDG_CONFIG_HOME =
    nyagos.getenv('XDG_CONFIG_HOME') or (home .. '\\.config'),
}

local function ext(fname)
  local p = nyagos.which(fname)
  return p and nyagos.stat(p).name or fname
end
-- }}}

-------------------------------------------------------------------------------
-- Fix nyagos.lua: {{{
local function include(fname)
  local chank, err = nyagos.loadfile(fname)
  if err then
    print(err)
  elseif chank then
    local ok, err = pcall(chank)
    if not ok then
      print(fname .. ': ' .. err)
    end
  else
    print(fname .. ':fail to load')
  end
end

local org_use = use
function use(name)
  local name = string.gsub(name, '%.lua$', '') .. '.lua'
  local script = nyagos.pathjoin(home, '.nyagos.d', name)
  if nyagos.stat(script) then
    include(script)
  else
    org_use(name)
  end
end
-- }}}

-------------------------------------------------------------------------------
-- Environment Variable: {{{
set {
  TZ   = 'JST-09',
  LANG = 'C',
  TERM = 'msys',
}

if nyagos.which('vim') then
  if nyagos.which('gvim') then
    set { EDITOR = ext('gvim') }
  else
    set { EDITOR = ext('vim') }
  end
  if nyagos.which('git') then
    set { GIT_EDITOR = '%EDITOR% --remote-wait-silent' }
  end
  if nyagos.which('hg') then
    set { HGEDITOR = '%EDITOR% --remote-wait-silent' }
  end
elseif nyagos.which('vi') then
  set { EDITOR = ext('vi') }
end

if nyagos.which('less') then
  set { PAGER = ext('less') }
elseif nyagos.which('lv') then
  set { PAGER = ext('lv') }
elseif nyagos.which('more') then
  set { PAGER = ext('more') }
end

if nyagos.which('less') then
  set {
    LESS         = '-r',
    LESSCHARSET  = 'utf-8',
    LESSHISTFILE = '%XDG_CACHE_HOME%\\lesshst',
  }
  if nyagos.which('lesspipe') then
    set {
      LESSOPEN  = '| lesspipe %s',
      LESSCLOSE = 'lesspipe %s %s',
    }
  elseif nyagos.which('lesspipe.sh') then
    set {
      LESSOPEN  = '| lesspipe.sh %s',
      LESSCLOSE = 'lesspipe.sh %s %s',
    }
  end
end
if nyagos.which('lv') then
  set { LV = '-Ia -c' }
end

if nyagos.which('git') then
  set { GIT_PAGER = '%PAGER%' }
  if nyagos.which('diff-highlight') then
    set { GIT_PAGER = ext('diff-highlight') .. ' | %GIT_PAGER%' }
  end
  if nyagos.which('nkf') then
    set { GIT_PAGER = ext('nkf') .. ' | %GIT_PAGER%' }
  end
end

if nyagos.which('hg') then
  set {
    HGENCODING = 'utf-8',
    HGPAGER    = '%PAGER%',
  }
  if nyagos.which('nkf') then
    set { HGPAGER = ext('nkf') .. ' | %HGPAGER%' }
  end
end
-- }}}

-------------------------------------------------------------------------------
-- Catalog: {{{
-- use 'autocd'
use 'dollar'
if nyagos.which('peco') then
  use 'peco'
end

share.ignoreeof = 1
use 'ignoreeof'
use 'vcvars'
-- }}}

-------------------------------------------------------------------------------
-- Prompt: {{{
nyagos.prompt = function(template)
  local user = nyagos.getenv('USERNAME')
  local host = nyagos.getenv('USERDOMAIN')
  local home = nyagos.getenv('HOME') or nyagos.getenv('USERPROFILE')
  local pat  = '^' .. home:gsub('[().%%+-*?%[%]^$]', '%%%0')
  local cwd  = nyagos.netdrivetounc(nyagos.getwd()):gsub(pat, '~')
  local root = nyagos.elevated() and '#' or '$$'
  return nyagos.default_prompt(
    '$e[32;1m' .. user .. '@' .. host ..
    ' $e[33;1m' .. cwd .. '\n$e[m' .. root .. ' ')
end
-- }}}

-------------------------------------------------------------------------------
-- Su: {{{
if nyagos.getenv('ConEmuDir') then
  share._clone = function(elv, org)
    local arch = nyagos.getenv('PROGRAMFILES(X86)') and '64' or ''
    local cemu = nyagos.getenv('ConEmuDir') .. '\\ConEmu' .. arch .. '.exe'
    local args = '-single -run {NYAGOS}' .. (elv and ' -cur_console:a' or '')
    local cwd  = nyagos.netdrivetounc(nyagos.getwd())
    local status, err = nyagos.shellexecute('open', cemu, args, cwd)
    if status then
      return status, err
    end
    if string.match(err,"^Error%(5%)") or string.match(err,"winapi error") then
      status, err = nyagos.exec(org)
    end
    return status, err
  end
  alias {
    clone = function()
      assert(share._clone(false, '__clone__'))
    end,
    su = function(args)
      if #args <= 0 then
        assert(share._clone(true, '__su__'))
        return
      end
      local arch = nyagos.getenv('PROGRAMFILES(X86)') and '64' or ''
      local cemu = nyagos.getenv('ConEmuDir') .. '\\ConEmu' .. arch .. '.exe'
      local args = '-single -cmd "' .. table.concat(args.rawargs, ' ') .. '" -cur_console:a'
      local cwd  = nyagos.netdrivetounc(nyagos.getwd())
      local status, err = nyagos.shellexecute('open', cemu, args, cwd)
      if status then
        return
      end
      if string.match(err,"^Error%(5%)") or string.match(err,"winapi error") then
        status, err = nyagos.exec('__su__ ' .. table.concat(args.rawargs, ' '))
      end
    end,
  }
end
-- }}}

-------------------------------------------------------------------------------
-- LS: {{{
share._llx = function(args, ...)
  local a = {...}
  local o = nyagos.eval('ll ' .. table.concat(args.rawargs, ' '))
  o:gsub('[^\n]+', function(s)
    for _, p in pairs(a) do
      if s:match(p) then
        nyagos.write(s .. '\n')
        return
      end
    end
  end)
end
alias {
  ls  = '__ls__ -o',
  ll  = 'ls -alF',
  lld = function(args)
    share._llx(args, '^d')
  end,
  llf = function(args)
    share._llx(args, '^-')
  end,
  lle = function(args)
    share._llx(args, '%*$')
  end,
  lls = function(args)
    share._llx(args, '@$', '->')
  end,
}
-- }}}

-------------------------------------------------------------------------------
-- Alias: {{{
alias {
  q     = '__exit__',
  clear = '__cls__',
  sudo  = 'su',
}
if nyagos.getenv('EDITOR') then
  alias { e = '%EDITOR%' }
end
if nyagos.getenv('PAGER') then
  alias { l = '%PAGER%' }
end
if nyagos.which('head') then
  alias { h = ext('head') }
end
if nyagos.which('tailf') then
  alias { t = ext('tailf') }
elseif nyagos.which('tail') then
  alias { t = ext('tail') .. ' -f' }
end
if nyagos.which('gvim') then
  alias {
    vi  = ext('gvim'),
    vim = ext('gvim')
  }
elseif nyagos.which('vim') then
  alias { vi = ext('vim') }
end
if nyagos.which('hub') then
  alias { git = ext('hub') }
end
if nyagos.getenv('HGPAGER') then
  alias { hg = ext('hg') .. ' --config pager.pager="%HGPAGER%"' }
end
if nyagos.which('grep') then
  alias { grep = ext('grep') .. ' --color' }
end
if nyagos.which('colordiff') then
  alias { diff = ext('colordiff') }
end
if nyagos.which('less') then
  alias { ltail = ext('less') .. ' +F -L' }
end

if nyagos.stat(prof .. '\\Apps\\PortableGit') then
  alias {
    find = '"' .. prof .. '\\Apps\\PortableGit\\usr\\bin\\find.exe"',
    sort = '"' .. prof .. '\\Apps\\PortableGit\\usr\\bin\\sort.exe"',
  }
elseif nyagos.stat(pf64 .. '\\Git') then
  alias {
    find = '"' .. pf64 .. '\\Git\\usr\\bin\\find.exe"',
    sort = '"' .. pf64 .. '\\Git\\usr\\bin\\sort.exe"',
  }
elseif nyagos.stat(pf32 .. '\\Git') then
  alias {
    find = '"' .. pf32 .. '\\Git\\usr\\bin\\find.exe"',
    sort = '"' .. pf32 .. '\\Git\\usr\\bin\\sort.exe"',
  }
end
if nyagos.which('mingw32-make') then
  alias { make = ext('mingw32-make') }
end
-- }}}

-------------------------------------------------------------------------------
-- Suffix: {{{
suffix.htm  = 'open'
suffix.html = 'open'
suffix.msi  = 'open'
suffix.js   = 'cscript //nologo'
suffix.vbs  = 'cscript //nologo'
suffix.wsf  = 'cscript //nologo'
suffix.ps1  = 'powershell -ExecutionPolicy RemoteSigned -file'
suffix.lua  = 'nyagos.exe -f'
if nyagos.which('sh') then
  suffix.sh = 'sh'
end
if nyagos.which('bash') then
  suffix.bash = 'bash'
end
if nyagos.which('zsh') then
  suffix.zsh = 'zsh'
end
if nyagos.which('gawk') then
  suffix.awk = 'gawk -f'
end
if nyagos.which('perl') then
  suffix.pl = 'perl'
end
if nyagos.which('ruby') then
  suffix.rb = 'ruby'
end
if nyagos.which('python') then
  suffix.py = 'python'
end
-- }}}

-------------------------------------------------------------------------------
-- Post Init: {{{
if nyagos.stat(home .. '\\.nyagos_local.lua') then
  include(home .. '\\.nyagos_local.lua')
elseif nyagos.stat(home .. '\\.nyagos.d\\nyagos_local.lua') then
  include(home .. '\\.nyagos.d\\nyagos_local.lua')
end
-- }}}

-- vim: ft=lua