#compdef lua -P lua[0-9.-]##

# Complete lua library names. We go out of our way here to support sub-modules
# (of the format foo.bar.baz), even though the way `lua -l` handles those isn't
# very nice, because it might be useful for informational purposes
(( $+functions[_lua_libraries] )) ||
_lua_libraries() {
  local p
  local -a tmp tmp2 pre

  # Lua will tell us its effective search paths in the error message
  tmp=( ${(f)"$( _call_program libraries $words[1] -l '"%%FAKE%%"' 2>&1 )"} )

  [[ $tmp == *'%%FAKE%%'* ]] && {
    tmp=( ${(@M)tmp:#*no file*'%%FAKE%%'*} )
    tmp=( ${(@)tmp##[[:space:]]#no file[[:space:]\'\"]##} )
    tmp=( ${(@)tmp%%[[:space:]\'\"]##} )

    for p in $tmp; do
      pre+=( ${(b)p%%'%%FAKE%%'*} )
      # Don't recurse infinitely into the current directory; we'll just trust
      # that all other paths are sensible
      if [[ $p == './%%FAKE%%'* ]]; then
        tmp2+=( ${~${${(b)p}/'%%FAKE%%'/'*'}}(#qN) )
      else
        tmp2+=( ${~${${(b)p}/'%%FAKE%%'/'**/*'}}(#qN) )
      fi
    done

    tmp=( $tmp2 )
    tmp=( ${(@)tmp%%(.lua|/init.lua|.so)} )
    tmp=( ${(@)tmp##(${~${(j<|>)pre}})} )
    tmp=( ${(@u)${(@)tmp//\//.}} )
  }

  _wanted -x libraries expl 'Lua library' compadd -a "$@" - tmp
}

# Stacking not supported, no arguments are exclusive except `-`
_arguments -S -A '-*' : \
  '*-e+[execute specified command string]:command string' \
  '-E[ignore environment variables]' \
  '-i[enter interactive mode]' \
  '*-l+[specify library or module to require]: :_lua_libraries' \
  '-v[display version information]' \
  '(1 -)-[stop argument parsing and execute script on stdin]' \
  '1:Lua script:_files' \
  '*:script argument'