# 文件路径：scripts/safe_path.py
import os

# 允许读取的目录
ALLOWED_READ_DIRS = [
    "/mnt/user-data/uploads",
    "/mnt/skills/public",
]

# 允许写入的目录
ALLOWED_WRITE_DIRS = [
    "/mnt/user-data/outputs",
    "/home/claude",
]

def is_safe_path(path: str, allowed_dirs: list) -> bool:
    """
    检查路径是否在允许的目录范围内

    防止 ../../../etc/passwd 类型的路径穿越攻击
    """
    # 解析为绝对路径（消除 .. 和符号链接）
    real_path = os.path.realpath(os.path.abspath(path))

    for allowed in allowed_dirs:
        real_allowed = os.path.realpath(allowed)
        # 检查 real_path 是否以 allowed 目录开头
        if real_path.startswith(real_allowed + os.sep) or real_path == real_allowed:
            return True
    return False

def safe_read_path(user_input: str) -> str:
    """验证读取路径，不合法时抛出异常"""
    if not is_safe_path(user_input, ALLOWED_READ_DIRS):
        raise PermissionError(
            f"拒绝访问：{user_input}\n"
            f"只允许读取以下目录：{ALLOWED_READ_DIRS}"
        )
    return os.path.realpath(user_input)

def safe_write_path(user_input: str) -> str:
    """验证写入路径，不合法时抛出异常"""
    if not is_safe_path(user_input, ALLOWED_WRITE_DIRS):
        raise PermissionError(
            f"拒绝写入：{user_input}\n"
            f"只允许写入以下目录：{ALLOWED_WRITE_DIRS}"
        )
    return os.path.realpath(user_input)

# 使用示例
if __name__ == "__main__":
    # 正常路径：通过
    ok_path = safe_read_path("/mnt/user-data/uploads/runoob.csv")
    print(f"通过：{ok_path}")

    # 穿越路径：拒绝
    try:
        bad_path = safe_read_path("/mnt/user-data/uploads/../../etc/passwd")
    except PermissionError as e:
        print(f"已拦截：{e}")

# 文件路径：scripts/config.py
# 安全地从多个来源读取配置，按优先级降序查找

import os

def get_secret(key: str, required: bool = True) -> str:
    """
    按优先级从以下来源读取密钥：
    1. 环境变量（最高优先级）
    2. /home/claude/.skill_secrets 文件（本地密钥文件）
    3. 若 required=True 且找不到，抛出异常
    """
    # 1. 环境变量
    value = os.environ.get(key)
    if value:
        return value

    # 2. 本地密钥文件（每行格式：KEY=VALUE）
    secrets_file = "/home/claude/.skill_secrets"
    if os.path.exists(secrets_file):
        with open(secrets_file) as f:
            for line in f:
                line = line.strip()
                if line.startswith(f"{key}="):
                    return line[len(key)+1:]

    # 3. 未找到
    if required:
        raise EnvironmentError(
            f"缺少必要的密钥：{key}\n"
            f"请设置环境变量：export {key}='你的密钥'"
        )
    return ""

# 文件路径：scripts/safe_exec.py
import subprocess
import shlex

def safe_shell_exec(template: str, user_input: str) -> str:
    """
    安全地将用户输入嵌入 Shell 命令

    错误做法：os.system(f"process {user_input}")   # 命令注入风险！
    正确做法：使用参数列表，让 subprocess 处理转义
    """
    # 使用列表而非字符串，subprocess 会自动处理转义
    cmd = ["python", "scripts/process.py", user_input]
    result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
    return result.stdout

# 如果必须构建字符串命令，使用 shlex.quote 转义用户输入
def safe_string_exec(user_filename: str) -> str:
    safe_name = shlex.quote(user_filename)   # 自动添加引号并转义特殊字符
    cmd = f"wc -l {safe_name}"
    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
    return result.stdout