1
0
Fork 0
mirror of synced 2024-11-30 04:35:35 -05:00
ultimate-vim/sources_non_forked/vim-go/autoload/go/util.vim

399 lines
11 KiB
VimL
Raw Normal View History

2015-07-13 06:22:46 -04:00
" PathSep returns the appropriate OS specific path separator.
2016-12-27 09:46:49 -05:00
function! go#util#PathSep() abort
2016-06-26 07:12:36 -04:00
if go#util#IsWin()
return '\'
endif
return '/'
2015-07-13 06:22:46 -04:00
endfunction
" PathListSep returns the appropriate OS specific path list separator.
2016-12-27 09:46:49 -05:00
function! go#util#PathListSep() abort
2016-06-26 07:12:36 -04:00
if go#util#IsWin()
return ";"
endif
return ":"
2015-07-13 06:22:46 -04:00
endfunction
" LineEnding returns the correct line ending, based on the current fileformat
2016-12-27 09:46:49 -05:00
function! go#util#LineEnding() abort
2016-06-26 07:12:36 -04:00
if &fileformat == 'dos'
return "\r\n"
elseif &fileformat == 'mac'
return "\r"
endif
2015-07-13 06:22:46 -04:00
2016-06-26 07:12:36 -04:00
return "\n"
2015-07-13 06:22:46 -04:00
endfunction
2016-07-03 07:53:59 -04:00
" Join joins any number of path elements into a single path, adding a
" Separator if necessary and returns the result
2016-12-27 09:46:49 -05:00
function! go#util#Join(...) abort
2016-07-03 07:53:59 -04:00
return join(a:000, go#util#PathSep())
endfunction
2015-07-13 06:22:46 -04:00
" IsWin returns 1 if current OS is Windows or 0 otherwise
2016-12-27 09:46:49 -05:00
function! go#util#IsWin() abort
2016-06-26 07:12:36 -04:00
let win = ['win16', 'win32', 'win64', 'win95']
for w in win
if (has(w))
return 1
endif
endfor
2016-02-20 08:13:10 -05:00
2016-06-26 07:12:36 -04:00
return 0
2015-07-13 06:22:46 -04:00
endfunction
" IsMac returns 1 if current OS is macOS or 0 otherwise.
function! go#util#IsMac() abort
return has('mac') ||
\ has('macunix') ||
\ has('gui_macvim') ||
\ go#util#System('uname') =~? '^darwin'
endfunction
2017-09-02 06:43:18 -04:00
" Checks if using:
" 1) Windows system,
" 2) And has cygpath executable,
" 3) And uses *sh* as 'shell'
function! go#util#IsUsingCygwinShell()
return go#util#IsWin() && executable('cygpath') && &shell =~ '.*sh.*'
endfunction
2016-12-27 09:46:49 -05:00
function! go#util#has_job() abort
" job was introduced in 7.4.xxx however there are multiple bug fixes and one
" of the latest is 8.0.0087 which is required for a stable async API.
return has('job') && has("patch-8.0.0087")
endfunction
2016-11-09 12:22:55 -05:00
let s:env_cache = {}
" env returns the go environment variable for the given key. Where key can be
" GOARCH, GOOS, GOROOT, etc... It caches the result and returns the cached
2017-03-07 12:04:28 -05:00
" version.
2016-12-27 09:46:49 -05:00
function! go#util#env(key) abort
2016-11-09 12:22:55 -05:00
let l:key = tolower(a:key)
if has_key(s:env_cache, l:key)
return s:env_cache[l:key]
endif
if executable('go')
let l:var = call('go#util#'.l:key, [])
if go#util#ShellError() != 0
call go#util#EchoError(printf("'go env %s' failed", toupper(l:key)))
return ''
endif
else
let l:var = eval("$".toupper(a:key))
endif
let s:env_cache[l:key] = l:var
return l:var
endfunction
2017-07-06 08:57:35 -04:00
" goarch returns 'go env GOARCH'. This is an internal function and shouldn't
" be used. Instead use 'go#util#env("goarch")'
2016-12-27 09:46:49 -05:00
function! go#util#goarch() abort
2016-06-26 07:12:36 -04:00
return substitute(go#util#System('go env GOARCH'), '\n', '', 'g')
2016-05-14 07:57:54 -04:00
endfunction
2017-07-06 08:57:35 -04:00
" goos returns 'go env GOOS'. This is an internal function and shouldn't
" be used. Instead use 'go#util#env("goos")'
2016-12-27 09:46:49 -05:00
function! go#util#goos() abort
2016-06-26 07:12:36 -04:00
return substitute(go#util#System('go env GOOS'), '\n', '', 'g')
2016-05-14 07:57:54 -04:00
endfunction
2017-07-06 08:57:35 -04:00
" goroot returns 'go env GOROOT'. This is an internal function and shouldn't
" be used. Instead use 'go#util#env("goroot")'
2016-12-27 09:46:49 -05:00
function! go#util#goroot() abort
2016-06-26 07:12:36 -04:00
return substitute(go#util#System('go env GOROOT'), '\n', '', 'g')
2016-05-14 07:57:54 -04:00
endfunction
2017-07-06 08:57:35 -04:00
" gopath returns 'go env GOPATH'. This is an internal function and shouldn't
" be used. Instead use 'go#util#env("gopath")'
2016-12-27 09:46:49 -05:00
function! go#util#gopath() abort
2016-06-26 07:12:36 -04:00
return substitute(go#util#System('go env GOPATH'), '\n', '', 'g')
2016-05-14 07:57:54 -04:00
endfunction
2016-12-27 09:46:49 -05:00
function! go#util#osarch() abort
2017-07-06 08:57:35 -04:00
return go#util#env("goos") . '_' . go#util#env("goarch")
2016-05-14 07:57:54 -04:00
endfunction
" Run a shell command.
"
" It will temporary set the shell to /bin/sh for Unix-like systems if possible,
" so that we always use a standard POSIX-compatible Bourne shell (and not e.g.
" csh, fish, etc.) See #988 and #1276.
function! s:system(cmd, ...) abort
2017-05-26 05:30:32 -04:00
" Preserve original shell and shellredir values
2016-08-02 08:48:32 -04:00
let l:shell = &shell
2017-05-26 05:30:32 -04:00
let l:shellredir = &shellredir
2016-08-02 08:48:32 -04:00
if !go#util#IsWin() && executable('/bin/sh')
2017-05-26 05:30:32 -04:00
set shell=/bin/sh shellredir=>%s\ 2>&1
2016-08-02 08:48:32 -04:00
endif
try
return call('system', [a:cmd] + a:000)
2016-08-02 08:48:32 -04:00
finally
2017-05-26 05:30:32 -04:00
" Restore original values
2016-08-02 08:48:32 -04:00
let &shell = l:shell
2017-05-26 05:30:32 -04:00
let &shellredir = l:shellredir
2016-08-02 08:48:32 -04:00
endtry
2016-05-14 07:57:54 -04:00
endfunction
" System runs a shell command "str". Every arguments after "str" is passed to
" stdin.
function! go#util#System(str, ...) abort
return call('s:system', [a:str] + a:000)
endfunction
" Exec runs a shell command "cmd", which must be a list, one argument per item.
" Every list entry will be automatically shell-escaped
" Every other argument is passed to stdin.
function! go#util#Exec(cmd, ...) abort
if len(a:cmd) == 0
call go#util#EchoError("go#util#Exec() called with empty a:cmd")
return
endif
" CheckBinPath will show a warning for us.
let l:bin = go#path#CheckBinPath(a:cmd[0])
if empty(l:bin)
return ["", 1]
endif
let l:out = call('s:system', [go#util#Shelljoin([l:bin] + a:cmd[1:])] + a:000)
return [l:out, go#util#ShellError()]
endfunction
2016-12-27 09:46:49 -05:00
function! go#util#ShellError() abort
return v:shell_error
2016-05-14 07:57:54 -04:00
endfunction
2015-07-13 06:22:46 -04:00
" StripPath strips the path's last character if it's a path separator.
" example: '/foo/bar/' -> '/foo/bar'
2016-12-27 09:46:49 -05:00
function! go#util#StripPathSep(path) abort
2016-06-26 07:12:36 -04:00
let last_char = strlen(a:path) - 1
if a:path[last_char] == go#util#PathSep()
return strpart(a:path, 0, last_char)
endif
2015-07-13 06:22:46 -04:00
2016-06-26 07:12:36 -04:00
return a:path
2015-07-13 06:22:46 -04:00
endfunction
2016-03-14 06:04:57 -04:00
" StripTrailingSlash strips the trailing slash from the given path list.
" example: ['/foo/bar/'] -> ['/foo/bar']
2016-12-27 09:46:49 -05:00
function! go#util#StripTrailingSlash(paths) abort
2016-03-14 06:04:57 -04:00
return map(copy(a:paths), 'go#util#StripPathSep(v:val)')
endfunction
2015-12-08 08:20:04 -05:00
" Shelljoin returns a shell-safe string representation of arglist. The
" {special} argument of shellescape() may optionally be passed.
2016-12-27 09:46:49 -05:00
function! go#util#Shelljoin(arglist, ...) abort
2016-06-26 07:12:36 -04:00
try
let ssl_save = &shellslash
set noshellslash
if a:0
return join(map(copy(a:arglist), 'shellescape(v:val, ' . a:1 . ')'), ' ')
endif
2016-02-20 08:13:10 -05:00
2016-07-16 14:30:35 -04:00
return join(map(copy(a:arglist), 'shellescape(v:val)'), ' ')
finally
let &shellslash = ssl_save
endtry
2015-12-16 08:53:53 -05:00
endfunction
2016-05-14 07:57:54 -04:00
fu! go#util#Shellescape(arg)
2016-07-16 14:30:35 -04:00
try
let ssl_save = &shellslash
set noshellslash
return shellescape(a:arg)
finally
let &shellslash = ssl_save
endtry
2016-05-14 07:57:54 -04:00
endf
2016-02-20 08:13:10 -05:00
" Shelllist returns a shell-safe representation of the items in the given
2015-12-16 08:53:53 -05:00
" arglist. The {special} argument of shellescape() may optionally be passed.
2016-12-27 09:46:49 -05:00
function! go#util#Shelllist(arglist, ...) abort
2016-07-16 14:30:35 -04:00
try
let ssl_save = &shellslash
set noshellslash
if a:0
return map(copy(a:arglist), 'shellescape(v:val, ' . a:1 . ')')
endif
return map(copy(a:arglist), 'shellescape(v:val)')
finally
let &shellslash = ssl_save
endtry
2015-12-16 08:53:53 -05:00
endfunction
2016-03-20 14:01:44 -04:00
" Returns the byte offset for line and column
2016-12-27 09:46:49 -05:00
function! go#util#Offset(line, col) abort
2016-07-16 14:30:35 -04:00
if &encoding != 'utf-8'
let sep = go#util#LineEnding()
let buf = a:line == 1 ? '' : (join(getline(1, a:line-1), sep) . sep)
let buf .= a:col == 1 ? '' : getline('.')[:a:col-2]
return len(iconv(buf, &encoding, 'utf-8'))
endif
return line2byte(a:line) + (a:col-2)
2016-03-20 14:01:44 -04:00
endfunction
"
" Returns the byte offset for the cursor
2016-12-27 09:46:49 -05:00
function! go#util#OffsetCursor() abort
2016-07-16 14:30:35 -04:00
return go#util#Offset(line('.'), col('.'))
2016-03-20 14:01:44 -04:00
endfunction
2016-04-12 04:31:09 -04:00
" Windo is like the built-in :windo, only it returns to the window the command
" was issued from
2016-12-27 09:46:49 -05:00
function! go#util#Windo(command) abort
2016-07-16 14:30:35 -04:00
let s:currentWindow = winnr()
try
execute "windo " . a:command
finally
execute s:currentWindow. "wincmd w"
unlet s:currentWindow
endtry
endfunction
" snippetcase converts the given word to given preferred snippet setting type
" case.
2016-12-27 09:46:49 -05:00
function! go#util#snippetcase(word) abort
2017-05-26 05:30:32 -04:00
let l:snippet_case = get(g:, 'go_addtags_transform', "snakecase")
2016-07-16 14:30:35 -04:00
if l:snippet_case == "snakecase"
return go#util#snakecase(a:word)
elseif l:snippet_case == "camelcase"
return go#util#camelcase(a:word)
else
return a:word " do nothing
endif
2016-04-12 04:31:09 -04:00
endfunction
2016-07-16 14:30:35 -04:00
" snakecase converts a string to snake case. i.e: FooBar -> foo_bar
" Copied from tpope/vim-abolish
2016-12-27 09:46:49 -05:00
function! go#util#snakecase(word) abort
let word = substitute(a:word, '::', '/', 'g')
let word = substitute(word, '\(\u\+\)\(\u\l\)', '\1_\2', 'g')
let word = substitute(word, '\(\l\|\d\)\(\u\)', '\1_\2', 'g')
let word = substitute(word, '[.-]', '_', 'g')
2016-07-16 14:30:35 -04:00
let word = tolower(word)
return word
endfunction
" camelcase converts a string to camel case. e.g. FooBar or foo_bar will become
" fooBar.
" Copied from tpope/vim-abolish.
2016-12-27 09:46:49 -05:00
function! go#util#camelcase(word) abort
2016-07-16 14:30:35 -04:00
let word = substitute(a:word, '-', '_', 'g')
if word !~# '_' && word =~# '\l'
return substitute(word, '^.', '\l&', '')
2016-07-16 14:30:35 -04:00
else
return substitute(word, '\C\(_\)\=\(.\)', '\=submatch(1)==""?tolower(submatch(2)) : toupper(submatch(2))','g')
2016-07-16 14:30:35 -04:00
endif
endfunction
" pascalcase converts a string to 'PascalCase'. e.g. fooBar or foo_bar will
" become FooBar.
function! go#util#pascalcase(word) abort
let word = go#util#camelcase(a:word)
return toupper(word[0]) . word[1:]
endfunction
" Echo a message to the screen and highlight it with the group in a:hi.
2016-12-27 09:46:49 -05:00
"
" The message can be a list or string; every line with be :echomsg'd separately.
function! s:echo(msg, hi)
let l:msg = []
if type(a:msg) != type([])
let l:msg = split(a:msg, "\n")
else
let l:msg = a:msg
endif
" Tabs display as ^I or <09>, so manually expand them.
let l:msg = map(l:msg, 'substitute(v:val, "\t", " ", "")')
exe 'echohl ' . a:hi
for line in l:msg
echom "vim-go: " . line
endfor
echohl None
2015-12-08 08:20:04 -05:00
endfunction
2015-12-16 08:53:53 -05:00
function! go#util#EchoSuccess(msg)
call s:echo(a:msg, 'Function')
endfunction
2015-12-16 08:53:53 -05:00
function! go#util#EchoError(msg)
call s:echo(a:msg, 'ErrorMsg')
2015-12-16 08:53:53 -05:00
endfunction
function! go#util#EchoWarning(msg)
call s:echo(a:msg, 'WarningMsg')
2015-12-16 08:53:53 -05:00
endfunction
function! go#util#EchoProgress(msg)
call s:echo(a:msg, 'Identifier')
2016-12-27 09:46:49 -05:00
endfunction
function! go#util#EchoInfo(msg)
call s:echo(a:msg, 'Debug')
2015-12-16 08:53:53 -05:00
endfunction
" Get all lines in the buffer as a a list.
2017-02-11 08:01:38 -05:00
function! go#util#GetLines()
let buf = getline(1, '$')
if &encoding != 'utf-8'
let buf = map(buf, 'iconv(v:val, &encoding, "utf-8")')
endif
if &l:fileformat == 'dos'
" XXX: line2byte() depend on 'fileformat' option.
" so if fileformat is 'dos', 'buf' must include '\r'.
let buf = map(buf, 'v:val."\r"')
endif
return buf
endfunction
" Convert the current buffer to the "archive" format of
" golang.org/x/tools/go/buildutil:
" https://godoc.org/golang.org/x/tools/go/buildutil#ParseOverlayArchive
"
" > The archive consists of a series of files. Each file consists of a name, a
" > decimal file size and the file contents, separated by newlinews. No newline
" > follows after the file contents.
function! go#util#archive()
let l:buffer = join(go#util#GetLines(), "\n")
return expand("%:p:gs!\\!/!") . "\n" . strlen(l:buffer) . "\n" . l:buffer
endfunction
" Make a named temporary directory which starts with "prefix".
"
" Unfortunately Vim's tempname() is not portable enough across various systems;
" see: https://github.com/mattn/vim-go/pull/3#discussion_r138084911
function! go#util#tempdir(prefix) abort
" See :help tempfile
if go#util#IsWin()
let l:dirs = [$TMP, $TEMP, 'c:\tmp', 'c:\temp']
else
let l:dirs = [$TMPDIR, '/tmp', './', $HOME]
endif
let l:dir = ''
for l:d in dirs
if !empty(l:d) && filewritable(l:d) == 2
let l:dir = l:d
break
endif
endfor
if l:dir == ''
echoerr 'Unable to find directory to store temporary directory in'
return
endif
" Not great randomness, but "good enough" for our purpose here.
let l:rnd = sha256(printf('%s%s', localtime(), fnamemodify(bufname(''), ":p")))
let l:tmp = printf("%s/%s%s", l:dir, a:prefix, l:rnd)
call mkdir(l:tmp, 'p', 0700)
return l:tmp
endfunction
2016-06-26 07:12:36 -04:00
" vim: sw=2 ts=2 et