from core.logging.debug import logging from core.utilities.utils import Base64Encode, ObfuscateJS, embed, IdentifyFileType from core.operations.genJS import GetJS from core.operations.genTemplate import SaveAsTemplate from core.visualization.visuals import VisualizePNG import os import re # Constants USING_CUSTOM_TEMPLATE = "Using custom {} template." INVALID_FILE_TYPE = "Invalid file type: Expected HTML or SVG." INVALID_CUSTOM_FILE = "Invalid custom file path." # Common Operations Function def GenerateObfuscatedJS(PathToZIPFile, args): EncodedString = Base64Encode(PathToZIPFile) JScode = GetJS(EncodedString, args.o, args.t, args.u) return ObfuscateJS(JScode) def HandleCustomTemplate(args, ObfuscatedCodeJS, OutputFile): FileType = IdentifyFileType(args.e) if FileType == 'html': logging.debug(USING_CUSTOM_TEMPLATE.format('HTML')) HandleCustomHTML(args.e, ObfuscatedCodeJS, OutputFile) elif FileType == 'svg': logging.debug(USING_CUSTOM_TEMPLATE.format('SVG')) HandleCustomSVG(args.e, ObfuscatedCodeJS, OutputFile) else: logging.error(INVALID_FILE_TYPE) raise ValueError(INVALID_FILE_TYPE) def HandleHTML(args, PathToZIPFile): logging.debug("Handling HTML type.") ObfuscatedCodeJS = GenerateObfuscatedJS(PathToZIPFile, args) if args.e: HandleCustomTemplate(args, ObfuscatedCodeJS, args.f) else: logging.debug("Using default HTML template.") SaveAsTemplate(ObfuscatedCodeJS, args.t, args.f) def HandleSVG(args, PathToZIPFile): logging.debug("Handling SVG type.") ObfuscatedCodeJS = GenerateObfuscatedJS(PathToZIPFile, args) if args.e: HandleCustomTemplate(args, ObfuscatedCodeJS, args.f) else: logging.debug("Using default SVG template.") SaveAsTemplate(ObfuscatedCodeJS, args.t, args.f) def HandleCustomHTML(CustomHTMLPath, ObfuscatedCodeJS, CustomHTMLEmbedded): try: with open(CustomHTMLPath, "r", encoding='utf-8') as file: ClonedHTML = file.read() index = ClonedHTML.rfind("") if index == -1: print("Warning: tag not found. Adding tags.") CustomFinalHTML = ClonedHTML + f"" else: CustomFinalHTML = ClonedHTML[:index] + f"" + ClonedHTML[index:] with open(CustomHTMLEmbedded, "w") as file: file.write(CustomFinalHTML) except IOError as e: logging.error(f"Custom HTML file operation failed: {e}") raise def HandleCustomSVG(CustomSVGPath, ObfuscatedCodeJS, CustomSVGEmbedded): try: with open(CustomSVGPath, "r", encoding='utf-8') as file: ClonedSVG = file.read() # Regex to find " match = re.search(Regex, ClonedSVG, re.DOTALL) if match: # Extract the existing script content and replace with new JS ScriptContent = match.group() NewScriptContent = re.sub(r"", f"", ScriptContent, flags=re.DOTALL) CustomFinalSVG = ClonedSVG.replace(ScriptContent, NewScriptContent) else: raise ValueError("Invalid SVG file: