X-Git-Url: https://git.rmz.io/dotfiles.git/blobdiff_plain/de55c1c46e484a81bed3730f7773d6745ef4405c..b51f3563d6cb0010825b2d2b9fe01e687f54345a:/vim/ycm_extra_conf.py

diff --git a/vim/ycm_extra_conf.py b/vim/ycm_extra_conf.py
index f02f017..2bd7b4e 100644
--- a/vim/ycm_extra_conf.py
+++ b/vim/ycm_extra_conf.py
@@ -1,17 +1,32 @@
 # 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.path
+from glob import glob
 import logging
 import ycm_core
+import difflib
 
 BASE_FLAGS = [
     '-Wall',
     '-std=c++1z',
-    '-xc++',
-    '-I/usr/lib/'
-    '-I/usr/include/'
+    '-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 = [
@@ -30,20 +45,31 @@ HEADER_EXTENSIONS = [
     '.hh'
 ]
 
+
+def similarity_ratio(s, t):
+    return difflib.SequenceMatcher(a=s.lower(), b=t.lower()).ratio()
+
+
+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
-    import Levenshtein
     logging.info("Trying to find some file close to: " + filename)
     db = json.load(open(dbpath))
     best_filename = ''
-    best_distance = 1 << 31
+    best_ratio = 0
     for entry in db:
-        entry_filename = os.path.normpath(
-            os.path.join(entry["directory"], entry["file"]))
-        distance = Levenshtein.distance(str(filename), str(entry_filename))
-        if distance < best_distance:
+        entry_filename = os.path.normpath(os.path.join(entry["directory"],
+                                                       entry["file"]))
+        ratio = similarity_ratio(str(filename), str(entry_filename))
+        if ratio > best_ratio:
             best_filename = entry_filename
-            best_distance = distance
+            best_ratio = ratio
     return best_filename
 
 def ok_compilation_info(info):
@@ -68,6 +94,24 @@ def get_compilation_info_for_file(dbpath, database, 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 find_nearest(path, target):
     candidates = [
         os.path.join(path, target),
@@ -83,6 +127,7 @@ def find_nearest(path, target):
         raise RuntimeError("Could not find " + target)
     return find_nearest(parent, target)
 
+
 def make_relative_paths_in_flags_absolute(flags, working_directory):
     if not working_directory:
         return list(flags)
@@ -107,15 +152,6 @@ def make_relative_paths_in_flags_absolute(flags, working_directory):
             new_flags.append(new_flag)
     return new_flags
 
-def flags_for_clang_complete(root):
-    try:
-        clang_complete_path = find_nearest(root, '.clang_complete')
-        clang_complete_flags = open(clang_complete_path, 'r').read().splitlines()
-        return clang_complete_flags
-    except Exception, err:
-        logging.info("Error while looking flags for .clang_complete in root: " + root)
-        logging.error(err)
-        return None
 
 def flags_for_include(root):
     try:
@@ -126,14 +162,15 @@ def flags_for_include(root):
                 real_path = os.path.join(dirroot, dir_path)
                 flags = flags + ["-I" + real_path]
         return flags
-    except Exception, err:
+    except Exception as err:
         logging.info("Error while looking flags for includes in root: " + root)
         logging.error(err)
         return None
 
+
 def flags_for_compilation_database(root, filename):
     try:
-        compilation_db_path = find_nearest(root, 'compile_commands.json')
+        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)
@@ -148,27 +185,31 @@ def flags_for_compilation_database(root, filename):
         return make_relative_paths_in_flags_absolute(
             compilation_info.compiler_flags_,
             compilation_info.compiler_working_dir_)
-    except Exception, err:
+    except Exception as err:
         logging.info("Error while trying to get flags for " + filename + " in compilation database")
         logging.error(err)
         return None
 
-def flags_for_file(filename):
-    root = os.path.realpath(filename)
+
+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
-        clang_flags = flags_for_clang_complete(root)
-        if clang_flags:
-            final_flags = final_flags + clang_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,
+        'flags': final_flags + EXTRA_FLAGS,
         'do_cache': True
     }
-
-FlagsForFile = flags_for_file