mirror of
https://github.com/mrexodia/phnt-single-header
synced 2026-06-08 16:10:51 +00:00
sup817ch
95 lines
3.5 KiB
Python
95 lines
3.5 KiB
Python
import os
|
|
import sys
|
|
import hashlib
|
|
import subprocess
|
|
import urllib.request
|
|
|
|
def exec(command, working_directory=None):
|
|
result = subprocess.run(
|
|
command,
|
|
shell=True,
|
|
cwd=working_directory
|
|
)
|
|
if result.returncode != 0:
|
|
raise Exception(f"Command '{command}' failed with exit code {result.returncode}")
|
|
|
|
def download_file(url, save_path):
|
|
with urllib.request.urlopen(url) as response:
|
|
with open(save_path, 'wb') as file:
|
|
file.write(response.read())
|
|
|
|
def sha256(file_path):
|
|
hash = hashlib.sha256()
|
|
with open(file_path, "rb") as file:
|
|
# Read the file in chunks to handle large files efficiently
|
|
chunk = 0
|
|
while chunk := file.read(8192):
|
|
hash.update(chunk)
|
|
return hash.hexdigest()
|
|
|
|
def main():
|
|
# Sanity checks
|
|
if not sys.platform.startswith("win"):
|
|
raise Exception("This script only works on Windows...")
|
|
if not os.path.exists(".git"):
|
|
raise Exception("Clone this repository with git to use it...")
|
|
|
|
# Get the current System Informer commit hash
|
|
if not os.path.exists("systeminformer"):
|
|
exec("git submodule update --init")
|
|
commit = subprocess.run(
|
|
"git rev-parse HEAD",
|
|
stdout=subprocess.PIPE,
|
|
text=True,
|
|
shell=True,
|
|
cwd="systeminformer"
|
|
)
|
|
if commit.returncode != 0:
|
|
raise Exception(f"Failed to get commit hash, exit code {commit.returncode}")
|
|
commit_hash = commit.stdout.strip()
|
|
|
|
# Download cpp-amalgamate
|
|
cpp_amalgamate = os.path.join(os.getcwd(), "cpp-amalgamate.exe")
|
|
if not os.path.exists(cpp_amalgamate):
|
|
print(f"Downloading cpp-amalgamate...")
|
|
download_file("https://github.com/Felerius/cpp-amalgamate/releases/download/1.0.1/cpp-amalgamate-x86_64-pc-windows-gnu.exe", cpp_amalgamate)
|
|
actual_hash = sha256(cpp_amalgamate)
|
|
expected_hash = "cdb689a610b67f267a1b28733f975431083150f6cd01adfd5914f989508b0522"
|
|
if actual_hash != expected_hash:
|
|
raise Exception(f"cpp-amalgamate.exe hash mismatch (actual: {actual_hash}, expected: {expected_hash})")
|
|
|
|
# Extract System Informer license
|
|
license_file = "systeminformer/LICENSE.txt"
|
|
if not os.path.exists(license_file):
|
|
raise Exception(f"License file not found: {license_file}")
|
|
with open(license_file, "r") as f:
|
|
license = f.read().strip()
|
|
|
|
# Create output folder
|
|
os.makedirs("out", exist_ok=True)
|
|
with open("out/LICENSE", "w") as f:
|
|
f.write(license)
|
|
out_path = "out/phnt.h"
|
|
exec(f"\"{cpp_amalgamate}\" -d systeminformer/phnt/include phnt_amalgamate.h -o \"{out_path}\"")
|
|
with open(out_path, "r") as f:
|
|
header = f.read()
|
|
header = header.replace(
|
|
"#if __has_include (",
|
|
"#if 1 // __has_include ("
|
|
)
|
|
with open(out_path, "w") as f:
|
|
f.write("/*\n\n")
|
|
f.write("+===========================================================+\n")
|
|
f.write("| THIS FILE WAS AUTOMATICALLY GENERATED |\n")
|
|
f.write("+===========================================================+\n")
|
|
f.write("| Source: https://github.com/winsiderss/systeminformer |\n")
|
|
f.write(f"| Commit: {commit_hash} |\n")
|
|
f.write("| Generator: https://github.com/mrexodia/phnt-single-header |\n")
|
|
f.write("+===========================================================+\n\n")
|
|
f.write(license)
|
|
f.write("\n*/\n\n")
|
|
f.write(header)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|