init

2025-08-15 20:12:35 -07:00
parent c686e60ec5
commit 387c694efe
14 changed files with 2724 additions and 0 deletions
@@ -0,0 +1,54 @@
+import random
+import string
+from utils.name_gen import NameGenerator
+
+class JunkGenerator:
+    def __init__(self, name_generator: NameGenerator):
+        self.name_generator = name_generator
+
+    def add_junk(self) -> str:
+        """
+        Generates random multi-line junk code, used to pad the final output.
+        """
+        var1 = self.name_generator.generate_name()
+        var2 = self.name_generator.generate_name()
+
+        example_strings = ["HelloWorld", "CryptoMagic", "XxX_Obf_42", "DataStorm", "RandomStr", "TestString"]
+        s1 = random.choice(example_strings)
+        s2 = random.choice(example_strings)
+
+        multi_line_junks = [
+            f"{var1} = chr({random.randint(65, 90)}) + chr({random.randint(65, 90)})",
+            f"{var1} = sum([{random.randint(1,10)} for _ in range({random.randint(1,5)})])",
+            f"{var1} = len(bytes([{random.randint(65,90)}, {random.randint(65,90)}, {random.randint(65,90)}]))",
+            f"if chr({random.randint(65,90)}).isupper(): {var1} = {random.randint(0, 999)}",
+            f"{var1} = bytes([{random.randint(65,90)}] * {random.randint(1,3)})",
+            f"{var1} = ord(chr({random.randint(65,90)})) - {random.randint(1,10)}",
+
+            f"{var1} = '{s1}'.lower()\n{var2} = '{s2}'.upper()",
+            f"{var1} = [chr(random.randint(65,90)) for _ in range({random.randint(2,5)})]\n{var2} = ''.join({var1})",
+            f"{var1} = {random.randint(1,10)} ** {random.randint(1,3)}\n{var2} = {var1} % {random.randint(1,10)}",
+            f"{var1} = [{random.randint(1,5)} for _ in range({random.randint(3,5)})]\nif len({var1}) > {random.randint(1,4)}: {var2} = sum({var1})",
+            f"{var1} = {{ {random.randint(1,10)} : '{s1}' }}\n{var2} = {var1}.get({random.randint(1,10)}, None)",
+            f"{var1} = set(range({random.randint(2,5)}))\n{var2} = set(range({random.randint(2,5)}))\n{var1}.intersection({var2})"
+        ]
+        return random.choice(multi_line_junks)
+
+    def generate_junk(self) -> str:
+        """
+        Generates a single line of junk code.
+        """
+        var_name = self.name_generator.generate_name()
+        junk_code = [
+            f"{var_name} = {random.randint(0, 100)}",
+            f"{var_name} = '{''.join(random.choices(string.ascii_letters, k=10))}'",
+            f"{var_name} = {random.randint(0, 100)} + {random.randint(0, 100)}",
+            f"{var_name} = {random.randint(0, 100)} * {random.randint(0, 100)}",
+            f"{var_name} = {random.randint(0, 100)} - {random.randint(0, 100)}",
+            f"{var_name} = {random.randint(0, 100)} / {random.randint(1, 100)}",
+            f"{var_name} = '{''.join(random.choices(string.ascii_letters + string.digits, k=15))}'",
+            f"{var_name} = {random.randint(0, 100)} % {random.randint(1, 100)}",
+            f"{var_name} = {random.randint(0, 100)} ** {random.randint(0, 5)}",
+            f"{var_name} = {random.randint(0, 100)} // {random.randint(1, 100)}"
+        ]
+        return random.choice(junk_code)