mirror of
https://github.com/silenceper/wechat.git
synced 2026-02-04 12:52:27 +08:00
258 lines
7.9 KiB
Go
258 lines
7.9 KiB
Go
package security
|
||
|
||
import (
|
||
context2 "context"
|
||
"fmt"
|
||
"strconv"
|
||
|
||
"github.com/silenceper/wechat/v2/miniprogram/context"
|
||
"github.com/silenceper/wechat/v2/util"
|
||
)
|
||
|
||
const (
|
||
mediaCheckAsyncURL = "https://api.weixin.qq.com/wxa/media_check_async?access_token=%s"
|
||
imageCheckURL = "https://api.weixin.qq.com/wxa/img_sec_check?access_token=%s"
|
||
msgCheckURL = "https://api.weixin.qq.com/wxa/msg_sec_check?access_token=%s"
|
||
)
|
||
|
||
// Security 内容安全
|
||
type Security struct {
|
||
*context.Context
|
||
}
|
||
|
||
// NewSecurity init
|
||
func NewSecurity(ctx *context.Context) *Security {
|
||
return &Security{ctx}
|
||
}
|
||
|
||
// MediaCheckAsyncV1Request 图片/音频异步校验请求参数
|
||
type MediaCheckAsyncV1Request struct {
|
||
MediaURL string `json:"media_url"` // 要检测的图片或音频的url，支持图片格式包括jpg, jepg, png, bmp, gif（取首帧），支持的音频格式包括mp3, aac, ac3, wma, flac, vorbis, opus, wav
|
||
MediaType uint8 `json:"media_type"` // 1:音频;2:图片
|
||
}
|
||
|
||
// MediaCheckAsyncV1 异步校验图片/音频是否含有违法违规内容
|
||
// Deprecated
|
||
// 在2021年9月1日停止更新，请尽快更新至 2.0 接口。建议使用 MediaCheckAsync
|
||
func (security *Security) MediaCheckAsyncV1(in *MediaCheckAsyncV1Request) (traceID string, err error) {
|
||
accessToken, err := security.GetAccessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
uri := fmt.Sprintf(mediaCheckAsyncURL, accessToken)
|
||
response, err := util.PostJSON(uri, in)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 使用通用方法返回错误
|
||
var res struct {
|
||
util.CommonError
|
||
TraceID string `json:"trace_id"`
|
||
}
|
||
err = util.DecodeWithError(response, &res, "MediaCheckAsyncV1")
|
||
return res.TraceID, err
|
||
}
|
||
|
||
// MediaCheckAsyncRequest 图片/音频异步校验请求参数
|
||
type MediaCheckAsyncRequest struct {
|
||
MediaURL string `json:"media_url"` // 要检测的图片或音频的url，支持图片格式包括jpg, jepg, png, bmp, gif（取首帧），支持的音频格式包括mp3, aac, ac3, wma, flac, vorbis, opus, wav
|
||
MediaType uint8 `json:"media_type"` // 1:音频;2:图片
|
||
OpenID string `json:"openid"` // 用户的openid（用户需在近两小时访问过小程序）
|
||
Scene uint8 `json:"scene"` // 场景枚举值（1 资料；2 评论；3 论坛；4 社交日志）
|
||
}
|
||
|
||
// MediaCheckAsync 异步校验图片/音频是否含有违法违规内容
|
||
func (security *Security) MediaCheckAsync(in *MediaCheckAsyncRequest) (traceID string, err error) {
|
||
return security.MediaCheckAsyncContext(context2.Background(), in)
|
||
}
|
||
|
||
// MediaCheckAsyncContext 异步校验图片/音频是否含有违法违规内容
|
||
func (security *Security) MediaCheckAsyncContext(ctx context2.Context, in *MediaCheckAsyncRequest) (traceID string, err error) {
|
||
accessToken, err := security.GetAccessTokenContext(ctx)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
var req struct {
|
||
MediaCheckAsyncRequest
|
||
Version uint `json:"version"` // 接口版本号，2.0版本为固定值2
|
||
}
|
||
req.MediaCheckAsyncRequest = *in
|
||
req.Version = 2
|
||
|
||
uri := fmt.Sprintf(mediaCheckAsyncURL, accessToken)
|
||
response, err := util.PostJSONContext(ctx, uri, req)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 使用通用方法返回错误
|
||
var res struct {
|
||
util.CommonError
|
||
TraceID string `json:"trace_id"`
|
||
}
|
||
err = util.DecodeWithError(response, &res, "MediaCheckAsync")
|
||
return res.TraceID, err
|
||
}
|
||
|
||
// ImageCheckV1 校验一张图片是否含有违法违规内容（同步）
|
||
// https://developers.weixin.qq.com/miniprogram/dev/framework/security.imgSecCheck.html
|
||
// Deprecated
|
||
// 在2021年9月1日停止更新。建议使用 MediaCheckAsync
|
||
func (security *Security) ImageCheckV1(filename string) (err error) {
|
||
accessToken, err := security.GetAccessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
uri := fmt.Sprintf(imageCheckURL, accessToken)
|
||
response, err := util.PostFile("media", filename, uri)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 使用通用方法返回错误
|
||
return util.DecodeWithCommonError(response, "ImageCheckV1")
|
||
}
|
||
|
||
// CheckSuggest 检查建议
|
||
type CheckSuggest string
|
||
|
||
const (
|
||
// CheckSuggestRisky 违规风险建议
|
||
CheckSuggestRisky CheckSuggest = "risky"
|
||
// CheckSuggestPass 安全
|
||
CheckSuggestPass CheckSuggest = "pass"
|
||
// CheckSuggestReview 需要审查
|
||
CheckSuggestReview CheckSuggest = "review"
|
||
)
|
||
|
||
// MsgScene 文本场景
|
||
type MsgScene uint8
|
||
|
||
const (
|
||
// MsgSceneMaterial 资料文件检查场景
|
||
MsgSceneMaterial MsgScene = iota + 1
|
||
// MsgSceneComment 评论
|
||
MsgSceneComment
|
||
// MsgSceneForum 论坛
|
||
MsgSceneForum
|
||
// MsgSceneSocialLog 社交日志
|
||
MsgSceneSocialLog
|
||
)
|
||
|
||
// CheckLabel 检查命中标签
|
||
type CheckLabel int
|
||
|
||
func (cl CheckLabel) String() string {
|
||
switch cl {
|
||
case 100:
|
||
return "正常"
|
||
case 10001:
|
||
return "广告"
|
||
case 20001:
|
||
return "时政"
|
||
case 20002:
|
||
return "色情"
|
||
case 20003:
|
||
return "辱骂"
|
||
case 20006:
|
||
return "违法犯罪"
|
||
case 20008:
|
||
return "欺诈"
|
||
case 20012:
|
||
return "低俗"
|
||
case 20013:
|
||
return "版权"
|
||
case 21000:
|
||
return "其他"
|
||
default:
|
||
return strconv.Itoa(int(cl))
|
||
}
|
||
}
|
||
|
||
// MsgCheckRequest 文本检查请求
|
||
type MsgCheckRequest struct {
|
||
OpenID string `json:"openid"` // 用户的openid（用户需在近两小时访问过小程序）
|
||
Scene MsgScene `json:"scene"` // 场景枚举值（1 资料；2 评论；3 论坛；4 社交日志）
|
||
Content string `json:"content"` // 需检测的文本内容，文本字数的上限为 2500 字，需使用 UTF-8 编码
|
||
Nickname string `json:"nickname"` // （非必填）用户昵称，需使用UTF-8编码
|
||
Title string `json:"title"` // （非必填）文本标题，需使用UTF-8编码
|
||
Signature string `json:"signature"` // （非必填）个性签名，该参数仅在资料类场景有效(scene=1)，需使用UTF-8编码
|
||
}
|
||
|
||
// MsgCheckResponse 文本检查响应
|
||
type MsgCheckResponse struct {
|
||
util.CommonError
|
||
TraceID string `json:"trace_id"` // 唯一请求标识
|
||
Result struct {
|
||
Suggest CheckSuggest `json:"suggest"` // 建议
|
||
Label CheckLabel `json:"label"` // 命中标签
|
||
} `json:"result"` // 综合结果
|
||
Detail []struct {
|
||
ErrCode int64 `json:"errcode"` // 错误码，仅当该值为0时，该项结果有效
|
||
Strategy string `json:"strategy"` // 策略类型
|
||
Suggest string `json:"suggest"` // 建议
|
||
Label CheckLabel `json:"label"` // 命中标签
|
||
Prob uint `json:"prob"` // 置信度。0-100，越高代表越有可能属于当前返回的标签（label）
|
||
Keyword string `json:"keyword"` // 命中的自定义关键词
|
||
} `json:"detail"` // 详细检测结果
|
||
}
|
||
|
||
// MsgCheckV1 检查一段文本是否含有违法违规内容
|
||
// Deprecated
|
||
// 在2021年9月1日停止更新，请尽快更新至 2.0 接口。建议使用 MsgCheck
|
||
func (security *Security) MsgCheckV1(content string) (res MsgCheckResponse, err error) {
|
||
accessToken, err := security.GetAccessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
var req struct {
|
||
Content string `json:"content"`
|
||
}
|
||
req.Content = content
|
||
|
||
uri := fmt.Sprintf(msgCheckURL, accessToken)
|
||
response, err := util.PostJSON(uri, req)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 使用通用方法返回错误
|
||
err = util.DecodeWithError(response, &res, "security.MsgCheckV1")
|
||
return
|
||
}
|
||
|
||
// MsgCheck 检查一段文本是否含有违法违规内容
|
||
func (security *Security) MsgCheck(in *MsgCheckRequest) (res MsgCheckResponse, err error) {
|
||
return security.MsgCheckContext(context2.Background(), in)
|
||
}
|
||
|
||
// MsgCheckContext 检查一段文本是否含有违法违规内容
|
||
func (security *Security) MsgCheckContext(ctx context2.Context, in *MsgCheckRequest) (res MsgCheckResponse, err error) {
|
||
accessToken, err := security.GetAccessTokenContext(ctx)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
var req struct {
|
||
MsgCheckRequest
|
||
Version uint `json:"version"`
|
||
}
|
||
req.MsgCheckRequest = *in
|
||
req.Version = 2
|
||
|
||
uri := fmt.Sprintf(msgCheckURL, accessToken)
|
||
response, err := util.PostJSONContext(ctx, uri, req)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
// 使用通用方法返回错误
|
||
err = util.DecodeWithError(response, &res, "security.MsgCheck")
|
||
return
|
||
}
|