1#!/bin/bash
  2#================================================================
  3# HEADER
  4#================================================================
  5#    Filename         send-ding.sh
  6#    Revision         0.0.1
  7#    Date             2020/06/08
  8#    Author           jiangliheng
  9#    Email            jiang_liheng@163.com
 10#    Website          https://jiangliheng.github.io/
 11#    Description      发送钉钉消息
 12#    Copyright        Copyright (c) jiangliheng
 13#    License          GNU General Public License
 14#
 15#================================================================
 16#
 17#  Version 0.0.1 2020/06/08
 18#     发送钉钉消息，支持 text，markdown 两种类型消息
 19#
 20#================================================================
 21#%名称(NAME)
 22#%       ${SCRIPT_NAME} - 发送钉钉消息
 23#%
 24#%概要(SYNOPSIS)
 25#%       sh ${SCRIPT_NAME} [options] <value> ...
 26#%
 27#%描述(DESCRIPTION)
 28#%       发送钉钉消息
 29#%
 30#%选项(OPTIONS)
 31#%     * -a <value>                 钉钉机器人 Webhook 地址的 access_token
 32#%     * -t <value>                 消息类型：text，markdown
 33#%     & -T <value>                 title，首屏会话透出的展示内容；消息类型（-t）为：markdown 时
 34#%     * -c <value>                 消息内容，content或者text
 35#%     & -m <value>                 被@人的手机号（在 content 里添加@人的手机号），多个参数用逗号隔开；如：138xxxx6666，182xxxx8888；与是否@所有人（-A）互斥，仅能选择一种方式
 36#%     & -A                         是否@所有人，即 isAtAll 参数设置为 ture；与被@人手机号（-m）互斥，仅能选择一种方式
 37#%       -v, --version              版本信息
 38#%       --help                     帮助信息
 39#%
 40#%     * 表示必输，& 表示条件必输，其余为可选
 41#%
 42#%示例(EXAMPLES)
 43#%
 44#%       1. 发送 text 消息类型，并@指定人
 45#%       sh ${SCRIPT_NAME} -a xxx -t text -c "我就是我, 是不一样的烟火" -m "138xxxx6666,182xxxx8888"
 46#%
 47#%       2. 发送 markdown 消息类型，并@所有人
 48#%       sh ${SCRIPT_NAME} -a xxx -t markdown -T "markdown 测试标题" -c "# 我就是我, 是不一样的烟火" -A
 49#%
 50#================================================================
 51# END_OF_HEADER
 52#================================================================
 53 
 54# header 总行数
 55SCRIPT_HEADSIZE=$(head -200 "${0}" |grep -n "^# END_OF_HEADER" | cut -f1 -d:)
 56# 脚本名称
 57SCRIPT_NAME="$(basename "${0}")"
 58# 版本
 59VERSION="0.0.1"
 60 
 61# usage
 62function usage() {
 63  head -"${SCRIPT_HEADSIZE:-99}" "${0}" 
 64  | grep -e "^#%" 
 65  | sed -e "s/^#%//g" -e "s/${SCRIPT_NAME}/${SCRIPT_NAME}/g" -e "s/${VERSION}/${VERSION}/g"
 66}
 67 
 68# 发送 ding 消息
 69function sendDingMessage() {
 70  curl -s "${1}" -H 'Content-Type: application/json' -d "${2}"
 71}
 72 
 73# 检查参数输入合法性
 74function checkParameters() {
 75  # -a，-t，-c 参数必输校验
 76  if [ -z "${ACCESS_TOKEN}" ] || [ -z "${MSG_TYPE}" ] || [ -z "${CONTENT}" ]
 77  then
 78    printf "Parameter [-a,-t,-c] is required!n"
 79    exit 1
 80  fi
 81 
 82  # -t 为：markdown 时，检验参数 -T 必输
 83  if [ "X${MSG_TYPE}" = "Xmarkdown" ] && [ -z "${TITLE}" ]
 84  then
 85    printf "When [-t] is 'markdown', you must enter the parameter [-T]!n"
 86    exit 1
 87  fi
 88 
 89  # -A 和 -m 互斥，仅能选择一种方式
 90  if [ "X${IS_AT_ALL}" = "Xtrue" ] && [ -n "${MOBILES}" ]
 91  then
 92    printf "Only one of the parameters [-A] and [-m] can be entered!n"
 93    exit 1
 94  fi
 95}
 96 
 97# markdown 消息内容
 98function markdownMessage() {
 99  # 标题
100  title=${1}
101  # 消息内容
102  text=${2}
103  # @ 方式
104  at=${3}
105 
106  # 判断是@所有人，还是指定人
107  if [ -z "${at}" ]; then
108    atJson=""
109  elif [ "X${at}" = "Xtrue" ]; then
110    atJson='"at": {
111        "isAtAll": true }'
112  else
113    # 判断是否多个手机号
114    result=$(echo "${at}" | grep ",")
115 
116    # N 个手机号
117    if [ "X${result}" != "X" ]; then
118      # 转换为手机号数组
119      mobileArray=(${at//,/ })
120      # 循环遍历数组，组织 json 格式字符串
121      for mobile in "${mobileArray[@]}"
122      do
123         mobiles="${mobile}",${mobiles}
124         # @ 指定人
125         atMobiles="@${mobile}",${atMobiles}
126      done
127 
128    # 1 个手机号
129    else
130      mobiles="${at}"
131      # @ 指定人
132      atMobiles="@${at}"
133    fi
134 
135    # @ json内容
136    atJson='"at": {
137        "atMobiles": [
138            '${mobiles/%,/}'
139        ]
140    }'
141 
142    # 内容信息添加 @指定人
143    text="${text}n${atMobiles/%,/}"
144  fi
145 
146  message='{
147       "msgtype": "markdown",
148       "markdown": {
149           "title":"'${title}'",
150           "text": "'${text}'"},
151        '${atJson}'
152   }'
153 
154   echo "${message}"
155}
156 
157# text 消息内容
158function textMessage() {
159  # 消息内容
160  text=${1}
161  # @ 方式
162  at=${2}
163 
164  # 判断是@所有人，还是指定人
165  if [ -z "${at}" ]; then
166    atJson=""
167  elif [ "X${at}" = "Xtrue" ]; then
168    atJson='"at": {
169        "isAtAll": true }'
170  else
171    # 判断是否多个手机号
172    result=$(echo "${at}" | grep ",")
173 
174    # N 个手机号
175    if [ "X${result}" != "X" ]; then
176      # 转换为手机号数组
177      mobileArray=(${at//,/ })
178      # 循环遍历数组，组织 json 格式字符串
179      for mobile in "${mobileArray[@]}"
180      do
181         mobiles="${mobile}",${mobiles}
182         # @ 指定人
183         atMobiles="@${mobile}",${atMobiles}
184      done
185 
186    # 1 个手机号
187    else
188      mobiles="${at}"
189      # @ 指定人
190      atMobiles="@${at}"
191    fi
192 
193    # @ json内容
194    atJson='"at": {
195        "atMobiles": [
196            '${mobiles/%,/}'
197        ]
198    }'
199 
200    # 内容信息添加 @指定人
201    text="${text}n${atMobiles/%,/}"
202  fi
203 
204  message='{
205       "msgtype": "text",
206       "text": {
207           "content": "'${text}'"},
208        '${atJson}'
209   }'
210 
211   echo "${message}"
212}
213 
214# 主方法
215function main() {
216 
217  # 检查参数输入合法性
218  checkParameters
219 
220  # 判断发送消息类型
221  case ${MSG_TYPE} in
222    markdown)
223      # 判断 @ 方式
224      if [ -n "${MOBILES}" ]; then
225        DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}" "${MOBILES}")
226      elif [ -n "${IS_AT_ALL}" ]; then
227        DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}" "${IS_AT_ALL}")
228      else
229        DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}")
230      fi
231      ;;
232    text)
233      if [ -n "${MOBILES}" ]; then
234        DING_MESSAGE=$(textMessage "${CONTENT}" "${MOBILES}")
235      elif [ -n "${IS_AT_ALL}" ]; then
236        DING_MESSAGE=$(textMessage "${CONTENT}" "${IS_AT_ALL}")
237      else
238        DING_MESSAGE=$(textMessage "${CONTENT}")
239      fi
240      ;;
241    *)
242      printf "Unsupported message type, currently only [text, markdown] are supported!"
243      exit 1
244      ;;
245  esac
246 
247  sendDingMessage "${DING_URL}" "${DING_MESSAGE}"
248}
249 
250# 判断参数个数
251if [ $# -eq 0 ];
252then
253  usage
254  exit 1
255fi
256 
257# getopt 命令行参数
258if ! ARGS=$(getopt -o vAa:t:T:c:m: --long help,version -n "${SCRIPT_NAME}" -- "$@")
259then
260  # 无效选项，则退出
261  exit 1
262fi
263 
264# 命令行参数格式化
265eval set -- "${ARGS}"
266 
267while [ -n "$1" ]
268do
269  case "$1" in
270    -a)
271      # Webhook access_token
272      ACCESS_TOKEN=$2
273      # 钉钉机器人 url 地址
274      DING_URL="https://oapi.dingtalk.com/robot/send?access_token=${ACCESS_TOKEN}"
275      shift 2
276      ;;
277 
278    -t)
279      MSG_TYPE=$2
280      shift 2
281      ;;
282 
283    -T)
284      TITLE=$2
285      shift 2
286      ;;
287 
288    -c)
289      CONTENT=$2
290      shift 2
291      ;;
292 
293    -m)
294      MOBILES=$2
295      shift 2
296      ;;
297 
298    -A)
299      IS_AT_ALL=true
300      shift 2
301      ;;
302 
303    -v|--version)
304      printf "%s version %sn" "${SCRIPT_NAME}" "${VERSION}"
305      exit 1
306      ;;
307 
308    --help)
309      usage
310      exit 1
311      ;;
312 
313    --)
314      shift
315      break
316      ;;
317 
318    *)
319      printf "%s is not an option!" "$1"
320      exit 1
321      ;;
322 
323  esac
324done
325 
326main