#!/bin/bash
# 通过 REST API 批量安装模型供应商插件
# 使用方法: bash install_plugins_curl.sh

# Dify API 地址（根据实际情况修改）
API_URL="http://localhost:5001"

# 需要登录获取 Authorization token
# 可以从浏览器开发者工具中获取
AUTH_TOKEN="${DIFY_AUTH_TOKEN:-}"

if [ -z "$AUTH_TOKEN" ]; then
    echo "请设置 DIFY_AUTH_TOKEN 环境变量"
    echo "可以从浏览器开发者工具 -> Network -> 任意请求的 Authorization header 获取"
    echo ""
    echo "export DIFY_AUTH_TOKEN='Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9...'"
    exit 1
fi

echo "批量安装模型供应商插件..."
echo ""

# 模型供应商插件列表
PLUGINS=(
    "langgenius/openai"
    "langgenius/anthropic"
    "langgenius/deepseek"
    "langgenius/google"
    "langgenius/azure_openai"
    "langgenius/ollama"
    "langgenius/zhipuai"
    "langgenius/tongyi"
    "langgenius/spark"
    "langgenius/minimax"
    "langgenius/moonshot"
    "langgenius/baichuan"
    "langgenius/yi"
    "langgenius/mistralai"
    "langgenius/groq"
    "langgenius/siliconflow"
    "langgenius/fireworks"
    "langgenius/volcengine_maas"
    "langgenius/tencent"
    "langgenius/wenxin"
)

# 构建 JSON 数组
JSON_PLUGINS=$(printf '%s\n' "${PLUGINS[@]}" | jq -R . | jq -s .)

echo "将安装以下插件:"
printf '  ✓ %s\n' "${PLUGINS[@]}"
echo ""

# 调用安装 API
RESPONSE=$(curl -s -X POST "${API_URL}/console/api/workspaces/current/plugin/install/marketplace" \
    -H "Authorization: ${AUTH_TOKEN}" \
    -H "Content-Type: application/json" \
    -d "{\"plugin_unique_identifiers\": ${JSON_PLUGINS}}")

echo "响应: ${RESPONSE}"
echo ""
echo "安装任务已提交！请等待几分钟，然后刷新设置页面查看模型供应商列表"