polymech - fw latest | web ui

mb-script/scripts/html_to_header.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+import os
+import sys
+import re
+
+def escape_char(char):
+    """Escape special characters for C++ strings"""
+    if char == '\n':
+        return '\\n'
+    elif char == '\r':
+        return '\\r'
+    elif char == '"':
+        return '\\"'
+    elif char == '\\':
+        return '\\\\'
+    return char
+
+def file_to_progmem_string(file_path, var_name):
+    """Convert a file to a C++ PROGMEM string declaration"""
+    with open(file_path, 'r', encoding='utf-8') as f:
+        content = f.read()
+    
+    # Escape special characters
+    escaped_content = ''.join(escape_char(c) for c in content)
+    
+    # Create the C++ code
+    progmem_string = f'const char {var_name}[] PROGMEM = R"rawliteral({escaped_content})rawliteral";'
+    
+    return progmem_string
+
+def create_header_file(input_file, output_file, var_name):
+    """Create a C++ header file with a PROGMEM string variable"""
+    progmem_string = file_to_progmem_string(input_file, var_name)
+    
+    file_name = os.path.basename(output_file).upper().replace('.', '_')
+    guard_name = f"{file_name}_H"
+    
+    header_content = f"""#ifndef {guard_name}
+#define {guard_name}
+
+#include <pgmspace.h>
+
+{progmem_string}
+
+#endif // {guard_name}
+"""
+    
+    with open(output_file, 'w', encoding='utf-8') as f:
+        f.write(header_content)
+    
+    print(f"Created header file: {output_file}")
+
+def process_file(input_file, output_file, var_name):
+    """Process a single file and create a header file"""
+    if not os.path.exists(os.path.dirname(output_file)):
+        os.makedirs(os.path.dirname(output_file))
+    
+    create_header_file(input_file, output_file, var_name)
+
+def process_directory(input_dir, output_dir):
+    """Process all files in a directory and create header files"""
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+    
+    for filename in os.listdir(input_dir):
+        input_file = os.path.join(input_dir, filename)
+        
+        if os.path.isfile(input_file):
+            # Generate variable name from filename
+            var_name = re.sub(r'[^a-zA-Z0-9_]', '_', os.path.splitext(filename)[0]).upper()
+            var_name = f"{var_name}_CONTENT"
+            
+            # Generate output filename
+            output_filename = f"{os.path.splitext(filename)[0]}_content.h"
+            output_file = os.path.join(output_dir, output_filename)
+            
+            create_header_file(input_file, output_file, var_name)
+
+if __name__ == "__main__":
+    if len(sys.argv) < 3:
+        print("Usage: python html_to_header.py <input_path> <output_path> [var_name]")
+        sys.exit(1)
+    
+    input_path = sys.argv[1]
+    output_path = sys.argv[2]
+    var_name = sys.argv[3] if len(sys.argv) > 3 else None
+    
+    if os.path.isfile(input_path):
+        if var_name is None:
+            # Generate variable name from filename
+            var_name = re.sub(r'[^a-zA-Z0-9_]', '_', os.path.splitext(os.path.basename(input_path))[0]).upper()
+            var_name = f"{var_name}_CONTENT"
+        process_file(input_path, output_path, var_name)
+    else:
+        process_directory(input_path, output_path)