|
from flask import Flask, request, jsonify, render_template, send_from_directory |
|
import os |
|
import subprocess |
|
import tempfile |
|
import shutil |
|
import sys |
|
|
|
app = Flask(__name__) |
|
|
|
|
|
temp_dir = tempfile.mkdtemp() |
|
current_dir = temp_dir |
|
|
|
def execute_command(command, cwd=None): |
|
"""Executes a command and returns the output.""" |
|
process = subprocess.Popen( |
|
command, |
|
shell=True, |
|
stdout=subprocess.PIPE, |
|
stderr=subprocess.PIPE, |
|
text=True, |
|
cwd=cwd or current_dir |
|
) |
|
stdout, stderr = process.communicate() |
|
return stdout + stderr |
|
|
|
@app.route("/") |
|
def index(): |
|
return render_template("index.html") |
|
|
|
@app.route("/execute", methods=["POST"]) |
|
def execute_code(): |
|
global current_dir |
|
command = request.json.get("code", "").strip() |
|
if not command: |
|
return jsonify({"result": "Error: No command provided."}) |
|
|
|
try: |
|
if command == "show files": |
|
files = os.listdir(current_dir) |
|
return jsonify({"result": "Files in current directory:\n" + "\n".join(files)}) |
|
elif command == "hide files": |
|
return jsonify({"result": "Files hidden."}) |
|
elif command.startswith("new file "): |
|
filename = command[9:].strip() |
|
filepath = os.path.join(current_dir, filename) |
|
with open(filepath, 'w') as f: |
|
pass |
|
return jsonify({"result": f"Created new file: {filename}"}) |
|
elif command.startswith("edit "): |
|
filename = command[5:].strip() |
|
filepath = os.path.join(current_dir, filename) |
|
if os.path.exists(filepath): |
|
return jsonify({"result": "Enter code:", "action": "edit", "filename": filename}) |
|
else: |
|
return jsonify({"result": f"Error: File {filename} not found."}) |
|
elif command.startswith("cd "): |
|
new_dir = os.path.join(current_dir, command[3:]) |
|
if os.path.isdir(new_dir): |
|
current_dir = os.path.abspath(new_dir) |
|
return jsonify({"result": f"Changed directory to: {current_dir}"}) |
|
else: |
|
return jsonify({"result": f"Error: Directory not found: {new_dir}"}) |
|
elif command.startswith("!"): |
|
result = execute_command(command[1:]) |
|
elif command.startswith("pip install"): |
|
result = execute_command(f"{sys.executable} -m {command}") |
|
elif command.startswith("git "): |
|
result = execute_command(command) |
|
else: |
|
if command.endswith(".py"): |
|
result = execute_command(f"{sys.executable} {command}") |
|
else: |
|
result = execute_command(f"{sys.executable} -c \"{command}\"") |
|
return jsonify({"result": result}) |
|
except Exception as e: |
|
return jsonify({"result": f"Error: {str(e)}"}) |
|
|
|
@app.route("/save_file", methods=["POST"]) |
|
def save_file(): |
|
filename = request.json.get("filename") |
|
content = request.json.get("content") |
|
filepath = os.path.join(current_dir, filename) |
|
with open(filepath, 'w') as f: |
|
f.write(content) |
|
return jsonify({"result": f"File {filename} saved successfully."}) |
|
|
|
@app.route("/cleanup", methods=["POST"]) |
|
def cleanup(): |
|
global temp_dir, current_dir |
|
if os.path.exists(temp_dir): |
|
shutil.rmtree(temp_dir) |
|
temp_dir = tempfile.mkdtemp() |
|
current_dir = temp_dir |
|
return jsonify({"result": "Temporary files cleaned up."}) |
|
|
|
@app.route("/list_files", methods=["GET"]) |
|
def list_files(): |
|
files = os.listdir(current_dir) |
|
return jsonify({"files": files}) |
|
|
|
@app.route("/download/<path:filename>", methods=["GET"]) |
|
def download_file(filename): |
|
return send_from_directory(current_dir, filename, as_attachment=True) |
|
|
|
if __name__ == "__main__": |
|
app.run(host="0.0.0.0", port=7860) |