diff --git a/tree_sitter/__init__.py b/tree_sitter/__init__.py index a2cbe4e..7c690de 100644 --- a/tree_sitter/__init__.py +++ b/tree_sitter/__init__.py @@ -1,96 +1,80 @@ from types import MethodType from ctypes import cdll, c_void_p from ctypes.util import find_library from distutils.ccompiler import new_compiler from tempfile import TemporaryDirectory from tree_sitter_binding import Parser, Tree import os.path as path import platform class Language: def build_library(output_path, repo_paths): """ Build a dynamic library at the given path, based on the parser repositories at the given paths. Returns `True` if the dynamic library was compiled and `False` if the library already existed and was modified more recently than any of the source files. """ output_mtime = 0 if path.exists(output_path): output_mtime = path.getmtime(output_path) if len(repo_paths) == 0: raise ValueError('Must provide at least one language folder') cpp = False source_paths = [] source_mtimes = [path.getmtime(__file__)] for repo_path in repo_paths: src_path = path.join(repo_path, 'src') source_paths.append(path.join(src_path, "parser.c")) source_mtimes.append(path.getmtime(source_paths[-1])) if path.exists(path.join(src_path, "scanner.cc")): cpp = True source_paths.append(path.join(src_path, "scanner.cc")) source_mtimes.append(path.getmtime(source_paths[-1])) elif path.exists(path.join(src_path, "scanner.c")): source_paths.append(path.join(src_path, "scanner.c")) source_mtimes.append(path.getmtime(source_paths[-1])) compiler = new_compiler() if cpp: if find_library('c++'): compiler.add_library('c++') elif find_library('stdc++'): compiler.add_library('stdc++') if max(source_mtimes) > output_mtime: with TemporaryDirectory(suffix = 'tree_sitter_language') as dir: object_paths = [] for source_path in source_paths: if platform.system() == 'Windows': flags = None else: flags = ['-fPIC'] if source_path.endswith('.c'): flags.append('-std=c99') object_paths.append(compiler.compile( [source_path], output_dir = dir, include_dirs = [path.dirname(source_path)], extra_preargs = flags )[0]) compiler.link_shared_object(object_paths, output_path) return True else: return False def __init__(self, library_path, name): """ Load the language with the given name from the dynamic library at the given path. """ self.name = name self.lib = cdll.LoadLibrary(library_path) language_function = getattr(self.lib, "tree_sitter_%s" % name) language_function.restype = c_void_p self.language_id = language_function() - - -# def edit(self, *, start_byte, old_end_byte, new_end_byte, start_point, old_end_point, new_end_point): -# self._edit( -# start_byte, -# old_end_byte, -# new_end_byte, -# start_point[0], -# old_end_point[0], -# new_end_point[0], -# start_column[0], -# old_end_column[0], -# new_end_column[0], -# ) -# -# Tree.edit = MethodType(edit, Tree)