linux环境企业机器人带模板v0.0.1

wecom_reboot_cli_tool.sh

@@ -0,0 +1,743 @@
+#!/bin/bash
+
+# 企业微信机器人增强版消息发送工具
+# 支持多种消息类型：文本、文件、图片、语音、Markdown、模板卡片
+# 作者: AI Assistant
+# 版本: 2.0
+
+# 配置文件路径
+CONFIG_FILE="$HOME/.wecom_config"
+
+# 颜色定义
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+PURPLE='\033[0;35m'
+CYAN='\033[0;36m'
+NC='\033[0m' # No Color
+
+# 打印彩色消息
+print_info() {
+    echo -e "${BLUE}[INFO]${NC} $1"
+}
+
+print_success() {
+    echo -e "${GREEN}[SUCCESS]${NC} $1"
+}
+
+print_warning() {
+    echo -e "${YELLOW}[WARNING]${NC} $1"
+}
+
+print_error() {
+    echo -e "${RED}[ERROR]${NC} $1"
+}
+
+print_debug() {
+    echo -e "${PURPLE}[DEBUG]${NC} $1"
+}
+
+print_title() {
+    echo -e "${CYAN}$1${NC}"
+}
+
+# 加载或创建配置
+load_config() {
+    if [ -f "$CONFIG_FILE" ]; then
+        source "$CONFIG_FILE"
+        if [ -n "$WEBHOOK_KEY" ]; then
+            print_info "已加载配置文件中的webhook key"
+            return 0
+        fi
+    fi
+    
+    print_warning "未找到配置文件或webhook key为空"
+    setup_webhook_key
+}
+
+# 设置webhook key
+setup_webhook_key() {
+    print_title "=== 企业微信机器人配置 ==="
+    echo "请输入你的企业微信机器人webhook key："
+    echo "（从 https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=YOUR_KEY 中提取key部分）"
+    echo ""
+    echo -n "Webhook Key: "
+    read webhook_key
+    
+    if [ -z "$webhook_key" ]; then
+        print_error "Webhook key不能为空！"
+        exit 1
+    fi
+    
+    # 保存配置
+    echo "WEBHOOK_KEY=\"$webhook_key\"" > "$CONFIG_FILE"
+    print_success "配置已保存到 $CONFIG_FILE"
+    
+    WEBHOOK_KEY="$webhook_key"
+    setup_urls
+}
+
+# 设置API URLs
+setup_urls() {
+    BASE_URL="https://qyapi.weixin.qq.com/cgi-bin/webhook"
+    SEND_URL="${BASE_URL}/send?key=${WEBHOOK_KEY}"
+    UPLOAD_URL="${BASE_URL}/upload_media?key=${WEBHOOK_KEY}"
+}
+
+# 发送文本消息
+send_text() {
+    local content="$1"
+    local mention="$2"
+    
+    local json_data
+    if [ -n "$mention" ]; then
+        json_data=$(cat << EOF
+{
+    "msgtype": "text",
+    "text": {
+        "content": "$content",
+        "mentioned_list": ["$mention"]
+    }
+}
+EOF
+)
+    else
+        json_data=$(cat << EOF
+{
+    "msgtype": "text",
+    "text": {
+        "content": "$content"
+    }
+}
+EOF
+)
+    fi
+    
+    send_request "$json_data" "文本消息"
+}
+
+# 上传文件/语音
+upload_media() {
+    local file_path="$1"
+    local file_type="${2:-file}"
+    
+    if [ ! -f "$file_path" ]; then
+        print_error "文件不存在: $file_path" >&2
+        return 1
+    fi
+    
+    # 检查文件大小
+    local file_size=$(get_file_size "$file_path")
+    if [ "$file_size" -le 5 ]; then
+        print_error "文件大小必须大于5个字节，当前大小: ${file_size} 字节" >&2
+        return 1
+    fi
+    
+    print_info "正在上传${file_type}文件: $file_path" >&2
+    local response=$(curl -s -X POST "${UPLOAD_URL}&type=${file_type}" \
+        -F "media=@$file_path")
+    
+    local errcode=$(echo "$response" | grep -o '"errcode":[0-9]*' | cut -d':' -f2)
+    if [ "$errcode" = "0" ]; then
+        local media_id=$(echo "$response" | grep -o '"media_id":"[^"]*"' | cut -d'"' -f4)
+        print_success "文件上传成功！media_id: $media_id" >&2
+        echo "$media_id"
+        return 0
+    else
+        print_error "文件上传失败: $response" >&2
+        return 1
+    fi
+}
+
+# 获取文件大小（跨平台兼容）
+get_file_size() {
+    local file_path="$1"
+    
+    if command -v stat >/dev/null 2>&1; then
+        # Linux系统
+        if stat -c%s "$file_path" >/dev/null 2>&1; then
+            stat -c%s "$file_path"
+        # macOS系统  
+        elif stat -f%z "$file_path" >/dev/null 2>&1; then
+            stat -f%z "$file_path"
+        else
+            wc -c < "$file_path" | tr -d ' '
+        fi
+    else
+        wc -c < "$file_path" | tr -d ' '
+    fi
+}
+
+# 发送文件消息
+send_file() {
+    local file_path="$1"
+    
+    local media_id=$(upload_media "$file_path" "file")
+    if [ $? -ne 0 ]; then
+        return 1
+    fi
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "file",
+    "file": {
+        "media_id": "$media_id"
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "文件消息"
+}
+
+# 发送语音消息
+send_voice() {
+    local voice_path="$1"
+    
+    # 检查文件格式（简单检查扩展名）
+    if [[ ! "$voice_path" =~ \.(amr|mp3|wav|m4a)$ ]]; then
+        print_warning "语音文件建议使用 .amr、.mp3、.wav 或 .m4a 格式"
+    fi
+    
+    local media_id=$(upload_media "$voice_path" "voice")
+    if [ $? -ne 0 ]; then
+        return 1
+    fi
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "voice",
+    "voice": {
+        "media_id": "$media_id"
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "语音消息"
+}
+
+# 发送图片消息
+send_image() {
+    local image_path="$1"
+    
+    if [ ! -f "$image_path" ]; then
+        print_error "图片文件不存在: $image_path"
+        return 1
+    fi
+    
+    # 检查文件格式
+    if [[ ! "$image_path" =~ \.(jpg|jpeg|png|gif)$ ]]; then
+        print_warning "图片文件建议使用 .jpg、.png 或 .gif 格式"
+    fi
+    
+    # 读取图片并转换为base64
+    if ! command -v base64 >/dev/null 2>&1; then
+        print_error "系统缺少base64命令，无法发送图片"
+        return 1
+    fi
+    
+    local file_size=$(get_file_size "$image_path")
+    if [ "$file_size" -gt 2097152 ]; then  # 2MB
+        print_error "图片文件不能超过2M，当前大小: $((file_size/1024))KB"
+        return 1
+    fi
+    
+    print_info "正在处理图片: $image_path"
+    local image_data=$(base64 "$image_path" | tr -d '\n')
+    local md5_hash
+    
+    if command -v md5sum >/dev/null 2>&1; then
+        md5_hash=$(md5sum "$image_path" | cut -d' ' -f1)
+    elif command -v md5 >/dev/null 2>&1; then
+        md5_hash=$(md5 -q "$image_path")
+    else
+        print_error "系统缺少md5计算工具，无法发送图片"
+        return 1
+    fi
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "image",
+    "image": {
+        "base64": "$image_data",
+        "md5": "$md5_hash"
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "图片消息"
+}
+
+# 发送Markdown消息
+send_markdown() {
+    local content="$1"
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "markdown",
+    "markdown": {
+        "content": "$content"
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "Markdown消息"
+}
+
+# 发送图文消息
+send_news() {
+    local title="$1"
+    local description="$2"
+    local url="$3"
+    local pic_url="$4"
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "news",
+    "news": {
+        "articles": [
+            {
+                "title": "$title",
+                "description": "$description",
+                "url": "$url",
+                "picurl": "$pic_url"
+            }
+        ]
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "图文消息"
+}
+
+# 发送文本通知模板卡片
+send_text_card() {
+    local title="$1"
+    local content="$2"
+    local source_desc="${3:-企业微信机器人}"
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "template_card",
+    "template_card": {
+        "card_type": "text_notice",
+        "source": {
+            "desc": "$source_desc",
+            "desc_color": 0
+        },
+        "main_title": {
+            "title": "$title"
+        },
+        "sub_title_text": "$content"
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "文本通知卡片"
+}
+
+# 发送图文展示模板卡片
+send_news_card() {
+    local title="$1"
+    local desc="$2"
+    local image_url="$3"
+    local jump_url="$4"
+    local source_desc="${5:-企业微信机器人}"
+    
+    local json_data=$(cat << EOF
+{
+    "msgtype": "template_card",
+    "template_card": {
+        "card_type": "news_notice",
+        "source": {
+            "desc": "$source_desc",
+            "desc_color": 0
+        },
+        "main_title": {
+            "title": "$title",
+            "desc": "$desc"
+        },
+        "card_image": {
+            "url": "$image_url",
+            "aspect_ratio": 2.25
+        },
+        "card_action": {
+            "type": 1,
+            "url": "$jump_url"
+        }
+    }
+}
+EOF
+)
+    
+    send_request "$json_data" "图文展示卡片"
+}
+
+# 发送HTTP请求
+send_request() {
+    local json_data="$1"
+    local msg_type="$2"
+    
+    print_info "正在发送${msg_type}..."
+    local response=$(curl -s -X POST "$SEND_URL" \
+        -H "Content-Type: application/json" \
+        -d "$json_data")
+    
+    local errcode=$(echo "$response" | grep -o '"errcode":[0-9]*' | cut -d':' -f2)
+    if [ "$errcode" = "0" ]; then
+        print_success "${msg_type}发送成功！"
+        return 0
+    else
+        print_error "${msg_type}发送失败: $response"
+        return 1
+    fi
+}
+
+# 发送组合消息
+send_combo() {
+    local text_content="$1"
+    local file_path="$2"
+    local mention="$3"
+    
+    print_info "开始发送组合消息（文本+文件）"
+    
+    if ! send_text "$text_content" "$mention"; then
+        return 1
+    fi
+    
+    sleep 1  # 避免发送太快
+    
+    if ! send_file "$file_path"; then
+        return 1
+    fi
+    
+    print_success "组合消息发送完成！"
+}
+
+# 显示帮助信息
+show_help() {
+    print_title "企业微信机器人增强版消息发送工具 v2.0"
+    echo ""
+    echo "用法: $0 [选项]"
+    echo ""
+    echo "基础消息选项:"
+    echo "  -t, --text TEXT           发送文本消息"
+    echo "  -f, --file FILE           发送文件"
+    echo "  -i, --image IMAGE         发送图片"
+    echo "  -v, --voice VOICE         发送语音"
+    echo "  -m, --markdown TEXT       发送Markdown消息"
+    echo "  -c, --combo TEXT FILE     发送文本+文件组合"
+    echo ""
+    echo "高级消息选项:"
+    echo "  -n, --news                发送图文消息（交互式输入）"
+    echo "  --text-card               发送文本通知卡片（交互式输入）"
+    echo "  --news-card               发送图文展示卡片（交互式输入）"
+    echo ""
+    echo "其他选项:"
+    echo "  -@ USERID                 @指定用户（与文本消息配合使用）"
+    echo "  --config                  重新配置webhook key"
+    echo "  --test                    测试连接"
+    echo "  -h, --help               显示此帮助信息"
+    echo ""
+    echo "示例:"
+    echo "  $0 -t \"Hello World\"                     # 发送文本消息"
+    echo "  $0 -f \"report.pdf\"                      # 发送文件"
+    echo "  $0 -i \"screenshot.png\"                  # 发送图片"
+    echo "  $0 -v \"voice.amr\"                       # 发送语音"
+    echo "  $0 -c \"请查看附件\" \"document.pdf\"        # 发送文本+文件"
+    echo "  $0 -t \"重要通知\" -@ \"@all\"               # @所有人"
+    echo "  $0 -n                                    # 交互式创建图文消息"
+    echo "  $0 --text-card                           # 交互式创建文本卡片"
+    echo ""
+    echo "配置文件位置: $CONFIG_FILE"
+}
+
+# 主菜单
+show_menu() {
+    print_title "=================================="
+    print_title "  企业微信机器人增强版发送工具 v2.0"
+    print_title "=================================="
+    echo "1.  发送文本消息"
+    echo "2.  发送文件"
+    echo "3.  发送图片"
+    echo "4.  发送语音"
+    echo "5.  发送文本+文件组合"
+    echo "6.  发送Markdown消息"
+    echo "7.  发送图文消息"
+    echo "8.  发送文本通知卡片"
+    echo "9.  发送图文展示卡片"
+    echo "10. 测试连接"
+    echo "11. 重新配置webhook key"
+    echo "12. 退出"
+    print_title "=================================="
+    echo -n "请选择操作 (1-12): "
+}
+
+# 测试连接
+test_connection() {
+    print_info "测试机器人连接..."
+    local test_msg="🤖 机器人连接测试 - $(date '+%Y-%m-%d %H:%M:%S')"
+    
+    if send_text "$test_msg"; then
+        print_success "机器人连接正常！"
+    else
+        print_error "机器人连接失败，请检查webhook key是否正确"
+    fi
+}
+
+# 交互式输入图文消息
+interactive_news() {
+    echo "=== 创建图文消息 ==="
+    echo -n "标题: "
+    read title
+    echo -n "描述: "
+    read description
+    echo -n "点击跳转URL: "
+    read url
+    echo -n "图片URL（可选）: "
+    read pic_url
+    
+    if [ -z "$title" ] || [ -z "$url" ]; then
+        print_error "标题和URL不能为空"
+        return 1
+    fi
+    
+    send_news "$title" "$description" "$url" "$pic_url"
+}
+
+# 交互式输入文本卡片
+interactive_text_card() {
+    echo "=== 创建文本通知卡片 ==="
+    echo -n "卡片标题: "
+    read title
+    echo -n "卡片内容: "
+    read content
+    echo -n "来源描述（可选，默认：企业微信机器人）: "
+    read source_desc
+    
+    if [ -z "$title" ]; then
+        print_error "卡片标题不能为空"
+        return 1
+    fi
+    
+    send_text_card "$title" "$content" "${source_desc:-企业微信机器人}"
+}
+
+# 交互式输入图文卡片
+interactive_news_card() {
+    echo "=== 创建图文展示卡片 ==="
+    echo -n "卡片标题: "
+    read title
+    echo -n "卡片描述: "
+    read desc
+    echo -n "图片URL: "
+    read image_url
+    echo -n "点击跳转URL: "
+    read jump_url
+    echo -n "来源描述（可选，默认：企业微信机器人）: "
+    read source_desc
+    
+    if [ -z "$title" ] || [ -z "$image_url" ] || [ -z "$jump_url" ]; then
+        print_error "标题、图片URL和跳转URL不能为空"
+        return 1
+    fi
+    
+    send_news_card "$title" "$desc" "$image_url" "$jump_url" "${source_desc:-企业微信机器人}"
+}
+
+# 主程序
+main() {
+    # 加载配置
+    load_config
+    setup_urls
+    
+    local text_content=""
+    local file_path=""
+    local mention=""
+    local markdown_content=""
+    local action=""
+    
+    # 解析命令行参数
+    while [[ $# -gt 0 ]]; do
+        case $1 in
+            -t|--text)
+                text_content="$2"
+                action="text"
+                shift 2
+                ;;
+            -f|--file)
+                file_path="$2"
+                action="file"
+                shift 2
+                ;;
+            -i|--image)
+                file_path="$2"
+                action="image"
+                shift 2
+                ;;
+            -v|--voice)
+                file_path="$2"
+                action="voice"
+                shift 2
+                ;;
+            -c|--combo)
+                text_content="$2"
+                file_path="$3"
+                action="combo"
+                shift 3
+                ;;
+            -m|--markdown)
+                markdown_content="$2"
+                action="markdown"
+                shift 2
+                ;;
+            -n|--news)
+                action="news"
+                shift
+                ;;
+            --text-card)
+                action="text-card"
+                shift
+                ;;
+            --news-card)
+                action="news-card"
+                shift
+                ;;
+            -@)
+                mention="$2"
+                shift 2
+                ;;
+            --config)
+                setup_webhook_key
+                exit 0
+                ;;
+            --test)
+                test_connection
+                exit 0
+                ;;
+            -h|--help)
+                show_help
+                exit 0
+                ;;
+            *)
+                print_error "未知选项: $1"
+                show_help
+                exit 1
+                ;;
+        esac
+    done
+    
+    # 如果没有参数，显示交互式菜单
+    if [ -z "$action" ]; then
+        while true; do
+            show_menu
+            read choice
+            echo ""
+            
+            case $choice in
+                1)
+                    echo -n "请输入文本内容: "
+                    read text_input
+                    echo -n "是否@某人（可选，直接回车跳过）: "
+                    read mention_input
+                    send_text "$text_input" "$mention_input"
+                    ;;
+                2)
+                    echo -n "请输入文件路径: "
+                    read file_input
+                    send_file "$file_input"
+                    ;;
+                3)
+                    echo -n "请输入图片路径: "
+                    read image_input
+                    send_image "$image_input"
+                    ;;
+                4)
+                    echo -n "请输入语音文件路径: "
+                    read voice_input
+                    send_voice "$voice_input"
+                    ;;
+                5)
+                    echo -n "请输入文本内容: "
+                    read text_input
+                    echo -n "请输入文件路径: "
+                    read file_input
+                    send_combo "$text_input" "$file_input"
+                    ;;
+                6)
+                    echo -n "请输入Markdown内容: "
+                    read markdown_input
+                    send_markdown "$markdown_input"
+                    ;;
+                7)
+                    interactive_news
+                    ;;
+                8)
+                    interactive_text_card
+                    ;;
+                9)
+                    interactive_news_card
+                    ;;
+                10)
+                    test_connection
+                    ;;
+                11)
+                    setup_webhook_key
+                    setup_urls
+                    ;;
+                12)
+                    print_info "再见！"
+                    break
+                    ;;
+                *)
+                    print_error "无效选择，请重新输入"
+                    ;;
+            esac
+            echo ""
+            echo "按任意键继续..."
+            read -n 1
+            clear
+        done
+        exit 0
+    fi
+    
+    # 执行相应的操作
+    case $action in
+        "text")
+            send_text "$text_content" "$mention"
+            ;;
+        "file")
+            send_file "$file_path"
+            ;;
+        "image")
+            send_image "$file_path"
+            ;;
+        "voice")
+            send_voice "$file_path"
+            ;;
+        "combo")
+            send_combo "$text_content" "$file_path" "$mention"
+            ;;
+        "markdown")
+            send_markdown "$markdown_content"
+            ;;
+        "news")
+            interactive_news
+            ;;
+        "text-card")
+            interactive_text_card
+            ;;
+        "news-card")
+            interactive_news_card
+            ;;
+    esac
+}
+
+# 运行主程序
+main "$@"