#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
会话管理 API
作者：小白 🐶
"""

import json
import subprocess
from flask import jsonify, request
from . import api

@api.route('/sessions')
def get_sessions():
    """获取会话列表"""
    try:
        # 调用 sessions_list 命令
        result = subprocess.run(
            ['sessions_list'],
            capture_output=True,
            text=True,
            timeout=10
        )
        
        if result.returncode == 0:
            data = json.loads(result.stdout)
            return jsonify({
                'success': True,
                'data': data
            })
        else:
            return jsonify({
                'success': False,
                'error': result.stderr
            })
    except Exception as e:
        return jsonify({
            'success': False,
            'error': str(e)
        })

@api.route('/sessions/<session_key>')
def get_session_detail(session_key):
    """获取会话详情"""
    try:
        # 获取会话历史
        result = subprocess.run(
            ['sessions_history', '--session-key', session_key, '--limit', '20'],
            capture_output=True,
            text=True,
            timeout=10
        )
        
        if result.returncode == 0:
            data = json.loads(result.stdout)
            return jsonify({
                'success': True,
                'data': data
            })
        else:
            return jsonify({
                'success': False,
                'error': result.stderr
            })
    except Exception as e:
        return jsonify({
            'success': False,
            'error': str(e)
        })

@api.route('/sessions/<session_key>/kill', methods=['POST'])
def kill_session(session_key):
    """终止会话"""
    try:
        # 调用 subagents kill 命令
        result = subprocess.run(
            ['subagents', 'kill', '--target', session_key],
            capture_output=True,
            text=True,
            timeout=10
        )
        
        return jsonify({
            'success': result.returncode == 0,
            'message': '会话已终止' if result.returncode == 0 else '终止失败',
            'output': result.stdout or result.stderr
        })
    except Exception as e:
        return jsonify({
            'success': False,
            'error': str(e)
        })