forked from aniani/vim
		
	
		
			
				
	
	
		
			873 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			VimL
		
	
	
	
	
	
			
		
		
	
	
			873 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			VimL
		
	
	
	
	
	
| " Vim completion script
 | |
| " Language:		Ruby
 | |
| " Maintainer:		Mark Guzman <segfault@hasno.info>
 | |
| " URL:			https://github.com/vim-ruby/vim-ruby
 | |
| " Release Coordinator:	Doug Kearns <dougkearns@gmail.com>
 | |
| " Last Change:		2020 Apr 12
 | |
| " ----------------------------------------------------------------------------
 | |
| "
 | |
| " Ruby IRB/Complete author: Keiju ISHITSUKA(keiju@ishitsuka.com)
 | |
| " ----------------------------------------------------------------------------
 | |
| 
 | |
| " {{{ requirement checks
 | |
| 
 | |
| function! s:ErrMsg(msg)
 | |
|     echohl ErrorMsg
 | |
|     echo a:msg
 | |
|     echohl None
 | |
| endfunction
 | |
| 
 | |
| if !has('ruby')
 | |
|     call s:ErrMsg( "Error: Rubycomplete requires vim compiled with +ruby" )
 | |
|     call s:ErrMsg( "Error: falling back to syntax completion" )
 | |
|     " lets fall back to syntax completion
 | |
|     setlocal omnifunc=syntaxcomplete#Complete
 | |
|     finish
 | |
| endif
 | |
| 
 | |
| if version < 700
 | |
|     call s:ErrMsg( "Error: Required vim >= 7.0" )
 | |
|     finish
 | |
| endif
 | |
| " }}} requirement checks
 | |
| 
 | |
| " {{{ configuration failsafe initialization
 | |
| if !exists("g:rubycomplete_rails")
 | |
|     let g:rubycomplete_rails = 0
 | |
| endif
 | |
| 
 | |
| if !exists("g:rubycomplete_classes_in_global")
 | |
|     let g:rubycomplete_classes_in_global = 0
 | |
| endif
 | |
| 
 | |
| if !exists("g:rubycomplete_buffer_loading")
 | |
|     let g:rubycomplete_buffer_loading = 0
 | |
| endif
 | |
| 
 | |
| if !exists("g:rubycomplete_include_object")
 | |
|     let g:rubycomplete_include_object = 0
 | |
| endif
 | |
| 
 | |
| if !exists("g:rubycomplete_include_objectspace")
 | |
|     let g:rubycomplete_include_objectspace = 0
 | |
| endif
 | |
| " }}} configuration failsafe initialization
 | |
| 
 | |
| " {{{ regex patterns
 | |
| 
 | |
| " Regex that defines the start-match for the 'end' keyword.
 | |
| let s:end_start_regex =
 | |
|       \ '\C\%(^\s*\|[=,*/%+\-|;{]\|<<\|>>\|:\s\)\s*\zs' .
 | |
|       \ '\<\%(module\|class\|if\|for\|while\|until\|case\|unless\|begin' .
 | |
|       \   '\|\%(\K\k*[!?]\?\s\+\)\=def\):\@!\>' .
 | |
|       \ '\|\%(^\|[^.:@$]\)\@<=\<do:\@!\>'
 | |
| 
 | |
| " Regex that defines the middle-match for the 'end' keyword.
 | |
| let s:end_middle_regex = '\<\%(ensure\|else\|\%(\%(^\|;\)\s*\)\@<=\<rescue:\@!\>\|when\|elsif\):\@!\>'
 | |
| 
 | |
| " Regex that defines the end-match for the 'end' keyword.
 | |
| let s:end_end_regex = '\%(^\|[^.:@$]\)\@<=\<end:\@!\>'
 | |
| 
 | |
| " }}} regex patterns
 | |
| 
 | |
| " {{{ vim-side support functions
 | |
| let s:rubycomplete_debug = 0
 | |
| 
 | |
| function! s:dprint(msg)
 | |
|     if s:rubycomplete_debug == 1
 | |
|         echom a:msg
 | |
|     endif
 | |
| endfunction
 | |
| 
 | |
| function! s:GetBufferRubyModule(name, ...)
 | |
|     if a:0 == 1
 | |
|         let [snum,enum] = s:GetBufferRubyEntity(a:name, "module", a:1)
 | |
|     else
 | |
|         let [snum,enum] = s:GetBufferRubyEntity(a:name, "module")
 | |
|     endif
 | |
|     return snum . '..' . enum
 | |
| endfunction
 | |
| 
 | |
| function! s:GetBufferRubyClass(name, ...)
 | |
|     if a:0 >= 1
 | |
|         let [snum,enum] = s:GetBufferRubyEntity(a:name, "class", a:1)
 | |
|     else
 | |
|         let [snum,enum] = s:GetBufferRubyEntity(a:name, "class")
 | |
|     endif
 | |
|     return snum . '..' . enum
 | |
| endfunction
 | |
| 
 | |
| function! s:GetBufferRubySingletonMethods(name)
 | |
| endfunction
 | |
| 
 | |
| function! s:GetBufferRubyEntity( name, type, ... )
 | |
|     let lastpos = getpos(".")
 | |
|     let lastline = lastpos
 | |
|     if (a:0 >= 1)
 | |
|         let lastline = [ 0, a:1, 0, 0 ]
 | |
|         call cursor( a:1, 0 )
 | |
|     endif
 | |
| 
 | |
|     let stopline = 1
 | |
| 
 | |
|     let crex = '^\s*\<' . a:type . '\>\s*\<' . escape(a:name, '*') . '\>\s*\(<\s*.*\s*\)\?'
 | |
|     let [lnum,lcol] = searchpos( crex, 'w' )
 | |
|     "let [lnum,lcol] = searchpairpos( crex . '\zs', '', '\(end\|}\)', 'w' )
 | |
| 
 | |
|     if lnum == 0 && lcol == 0
 | |
|         call cursor(lastpos[1], lastpos[2])
 | |
|         return [0,0]
 | |
|     endif
 | |
| 
 | |
|     let curpos = getpos(".")
 | |
|     let [enum,ecol] = searchpairpos( s:end_start_regex, s:end_middle_regex, s:end_end_regex, 'W' )
 | |
|     call cursor(lastpos[1], lastpos[2])
 | |
| 
 | |
|     if lnum > enum
 | |
|         return [0,0]
 | |
|     endif
 | |
|     " we found a the class def
 | |
|     return [lnum,enum]
 | |
| endfunction
 | |
| 
 | |
| function! s:IsInClassDef()
 | |
|     return s:IsPosInClassDef( line('.') )
 | |
| endfunction
 | |
| 
 | |
| function! s:IsPosInClassDef(pos)
 | |
|     let [snum,enum] = s:GetBufferRubyEntity( '.*', "class" )
 | |
|     let ret = 'nil'
 | |
| 
 | |
|     if snum < a:pos && a:pos < enum
 | |
|         let ret = snum . '..' . enum
 | |
|     endif
 | |
| 
 | |
|     return ret
 | |
| endfunction
 | |
| 
 | |
| function! s:IsInComment(pos)
 | |
|     let stack = synstack(a:pos[0], a:pos[1])
 | |
|     if !empty(stack)
 | |
|         return synIDattr(stack[0], 'name') =~ 'ruby\%(.*Comment\|Documentation\)'
 | |
|     else
 | |
|         return 0
 | |
|     endif
 | |
| endfunction
 | |
| 
 | |
| function! s:GetRubyVarType(v)
 | |
|     let stopline = 1
 | |
|     let vtp = ''
 | |
|     let curpos = getpos('.')
 | |
|     let sstr = '^\s*#\s*@var\s*'.escape(a:v, '*').'\>\s\+[^ \t]\+\s*$'
 | |
|     let [lnum,lcol] = searchpos(sstr,'nb',stopline)
 | |
|     if lnum != 0 && lcol != 0
 | |
|         call setpos('.',curpos)
 | |
|         let str = getline(lnum)
 | |
|         let vtp = substitute(str,sstr,'\1','')
 | |
|         return vtp
 | |
|     endif
 | |
|     call setpos('.',curpos)
 | |
|     let ctors = '\(now\|new\|open\|get_instance'
 | |
|     if exists('g:rubycomplete_rails') && g:rubycomplete_rails == 1 && s:rubycomplete_rails_loaded == 1
 | |
|         let ctors = ctors.'\|find\|create'
 | |
|     else
 | |
|     endif
 | |
|     let ctors = ctors.'\)'
 | |
| 
 | |
|     let fstr = '=\s*\([^ \t]\+.' . ctors .'\>\|[\[{"''/]\|%[xwQqr][(\[{@]\|[A-Za-z0-9@:\-()\.]\+...\?\|lambda\|&\)'
 | |
|     let sstr = ''.escape(a:v, '*').'\>\s*[+\-*/]*'.fstr
 | |
|     let pos = searchpos(sstr,'bW')
 | |
|     while pos != [0,0] && s:IsInComment(pos)
 | |
|         let pos = searchpos(sstr,'bW')
 | |
|     endwhile
 | |
|     if pos != [0,0]
 | |
|         let [lnum, col] = pos
 | |
|         let str = matchstr(getline(lnum),fstr,col)
 | |
|         let str = substitute(str,'^=\s*','','')
 | |
| 
 | |
|         call setpos('.',pos)
 | |
|         if str == '"' || str == '''' || stridx(tolower(str), '%q[') != -1
 | |
|             return 'String'
 | |
|         elseif str == '[' || stridx(str, '%w[') != -1
 | |
|             return 'Array'
 | |
|         elseif str == '{'
 | |
|             return 'Hash'
 | |
|         elseif str == '/' || str == '%r{'
 | |
|             return 'Regexp'
 | |
|         elseif strlen(str) >= 4 && stridx(str,'..') != -1
 | |
|             return 'Range'
 | |
|         elseif stridx(str, 'lambda') != -1 || str == '&'
 | |
|             return 'Proc'
 | |
|         elseif strlen(str) > 4
 | |
|             let l = stridx(str,'.')
 | |
|             return str[0:l-1]
 | |
|         end
 | |
|         return ''
 | |
|     endif
 | |
|     call setpos('.',curpos)
 | |
|     return ''
 | |
| endfunction
 | |
| 
 | |
| "}}} vim-side support functions
 | |
| 
 | |
| "{{{ vim-side completion function
 | |
| function! rubycomplete#Init()
 | |
|     execute "ruby VimRubyCompletion.preload_rails"
 | |
| endfunction
 | |
| 
 | |
| function! rubycomplete#Complete(findstart, base)
 | |
|      "findstart = 1 when we need to get the text length
 | |
|     if a:findstart
 | |
|         let line = getline('.')
 | |
|         let idx = col('.')
 | |
|         while idx > 0
 | |
|             let idx -= 1
 | |
|             let c = line[idx-1]
 | |
|             if c =~ '\w'
 | |
|                 continue
 | |
|             elseif ! c =~ '\.'
 | |
|                 let idx = -1
 | |
|                 break
 | |
|             else
 | |
|                 break
 | |
|             endif
 | |
|         endwhile
 | |
| 
 | |
|         return idx
 | |
|     "findstart = 0 when we need to return the list of completions
 | |
|     else
 | |
|         let g:rubycomplete_completions = []
 | |
|         execute "ruby VimRubyCompletion.get_completions('" . a:base . "')"
 | |
|         return g:rubycomplete_completions
 | |
|     endif
 | |
| endfunction
 | |
| "}}} vim-side completion function
 | |
| 
 | |
| "{{{ ruby-side code
 | |
| function! s:DefRuby()
 | |
| ruby << RUBYEOF
 | |
| # {{{ ruby completion
 | |
| 
 | |
| begin
 | |
|     require 'rubygems' # let's assume this is safe...?
 | |
| rescue Exception
 | |
|     #ignore?
 | |
| end
 | |
| class VimRubyCompletion
 | |
| # {{{ constants
 | |
|   @@debug = false
 | |
|   @@ReservedWords = [
 | |
|         "BEGIN", "END",
 | |
|         "alias", "and",
 | |
|         "begin", "break",
 | |
|         "case", "class",
 | |
|         "def", "defined", "do",
 | |
|         "else", "elsif", "end", "ensure",
 | |
|         "false", "for",
 | |
|         "if", "in",
 | |
|         "module",
 | |
|         "next", "nil", "not",
 | |
|         "or",
 | |
|         "redo", "rescue", "retry", "return",
 | |
|         "self", "super",
 | |
|         "then", "true",
 | |
|         "undef", "unless", "until",
 | |
|         "when", "while",
 | |
|         "yield",
 | |
|       ]
 | |
| 
 | |
|   @@Operators = [ "%", "&", "*", "**", "+",  "-",  "/",
 | |
|         "<", "<<", "<=", "<=>", "==", "===", "=~", ">", ">=", ">>",
 | |
|         "[]", "[]=", "^", ]
 | |
| # }}} constants
 | |
| 
 | |
| # {{{ buffer analysis magic
 | |
|   def load_requires
 | |
| 
 | |
|     custom_paths = VIM::evaluate("get(g:, 'rubycomplete_load_paths', [])")
 | |
| 
 | |
|     if !custom_paths.empty?
 | |
|       $LOAD_PATH.concat(custom_paths).uniq!
 | |
|     end
 | |
| 
 | |
|     buf = VIM::Buffer.current
 | |
|     enum = buf.line_number
 | |
|     nums = Range.new( 1, enum )
 | |
|     nums.each do |x|
 | |
| 
 | |
|       ln = buf[x]
 | |
|       begin
 | |
|         if /.*require_relative\s*(.*)$/.match( ln )
 | |
|           eval( "require %s" % File.expand_path($1) )
 | |
|         elsif /.*require\s*(["'].*?["'])/.match( ln )
 | |
|           eval( "require %s" % $1 )
 | |
|         end
 | |
|       rescue Exception => e
 | |
|         dprint e.inspect
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def load_gems
 | |
|     fpath = VIM::evaluate("get(g:, 'rubycomplete_gemfile_path', 'Gemfile')")
 | |
|     return unless File.file?(fpath) && File.readable?(fpath)
 | |
|     want_bundler = VIM::evaluate("get(g:, 'rubycomplete_use_bundler')")
 | |
|     parse_file = !want_bundler
 | |
|     begin
 | |
|       require 'bundler'
 | |
|       Bundler.setup
 | |
|       Bundler.require
 | |
|     rescue Exception
 | |
|       parse_file = true
 | |
|     end
 | |
|     if parse_file
 | |
|       File.new(fpath).each_line do |line|
 | |
|         begin
 | |
|           require $1 if /\s*gem\s*['"]([^'"]+)/.match(line)
 | |
|         rescue Exception
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def load_buffer_class(name)
 | |
|     dprint "load_buffer_class(%s) START" % name
 | |
|     classdef = get_buffer_entity(name, 's:GetBufferRubyClass("%s")')
 | |
|     return if classdef == nil
 | |
| 
 | |
|     pare = /^\s*class\s*(.*)\s*<\s*(.*)\s*\n/.match( classdef )
 | |
|     load_buffer_class( $2 ) if pare != nil  && $2 != name # load parent class if needed
 | |
| 
 | |
|     mixre = /.*\n\s*(include|prepend)\s*(.*)\s*\n/.match( classdef )
 | |
|     load_buffer_module( $2 ) if mixre != nil && $2 != name # load mixins if needed
 | |
| 
 | |
|     begin
 | |
|       eval classdef
 | |
|     rescue Exception
 | |
|       VIM::evaluate( "s:ErrMsg( 'Problem loading class \"%s\", was it already completed?' )" % name )
 | |
|     end
 | |
|     dprint "load_buffer_class(%s) END" % name
 | |
|   end
 | |
| 
 | |
|   def load_buffer_module(name)
 | |
|     dprint "load_buffer_module(%s) START" % name
 | |
|     classdef = get_buffer_entity(name, 's:GetBufferRubyModule("%s")')
 | |
|     return if classdef == nil
 | |
| 
 | |
|     begin
 | |
|       eval classdef
 | |
|     rescue Exception
 | |
|       VIM::evaluate( "s:ErrMsg( 'Problem loading module \"%s\", was it already completed?' )" % name )
 | |
|     end
 | |
|     dprint "load_buffer_module(%s) END" % name
 | |
|   end
 | |
| 
 | |
|   def get_buffer_entity(name, vimfun)
 | |
|     loading_allowed = VIM::evaluate("exists('g:rubycomplete_buffer_loading') && g:rubycomplete_buffer_loading")
 | |
|     return nil if loading_allowed.to_i.zero?
 | |
|     return nil if /(\"|\')+/.match( name )
 | |
|     buf = VIM::Buffer.current
 | |
|     nums = eval( VIM::evaluate( vimfun % name ) )
 | |
|     return nil if nums == nil
 | |
|     return nil if nums.min == nums.max && nums.min == 0
 | |
| 
 | |
|     dprint "get_buffer_entity START"
 | |
|     visited = []
 | |
|     clscnt = 0
 | |
|     bufname = VIM::Buffer.current.name
 | |
|     classdef = ""
 | |
|     cur_line = VIM::Buffer.current.line_number
 | |
|     while (nums != nil && !(nums.min == 0 && nums.max == 0) )
 | |
|       dprint "visited: %s" % visited.to_s
 | |
|       break if visited.index( nums )
 | |
|       visited << nums
 | |
| 
 | |
|       nums.each do |x|
 | |
|         if x != cur_line
 | |
|           next if x == 0
 | |
|           ln = buf[x]
 | |
|           is_const = false
 | |
|           if /^\s*(module|class|def|include)\s+/.match(ln) || is_const = /^\s*?[A-Z]([A-z]|[1-9])*\s*?[|]{0,2}=\s*?.+\s*?/.match(ln)
 | |
|             clscnt += 1 if /class|module/.match($1)
 | |
|             # We must make sure to load each constant only once to avoid errors
 | |
|             if is_const
 | |
|                 ln.gsub!(/\s*?[|]{0,2}=\s*?/, '||=')
 | |
|             end
 | |
|             #dprint "\$1$1
 | |
|             classdef += "%s\n" % ln
 | |
|             classdef += "end\n" if /def\s+/.match(ln)
 | |
|             dprint ln
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       nm = "%s(::.*)*\", %s, \"" % [ name, nums.last ]
 | |
|       nums = eval( VIM::evaluate( vimfun % nm ) )
 | |
|       dprint "nm: \"%s\"" % nm
 | |
|       dprint "vimfun: %s" % (vimfun % nm)
 | |
|       dprint "got nums: %s" % nums.to_s
 | |
|     end
 | |
|     if classdef.length > 1
 | |
|         classdef += "end\n"*clscnt
 | |
|         # classdef = "class %s\n%s\nend\n" % [ bufname.gsub( /\/|\\/, "_" ), classdef ]
 | |
|     end
 | |
| 
 | |
|     dprint "get_buffer_entity END"
 | |
|     dprint "classdef====start"
 | |
|     lns = classdef.split( "\n" )
 | |
|     lns.each { |x| dprint x }
 | |
|     dprint "classdef====end"
 | |
|     return classdef
 | |
|   end
 | |
| 
 | |
|   def get_var_type( receiver )
 | |
|     if /(\"|\')+/.match( receiver )
 | |
|       "String"
 | |
|     else
 | |
|       VIM::evaluate("s:GetRubyVarType('%s')" % receiver)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def dprint( txt )
 | |
|     print txt if @@debug
 | |
|   end
 | |
| 
 | |
|   def escape_vim_singlequote_string(str)
 | |
|     str.to_s.gsub(/'/,"\\'")
 | |
|   end
 | |
| 
 | |
|   def get_buffer_entity_list( type )
 | |
|     # this will be a little expensive.
 | |
|     loading_allowed = VIM::evaluate("exists('g:rubycomplete_buffer_loading') && g:rubycomplete_buffer_loading")
 | |
|     allow_aggressive_load = VIM::evaluate("exists('g:rubycomplete_classes_in_global') && g:rubycomplete_classes_in_global")
 | |
|     return [] if allow_aggressive_load.to_i.zero? || loading_allowed.to_i.zero?
 | |
| 
 | |
|     buf = VIM::Buffer.current
 | |
|     eob = buf.length
 | |
|     ret = []
 | |
|     rg = 1..eob
 | |
|     re = eval( "/^\s*%s\s*([A-Za-z0-9_:-]*)(\s*<\s*([A-Za-z0-9_:-]*))?\s*/" % type )
 | |
| 
 | |
|     rg.each do |x|
 | |
|       if re.match( buf[x] )
 | |
|         next if type == "def" && eval( VIM::evaluate("s:IsPosInClassDef(%s)" % x) ) != nil
 | |
|         ret.push $1
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     return ret
 | |
|   end
 | |
| 
 | |
|   def get_buffer_modules
 | |
|     return get_buffer_entity_list( "modules" )
 | |
|   end
 | |
| 
 | |
|   def get_buffer_methods
 | |
|     return get_buffer_entity_list( "def" )
 | |
|   end
 | |
| 
 | |
|   def get_buffer_classes
 | |
|     return get_buffer_entity_list( "class" )
 | |
|   end
 | |
| 
 | |
|   def load_rails
 | |
|     allow_rails = VIM::evaluate("exists('g:rubycomplete_rails') && g:rubycomplete_rails")
 | |
|     return if allow_rails.to_i.zero?
 | |
| 
 | |
|     buf_path = VIM::evaluate('expand("%:p")')
 | |
|     file_name = VIM::evaluate('expand("%:t")')
 | |
|     vim_dir = VIM::evaluate('getcwd()')
 | |
|     file_dir = buf_path.gsub( file_name, '' )
 | |
|     file_dir.gsub!( /\\/, "/" )
 | |
|     vim_dir.gsub!( /\\/, "/" )
 | |
|     vim_dir << "/"
 | |
|     dirs = [ vim_dir, file_dir ]
 | |
|     sdirs = [ "", "./", "../", "../../", "../../../", "../../../../" ]
 | |
|     rails_base = nil
 | |
| 
 | |
|     dirs.each do |dir|
 | |
|       sdirs.each do |sub|
 | |
|         trail = "%s%s" % [ dir, sub ]
 | |
|         tcfg = "%sconfig" % trail
 | |
| 
 | |
|         if File.exists?( tcfg )
 | |
|           rails_base = trail
 | |
|           break
 | |
|         end
 | |
|       end
 | |
|       break if rails_base
 | |
|     end
 | |
| 
 | |
|     return if rails_base == nil
 | |
|     $:.push rails_base unless $:.index( rails_base )
 | |
| 
 | |
|     bootfile = rails_base + "config/boot.rb"
 | |
|     envfile = rails_base + "config/environment.rb"
 | |
|     if File.exists?( bootfile ) && File.exists?( envfile )
 | |
|       begin
 | |
|         require bootfile
 | |
|         require envfile
 | |
|         begin
 | |
|           require 'console_app'
 | |
|           require 'console_with_helpers'
 | |
|         rescue Exception
 | |
|           dprint "Rails 1.1+ Error %s" % $!
 | |
|           # assume 1.0
 | |
|         end
 | |
|         #eval( "Rails::Initializer.run" ) #not necessary?
 | |
|         VIM::command('let s:rubycomplete_rails_loaded = 1')
 | |
|         dprint "rails loaded"
 | |
|       rescue Exception
 | |
|         dprint "Rails Error %s" % $!
 | |
|         VIM::evaluate( "s:ErrMsg('Error loading rails environment')" )
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def get_rails_helpers
 | |
|     allow_rails = VIM::evaluate("exists('g:rubycomplete_rails') && g:rubycomplete_rails")
 | |
|     rails_loaded = VIM::evaluate('s:rubycomplete_rails_loaded')
 | |
|     return [] if allow_rails.to_i.zero? || rails_loaded.to_i.zero?
 | |
| 
 | |
|     buf_path = VIM::evaluate('expand("%:p")')
 | |
|     buf_path.gsub!( /\\/, "/" )
 | |
|     path_elm = buf_path.split( "/" )
 | |
|     dprint "buf_path: %s" % buf_path
 | |
|     types = [ "app", "db", "lib", "test", "components", "script" ]
 | |
| 
 | |
|     i = nil
 | |
|     ret = []
 | |
|     type = nil
 | |
|     types.each do |t|
 | |
|       i = path_elm.index( t )
 | |
|       break if i
 | |
|     end
 | |
|     type = path_elm[i]
 | |
|     type.downcase!
 | |
| 
 | |
|     dprint "type: %s" % type
 | |
|     case type
 | |
|       when "app"
 | |
|         i += 1
 | |
|         subtype = path_elm[i]
 | |
|         subtype.downcase!
 | |
| 
 | |
|         dprint "subtype: %s" % subtype
 | |
|         case subtype
 | |
|           when "views"
 | |
|             ret += ActionView::Base.instance_methods
 | |
|             ret += ActionView::Base.methods
 | |
|           when "controllers"
 | |
|             ret += ActionController::Base.instance_methods
 | |
|             ret += ActionController::Base.methods
 | |
|           when "models"
 | |
|             ret += ActiveRecord::Base.instance_methods
 | |
|             ret += ActiveRecord::Base.methods
 | |
|         end
 | |
| 
 | |
|       when "db"
 | |
|         ret += ActiveRecord::ConnectionAdapters::SchemaStatements.instance_methods
 | |
|         ret += ActiveRecord::ConnectionAdapters::SchemaStatements.methods
 | |
|     end
 | |
| 
 | |
|     return ret
 | |
|   end
 | |
| 
 | |
|   def add_rails_columns( cls )
 | |
|     allow_rails = VIM::evaluate("exists('g:rubycomplete_rails') && g:rubycomplete_rails")
 | |
|     rails_loaded = VIM::evaluate('s:rubycomplete_rails_loaded')
 | |
|     return [] if allow_rails.to_i.zero? || rails_loaded.to_i.zero?
 | |
| 
 | |
|     begin
 | |
|         eval( "#{cls}.establish_connection" )
 | |
|         return [] unless eval( "#{cls}.ancestors.include?(ActiveRecord::Base).to_s" )
 | |
|         col = eval( "#{cls}.column_names" )
 | |
|         return col if col
 | |
|     rescue
 | |
|         dprint "add_rails_columns err: (cls: %s) %s" % [ cls, $! ]
 | |
|         return []
 | |
|     end
 | |
|     return []
 | |
|   end
 | |
| 
 | |
|   def clean_sel(sel, msg)
 | |
|     ret = sel.reject{|x|x.nil?}.uniq
 | |
|     ret = ret.grep(/^#{Regexp.quote(msg)}/) if msg != nil
 | |
|     ret
 | |
|   end
 | |
| 
 | |
|   def get_rails_view_methods
 | |
|     allow_rails = VIM::evaluate("exists('g:rubycomplete_rails') && g:rubycomplete_rails")
 | |
|     rails_loaded = VIM::evaluate('s:rubycomplete_rails_loaded')
 | |
|     return [] if allow_rails.to_i.zero? || rails_loaded.to_i.zero?
 | |
| 
 | |
|     buf_path = VIM::evaluate('expand("%:p")')
 | |
|     buf_path.gsub!( /\\/, "/" )
 | |
|     pelm = buf_path.split( "/" )
 | |
|     idx = pelm.index( "views" )
 | |
| 
 | |
|     return [] unless idx
 | |
|     idx += 1
 | |
| 
 | |
|     clspl = pelm[idx].camelize.pluralize
 | |
|     cls = clspl.singularize
 | |
| 
 | |
|     ret = []
 | |
|     begin
 | |
|       ret += eval( "#{cls}.instance_methods" )
 | |
|       ret += eval( "#{clspl}Helper.instance_methods" )
 | |
|     rescue Exception
 | |
|       dprint "Error: Unable to load rails view helpers for %s: %s" % [ cls, $! ]
 | |
|     end
 | |
| 
 | |
|     return ret
 | |
|   end
 | |
| # }}} buffer analysis magic
 | |
| 
 | |
| # {{{ main completion code
 | |
|   def self.preload_rails
 | |
|     a = VimRubyCompletion.new
 | |
|     if VIM::evaluate("has('nvim')") == 0
 | |
|       require 'thread'
 | |
|       Thread.new(a) do |b|
 | |
|         begin
 | |
|         b.load_rails
 | |
|         rescue
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|     a.load_rails
 | |
|   rescue
 | |
|   end
 | |
| 
 | |
|   def self.get_completions(base)
 | |
|     b = VimRubyCompletion.new
 | |
|     b.get_completions base
 | |
|   end
 | |
| 
 | |
|   def get_completions(base)
 | |
|     loading_allowed = VIM::evaluate("exists('g:rubycomplete_buffer_loading') && g:rubycomplete_buffer_loading")
 | |
|     if loading_allowed.to_i == 1
 | |
|       load_requires
 | |
|       load_rails
 | |
|     end
 | |
| 
 | |
|     want_gems = VIM::evaluate("get(g:, 'rubycomplete_load_gemfile')")
 | |
|     load_gems unless want_gems.to_i.zero?
 | |
| 
 | |
|     input = VIM::Buffer.current.line
 | |
|     cpos = VIM::Window.current.cursor[1] - 1
 | |
|     input = input[0..cpos]
 | |
|     input += base
 | |
|     input.sub!(/.*[ \t\n\"\\'`><=;|&{(]/, '') # Readline.basic_word_break_characters
 | |
|     input.sub!(/self\./, '')
 | |
|     input.sub!(/.*((\.\.[\[(]?)|([\[(]))/, '')
 | |
| 
 | |
|     dprint 'input %s' % input
 | |
|     message = nil
 | |
|     receiver = nil
 | |
|     methods = []
 | |
|     variables = []
 | |
|     classes = []
 | |
|     constants = []
 | |
| 
 | |
|     case input
 | |
|       when /^(\/[^\/]*\/)\.([^.]*)$/ # Regexp
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($2)
 | |
|         methods = Regexp.instance_methods(true)
 | |
| 
 | |
|       when /^([^\]]*\])\.([^.]*)$/ # Array
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($2)
 | |
|         methods = Array.instance_methods(true)
 | |
| 
 | |
|       when /^([^\}]*\})\.([^.]*)$/ # Proc or Hash
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($2)
 | |
|         methods = Proc.instance_methods(true) | Hash.instance_methods(true)
 | |
| 
 | |
|       when /^(:[^:.]*)$/ # Symbol
 | |
|         dprint "symbol"
 | |
|         if Symbol.respond_to?(:all_symbols)
 | |
|           receiver = $1
 | |
|           message = $1.sub( /:/, '' )
 | |
|           methods = Symbol.all_symbols.collect{|s| s.id2name}
 | |
|           methods.delete_if { |c| c.match( /'/ ) }
 | |
|         end
 | |
| 
 | |
|       when /^::([A-Z][^:\.\(]*)?$/ # Absolute Constant or class methods
 | |
|         dprint "const or cls"
 | |
|         receiver = $1
 | |
|         methods = Object.constants.collect{ |c| c.to_s }.grep(/^#{receiver}/)
 | |
| 
 | |
|       when /^(((::)?[A-Z][^:.\(]*)+?)::?([^:.]*)$/ # Constant or class methods
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($4)
 | |
|         dprint "const or cls 2 [recv: \'%s\', msg: \'%s\']" % [ receiver, message ]
 | |
|         load_buffer_class( receiver )
 | |
|         load_buffer_module( receiver )
 | |
|         begin
 | |
|           constants = eval("#{receiver}.constants").collect{ |c| c.to_s }.grep(/^#{message}/)
 | |
|           methods = eval("#{receiver}.methods").collect{ |m| m.to_s }.grep(/^#{message}/)
 | |
|         rescue Exception
 | |
|           dprint "exception: %s" % $!
 | |
|           constants = []
 | |
|           methods = []
 | |
|         end
 | |
| 
 | |
|       when /^(:[^:.]+)\.([^.]*)$/ # Symbol
 | |
|         dprint "symbol"
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($2)
 | |
|         methods = Symbol.instance_methods(true)
 | |
| 
 | |
|       when /^([0-9_]+(\.[0-9_]+)?(e[0-9]+)?)\.([^.]*)$/ # Numeric
 | |
|         dprint "numeric"
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($4)
 | |
|         begin
 | |
|           methods = eval(receiver).methods
 | |
|         rescue Exception
 | |
|           methods = []
 | |
|         end
 | |
| 
 | |
|       when /^(\$[^.]*)$/ #global
 | |
|         dprint "global"
 | |
|         methods = global_variables.grep(Regexp.new(Regexp.quote($1)))
 | |
| 
 | |
|       when /^((\.?[^.]+)+?)\.([^.]*)$/ # variable
 | |
|         dprint "variable"
 | |
|         receiver = $1
 | |
|         message = Regexp.quote($3)
 | |
|         load_buffer_class( receiver )
 | |
| 
 | |
|         cv = eval("self.class.constants")
 | |
|         vartype = get_var_type( receiver )
 | |
|         dprint "vartype: %s" % vartype
 | |
| 
 | |
|         invalid_vartype = ['', "gets"]
 | |
|         if !invalid_vartype.include?(vartype)
 | |
|           load_buffer_class( vartype )
 | |
| 
 | |
|           begin
 | |
|             methods = eval("#{vartype}.instance_methods")
 | |
|             variables = eval("#{vartype}.instance_variables")
 | |
|           rescue Exception
 | |
|             dprint "load_buffer_class err: %s" % $!
 | |
|           end
 | |
|         elsif (cv).include?(receiver)
 | |
|           # foo.func and foo is local var.
 | |
|           methods = eval("#{receiver}.methods")
 | |
|           vartype = receiver
 | |
|         elsif /^[A-Z]/ =~ receiver and /\./ !~ receiver
 | |
|           vartype = receiver
 | |
|           # Foo::Bar.func
 | |
|           begin
 | |
|             methods = eval("#{receiver}.methods")
 | |
|           rescue Exception
 | |
|           end
 | |
|         else
 | |
|           # func1.func2
 | |
|           ObjectSpace.each_object(Module){|m|
 | |
|             next if m.name != "IRB::Context" and
 | |
|               /^(IRB|SLex|RubyLex|RubyToken)/ =~ m.name
 | |
|             methods.concat m.instance_methods(false)
 | |
|           }
 | |
|         end
 | |
|         variables += add_rails_columns( "#{vartype}" ) if vartype && !invalid_vartype.include?(vartype)
 | |
| 
 | |
|       when /^\(?\s*[A-Za-z0-9:^@.%\/+*\(\)]+\.\.\.?[A-Za-z0-9:^@.%\/+*\(\)]+\s*\)?\.([^.]*)/
 | |
|         message = $1
 | |
|         methods = Range.instance_methods(true)
 | |
| 
 | |
|       when /^\.([^.]*)$/ # unknown(maybe String)
 | |
|         message = Regexp.quote($1)
 | |
|         methods = String.instance_methods(true)
 | |
| 
 | |
|     else
 | |
|       dprint "default/other"
 | |
|       inclass = eval( VIM::evaluate("s:IsInClassDef()") )
 | |
| 
 | |
|       if inclass != nil
 | |
|         dprint "inclass"
 | |
|         classdef = "%s\n" % VIM::Buffer.current[ inclass.min ]
 | |
|         found = /^\s*class\s*([A-Za-z0-9_-]*)(\s*<\s*([A-Za-z0-9_:-]*))?\s*\n$/.match( classdef )
 | |
| 
 | |
|         if found != nil
 | |
|           receiver = $1
 | |
|           message = input
 | |
|           load_buffer_class( receiver )
 | |
|           begin
 | |
|             methods = eval( "#{receiver}.instance_methods" )
 | |
|             variables += add_rails_columns( "#{receiver}" )
 | |
|           rescue Exception
 | |
|             found = nil
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       if inclass == nil || found == nil
 | |
|         dprint "inclass == nil"
 | |
|         methods = get_buffer_methods
 | |
|         methods += get_rails_view_methods
 | |
| 
 | |
|         cls_const = Class.constants
 | |
|         constants = cls_const.select { |c| /^[A-Z_-]+$/.match( c ) }
 | |
|         classes = eval("self.class.constants") - constants
 | |
|         classes += get_buffer_classes
 | |
|         classes += get_buffer_modules
 | |
| 
 | |
|         include_objectspace = VIM::evaluate("exists('g:rubycomplete_include_objectspace') && g:rubycomplete_include_objectspace")
 | |
|         ObjectSpace.each_object(Class) { |cls| classes << cls.to_s } if include_objectspace == "1"
 | |
|         message = receiver = input
 | |
|       end
 | |
| 
 | |
|       methods += get_rails_helpers
 | |
|       methods += Kernel.public_methods
 | |
|     end
 | |
| 
 | |
|     include_object = VIM::evaluate("exists('g:rubycomplete_include_object') && g:rubycomplete_include_object")
 | |
|     methods = clean_sel( methods, message )
 | |
|     methods = (methods-Object.instance_methods) if include_object == "0"
 | |
|     rbcmeth = (VimRubyCompletion.instance_methods-Object.instance_methods) # lets remove those rubycomplete methods
 | |
|     methods = (methods-rbcmeth)
 | |
| 
 | |
|     variables = clean_sel( variables, message )
 | |
|     classes = clean_sel( classes, message ) - ["VimRubyCompletion"]
 | |
|     constants = clean_sel( constants, message )
 | |
| 
 | |
|     valid = []
 | |
|     valid += methods.collect { |m| { :name => m.to_s, :type => 'm' } }
 | |
|     valid += variables.collect { |v| { :name => v.to_s, :type => 'v' } }
 | |
|     valid += classes.collect { |c| { :name => c.to_s, :type => 't' } }
 | |
|     valid += constants.collect { |d| { :name => d.to_s, :type => 'd' } }
 | |
|     valid.sort! { |x,y| x[:name] <=> y[:name] }
 | |
| 
 | |
|     outp = ""
 | |
| 
 | |
|     rg = 0..valid.length
 | |
|     rg.step(150) do |x|
 | |
|       stpos = 0+x
 | |
|       enpos = 150+x
 | |
|       valid[stpos..enpos].each { |c| outp += "{'word':'%s','item':'%s','kind':'%s'}," % [ c[:name], c[:name], c[:type] ].map{|x|escape_vim_singlequote_string(x)} }
 | |
|       outp.sub!(/,$/, '')
 | |
| 
 | |
|       VIM::command("call extend(g:rubycomplete_completions, [%s])" % outp)
 | |
|       outp = ""
 | |
|     end
 | |
|   end
 | |
| # }}} main completion code
 | |
| 
 | |
| end # VimRubyCompletion
 | |
| # }}} ruby completion
 | |
| RUBYEOF
 | |
| endfunction
 | |
| 
 | |
| let s:rubycomplete_rails_loaded = 0
 | |
| 
 | |
| call s:DefRuby()
 | |
| "}}} ruby-side code
 | |
| 
 | |
| " vim:tw=78:sw=4:ts=8:et:fdm=marker:ft=vim:norl:
 |