#!/bin/bash
# 2号机: 添加学习任务到青龙任务队列

QUEUE_FILE="/root/.openclaw/workspace/memory/task_queue.txt"

usage() {
  echo "用法: $0 <主题> <类型> [数量] [时间]"
  echo "  主题: 学习主题关键词"
  echo "  类型: article(文章) / novel(小说) / stock(股票)"
  echo "  数量: 默认 3"
  echo "  时间: 执行时间 (默认: 下一小时整点)"
  echo ""
  echo "示例:"
  echo "  $0 '机器学习' article 5"
  echo "  $0 'AAPL' stock 1 '2026-02-07T09:00:00'"
}

# 参数
TOPIC="$1"
TYPE="${2:-article}"
COUNT="${3:-3}"
TIME="${4:-}"

if [ -z "$TOPIC" ]; then
  usage
  exit 1
fi

# 默认时间: 下一小时整点
if [ -z "$TIME" ]; then
  TIME=$(date -d "+1 hour" '+%Y-%m-%dT%H:00:00')
fi

# 格式验证
if ! [[ "$TIME" =~ ^[0-9]{4}-[0-9]{2}-[0-9]{2}T[0-9]{2}:[0-9]{2}:[0-9]{2}$ ]]; then
  echo "✗ 时间格式错误，应为: YYYY-MM-DDTHH:MM:SS"
  exit 1
fi

# 添加任务
echo "$TIME|$TOPIC|$TYPE|$COUNT" >> "$QUEUE_FILE"

echo "✅ 任务已添加"
echo ""
echo "任务队列: $QUEUE_FILE"
echo "执行时间: $TIME"
echo "主题: $TOPIC"
echo "类型: $TYPE"
echo "数量: $COUNT"
echo ""
echo "要查看队列吗? (cat $QUEUE_FILE)"