1 files changed, 26 insertions, 0 deletions
diff --git a/hooks/bake_version.py b/hooks/bake_version.py
new file mode 100644
index 0000000..4d43228
--- /dev/null
+++ b/hooks/bake_version.py
@@ -0,0 +1,26 @@
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+
+import os
+import re
+
+filename = os.path.normpath(os.path.join("sploit","__init__.py"))
+
+#put the file back when the build ends
+class RestoreVersionFile:
+    def __init__(self,contents):
+        self.contents = contents
+    def __del__(self):
+        with open(filename,"w") as f:
+            f.write(self.contents)
+
+class BakeVersionBuildHook(BuildHookInterface):
+    def initialize(self,version,build_data):
+        with open(filename,"r") as f:
+            self.restore = RestoreVersionFile(f.read())
+        pattern = r'(?i)^__version__ *= *(?P<version>.+?)$'
+        match = re.search(pattern, self.restore.contents, flags=re.MULTILINE)
+        if not match:
+            raise ValueError("regex of version file failed")
+        span = match.span('version')
+        with open(filename,"w") as f:
+            f.write(f'{self.restore.contents[:span[0]]}"v{self.metadata.version}"{self.restore.contents[span[1]:]}')