#!/bin/bash
# 小红书增强版发布器快速启动脚本

set -e

echo "=========================================="
echo "小红书增强版发布器快速启动"
echo "=========================================="

# 检查Python环境
echo "1. 检查Python环境..."
python3 --version
if [ $? -ne 0 ]; then
    echo "❌ Python3未安装"
    exit 1
fi

# 检查依赖
echo -e "\n2. 检查依赖..."
if ! python3 -c "import pillow" 2>/dev/null; then
    echo "安装Pillow..."
    pip install pillow
fi

if ! python3 -c "import requests" 2>/dev/null; then
    echo "安装requests..."
    pip install requests
fi

# 检查小红书CLI
echo -e "\n3. 检查小红书CLI..."
if ! command -v xhs &> /dev/null; then
    echo "❌ 小红书CLI未安装"
    echo "安装命令: uv tool install xiaohongshu-cli"
    echo "或: pipx install xiaohongshu-cli"
    exit 1
else
    echo "✅ 小红书CLI已安装"
    xhs --version
fi

# 检查登录状态
echo -e "\n4. 检查小红书登录状态..."
xhs status --yaml >/dev/null 2>&1
if [ $? -eq 0 ]; then
    echo "✅ 小红书已登录"
else
    echo "⚠️ 小红书未登录"
    echo "请运行: xhs login"
    echo "或: xhs login --qrcode"
fi

# 复制配置文件
echo -e "\n5. 准备配置文件..."
if [ ! -f "xiaohongshu_enhanced_config.json" ]; then
    echo "❌ 增强版配置文件不存在"
    exit 1
fi

# 检查API密钥配置
echo -e "\n6. 检查API密钥配置..."
API_KEY=$(python3 -c "import json; data=json.load(open('xiaohongshu_enhanced_config.json')); print(data.get('image_generation', {}).get('api_key', '未配置'))")
if [ "$API_KEY" = "未配置" ] || [ -z "$API_KEY" ]; then
    echo "⚠️ API密钥未配置"
    echo "请编辑 xiaohongshu_enhanced_config.json 文件"
    echo "添加你的豆包Seedream API密钥"
else
    echo "✅ API密钥已配置"
fi

# 创建输出目录
echo -e "\n7. 创建输出目录..."
OUTPUT_DIR=$(python3 -c "import json; data=json.load(open('xiaohongshu_enhanced_config.json')); print(data.get('image_generation', {}).get('output_dir', '/home/ubuntu/xhs_generated_images'))")
mkdir -p "$OUTPUT_DIR"
echo "输出目录: $OUTPUT_DIR"

# 测试功能
echo -e "\n8. 测试基本功能..."
echo -e "\n测试内容分析..."
python3 xiaohongshu_enhanced_publisher.py analyze --content "高效学习法分享" 2>&1 | grep -A5 "内容分析结果"

echo -e "\n测试配置查看..."
python3 xiaohongshu_enhanced_publisher.py config 2>&1 | head -10

# 创建示例文件
echo -e "\n9. 创建示例文件..."
cat > /tmp/example_article.md << 'EOF'
# 高效学习法分享

掌握高效学习法，让学习事半功倍！

## 核心方法

1. **番茄工作法**
   - 25分钟专注学习
   - 5分钟短暂休息
   - 每4个番茄钟长休息

2. **费曼学习法**
   - 选择一个概念
   - 教授给他人
   - 回顾和简化
   - 组织和完善

坚持使用这些方法，学习效率提升300%！
EOF

echo "示例文件已创建: /tmp/example_article.md"

# 显示使用说明
echo -e "\n=========================================="
echo "使用说明"
echo "=========================================="

echo -e "\n📝 基本发布:"
echo "  python3 xiaohongshu_enhanced_publisher.py publish --markdown /tmp/example_article.md"

echo -e "\n🎨 指定风格发布:"
echo "  python3 xiaohongshu_enhanced_publisher.py publish \\"
echo "    --markdown /tmp/example_article.md \\"
echo "    --style notion \\"
echo "    --layout dense"

echo -e "\n📁 批量处理:"
echo "  python3 xiaohongshu_enhanced_publisher.py batch --directory ./articles/"

echo -e "\n🔍 内容分析:"
echo "  python3 xiaohongshu_enhanced_publisher.py analyze --content \"你的内容\""

echo -e "\n🖼️ 图片生成:"
echo "  python3 xiaohongshu_enhanced_publisher.py generate --content \"你的内容\" --style notion"

echo -e "\n⚙️ 查看配置:"
echo "  python3 xiaohongshu_enhanced_publisher.py config"
echo "  python3 xiaohongshu_enhanced_publisher.py config --show"

echo -e "\n🧪 测试功能:"
echo "  python3 xiaohongshu_enhanced_publisher.py test"

echo -e "\n=========================================="
echo "下一步操作"
echo "=========================================="

echo -e "\n1. 配置API密钥:"
echo "   编辑 xiaohongshu_enhanced_config.json 文件"
echo "   添加豆包Seedream API密钥"

echo -e "\n2. 登录小红书:"
echo "   xhs login  # 自动检测浏览器cookies"
echo "   xhs login --qrcode  # 二维码登录"

echo -e "\n3. 测试完整流程:"
echo "   python3 xiaohongshu_enhanced_publisher.py test"

echo -e "\n4. 查看详细指南:"
echo "   阅读 ENHANCED_GUIDE.md 文件"

echo -e "\n5. 查看集成示例:"
echo "   python3 integration_example.py"

echo -e "\n6. 查看集成总结:"
echo "   阅读 INTEGRATION_SUMMARY.md 文件"

echo -e "\n=========================================="
echo "支持的功能"
echo "=========================================="

echo -e "\n🎨 支持的风格:"
echo "  - 知识风 (notion) - 干货分享"
echo "  - 可爱风 (cute) - 生活分享"
echo "  - 清新风 (fresh) - 健康生活"
echo "  - 极简风 (minimal) - 专业内容"
echo "  - 温暖风 (warm) - 情感分享"
echo "  - 醒目风 (bold) - 重要提醒"
echo "  - 复古风 (retro) - 怀旧内容"
echo "  - 流行风 (pop) - 时尚内容"

echo -e "\n📐 支持的布局:"
echo "  - 稀疏布局 (sparse) - 封面/金句"
echo "  - 平衡布局 (balanced) - 常规内容"
echo "  - 密集布局 (dense) - 知识卡片"
echo "  - 列表布局 (list) - 清单/要点"
echo "  - 对比布局 (comparison) - 产品对比"
echo "  - 流程布局 (flow) - 教程步骤"

echo -e "\n⚡ 预设组合:"
echo "  - 知识卡片 (notion + dense)"
echo "  - 可爱分享 (cute + balanced)"
echo "  - 教程步骤 (fresh + flow)"
echo "  - 产品对比 (fresh + comparison)"
echo "  - 清单列表 (notion + list)"
echo "  - 极简金句 (minimal + sparse)"

echo -e "\n=========================================="
echo "快速开始示例"
echo "=========================================="

echo -e "\n# 1. 分析内容"
echo 'python3 xiaohongshu_enhanced_publisher.py analyze --content "高效学习法分享"'

echo -e "\n# 2. 生成图片"
echo 'python3 xiaohongshu_enhanced_publisher.py generate --content "高效学习法分享" --style notion --layout dense'

echo -e "\n# 3. 发布内容"
echo "python3 xiaohongshu_enhanced_publisher.py publish --markdown /tmp/example_article.md --style notion --layout dense"

echo -e "\n=========================================="
echo "完成！现在可以开始使用增强版小红书发布器了。"
echo "=========================================="

echo -e "\n💡 提示:"
echo "- 首次使用建议先运行测试: python3 xiaohongshu_enhanced_publisher.py test"
echo "- 查看详细日志了解运行状态"
echo "- 定期检查小红书登录状态: xhs status"
echo "- 备份重要配置文件"

exit 0