Apache Cassandra adapter support

README.markdown

@@ -15,6 +15,7 @@ take on [dbext.vim][], improving on it on the following ways:
   - jq
   - MongoDB
   - MySQL
+  - Cassandra
   - MariaDB
   - Oracle
   - osquery

autoload/db/adapter/cassandra.vim

@@ -0,0 +1,52 @@
+function! db#adapter#cassandra#canonicalize(url) abort
+  let url = substitute(a:url, '^cassandra\d*:/\@!', 'cassandra:///', '')
+  " JDBC
+  let url = substitute(url, '//address=(\(.*\))\(/[^#]*\)', '\="//".submatch(2)."&".substitute(submatch(1), ")(", "\\&", "g")', '')
+  let url = substitute(url, '[&?]', '?', '')
+  return db#url#absorb_params(url, {
+        \ 'user': 'user',
+        \ 'password': 'password',
+        \ 'host': 'host',
+        \ 'port': 'port'})
+endfunction
+
+function! s:command_for_url(url) abort
+  let params = db#url#parse(a:url).params
+  let command = ['cqlsh']
+
+  for i in keys(params)
+    let command += ['--'.i.'='.params[i]]
+  endfor
+
+  return command + db#url#as_argv(a:url, '', '', '', '-u ', '-p', '-k')
+endfunction
+
+function! db#adapter#cassandra#interactive(url) abort
+  return s:command_for_url(a:url)
+endfunction
+
+" function! db#adapter#cassandra#filter(url) abort
+"   return s:command_for_url(a:url) + ['-t']
+" endfunction
+
+function! db#adapter#cassandra#auth_pattern() abort
+  return '^ERROR 104[45] '
+endfunction
+
+function! db#adapter#cassandra#complete_opaque(url) abort
+  return db#adapter#cassandra#complete_database('cassandra:///')
+endfunction
+
+function! db#adapter#cassandra#complete_database(url) abort
+  let params = db#url#parse(a:url)
+  let query = "SELECT keyspace_name FROM system_schema.keyspaces WHERE keyspace_name='" . params['scheme'] . "'"
+  let tables = filter(map(db#systemlist(s:command_for_url(a:url) + ['-e', query]), 'trim(v:val)'), '!empty(v:val)')
+  return tables[2:-2]
+endfunction
+
+function! db#adapter#cassandra#tables(url) abort
+  let params = db#url#parse(a:url)
+  let query = "SELECT table_name FROM system_schema.tables WHERE keyspace_name='" . params['path'][1:] . "'"
+  let tables = filter(map(db#systemlist(s:command_for_url(a:url) + ['-e', query]), 'trim(v:val)'), '!empty(v:val)')
+  return tables[2:-2]
+endfunction

doc/dadbod.txt

@@ -148,6 +148,16 @@ MySQL ~
     mysql:[<database>]
 <
 Query parameters such as `login-path`, `protocol`, `ssl-ca`, `ssl-cert`, and
+`ssl-key` are passed as the corresponding command line options.
+
+                                                *dadbod-cassandra*
+Cassandra ~
+>
+    cassandra://[<user>[:<password>]@][<host>[:<port>]]/[<database>]
+    cassandra://[<user>[:<password>]@][<%2Fsocket%2Fpath>]/[<database>]
+    cassandra:[<database>]
+<
+Query parameters such as `login-path`, `protocol`, `ssl-ca`, `ssl-cert`, and
 `ssl-key` are passed as the corresponding command line options.
 
                                                 *dadbod-oracle*