Add support for languages with a dash in its name

Author: cedricrupb

code_ast/ast.py

@@ -33,7 +33,7 @@ def code(self):
     def __repr__(self):
 
         lang = self.config.lang
-        lang_name = lang[0].upper() + lang[1:]
+        lang_name = "".join((lang_part[0].upper() + lang_part[1:] for lang_part in lang.split("-")))
 
         ast_repr = ast_to_str(self.source_tree, indent = 1)
 

code_ast/parsers.py

@@ -50,7 +50,7 @@ def load_language(lang):
     source_lang_path   = os.path.join(cache_path, "tree-sitter-%s" % lang)
 
     if os.path.isfile(compiled_lang_path):
-        return Language(compiled_lang_path, lang)
+        return Language(compiled_lang_path, _lang_to_fnname(lang))
 
     if os.path.exists(source_lang_path) and os.path.isdir(source_lang_path):
         logger.warn("Compiling language for %s" % lang)
@@ -196,6 +196,14 @@ def _compile_lang(source_path, compiled_path):
     )
 
 
+def _lang_to_fnname(lang):
+    """
+    dash is not supported for function names. Therefore,
+    we assume that dashes represented by underscores.
+    """
+    return lang.replace("-", "_")
+
+
 # Auto Clone from Github --------------------------------
 
 def _exists_url(url):