code-intelligence 3.2.0__py3-none-win_amd64.whl → 3.3.1__py3-none-win_amd64.whl

code_intelligence/__init__.py

@@ -3,6 +3,7 @@ from .goober import (
     Definition_Link,
     scan_file,
     find_symbols_in_text,
+    init,
     stop,
     clear,
 )
@@ -12,7 +13,7 @@ from .goober import (
 # SHAME   - Bump when fixing things too embarrassing to admit.
 #       PROUD.DEFAULT.SHAME
 #            \   |   /
-__version__ = "3.2.0"
+__version__ = "3.3.1"
 __version_info__ = tuple(int(i) for i in __version__.split('.'))
 
 __all__ = [
@@ -20,6 +21,7 @@ __all__ = [
     "Definition_Link",
     "scan_file",
     "find_symbols_in_text",
+    "init",
     "stop",
     "clear",
 ]

code_intelligence/goober.py

@@ -41,6 +41,9 @@ _lib.enqueue_stop.restype = None
 _lib.enqueue_clear.argtypes = []
 _lib.enqueue_clear.restype = None
 
+_lib.start_workers.argtypes = []
+_lib.start_workers.restype = None
+
 _lib.get_pending_tasks_count.argtypes = []
 _lib.get_pending_tasks_count.restype = C.c_uint64
 
@@ -76,33 +79,31 @@ class Definition_Link:
 C_NO_CALLBACK = SCAN_CALLBACK(0)
 
 
-def _wrap_data_ptr(callback, data):
-    full_data = (callback, data)
-
+def _wrap_data_ptr(tup: tuple):
     # MANUALLY INCREMENT THE REFERENCE COUNT
     # We are now responsible for its lifetime
-    C.pythonapi.Py_IncRef(full_data)
+    C.pythonapi.Py_IncRef(C.py_object(tup))
 
-    return id(full_data)
+    return id(tup) # C.cast(obj, C.c_void_p)
 
 
-def _unwrap_data_ptr(data_ptr):
+def _unwrap_data_ptr(data_ptr) -> tuple:
     assert data_ptr != 0, "we got NULL"
 
-    full_data = C.cast(data_ptr, C.py_object).value
+    tup = C.cast(data_ptr, C.py_object).value
 
     # MANUALLY DECREMENT THE REFERENCE COUNT.
     # We are fulfilling our promise
-    C.pythonapi.Py_DecRef(full_data)
+    C.pythonapi.Py_DecRef(C.py_object(tup))
 
-    return full_data
+    return tup
 
 
 def scan_file(file, language: Language, callback=None, data=None) -> bool:
     if callback is None:
         c_callback, data_ptr = C_NO_CALLBACK, 0
     else:
-        c_callback, data_ptr = C_AFTER_SCAN, _wrap_data_ptr(callback, data)
+        c_callback, data_ptr = C_AFTER_SCAN, _wrap_data_ptr((callback, data))
 
     # print(f"[PY] sending next request {file}")
     _lib.enqueue_scan_file(c_callback, data_ptr, os.fsencode(file), language)
@@ -122,7 +123,7 @@ def find_symbols_in_text(text: str, callback=None, data=None) -> bool:
     if callback is None:
         c_callback, data_ptr = C_NO_CALLBACK, 0
     else:
-        c_callback, data_ptr = C_AFTER_FIND, _wrap_data_ptr(callback, data)
+        c_callback, data_ptr = C_AFTER_FIND, _wrap_data_ptr((callback, data))
 
     # print("[PY] sending find_symbols_in_text")
     _lib.enqueue_find_symbols_in_text(c_callback, data_ptr, text.encode())
@@ -156,6 +157,10 @@ def pending_tasks() -> int:
     return _lib.get_pending_tasks_count()
 
 
+def init():
+    _lib.start_workers()
+
+
 def stop():
     _lib.enqueue_stop()
 
@@ -199,9 +204,9 @@ if __name__ == "__main__":
         path = Path(folder)
         if path.is_dir():
             for file in path.rglob("*.py"):
-                scan_file(file, scan_callback, file)
+                scan_file(file, Language.PYTHON, scan_callback, file)
         else:
-            scan_file(path, scan_callback, path)
+            scan_file(path, Language.PYTHON, scan_callback, path)
 
     text = sys.argv[1]
     find_symbols_in_text(text, find_callback, data=(text, "before"))

code_intelligence/libcode_intelligence.h

@@ -108,6 +108,7 @@ typedef struct { void *data; GoInt len; GoInt cap; } GoSlice;
 extern "C" {
 #endif
 
+extern __declspec(dllexport) void start_workers(void);
 extern __declspec(dllexport) void enqueue_scan_file(scan_callback_t callback, uintptr_t data, char* c_path, GoUint64 language);
 extern __declspec(dllexport) void enqueue_find_symbols_in_text(find_callback_t callback, uintptr_t data, char* c_text);
 extern __declspec(dllexport) void enqueue_stop(void);

{code_intelligence-3.2.0.dist-info → code_intelligence-3.3.1.dist-info}/METADATA

@@ -1,4 +1,4 @@
 Metadata-Version: 2.1
 Name: code_intelligence
-Version: 3.2.0
+Version: 3.3.1
 Summary: Code intelligence library

code_intelligence-3.3.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+code_intelligence-3.3.1.dist-info/METADATA,sha256=w-DvM6pR4uWVpKsguom5rlXqeWBYdtudT4Rrf5-_AFg,100
+code_intelligence-3.3.1.dist-info/RECORD,,
+code_intelligence-3.3.1.dist-info/WHEEL,sha256=aD7J6E7Slys9i0J65v6Ehfi0DHYa5rImRhUDp62Iwh0,98
+code_intelligence/__init__.py,sha256=iG0BcJz8lbGitiiOrnzZ7R9fIY9blHehoQ9Kmpu1PBw,587
+code_intelligence/goober.py,sha256=4MYEegk26-GfaOOsd8ufREYqFRq0nNmDs6jiBq-kD6s,6510
+code_intelligence/libcode_intelligence.h,sha256=9iy0xXcOpq366DuiKw1cw4-7VNz-QPAYBNJGZKEh12M,3169
+code_intelligence/libcode_intelligence.so,sha256=gbNPCTEzVkKwp1m7h16dQqyWvnhosxpXCWEUbRTqwh0,2840576
+code_intelligence/symdb.py,sha256=gRq3-xr2oYIzudgwDwMj3Zl5Pb0B8tDdnGKUB0WIV_o,32427

code_intelligence-3.2.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-code_intelligence-3.2.0.dist-info/METADATA,sha256=V1bD7fBzmNxapeMHm7mj6uXPfKdeZkUut0CMoFw08K4,100
-code_intelligence-3.2.0.dist-info/RECORD,,
-code_intelligence-3.2.0.dist-info/WHEEL,sha256=aD7J6E7Slys9i0J65v6Ehfi0DHYa5rImRhUDp62Iwh0,98
-code_intelligence/__init__.py,sha256=GeMGefLs44gKys7KF9QBqaUKip-vahzACFkgAQ1WO50,563
-code_intelligence/goober.py,sha256=oKCO03I8_f68xf8E068dL5WCl7JeFhnlS_NrRVyzSA0,6367
-code_intelligence/libcode_intelligence.h,sha256=GE8YrqsyrBQbWRIb0FfiWPRCG5W6c5HcZ3sjIOlp2bU,3114
-code_intelligence/libcode_intelligence.so,sha256=eyZbp26Dt45Zoj-Kzgr9lsoylOzZzVqZI4xu0ht2bLM,2839552
-code_intelligence/symdb.py,sha256=gRq3-xr2oYIzudgwDwMj3Zl5Pb0B8tDdnGKUB0WIV_o,32427