Merge commit '76265755a1add77121c8f9dabb3e9bb70fe9a972' as '.vim/bundle/ale'

[etc/vim.git] / .vim / bundle / ale / ale_linters / sql / sqlfluff.vim

diff --git a/.vim/bundle/ale/ale_linters/sql/sqlfluff.vim b/.vim/bundle/ale/ale_linters/sql/sqlfluff.vim
new file mode 100644 (file)
index 0000000..91d69c1
--- /dev/null
+++ b/.vim/bundle/ale/ale_linters/sql/sqlfluff.vim
@@ -0,0 +1,108 @@
+" Author: Carl Smedstad <carl.smedstad at protonmail dot com>
+" Description: sqlfluff for SQL files
+
+let g:ale_sql_sqlfluff_executable =
+\   get(g:, 'ale_sql_sqlfluff_executable', 'sqlfluff')
+
+let g:ale_sql_sqlfluff_options =
+\   get(g:, 'ale_sql_sqlfluff_options', '')
+
+function! ale_linters#sql#sqlfluff#Executable(buffer) abort
+    return ale#Var(a:buffer, 'sql_sqlfluff_executable')
+endfunction
+
+function! ale_linters#sql#sqlfluff#Command(buffer, version) abort
+    let l:executable = ale_linters#sql#sqlfluff#Executable(a:buffer)
+    let l:options = ale#Var(a:buffer, 'sql_sqlfluff_options')
+
+    let l:cmd =
+    \    ale#Escape(l:executable)
+    \    . ' lint'
+
+    let l:config_file = ale#path#FindNearestFile(a:buffer, '.sqlfluff')
+
+    if !empty(l:config_file)
+        let l:cmd .= ' --config ' . ale#Escape(l:config_file)
+    else
+        let l:cmd .= ' --dialect ansi'
+    endif
+
+    let l:cmd .=
+    \   ' --format json '
+    \   . l:options
+    \   . ' %t'
+
+    return l:cmd
+endfunction
+
+function! ale_linters#sql#sqlfluff#Handle(buffer, version, lines) abort
+    let l:output = []
+    let l:json_lines = ale#util#FuzzyJSONDecode(a:lines, [])
+
+    if empty(l:json_lines)
+        return l:output
+    endif
+
+    let l:json = l:json_lines[0]
+
+    " if there's no warning, 'result' is `null`.
+    if empty(get(l:json, 'violations'))
+        return l:output
+    endif
+
+    if ale#semver#GTE(a:version, [3, 0, 0])
+        for l:violation in get(l:json, 'violations', [])
+            let l:err = {
+            \   'filename': l:json.filepath,
+            \   'lnum': l:violation.start_line_no,
+            \   'col': l:violation.start_line_pos,
+            \   'text': l:violation.description,
+            \   'code': l:violation.code,
+            \   'type': 'W',
+            \}
+
+            if has_key(l:violation, 'end_line_no')
+                let l:err.end_lnum = l:violation.end_line_no
+            endif
+
+            if has_key(l:violation, 'end_line_pos')
+                let l:err.end_col = l:violation.end_line_pos
+            endif
+
+            call add(l:output, l:err)
+        endfor
+    else
+        for l:violation in get(l:json, 'violations', [])
+            call add(l:output, {
+            \   'filename': l:json.filepath,
+            \   'lnum': l:violation.line_no,
+            \   'col': l:violation.line_pos,
+            \   'text': l:violation.description,
+            \   'code': l:violation.code,
+            \   'type': 'W',
+            \})
+        endfor
+    endif
+
+    return l:output
+endfunction
+
+call ale#linter#Define('sql', {
+\   'name': 'sqlfluff',
+\   'executable': function('ale_linters#sql#sqlfluff#Executable'),
+\   'command': {buffer -> ale#semver#RunWithVersionCheck(
+\       buffer,
+\       ale_linters#sql#sqlfluff#Executable(buffer),
+\       '%e --version',
+\       function('ale_linters#sql#sqlfluff#Command'),
+\   )},
+\   'callback': {buffer, lines -> ale#semver#RunWithVersionCheck(
+\       buffer,
+\       ale_linters#sql#sqlfluff#Executable(buffer),
+\       '%e --version',
+\       {buffer, version -> ale_linters#sql#sqlfluff#Handle(
+\           buffer,
+\           l:version,
+\           lines)},
+\   )},
+\})