飞书/钉钉群机器人通知发送的案例

jetsung

飞书/钉钉群机器人通知发送的案例

创建文件夹 test

将下述代码粘贴至 test 目录下的 main.go 文件中

执行 go mod tidy test，创建 module 方式

使用 go run main.go -u "https://open.feishu.cn/xxx" -s "Secret" -m "Message" -t 1234567890 方式发送通知消息（-t 部分可省略）

执行 go build 生成 test 可执行文件

package main

import (
	"crypto/hmac"
	"crypto/sha256"
	"encoding/base64"
	"flag"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
	"time"

	"log"
)

var (
	webhookUrl string
	message    string
	secret     string
	ts         int64
)

func sendMsg(apiUrl, msg string) {
	// json
	contentType := "application/json"

	if ts == 0 {
		ts = time.Now().Unix()
	}

	var sign string

	// data
	sendData1 := `{
		"msg_type": "text",
		"content": {"text": "消息通知: %s "},
		"sign": "%s",
		"timestamp": %d
	}`
	sendData2 := `{
  "msgtype": "text",
  "text": {
    "content": "消息通知： %s "
  },
  "at": {
    "isAtAll": false
  }
}`
	var sendData string
	if strings.HasPrefix(apiUrl, "https://oapi.dingtalk.com") {
		sendData = sendData2

		if ts <= 10e9 {
			ts *= 1000
		}
		sign, _ = GenSign(secret, ts, 2)
		sendData = fmt.Sprintf(sendData2, message)
		apiUrl = fmt.Sprintf("%s&timestamp=%d&sign=%s", apiUrl, ts, sign)
	} else if strings.HasPrefix(apiUrl, "https://open.feishu.cn") {
		sign, _ = GenSign(secret, ts, 1)
		sendData = fmt.Sprintf(sendData1, message, sign, ts)
	} else {
		fmt.Printf("此方式（%s）不支持\n", apiUrl)
		return
	}

	fmt.Printf(`%s

%s

timestamp: %d
secret: %s
sign: %s
`, apiUrl, sendData, ts, secret, sign)

	// request
	result, err := http.Post(apiUrl, contentType, strings.NewReader(sendData))
	if err != nil {
		fmt.Printf("post failed, err:%v\n", err)
		return
	}
	defer result.Body.Close()

	rspBody, err := ioutil.ReadAll(result.Body)
	if err != nil {
		log.Fatalf("ReadAll failed, reqBody: %s, err: %v\n", rspBody, err)
		return
	}

	fmt.Println("结果：" + string(rspBody))
}

func main() {
	flag.StringVar(&webhookUrl, "u", "", "飞书/钉钉webhook地址")
	flag.StringVar(&secret, "s", "", "密钥")
	flag.StringVar(&message, "m", "", "需要发送的消息内容")
	flag.Int64Var(&ts, "t", 0, "时间戳")

	flag.Parse()
	//flag.Usage()
	sendMsg(webhookUrl, message)
}

// GenSign 签名
//	mode: 1.飞书 2.钉钉
func GenSign(secret string, timestamp int64, mode int) (string, error) {
	stringToSign := fmt.Sprintf("%d\n%s", timestamp, secret)

	var data []byte
	var signStr string

	if mode == 1 {
		signStr = stringToSign
	} else {
		signStr = secret
		data = []byte(stringToSign)
	}

	h := hmac.New(sha256.New, []byte(signStr))
	_, err := h.Write(data)
	if err != nil {
		return "", err
	}

	signature := base64.StdEncoding.EncodeToString(h.Sum(nil))
	return signature, nil
}

jetsung

Shell 语言

bash main.sh -u "https://open.feishu.cn/xxx" -s "Secret" -m "Message" -t 1234567890 （-t 部分可省略）

#!/usr/bin/env bash

set -euo pipefail

judgment_parameters() {
  while [[ "$#" -gt '0' ]]; do
    case "$1" in
    '-h' | '--help')
      HELP='1'
      break
      ;;
    -m | --message)
      if [[ -z "${2:?error: Please specify the correct message.}" ]]; then
        exit 1
      fi
      MESSAGE="$2"
      shift
      ;;
    -s | --secret)
      if [[ -z "${2:?error: Please specify the correct secret.}" ]]; then
        exit 1
      fi
      SECRET="$2"
      shift
      ;;
    -t | --timestamp)
      if [[ -z "${2:?error: Please specify the correct timestamp.}" ]]; then
        exit 1
      fi
      TIMESTAMP="$2"
      shift
      ;;
    -u | --url)
      if [[ -z "${2:?error: Please specify the correct url.}" ]]; then
        exit 1
      fi
      URL="$2"
      shift
      ;;
    *)
      echo "$0: unknown option -- -"
      exit 1
      ;;
    esac
    shift
  done
}

show_help() {
  echo "usage: $0 [ options ]"
  echo '  -h, --help      Show help'
  echo '  -m, --message   Push message'
  echo '  -s, --secret    Push secret'
  echo '  -t, --timestamp Push timestamp'
  echo '  -u, --url       Push url'
  exit 0
}

send_message() {
  SEND_TEMPLATE_DINGTALK='{
  "msgtype": "text",
  "text": {
    "content": "消息通知： %s "
  },
  "at": {
    "isAtAll": false
  }
}'

  SEND_TEMPLATE_FEISHU='{
		"msg_type": "text",
		"content": {"text": "消息通知: %s "},
		"sign": "%s",
		"timestamp": %d
	}'

  # shellcheck disable=SC2059
  if [[ "$DOMAIN" == "oapi.dingtalk.com" ]]; then
    [[ ${#TIMESTAMP} -eq 13 ]] || TIMESTAMP="${TIMESTAMP}000"
    signature 1
    message=$(printf "$SEND_TEMPLATE_DINGTALK" "$MESSAGE")
    URL=$(printf "${URL}&timestamp=%s&sign=%s" "$TIMESTAMP" "$SIGN")
  else
    signature 2
    message=$(printf "$SEND_TEMPLATE_FEISHU" "$MESSAGE" "$SIGN" "$TIMESTAMP")
  fi

  curl -XPOST -s -L "$URL" -H "Content-Type:application/json" -H "charset:utf-8" -d "$message"
}

signature() {
  string_to_sign="$TIMESTAMP\n$SECRET"

  if [[ "${1:-1}" == 1 ]]; then
    sign_str="$SECRET"
    data="$string_to_sign"
  else
    sign_str="$string_to_sign"
    data=""
  fi

  TMP_PATH=$(mktemp)
  # shellcheck disable=SC2059
  printf "$sign_str" >"$TMP_PATH"
  # shellcheck disable=SC2059
  SIGN=$(printf "$data" | openssl dgst -sha256 -hmac "$(cat "$TMP_PATH")" -binary | base64)
  rm -rf "$TMP_PATH"
}

main() {
  judgment_parameters "$@"

  [[ "${HELP:-}" -eq '1' ]] && show_help

  if [[ -z "$MESSAGE" ]]; then
    echo "error: Please specify the correct message."
    exit 1
  fi

  if [[ -z "$URL" ]]; then
    echo "error: Please specify the correct url."
    exit 1
  fi

  DOMAIN=$(echo "$URL" | awk -F[/:] '{print $4}')
  if [[ "$DOMAIN" != "oapi.dingtalk.com" ]] &&
    [[ "$DOMAIN" != "open.feishu.cn" ]] &&
    [[ "$DOMAIN" != "open.larksuite.com" ]]; then
    echo "error: Please specify the correct url."
    exit 1
  fi

  [[ -n "${TIMESTAMP-}" ]] || TIMESTAMP="$(date +%s)"

  send_message
}

main "$@"