openclaw-monitor/code/backend/api/views.py

"""
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

@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):
    """获取龙虾有日记（成才之路）的日期列表"""
    # 日记文件目录
    diary_dir = Path(f'/home/node/.openclaw/workspace/flying-hero/memory/成才之路')
    
    # 获取所有日记文件
    dates = []
    if diary_dir.exists():
        for file in diary_dir.glob('*.md'):
            # 提取日期 (YYYY-MM-DD.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 = list(set(dates))  # 去重
    dates.sort(reverse=True)
    return Response({'dates': dates})

@api_view(['GET'])
def lobster_diary_detail(request, lobster_id, date):
    """获取指定日期的日记内容（成才之路）"""
    # 优先查找故事版，其次技术版，再其次普通版
    diary_file = Path(f'/home/node/.openclaw/workspace/flying-hero/memory/成才之路/{date}-故事版.md')
    if not diary_file.exists():
        diary_file = Path(f'/home/node/.openclaw/workspace/flying-hero/memory/成才之路/{date}-技术版.md')
    if not diary_file.exists():
        diary_file = Path(f'/home/node/.openclaw/workspace/flying-hero/memory/成才之路/{date}.md')
    
    if not diary_file.exists():
        return Response({'error': '该日期没有日记'}, status=404)
    
    content = diary_file.read_text(encoding='utf-8')
    return Response({
        'date': date,
        'content': content
    })