# workspace_tree.dspy - 返回 Tree widget 需要的 {id, parentid, label, path} 格式

uid = await get_user()
if not uid:
    uid = 'user-01'

dbname = get_module_dbname('pipeline-sdlc')
workspace_base = '/d/pipeline/workspaces'
import os

async with DBPools().sqlorContext(dbname) as sor:
    recs = await sor.sqlExe(
        "SELECT current_project_id FROM pipeline_agent_settings WHERE user_id=${u}$",
        {"u": uid})
    pid = getattr(recs[0], 'current_project_id', '') if recs else ''

    ws_dir = ''
    if pid:
        proj = await sor.sqlExe(
            "SELECT name, org_id, workspace_dir FROM sd_projects WHERE id=${p}$",
            {"p": pid})
        if proj:
            pname = getattr(proj[0], 'name', '')
            org_id = getattr(proj[0], 'org_id', '0') or '0'
            ws = getattr(proj[0], 'workspace_dir', '') or ''
            ws_dir = ws if ws.startswith('/') else workspace_base + '/' + org_id + '/' + pname

if not ws_dir or not os.path.isdir(ws_dir):
    return [{"id": "__root__", "parentid": "", "label": "📁 工作空间（不可用）", "path": ""}]

# 递归扫描目录，id 只含相对路径，full path 存 path 字段
items = [{"id": "__root__", "parentid": "", "label": "📁 " + os.path.basename(ws_dir), "path": ws_dir, "is_leaf": False}]

def scan_dir(path, parent_id, rel_prefix):
    try:
        entries = sorted(os.listdir(path))
    except Exception:
        return
    for name in entries:
        full = os.path.join(path, name)
        if name.startswith('.') or name == '__pycache__':
            continue
        rel_path = rel_prefix + '/' + name if rel_prefix else name
        if os.path.isdir(full):
            items.append({
                "id": rel_path,
                "parentid": parent_id,
                "label": "📁 " + name,
                "path": full,
                "is_leaf": False
            })
            scan_dir(full, rel_path, rel_path)
        else:
            items.append({
                "id": rel_path,
                "parentid": parent_id,
                "label": "📄 " + name,
                "path": full,
                "is_leaf": True
            })

scan_dir(ws_dir, "__root__", "")
return items