attendance.sh

#!/bin/bash

#------------------------------------------------------------------------------
# 常量
#------------------------------------------------------------------------------

HYPERGRYPH_OAUTH_URL='https://as.hypergryph.com/user/oauth2/v2/grant'
SKLAND_AUTH_URL='https://zonai.skland.com/api/v1/user/auth/generate_cred_by_code'
SKLAND_BINDING_URL='https://zonai.skland.com/api/v1/game/player/binding'
SKLAND_CHECKIN_URL='https://zonai.skland.com/api/v1/score/checkin'
SKLAND_ATTENDANCE_URL='https://zonai.skland.com/api/v1/game/attendance'

SKLAND_USER_AGENT='User-Agent: Skland/1.21.0 (com.hypergryph.skland; build:102100065; Android 34; ) Okhttp/4.11.0'

declare -A SKLAND_BOARD_MAP
SKLAND_BOARD_MAP[1]='明日方舟'
SKLAND_BOARD_MAP[2]='来自星辰'
SKLAND_BOARD_MAP[3]='明日方舟: 终末地'
SKLAND_BOARD_MAP[4]='泡姆泡姆'
SKLAND_BOARD_MAP[100]='纳斯特港'
SKLAND_BOARD_MAP[101]='开拓芯'

#------------------------------------------------------------------------------
# 全局变量
#------------------------------------------------------------------------------

_root_pid=$$

#------------------------------------------------------------------------------
# 日志函数
#------------------------------------------------------------------------------

# Debug 日志
# (str)
debug() {
  if [ -z "$SKLAND_DEBUG" ]; then
    return 0
  fi

  local timestamp=$(date +"%Y-%m-%dT%H:%M:%S%z")
  local spid=$(printf %5d $$)
  echo "[$timestamp] [$spid] [DEBUG] $1" >/proc/$_root_pid/fd/1
}

# Info 日志
# (str)
info() {
  local timestamp=$(date +"%Y-%m-%dT%H:%M:%S%z")
  local spid=$(printf %5d $$)
  echo "[$timestamp] [$spid] [ INFO] $1" >/proc/$_root_pid/fd/1
}

# Error 日志
# (str)
error() {
  local timestamp=$(date +"%Y-%m-%dT%H:%M:%S%z")
  local spid=$(printf %5d $$)
  echo "[$timestamp] [$spid] [ERROR] $1" >/proc/$_root_pid/fd/2
}

#------------------------------------------------------------------------------
# 工具函数
#------------------------------------------------------------------------------

# 生成 API 签名
# (token, path, data, query) -> signature, header
generate_signature() {
  # 初始化 header
  local platform='1'
  local timestamp=$(($(date +%s) - 2))
  local d_id=''
  local v_name='1.21.0'
  local header=$(
    jq \
      -cnM \
      --arg platform  "$platform" \
      --arg timestamp "$timestamp" \
      --arg d_id      "$d_id" \
      --arg v_name    "$v_name" \
      '{platform: $platform, timestamp: $timestamp, dId: $d_id, vName: $v_name}'
  )

  # 生成签名
  local str="$2$4$3$timestamp$header"
  local hmac_sha256=$(echo -n "$str" | openssl dgst -sha256 -hmac "$1" | awk '{print $2}')
  local sign=$(echo -n "$hmac_sha256" | md5sum | awk '{print $1}')
  debug "[generate_signature] sign=$sign, header=$header"

  # 返回签名和 header
  echo "$sign"
  echo "$header"
}

# 名字打码
# (name, replace) -> privacy_name
get_privacy_name() {
  local privacy=$(sed "s#.#$2#g" <<< "${1:1:-1}")
  echo "${1:0:1}$privacy${1: -1}"
}

#------------------------------------------------------------------------------
# 通知推送 API
#------------------------------------------------------------------------------

# 初始化
notification_init() {
  local timestamp=$(date +"%Y-%m-%d %H:%M:%S %Z")
  echo -ne "# 森空岛每日签到\n\n> $timestamp" >/tmp/skland-daily.log
  echo -ne "<h1>森空岛每日签到</h1><blockquote>$timestamp</blockquote>" >/tmp/skland-daily-mail.log
}

# 添加推送内容记录
# (str, mstr)
notification_add() {
  echo -ne "\n\n$1" >>/tmp/skland-daily.log
  echo -ne "$2" >>/tmp/skland-daily-mail.log
}

# Bark 推送 API
# (url, title, content)
notification_bark() {
  # 发送请求
  local body=$(
    jq \
      -cnMS \
      --arg title "$2" \
      --arg body  "$3" \
      --arg group 'Skland' \
      '{title: $title, body: $body, group: $group}'
  )
  local response=$(
    curl \
      -s \
      -H 'Content-Type: application/json' \
      -d "$body" \
      "$1"
  )
  debug "[notification_bark] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问 Bark API: curl_exit=$?"
    return 1
  fi

  # 打印信息
  info 'Bark 推送成功'
}

# Server 酱推送 API
# (key, title, content)
notification_server_chan() {
  # 发送请求
  local body=$(
    jq \
      -cnMS \
      --arg title "$2" \
      --arg desp  "$3" \
      '{title: $title, desp: $desp}'
  )
  local response=$(
    curl \
      -s \
      -H 'Content-Type: application/json' \
      -d "$body" \
      "https://sctapi.ftqq.com/$1.send"
  )
  debug "[notification_server_chan] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问 Server 酱 API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  if [[ $(jq '.code == 0' <<< "$response") == 'true' ]]; then
    info 'Server 酱推送成功'
  else
    error "Server 酱推送失败: response=$response"
    return 1
  fi
}

# SMTP 推送
# (to, subject, content)
notification_smtp() {
  # 发送邮件
  if [[ "$DOCKER" == '∞X3XckwT1ztOA2da∞' ]]; then
    echo $3 | mutt -s "$2" "$1" >/tmp/skland-daily-mail-send.log 2>&1
  else
    echo $3 | mailx -a 'Content-Type: text/html' -s "$2" "$1" >/tmp/skland-daily-mail-send.log 2>&1
  fi

  # 处理返回值
  local exit_code=$?
  local msg=$(cat /tmp/skland-daily-mail-send.log)
  rm -f /tmp/skland-daily-mail-send.log

  # 检查返回值
  if [ $exit_code -ne 0 ]; then
    error "无法发送邮件: mail_exit=$exit_code, msg=$msg"
    return 1
  else
    info 'SMTP 推送成功'
  fi
}

# 执行推送
notification_execute() {
  # 获取记录
  local content=$(cat /tmp/skland-daily.log)
  local mail_content=$(cat /tmp/skland-daily-mail.log)
  rm -f /tmp/skland-daily.log /tmp/skland-daily-mail.log

  # Bark 推送
  if [[ -n "$BARK_URL" ]]; then
    notification_bark "$BARK_URL" '【森空岛每日签到】' "$content"
  fi

  # Server 酱推送
  if [[ -n "$SERVERCHAN_SENDKEY" ]]; then
    notification_server_chan "$SERVERCHAN_SENDKEY" '【森空岛每日签到】' "$content"
  fi

  # SMTP 推送
  if [[ -n "$SMTP_TO" ]]; then
    notification_smtp "$SMTP_TO" '【森空岛每日签到】' "$mail_content"
  fi
}

#------------------------------------------------------------------------------
# 鹰角网络通行证 API
#------------------------------------------------------------------------------

# OAuth 登陆
# (token) -> code
hypergryph_auth() {
  # 发送请求
  local body=$(
    jq \
      -cnMS \
      --arg     app_code '4ca99fa6b56cc2ba' \
      --arg     token    "$1" \
      --argjson type     '0' \
      '{appCode: $app_code, token: $token, type: $type}'
  )
  local response=$(
    curl \
      -s \
      -H "$SKLAND_USER_AGENT" \
      -H 'Accept-Encoding: gzip' \
      -H 'Connection: close' \
      -H 'Content-Type: application/json' \
      -d "$body" \
      "$HYPERGRYPH_OAUTH_URL"
  )
  debug "[hypergryph_auth] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问 OAuth API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  if [[ $(jq '.status != 0' <<< "$response") == 'true' ]]; then
    local msg=$(jq -r '.msg' <<< "$response")
    error "无法登陆鹰角网络通行证: msg=$msg"
    return 1
  fi

  # 返回 code
  echo $(jq -r '.data.code' <<< "$response")
}

#------------------------------------------------------------------------------
# 森空岛 API
#------------------------------------------------------------------------------

# 鉴权
# (code) -> cred, token
skland_auth() {
  # 发送请求
  local body=$(
    jq \
      -cnMS \
      --arg     code "$1" \
      --argjson kind '1' \
      '{code: $code, kind: $kind}'
  )
  local response=$(
    curl \
      -s \
      -H "$SKLAND_USER_AGENT" \
      -H 'Accept-Encoding: gzip' \
      -H 'Connection: close' \
      -H 'Content-Type: application/json' \
      -d "$body" \
      "$SKLAND_AUTH_URL"
  )
  debug "[skland_auth] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问鉴权 API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  if [[ $(jq '.code != 0' <<< "$response") == 'true' ]]; then
    local msg=$(jq -r '.message' <<< "$response")
    error "无法鉴权: message=$msg"
    return 1
  fi

  # 返回 cred 和 token
  echo $(jq -r '.data.cred' <<< "$response")
  echo $(jq -r '.data.token' <<< "$response")
}

# 获得角色绑定信息
# (cred, token) -> list
skland_get_binding() {
  # 获得签名和 header
  local tmp=$(generate_signature "$2" '/api/v1/game/player/binding')
  local sign=$(awk 'NR==1' <<< "$tmp")
  local header=$(awk 'NR==2' <<< "$tmp")

  # 发送请求
  local response=$(
    curl \
      -s \
      -H "$SKLAND_USER_AGENT" \
      -H 'Accept-Encoding: gzip' \
      -H 'Connection: close' \
      -H 'Content-Type: application/json' \
      -H "Platform: $(jq -r '.platform' <<< "$header")" \
      -H "Timestamp: $(jq -r '.timestamp' <<< "$header")" \
      -H "Did: $(jq -r '.dId' <<< "$header")" \
      -H "Vname: $(jq -r '.vName' <<< "$header")" \
      -H "Sign: $sign" \
      -H "Cred: $1" \
      "$SKLAND_BINDING_URL"
  )
  debug "[skland_get_binding] response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问角色绑定 API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  if [[ $(jq '.code != 0' <<< "$response") == 'true' ]]; then
    local msg=$(jq -r '.message' <<< "$response")
    error "无法获取角色绑定: message=$msg"
    return 1
  fi

  # 返回角色绑定表
  echo $(jq -c '.data.list | map(.bindingList) | flatten(1)' <<< "$response")
}

# 森空岛检票
# (cred, token, id)
skland_check_in() {
  # 获得签名和 header
  local body=$(
    jq \
      -cnMS \
      --argjson game_id "$3" \
      '{gameId: $game_id}'
  )
  local tmp=$(generate_signature "$2" '/api/v1/score/checkin' "$body")
  local sign=$(awk 'NR==1' <<< "$tmp")
  local header=$(awk 'NR==2' <<< "$tmp")

  # 发送请求
  local response=$(
    curl \
      -s \
      -H "$SKLAND_USER_AGENT" \
      -H 'Accept-Encoding: gzip' \
      -H 'Connection: close' \
      -H 'Content-Type: application/json' \
      -H "Platform: $(jq -r '.platform' <<< "$header")" \
      -H "Timestamp: $(jq -r '.timestamp' <<< "$header")" \
      -H "Did: $(jq -r '.dId' <<< "$header")" \
      -H "Vname: $(jq -r '.vName' <<< "$header")" \
      -H "Sign: $sign" \
      -H "Cred: $1" \
      -d "$body" \
      "$SKLAND_CHECKIN_URL"
  )
  debug "[skland_check_in] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问检票 API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  local board="${SKLAND_BOARD_MAP[$3]}"
  if [[ $(jq '.code == 0' <<< "$response") == 'true' ]]; then
    notification_add "版面【$board】检票成功" "<p>版面【$board】检票成功</p>"
    info "版面【$board】检票成功"
  else
    local msg=$(jq -r '.message' <<< "$response")
    notification_add "版面【$board】检票失败，错误信息：$msg" "<p>版面【$board】检票失败，错误信息：$msg</p>"
    error "版面【$board】检票失败: message=$msg"
    return 1
  fi
}

# 明日方舟签到
# (cred, token, item)
skland_attendance() {
  # 提取角色数据
  local uid=$(jq -r '.uid' <<< "$3")
  local cid=$(jq -r '.channelMasterId' <<< "$3")
  local cname=$(jq -r '.channelName' <<< "$3")
  local nname=$(jq -r '.nickName' <<< "$3")
  local pname=$(get_privacy_name "$nname" '*')
  local mdpname=$(get_privacy_name "$nname" '\\*')

  # 获得签名和 header
  local body=$(
    jq \
      -cnMS \
      --arg uid "$uid" \
      --arg game_id "$cid" \
      '{uid: $uid, gameId: $game_id}'
  )
  local tmp=$(generate_signature "$2" '/api/v1/game/attendance' "$body")
  local sign=$(awk 'NR==1' <<< "$tmp")
  local header=$(awk 'NR==2' <<< "$tmp")

  # 发送请求
  local response=$(
    curl \
      -s \
      --compressed \
      -H "$SKLAND_USER_AGENT" \
      -H 'Accept-Encoding: gzip' \
      -H 'Connection: close' \
      -H 'Content-Type: application/json' \
      -H "Platform: $(jq -r '.platform' <<< "$header")" \
      -H "Timestamp: $(jq -r '.timestamp' <<< "$header")" \
      -H "Did: $(jq -r '.dId' <<< "$header")" \
      -H "Vname: $(jq -r '.vName' <<< "$header")" \
      -H "Sign: $sign" \
      -H "Cred: $1" \
      -d "$body" \
      "$SKLAND_ATTENDANCE_URL"
  )
  debug "[skland_attendance] body=$body, response=$response"

  # 检查 CURL 返回值
  if [ $? -ne 0 ]; then
    error "无法访问签到 API: curl_exit=$?"
    return 1
  fi

  # 检查响应
  if [[ $(jq '.code == 0' <<< "$response") == 'true' ]]; then
    local awards=$(jq '.data.awards | map([.resource.name, .count]) | map(join(" x")) | join(", ")' <<< "$response")
    notification_add "$cname - $mdpname 签到成功，获得了：$awards" "<p>$cname - $pname 签到成功，获得了：$awards</p>"
    info "$cname - $pname 签到成功，获得了：$awards"
  else
    local msg=$(jq -r '.message' <<< "$response")
    notification_add "$cname - $mdpname 签到失败，错误信息：$msg" "<p>$cname - $pname 签到失败，错误信息：$msg</p>"
    error "$cname - $pname 签到失败: message=$msg"
    return 1
  fi
}

#------------------------------------------------------------------------------
# 签到流程函数
#------------------------------------------------------------------------------

# 进行签到
# (token)
do_attendance() {
  debug "[do_attendance] token=$1"

  # 初始化推送
  notification_init

  # OAuth 登陆
  local code=$(hypergryph_auth "$1")
  if [[ -z "$code" ]]; then
    notification_add '## 错误\n\n无法登陆鹰角网络通行证' '<h2>错误</h2><p>无法登陆鹰角网络通行证</p>'
    notification_execute
    return 1
  fi
  info 'OAuth 登陆成功'

  # 森空岛鉴权
  local tmp=$(skland_auth "$code")
  if [[ -z "$tmp" ]]; then
    notification_add '## 错误\n\n无法登陆森空岛' '<h2>错误</h2><p>无法登陆森空岛</p>'
    notification_execute
    return 1
  fi
  local cred=$(awk 'NR==1' <<< "$tmp")
  local token=$(awk 'NR==2' <<< "$tmp")
  info '森空岛鉴权成功'

  # 获取角色绑定
  local list=$(skland_get_binding "$cred" "$token")
  if [[ -z "$list" ]]; then
    notification_add '## 错误\n\n无法获取角色绑定' '<h2>错误</h2><p>无法获取角色绑定</p>'
    notification_execute
    return 1
  fi
  debug "[do_attendance] list=$list"
  info '角色绑定获取成功'

  # 森空岛检票
  notification_add '## 森空岛各版面每日检票' '<h2>森空岛各版面每日检票</h2>'

  declare id
  for id in ${!SKLAND_BOARD_MAP[*]}; do
    skland_check_in "$cred" "$token" "$id" &
  done
  wait
  info '森空岛检票完毕'

  # 明日方舟签到
  notification_add '## 明日方舟签到' '<h2>明日方舟签到</h2>'
  local list_len=$(jq 'length' <<< "$list")
  local i=0
  while [ $i -lt $list_len ]; do
    local item=$(jq -c ".[$i]" <<< "$list")
    skland_attendance "$cred" "$token" "$item" &
    i=$(($i + 1))
  done
  wait
  info '明日方舟签到完毕'

  # 推送通知
  notification_execute
}

#------------------------------------------------------------------------------
# 主入口
#------------------------------------------------------------------------------

main() {
  # 检查环境变量
  if [[ -z "$SKLAND_TOKEN" ]]; then
    error '环境变量 "SKLAND_TOKEN" 未定义'
    exit 1
  fi

  # 配置 Docker 环境
  if [[ "$DOCKER" == '∞X3XckwT1ztOA2da∞' ]]; then
    # 写入/获取入口 PID
    if [ -f /run/attendance.pid ]; then
      _root_pid=$(cat /run/attendance.pid)
    else
      echo -n $$ >/run/attendance.pid
    fi

    # 启动计划任务
    if [[ $$ == $_root_pid ]]; then
      # 配置邮件服务
      if [[ -n "$SMTP_TO" ]]; then
        echo -e "# Skland\n\
account      skland\n\
host         $SMTP_HOST\n\
port         $SMTP_PORT\n\
tls_starttls $SMTP_START_TLS\n\
from         $SMTP_FROM\n\
user         $SMTP_USER\n\
password     $SMTP_PASSWD\n\n\
account default: skland" >>/etc/msmtprc
        echo -e "set realname=$SMTP_REAL_NAME\n\
set from=$SMTP_FROM" >>/root/.muttrc
        info '邮件服务配置完成'
      fi

      # 启动计划任务
      info '计划任务启动'
      /attendance.sh &
      crond -f
      exit 0
    fi
  fi

  # 执行多账号签到任务
  declare -a tokens
  IFS=',' read -ra tokens <<< "$SKLAND_TOKEN"

  local i=1
  declare token
  for token in ${tokens[*]}; do
    info "正在处理账号 #$i"
    do_attendance "$token"
    i=$(($i + 1))
  done
}
main