From 15c1ff399116ecd5c019f94cf84343628f5866f3 Mon Sep 17 00:00:00 2001
From: Meow King <mr.meowking@anche.no>
Date: Thu, 8 Aug 2024 14:45:10 +0800
Subject: [PATCH] chore: genreate stub files for bit_reader and disassembler
 module

---
 CMakeLists.txt | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index b18cffc..9f125bc 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -186,6 +186,22 @@ nanobind_add_stub(
   DEPENDS llvmpym_ext
 )
 
+nanobind_add_stub(
+  llvmpym_ext_stub_bit_reader
+  MODULE llvmpym_ext.bit_reader
+  OUTPUT bit_reader.pyi
+  PYTHON_PATH $<TARGET_FILE_DIR:llvmpym_ext>
+  DEPENDS llvmpym_ext
+)
+
+nanobind_add_stub(
+  llvmpym_ext_stub_disassembler
+  MODULE llvmpym_ext.disassembler
+  OUTPUT disassembler.pyi
+  PYTHON_PATH $<TARGET_FILE_DIR:llvmpym_ext>
+  DEPENDS llvmpym_ext
+)
+
 
 # Install directive for scikit-build-core
 install(TARGETS llvmpym_ext LIBRARY DESTINATION ${SKBUILD_PROJECT_NAME})
@@ -200,5 +216,7 @@ install(FILES
   ${CMAKE_BINARY_DIR}/analysis.pyi
   ${CMAKE_BINARY_DIR}/target.pyi
   ${CMAKE_BINARY_DIR}/target_machine.pyi
+  ${CMAKE_BINARY_DIR}/bit_reader.pyi
+  ${CMAKE_BINARY_DIR}/disassembler.pyi
   
   DESTINATION ${SKBUILD_PROJECT_NAME}/llvmpym_ext)