121 lines
4.7 KiB
Python
121 lines
4.7 KiB
Python
# myapp/views/vscode.py
|
|
import re
|
|
import os, uuid, json
|
|
from flask import Blueprint, current_app, session, redirect, url_for, render_template, request, jsonify
|
|
from ..services.backboard_service import realtime_response
|
|
from ..extension_ase.ase_client.manager import ChatManager
|
|
import subprocess
|
|
|
|
bp = Blueprint("vscode", __name__)
|
|
|
|
@bp.route("/desktop/<user_uuid>/<course_id>/<chapter_name>/<lesson_name>")
|
|
def desktop(user_uuid, course_id, chapter_name, lesson_name):
|
|
username2uuid = current_app.extensions["username2uuid"]
|
|
uuid2username = current_app.extensions["uuid2username"]
|
|
userid_recorder = current_app.extensions["userid_recorder"]
|
|
if "user_uuid" not in session:
|
|
return redirect(url_for("auth.login"))
|
|
# 全局映射
|
|
user_id = uuid2username[user_uuid]
|
|
userid_recorder[f"{user_id}&{course_id}"] = session["user_uuid"]
|
|
|
|
user_home = os.path.join("/home", user_id)
|
|
try:
|
|
subprocess.run(["sudo", "useradd", "-m", "-d", user_home, user_id]) # 创建用户
|
|
subprocess.run(["sudo", "chown", "-R", user_id, user_home]) # 设置目录权限为新用户
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error creating user {user_id}: {e}")
|
|
path_dir = os.path.join("/home", user_id, course_id, chapter_name, lesson_name)
|
|
try:
|
|
# 使用 sudo 执行 mkdir 命令来创建目录
|
|
subprocess.run(["sudo", "-u", user_id, "mkdir", "-p", path_dir], check=True)
|
|
print(f"Directory {path_dir} created successfully for user {user_id}")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error creating directory {path_dir} details {str(e)}")
|
|
|
|
chatmanager = ChatManager()
|
|
current_app.extensions["user_uuid2chatmanager"][user_uuid] = chatmanager
|
|
code_server_port = 10000 + int(uuid.uuid4().int % 10000)
|
|
chatmanager.start_code_server(user_uuid, user_id, path_dir, code_server_port)
|
|
|
|
|
|
|
|
caddy_conf = f"""
|
|
handle_path /vsc{user_id}/* {{
|
|
reverse_proxy localhost:{code_server_port}
|
|
}}
|
|
"""
|
|
caddy_config_path = "/etc/caddy/Caddyfile"
|
|
with open(caddy_config_path, "r") as f:
|
|
lines = f.readlines()
|
|
pattern = re.compile(
|
|
r'handle_path\s+/vsc' + re.escape(user_uuid) + r'/\* {.*?reverse_proxy.*?}',
|
|
re.DOTALL
|
|
)
|
|
content = "".join(lines)
|
|
if pattern.search(content):
|
|
new_content = pattern.sub('', content)
|
|
lines = new_content.splitlines(keepends=True)
|
|
if len(lines) >= 2:
|
|
insert_position = len(lines) - 2
|
|
else:
|
|
insert_position = len(lines)
|
|
if caddy_conf[-1] not in ['\n', '\r']:
|
|
caddy_conf += '\n'
|
|
lines.insert(insert_position, caddy_conf)
|
|
with open(caddy_config_path, "w") as f:
|
|
f.writelines(lines)
|
|
|
|
try:
|
|
subprocess.run(["sudo", "caddy", "reload", "--config", "/etc/caddy/Caddyfile"])
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error reload caddy {e}")
|
|
|
|
cfg = current_app.config["VSCODE_WEB_PATH"]
|
|
path_for_vscode = path_dir
|
|
|
|
config_path = os.path.join(path_dir, ".config")
|
|
tmpd = {
|
|
"user_uuid": session["user_uuid"],
|
|
"user_id": user_id,
|
|
"course_id": course_id,
|
|
"chapter_name": chapter_name,
|
|
"lesson_name": lesson_name,
|
|
"path": path_for_vscode
|
|
}
|
|
config_json = json.dumps(tmpd, ensure_ascii=False)
|
|
|
|
# 使用 sudo 和 tee 命令将 JSON 数据写入文件
|
|
subprocess.run(
|
|
["sudo", "-u", user_id, "bash", "-c", f"echo '{config_json}' | tee {config_path} > /dev/null"],
|
|
check=True
|
|
)
|
|
# 确保文件的所有权归 user_id
|
|
subprocess.run(
|
|
["sudo", "chown", f"{user_id}:{user_id}", config_path],
|
|
check=True
|
|
)
|
|
return render_template(
|
|
"desktop.html",
|
|
vscode_web_url=current_app.config["VSCODE_WEB_URL"]+f"/vsc{user_id}",
|
|
user_uuid=session["user_uuid"], course_id=course_id, chapter_name=chapter_name, lesson_name=lesson_name,
|
|
workspace_path=path_for_vscode,
|
|
)
|
|
|
|
@bp.route("/desktop_nouser/<course_id>/<chapter_name>/<lesson_name>")
|
|
def desktop_nouser(course_id, chapter_name, lesson_name):
|
|
if "user_uuid" not in session:
|
|
return redirect(url_for("auth.login")) # 如果你有 auth 蓝图
|
|
user_uuid = session["user_uuid"]
|
|
return redirect(url_for("vscode.desktop", user_uuid=user_uuid, course_id=course_id, chapter_name=chapter_name, lesson_name=lesson_name))
|
|
|
|
@bp.route("/vscode_data", methods=["POST"])
|
|
def vscode_data():
|
|
# data = request.get_json(force=True)
|
|
# config = data.get("config") or {}
|
|
|
|
# realtime_response(config, data) # 调用服务层逻辑
|
|
# return jsonify({"status": "success", "received": data})
|
|
|
|
return jsonify({"status": "success", "received": "success"})
|