vim: update ycm_extra_conf with my changes

[dotfiles.git] / vim / ycm_extra_conf.py

diff --git a/vim/ycm_extra_conf.py b/vim/ycm_extra_conf.py
index ac15c2297f5316fdf57ff37e29ffae07cfc694d2..deeac5ee9b1d153070041042e6921a3b6c85247d 100644 (file)
--- a/vim/ycm_extra_conf.py
+++ b/vim/ycm_extra_conf.py
@@ -1,83 +1,231 @@
+# https://github.com/Valloric/ycmd/blob/master/cpp/ycm/.ycm_extra_conf.py
+# https://jonasdevlieghere.com/a-better-youcompleteme-config/
+# https://github.com/arximboldi/dotfiles/blob/master/emacs/.ycm_extra_conf.py
+

 import os
 import os

-import shlex
-import subprocess
+import os.path
+from glob import glob
+import logging

 import ycm_core
 
 import ycm_core
 

-def FlagsForFile( filename, **kwargs ):
-  client_data = kwargs['client_data']
-  cwd = client_data['getcwd()']
-  # These are the compilation flags that will be used in case there's no
-  # compilation database set (by default, one is not set).
-  flags = [
-  '-Wall',
-  # '-Wextra',
-  # '-Wshadow',
-  # '-Werror',
-  # '-Wc++98-compat',
-  # '-Wno-long-long',
-  # '-Wno-variadic-macros',
-  # '-fexceptions',
-  # '-DNDEBUG',
-  '-std=c++11',
-  '-stdlib=libstdc++',
-  '-x', 'c++',
-  '-I', '.',
-  '-I', './include',
-  ]
-  flags += rospack()
-  relative_to = cwd
-  final_flags = MakeRelativePathsInFlagsAbsolute( flags, relative_to )
-
-  return {
-    'flags': final_flags,
-    'do_cache': True
-  }
-
-# This function makes it easy to pull in additional flags from rospack
-def rospack():
-  cmd = ['rospack', 'cflags-only-I']
-  try:
-    out = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE).stdout
-  except:
-    return []
-  line = out.readline()[:-1].split(" ")
-  includes = []
-  for include in line:
-    if include.startswith(os.path.expanduser('~')):
-      includes += ['-I', include]
+BASE_FLAGS = [
+    '-Wall',
+    '-std=c++1z',
+    '-x', 'c++',
+    '-isystem', '/usr/include',
+    '-isystem', '/usr/local/include',
+]
+
+EXTRA_FLAGS = [
+    '-Wall',
+    '-Wextra',
+    # '-Wshadow',
+    # '-Werror',
+    # '-Wc++98-compat',
+    # '-Wno-long-long',
+    # '-Wno-variadic-macros',
+    # '-fexceptions',
+    # '-DNDEBUG',
+]
+
+SOURCE_EXTENSIONS = [
+    '.cpp',
+    '.cxx',
+    '.cc',
+    '.c',
+    '.m',
+    '.mm'
+]
+
+HEADER_EXTENSIONS = [
+    '.h',
+    '.hxx',
+    '.hpp',
+    '.hh'
+]
+
+
+# Implementation taken from
+# https://en.wikibooks.org/wiki/Algorithm_Implementation/Strings/Levenshtein_distance#Python
+def levenshtein(s, t):
+    ''' From Wikipedia article; Iterative with two matrix rows. '''
+    if s == t: return 0
+    elif len(s) == 0: return len(t)
+    elif len(t) == 0: return len(s)
+    v0 = [None] * (len(t) + 1)
+    v1 = [None] * (len(t) + 1)
+    for i in range(len(v0)):
+        v0[i] = i
+    for i in range(len(s)):
+        v1[0] = i + 1
+        for j in range(len(t)):
+            cost = 0 if s[i] == t[j] else 1
+            v1[j + 1] = min(v1[j] + 1, v0[j + 1] + 1, v0[j] + cost)
+        for j in range(len(v0)):
+            v0[j] = v1[j]
+
+        return v1[len(t)]
+
+def generate_qt_flags():
+    flags = ['-isystem', '/usr/include/qt/']
+    for p in glob('/usr/include/qt/*/'):
+        flags += ['-isystem', p]
+    return flags
+
+
+def find_similar_file_in_database(dbpath, filename):
+    import json
+    logging.info("Trying to find some file close to: " + filename)
+    db = json.load(open(dbpath))
+    best_filename = ''
+    best_distance = 1 << 31
+    for entry in db:
+        entry_filename = os.path.normpath(
+            os.path.join(entry["directory"], entry["file"]))
+        distance = levenshtein(str(filename), str(entry_filename))
+        if distance < best_distance:
+            best_filename = entry_filename
+            best_distance = distance
+    return best_filename
+
+def ok_compilation_info(info):
+    return bool(info.compiler_flags_)
+
+def get_compilation_info_for_file(dbpath, database, filename):
+    info = database.GetCompilationInfoForFile(filename)
+    if ok_compilation_info(info):
+        logging.info("Flags for file where found in database: " + filename)
+        return info

     else:
     else:

-      includes += ['-isystem', include]
-  return filter(lambda a: a != ' ', includes)
+        logging.info("Flags for file not found in database: " + filename)
+        basename = os.path.splitext(filename)[0]
+        for extension in SOURCE_EXTENSIONS:
+            replacement_file = basename + extension
+            logging.info("Trying to replace extension with: " + extension)
+            info = database.GetCompilationInfoForFile(replacement_file)
+            if ok_compilation_info(info):
+                logging.info("Replacing header with: " + replacement_file)
+                return info
+        replacement_file = find_similar_file_in_database(dbpath, filename)
+        logging.info("Replacing header with: " + replacement_file)
+        return database.GetCompilationInfoForFile(replacement_file)
+
+
+def find_nearest_compilation_database(root='.'):
+    dirs = glob(root + '/*/compile_commands.json', recursive=True)
+
+    if len(dirs) == 1:
+        return dirs[0]
+    elif len(dirs) > 1:
+        logging.info("Multiple compilation databases found!")
+        logging.info(dirs)
+        logging.info("Selecting first: %s" % (dir))
+        return dirs[0]
+
+    parent = os.path.dirname(os.path.abspath(root))
+    if parent == root:
+        raise RuntimeError("Could not find compile_commands.json")
+    return find_nearest_compilation_database(parent)

 
 
 
 

-def DirectoryOfThisScript():
-  return os.path.dirname( os.path.abspath( __file__ ) )
+def find_nearest(path, target):
+    candidates = [
+        os.path.join(path, target),
+        os.path.join(path, 'build', target),
+        os.path.join(path, 'output', target),
+    ]
+    for candidate in candidates:
+        if os.path.isfile(candidate) or os.path.isdir(candidate):
+            logging.info("Found nearest " + target + " at " + candidate)
+            return candidate
+    parent = os.path.dirname(os.path.abspath(path))
+    if parent == path:
+        raise RuntimeError("Could not find " + target)
+    return find_nearest(parent, target)

 
 

-def MakeRelativePathsInFlagsAbsolute( flags, working_directory ):
-  if not working_directory:
-    return list( flags )
-  new_flags = []
-  make_next_absolute = False
-  path_flags = [ '-isystem', '-I', '-iquote', '--sysroot=' ]
-  for flag in flags:
-    new_flag = flag

 
 

-    if make_next_absolute:
-      make_next_absolute = False
-      if not flag.startswith( '/' ):
-        new_flag = os.path.join( working_directory, flag )
+def make_relative_paths_in_flags_absolute(flags, working_directory):
+    if not working_directory:
+        return list(flags)
+    new_flags = []
+    make_next_absolute = False
+    path_flags = [ '-isystem', '-I', '-iquote', '--sysroot=' ]
+    for flag in flags:
+        new_flag = flag
+        if make_next_absolute:
+            make_next_absolute = False
+            if not flag.startswith('/'):
+                new_flag = os.path.join(working_directory, flag)
+        for path_flag in path_flags:
+            if flag == path_flag:
+                make_next_absolute = True
+                break
+            if flag.startswith(path_flag):
+                path = flag[ len(path_flag): ]
+                new_flag = path_flag + os.path.join(working_directory, path)
+                break
+        if new_flag:
+            new_flags.append(new_flag)
+    return new_flags

 
 

-    for path_flag in path_flags:
-      if flag == path_flag:
-        make_next_absolute = True
-        break

 
 

-      if flag.startswith( path_flag ):
-        path = flag[ len( path_flag ): ]
-        new_flag = path_flag + os.path.join( working_directory, path )
-        break
+def flags_for_include(root):
+    try:
+        include_path = find_nearest(root, 'include')
+        flags = []
+        for dirroot, dirnames, filenames in os.walk(include_path):
+            for dir_path in dirnames:
+                real_path = os.path.join(dirroot, dir_path)
+                flags = flags + ["-I" + real_path]
+        return flags
+    except Exception as err:
+        logging.info("Error while looking flags for includes in root: " + root)
+        logging.error(err)
+        return None

 
 

-    if new_flag:
-      new_flags.append( new_flag )
-  return new_flags

 
 

+def flags_for_compilation_database(root, filename):
+    try:
+        compilation_db_path = find_nearest_compilation_database(root)
+        compilation_db_dir = os.path.dirname(compilation_db_path)
+        logging.info("Set compilation database directory to " + compilation_db_dir)
+        compilation_db = ycm_core.CompilationDatabase(compilation_db_dir)
+        if not compilation_db:
+            logging.info("Compilation database file found but unable to load")
+            return None
+        compilation_info = get_compilation_info_for_file(
+            compilation_db_path, compilation_db, filename)
+        if not compilation_info:
+            logging.info("No compilation info for " + filename + " in compilation database")
+            return None
+        return make_relative_paths_in_flags_absolute(
+            compilation_info.compiler_flags_,
+            compilation_info.compiler_working_dir_)
+    except Exception as err:
+        logging.info("Error while trying to get flags for " + filename + " in compilation database")
+        logging.error(err)
+        return None
+
+
+def FlagsForFile(filename, **kwargs):
+    client_data = kwargs['client_data']
+    root = client_data['getcwd()']
+
+    compilation_db_flags = flags_for_compilation_database(root, filename)
+    if compilation_db_flags:
+        final_flags = compilation_db_flags
+    else:
+        final_flags = BASE_FLAGS
+        include_flags = flags_for_include(root)
+        if include_flags:
+            final_flags = final_flags + include_flags
+
+    final_flags += generate_qt_flags()
+    final_flags += [
+        '-I', root,
+        '-I', root + '/include',
+        ]
+    return {
+        'flags': final_flags + EXTRA_FLAGS,
+        'do_cache': True
+    }