annotate runtime/autoload/sqlcomplete.vim @ 819:23f82b5d2814 v7.0c10

updated for version 7.0c10
author vimboss
date Wed, 05 Apr 2006 20:41:53 +0000
parents 1f929f3ca806
children 2c885fab04e3
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
1 " Vim completion script
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
2 " Language: SQL
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
3 " Maintainer: David Fishburn <fishburn@ianywhere.com>
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
4 " Version: 2.0
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
5 " Last Change: Mon Apr 03 2006 10:21:36 PM
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
6
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
7 " Set completion with CTRL-X CTRL-O to autoloaded function.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
8 " This check is in place in case this script is
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
9 " sourced directly instead of using the autoload feature.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
10 if exists('&omnifunc')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
11 " Do not set the option if already set since this
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
12 " results in an E117 warning.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
13 if &omnifunc == ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
14 setlocal omnifunc=sqlcomplete#Complete
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
15 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
16 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
17
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
18 if exists('g:loaded_sql_completion')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
19 finish
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
20 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
21 let g:loaded_sql_completion = 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
22
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
23 " Maintains filename of dictionary
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
24 let s:sql_file_table = ""
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
25 let s:sql_file_procedure = ""
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
26 let s:sql_file_view = ""
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
27
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
28 " Define various arrays to be used for caching
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
29 let s:tbl_name = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
30 let s:tbl_alias = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
31 let s:tbl_cols = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
32 let s:syn_list = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
33 let s:syn_value = []
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
34
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
35 " Used in conjunction with the syntaxcomplete plugin
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
36 let s:save_inc = ""
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
37 let s:save_exc = ""
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
38 if exists('g:omni_syntax_group_include_sql')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
39 let s:save_inc = g:omni_syntax_group_include_sql
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
40 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
41 if exists('g:omni_syntax_group_exclude_sql')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
42 let s:save_exc = g:omni_syntax_group_exclude_sql
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
43 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
44
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
45 " Used with the column list
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
46 let s:save_prev_table = ""
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
47
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
48 " Default the option to verify table alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
49 if !exists('g:omni_sql_use_tbl_alias')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
50 let g:omni_sql_use_tbl_alias = 'a'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
51 endif
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
52 " Default syntax items to precache
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
53 if !exists('g:omni_sql_precache_syntax_groups')
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
54 let g:omni_sql_precache_syntax_groups = [
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
55 \ 'syntax',
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
56 \ 'sqlKeyword',
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
57 \ 'sqlFunction',
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
58 \ 'sqlOption',
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
59 \ 'sqlType',
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
60 \ 'sqlStatement'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
61 \ ]
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
62 endif
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
63
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
64 " This function is used for the 'omnifunc' option.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
65 function! sqlcomplete#Complete(findstart, base)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
66
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
67 " Default to table name completion
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
68 let compl_type = 'table'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
69 " Allow maps to specify what type of object completion they want
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
70 if exists('b:sql_compl_type')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
71 let compl_type = b:sql_compl_type
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
72 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
73
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
74 " First pass through this function determines how much of the line should
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
75 " be replaced by whatever is chosen from the completion list
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
76 if a:findstart
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
77 " Locate the start of the item, including "."
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
78 let line = getline('.')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
79 let start = col('.') - 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
80 let lastword = -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
81 while start > 0
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
82 if line[start - 1] =~ '\w'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
83 let start -= 1
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
84 elseif line[start - 1] =~ '\.' && compl_type =~ 'column'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
85 " If the completion type is column then assume we are looking
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
86 " for column completion column_type can be either
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
87 " 'column' or 'column_csv'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
88 if lastword == -1 && compl_type == 'column'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
89 " Do not replace the table name prefix or alias
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
90 " if completing only a single column name
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
91 let lastword = start
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
92 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
93 let start -= 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
94 else
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
95 break
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
96 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
97 endwhile
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
98
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
99 " Return the column of the last word, which is going to be changed.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
100 " Remember the text that comes before it in s:prepended.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
101 if lastword == -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
102 let s:prepended = ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
103 return start
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
104 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
105 let s:prepended = strpart(line, start, lastword - start)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
106 return lastword
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
107 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
108
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
109 " Second pass through this function will determine what data to put inside
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
110 " of the completion list
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
111 " s:prepended is set by the first pass
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
112 let base = s:prepended . a:base
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
113
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
114 " Default the completion list to an empty list
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
115 let compl_list = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
116
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
117 " Default to table name completion
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
118 let compl_type = 'table'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
119 " Allow maps to specify what type of object completion they want
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
120 if exists('b:sql_compl_type')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
121 let compl_type = b:sql_compl_type
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
122 unlet b:sql_compl_type
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
123 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
124
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
125 if compl_type == 'tableReset'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
126 let compl_type = 'table'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
127 let base = ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
128 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
129
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
130 if compl_type == 'table' ||
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
131 \ compl_type == 'procedure' ||
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
132 \ compl_type == 'view'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
133
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
134 " This type of completion relies upon the dbext.vim plugin
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
135 if s:SQLCCheck4dbext() == -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
136 return []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
137 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
138
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
139 if s:sql_file_{compl_type} == ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
140 let compl_type = substitute(compl_type, '\w\+', '\u&', '')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
141 let s:sql_file_{compl_type} = DB_getDictionaryName(compl_type)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
142 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
143 let s:sql_file_{compl_type} = DB_getDictionaryName(compl_type)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
144 if s:sql_file_{compl_type} != ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
145 if filereadable(s:sql_file_{compl_type})
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
146 let compl_list = readfile(s:sql_file_{compl_type})
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
147 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
148 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
149 elseif compl_type == 'column'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
150
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
151 " This type of completion relies upon the dbext.vim plugin
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
152 if s:SQLCCheck4dbext() == -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
153 return []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
154 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
155
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
156 if base == ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
157 " The last time we displayed a column list we stored
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
158 " the table name. If the user selects a column list
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
159 " without a table name of alias present, assume they want
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
160 " the previous column list displayed.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
161 let base = s:save_prev_table
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
162 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
163
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
164 if base != ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
165 let compl_list = s:SQLCGetColumns(base, '')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
166 let s:save_prev_table = base
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
167 let base = ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
168 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
169 elseif compl_type == 'column_csv'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
170
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
171 " This type of completion relies upon the dbext.vim plugin
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
172 if s:SQLCCheck4dbext() == -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
173 return []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
174 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
175
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
176 if base == ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
177 " The last time we displayed a column list we stored
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
178 " the table name. If the user selects a column list
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
179 " without a table name of alias present, assume they want
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
180 " the previous column list displayed.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
181 let base = s:save_prev_table
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
182 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
183
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
184 if base != ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
185 let compl_list = s:SQLCGetColumns(base, 'csv')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
186 let s:save_prev_table = base
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
187 " Join the column array into 1 single element array
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
188 " but make the columns column separated
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
189 let compl_list = [join(compl_list, ', ')]
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
190 let base = ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
191 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
192 elseif compl_type == 'resetCache'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
193 " Reset all cached items
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
194 let s:tbl_name = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
195 let s:tbl_alias = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
196 let s:tbl_cols = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
197 let s:syn_list = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
198 let s:syn_value = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
199 else
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
200 let compl_list = s:SQLCGetSyntaxList(compl_type)
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
201 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
202
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
203 if base != ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
204 " Filter the list based on the first few characters the user
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
205 " entered
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
206 let expr = 'v:val =~ "^'.base.'"'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
207 let compl_list = filter(copy(compl_list), expr)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
208 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
209
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
210 if exists('b:sql_compl_savefunc') && b:sql_compl_savefunc != ""
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
211 let &omnifunc = b:sql_compl_savefunc
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
212 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
213
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
214 return compl_list
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
215 endfunc
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
216
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
217 function! s:SQLCWarningMsg(msg)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
218 echohl WarningMsg
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
219 echomsg a:msg
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
220 echohl None
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
221 endfunction
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
222
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
223 function! s:SQLCErrorMsg(msg)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
224 echohl ErrorMsg
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
225 echomsg a:msg
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
226 echohl None
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
227 endfunction
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
228
819
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
229 function! sqlcomplete#PreCacheSyntax(...)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
230 let syn_group_arr = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
231 if a:0 > 0
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
232 let syn_group_arr = a:1
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
233 else
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
234 let syn_group_arr = g:omni_sql_precache_syntax_groups
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
235 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
236 if !empty(syn_group_arr)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
237 for group_name in syn_group_arr
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
238 call s:SQLCGetSyntaxList(group_name)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
239 endfor
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
240 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
241 endfunction
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
242
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
243 function! sqlcomplete#Map(type)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
244 " Tell the SQL plugin what you want to complete
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
245 let b:sql_compl_type=a:type
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
246 " Record previous omnifunc, if the SQL completion
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
247 " is being used in conjunction with other filetype
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
248 " completion plugins
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
249 if &omnifunc != "" && &omnifunc != 'sqlcomplete#Complete'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
250 " Record the previous omnifunc, the plugin
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
251 " will automatically set this back so that it
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
252 " does not interfere with other ftplugins settings
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
253 let b:sql_compl_savefunc=&omnifunc
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
254 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
255 " Set the OMNI func for the SQL completion plugin
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
256 let &omnifunc='sqlcomplete#Complete'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
257 endfunction
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
258
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
259 function! s:SQLCGetSyntaxList(syn_group)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
260 let syn_group = a:syn_group
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
261 let compl_list = []
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
262
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
263 " Check if we have already cached the syntax list
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
264 let list_idx = index(s:syn_list, syn_group, 0, &ignorecase)
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
265 if list_idx > -1
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
266 " Return previously cached value
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
267 let compl_list = s:syn_value[list_idx]
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
268 else
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
269 " Request the syntax list items from the
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
270 " syntax completion plugin
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
271 if syn_group == 'syntax'
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
272 " Handle this special case. This allows the user
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
273 " to indicate they want all the syntax items available,
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
274 " so do not specify a specific include list.
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
275 let g:omni_syntax_group_include_sql = ''
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
276 else
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
277 " The user has specified a specific syntax group
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
278 let g:omni_syntax_group_include_sql = syn_group
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
279 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
280 let g:omni_syntax_group_exclude_sql = ''
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
281 let syn_value = OmniSyntaxList()
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
282 let g:omni_syntax_group_include_sql = s:save_inc
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
283 let g:omni_syntax_group_exclude_sql = s:save_exc
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
284 " Cache these values for later use
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
285 let s:syn_list = add( s:syn_list, syn_group )
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
286 let s:syn_value = add( s:syn_value, syn_value )
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
287 let compl_list = syn_value
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
288 endif
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
289
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
290 return compl_list
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
291 endfunction
23f82b5d2814 updated for version 7.0c10
vimboss
parents: 818
diff changeset
292
818
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
293 function! s:SQLCCheck4dbext()
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
294 if !exists('g:loaded_dbext')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
295 let msg = "The dbext plugin must be loaded for dynamic SQL completion"
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
296 call s:SQLCErrorMsg(msg)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
297 " Leave time for the user to read the error message
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
298 :sleep 2
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
299 return -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
300 elseif g:loaded_dbext < 210
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
301 let msg = "The dbext plugin must be at least version 2.10 " .
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
302 \ " for dynamic SQL completion"
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
303 call s:SQLCErrorMsg(msg)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
304 " Leave time for the user to read the error message
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
305 :sleep 2
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
306 return -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
307 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
308 return 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
309 endfunction
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
310
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
311 function! s:SQLCAddAlias(table_name, table_alias, cols)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
312 let table_name = a:table_name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
313 let table_alias = a:table_alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
314 let cols = a:cols
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
315
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
316 if g:omni_sql_use_tbl_alias != 'n'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
317 if table_alias == ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
318 if 'da' =~? g:omni_sql_use_tbl_alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
319 if table_name =~ '_'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
320 " Treat _ as separators since people often use these
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
321 " for word separators
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
322 let save_keyword = &iskeyword
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
323 setlocal iskeyword-=_
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
324
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
325 " Get the first letter of each word
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
326 " [[:alpha:]] is used instead of \w
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
327 " to catch extended accented characters
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
328 "
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
329 let table_alias = substitute(
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
330 \ table_name,
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
331 \ '\<[[:alpha:]]\+\>_\?',
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
332 \ '\=strpart(submatch(0), 0, 1)',
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
333 \ 'g'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
334 \ )
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
335 " Restore original value
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
336 let &iskeyword = save_keyword
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
337 elseif table_name =~ '\u\U'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
338 let initials = substitute(
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
339 \ table_name, '\(\u\)\U*', '\1', 'g')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
340 else
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
341 let table_alias = strpart(table_name, 0, 1)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
342 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
343 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
344 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
345 if table_alias != ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
346 " Following a word character, make sure there is a . and no spaces
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
347 let table_alias = substitute(table_alias, '\w\zs\.\?\s*$', '.', '')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
348 if 'a' =~? g:omni_sql_use_tbl_alias && a:table_alias == ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
349 let table_alias = inputdialog("Enter table alias:", table_alias)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
350 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
351 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
352 if table_alias != ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
353 let cols = substitute(cols, '\<\w', table_alias.'&', 'g')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
354 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
355 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
356
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
357 return cols
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
358 endfunction
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
359
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
360 function! s:SQLCGetColumns(table_name, list_type)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
361 let table_name = matchstr(a:table_name, '^\w\+')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
362 let table_cols = []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
363 let table_alias = ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
364 let move_to_top = 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
365
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
366 if g:loaded_dbext >= 210
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
367 let saveSettingAlias = DB_listOption('use_tbl_alias')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
368 exec 'DBSetOption use_tbl_alias=n'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
369 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
370
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
371 " Check if we have already cached the column list for this table
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
372 " by its name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
373 let list_idx = index(s:tbl_name, table_name, 0, &ignorecase)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
374 if list_idx > -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
375 let table_cols = split(s:tbl_cols[list_idx])
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
376 else
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
377 " Check if we have already cached the column list for this table
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
378 " by its alias, assuming the table_name provided was actually
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
379 " the alias for the table instead
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
380 " select *
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
381 " from area a
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
382 " where a.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
383 let list_idx = index(s:tbl_alias, table_name, 0, &ignorecase)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
384 if list_idx > -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
385 let table_alias = table_name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
386 let table_name = s:tbl_name[list_idx]
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
387 let table_cols = split(s:tbl_cols[list_idx])
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
388 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
389 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
390
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
391 " If we have not found a cached copy of the table
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
392 " And the table ends in a "." or we are looking for a column list
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
393 " if list_idx == -1 && (a:table_name =~ '\.' || b:sql_compl_type =~ 'column')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
394 " if list_idx == -1 && (a:table_name =~ '\.' || a:list_type =~ 'csv')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
395 if list_idx == -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
396 let saveY = @y
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
397 let saveSearch = @/
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
398 let saveWScan = &wrapscan
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
399 let curline = line(".")
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
400 let curcol = col(".")
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
401
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
402 " Do not let searchs wrap
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
403 setlocal nowrapscan
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
404 " If . was entered, look at the word just before the .
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
405 " We are looking for something like this:
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
406 " select *
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
407 " from customer c
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
408 " where c.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
409 " So when . is pressed, we need to find 'c'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
410 "
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
411
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
412 " Search backwards to the beginning of the statement
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
413 " and do NOT wrap
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
414 " exec 'silent! normal! v?\<\(select\|update\|delete\|;\)\>'."\n".'"yy'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
415 exec 'silent! normal! ?\<\(select\|update\|delete\|;\)\>'."\n"
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
416
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
417 " Start characterwise visual mode
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
418 " Advance right one character
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
419 " Search foward until one of the following:
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
420 " 1. Another select/update/delete statement
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
421 " 2. A ; at the end of a line (the delimiter)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
422 " 3. The end of the file (incase no delimiter)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
423 " Yank the visually selected text into the "y register.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
424 exec 'silent! normal! vl/\(\<select\>\|\<update\>\|\<delete\>\|;\s*$\|\%$\)'."\n".'"yy'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
425
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
426 let query = @y
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
427 let query = substitute(query, "\n", ' ', 'g')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
428 let found = 0
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
429
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
430 " if query =~? '^\(select\|update\|delete\)'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
431 if query =~? '^\(select\)'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
432 let found = 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
433 " \(\(\<\w\+\>\)\.\)\? -
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
434 " 'from.\{-}' - Starting at the from clause
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
435 " '\zs\(\(\<\w\+\>\)\.\)\?' - Get the owner name (optional)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
436 " '\<\w\+\>\ze' - Get the table name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
437 " '\s\+\<'.table_name.'\>' - Followed by the alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
438 " '\s*\.\@!.*' - Cannot be followed by a .
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
439 " '\(\<where\>\|$\)' - Must be followed by a WHERE clause
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
440 " '.*' - Exclude the rest of the line in the match
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
441 let table_name_new = matchstr(@y,
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
442 \ 'from.\{-}'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
443 \ '\zs\(\(\<\w\+\>\)\.\)\?'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
444 \ '\<\w\+\>\ze'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
445 \ '\s\+\%(as\s\+\)\?\<'.table_name.'\>'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
446 \ '\s*\.\@!.*'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
447 \ '\(\<where\>\|$\)'.
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
448 \ '.*'
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
449 \ )
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
450 if table_name_new != ''
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
451 let table_alias = table_name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
452 let table_name = table_name_new
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
453
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
454 let list_idx = index(s:tbl_name, table_name, 0, &ignorecase)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
455 if list_idx > -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
456 let table_cols = split(s:tbl_cols[list_idx])
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
457 let s:tbl_name[list_idx] = table_name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
458 let s:tbl_alias[list_idx] = table_alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
459 else
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
460 let list_idx = index(s:tbl_alias, table_name, 0, &ignorecase)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
461 if list_idx > -1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
462 let table_cols = split(s:tbl_cols[list_idx])
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
463 let s:tbl_name[list_idx] = table_name
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
464 let s:tbl_alias[list_idx] = table_alias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
465 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
466 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
467
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
468 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
469 else
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
470 " Simply assume it is a table name provided with a . on the end
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
471 let found = 1
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
472 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
473
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
474 let @y = saveY
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
475 let @/ = saveSearch
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
476 let &wrapscan = saveWScan
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
477
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
478 " Return to previous location
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
479 call cursor(curline, curcol)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
480
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
481 if found == 0
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
482 if g:loaded_dbext > 201
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
483 exec 'DBSetOption use_tbl_alias='.saveSettingAlias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
484 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
485
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
486 " Not a SQL statement, do not display a list
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
487 return []
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
488 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
489 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
490
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
491 if empty(table_cols)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
492 " Specify silent mode, no messages to the user (tbl, 1)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
493 " Specify do not comma separate (tbl, 1, 1)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
494 let table_cols_str = DB_getListColumn(table_name, 1, 1)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
495
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
496 if table_cols_str != ""
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
497 let s:tbl_name = add( s:tbl_name, table_name )
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
498 let s:tbl_alias = add( s:tbl_alias, table_alias )
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
499 let s:tbl_cols = add( s:tbl_cols, table_cols_str )
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
500 let table_cols = split(table_cols_str)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
501 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
502
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
503 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
504
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
505 if g:loaded_dbext > 201
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
506 exec 'DBSetOption use_tbl_alias='.saveSettingAlias
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
507 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
508
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
509 if a:list_type == 'csv' && !empty(table_cols)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
510 let cols = join(table_cols, ', ')
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
511 let cols = s:SQLCAddAlias(table_name, table_alias, cols)
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
512 let table_cols = [cols]
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
513 endif
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
514
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
515 return table_cols
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
516 endfunction
1f929f3ca806 updated for version 7.0c03
vimboss
parents:
diff changeset
517