"""
API views for lobster monitoring.
"""
from rest_framework.decorators import api_view
from rest_framework.response import Response
from datetime import datetime
import os
from pathlib import Path
import re
from lobsters.models import Lobster, LobsterDiary

@api_view(['GET'])
def lobster_list(request):
    """获取所有龙虾状态"""
    lobsters = Lobster.objects.all()
    result = []
    for lobster in lobsters:
        result.append({
            'id': lobster.id,
            'name': lobster.name,
            'emoji': lobster.emoji,
            'port': lobster.port,
            'specialty': lobster.specialty,
            'container': lobster.container,
            'app_name': lobster.app_name,
            'app_id': lobster.app_id,
            'status': 'healthy',
            'last_check': datetime.now().isoformat()
        })
    return Response(result)

@api_view(['GET'])
def lobster_detail(request, lobster_id):
    """获取单个龙虾详情"""
    try:
        lobster = Lobster.objects.get(id=lobster_id)
        return Response({
            'id': lobster.id,
            'name': lobster.name,
            'emoji': lobster.emoji,
            'port': lobster.port,
            'specialty': lobster.specialty,
            'container': lobster.container,
            'app_name': lobster.app_name,
            'app_id': lobster.app_id,
            'status': 'healthy',
            'workspace': f'/home/node/.openclaw/workspace/{lobster.name.lower()}',
            'last_check': datetime.now().isoformat()
        })
    except Lobster.DoesNotExist:
        return Response({'error': '龙虾不存在'}, status=404)

@api_view(['GET'])
def lobster_memory(request, lobster_id):
    """获取龙虾记忆"""
    # 这里简化处理，实际应该读取文件
    return Response({
        'lobster_id': lobster_id,
        'memory': '# 长期记忆\n\n记忆内容加载中...',
        'daily_memories': []
    })

@api_view(['GET'])
def tools_list(request):
    """获取工具列表"""
    tools = [
        {
            'name': 'Git 版本控制',
            'status': 'running',
            'description': '代码版本管理服务',
            'url': 'https://xjp.datalibstar.com/flying-hero/openclaw-monitor.git'
        }
    ]
    return Response(tools)

@api_view(['GET'])
def lobster_memory_dates(request, lobster_id):
    """获取龙虾有日记的日期列表"""
    # 获取龙虾工作区
    lobster_map = {
        1: 'flying-hero',
        2: 'daotong',
        3: 'coder',
        4: 'web',
        5: 'physics',
        6: 'watcher',
    }
    
    lobster_name = lobster_map.get(lobster_id)
    if not lobster_name:
        return Response({'error': '龙虾不存在'}, status=404)
    
    # 记忆文件目录
    memory_dir = Path(f'/home/node/.openclaw/workspace/flying-hero/memory')
    
    # 获取所有日记文件
    dates = []
    if memory_dir.exists():
        for file in memory_dir.glob('*.md'):
            # 提取日期 (YYYY-MM-DD.md)
            match = re.match(r'(\d{4}-\d{2}-\d{2})\.md', file.name)
            if match:
                dates.append(match.group(1))
    
    dates.sort(reverse=True)
    return Response({'dates': dates})

@api_view(['GET'])
def lobster_memory_detail(request, lobster_id, date):
    """获取指定日期的日记内容"""
    memory_file = Path(f'/home/node/.openclaw/workspace/flying-hero/memory/{date}.md')
    
    if not memory_file.exists():
        return Response({'error': '该日期没有日记'}, status=404)
    
    content = memory_file.read_text(encoding='utf-8')
    return Response({
        'date': date,
        'content': content
    })

@api_view(['GET'])
def lobster_diary_dates(request, lobster_id):
    """获取龙虾有日记（成才之路）的日期列表 - 从数据库读取"""
    try:
        lobster = Lobster.objects.get(id=lobster_id)
    except Lobster.DoesNotExist:
        return Response({'error': '龙虾不存在'}, status=404)
    
    # 从数据库查询日记日期
    diaries = LobsterDiary.objects.filter(
        lobster=lobster,
        category='chengcai'
    ).values_list('date', flat=True).distinct()
    
    dates = [str(date) for date in sorted(diaries, reverse=True)]
    return Response({'dates': dates})

@api_view(['GET'])
def lobster_diary_detail(request, lobster_id, date):
    """获取指定日期的日记内容（成才之路） - 从数据库读取"""
    try:
        lobster = Lobster.objects.get(id=lobster_id)
    except Lobster.DoesNotExist:
        return Response({'error': '龙虾不存在'}, status=404)
    
    # 从数据库查询日记
    try:
        diary = LobsterDiary.objects.get(
            lobster=lobster,
            date=date,
            category='chengcai'
        )
        return Response({
            'date': str(diary.date),
            'content': diary.content,
            'title': diary.title,
            'tags': diary.tags,
        })
    except LobsterDiary.DoesNotExist:
        return Response({'error': '该日期没有日记'}, status=404)