76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package watermark
|
||
|
||
import (
|
||
"context"
|
||
"net/http"
|
||
"net/url"
|
||
"strconv"
|
||
"time"
|
||
|
||
"go-common/app/interface/main/creative/conf"
|
||
"go-common/app/interface/main/creative/dao/tool"
|
||
"go-common/app/interface/main/creative/model/watermark"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
const (
|
||
_genWm = "/x/internal/image/gen"
|
||
// upload_key true string 上传的key(需要拿到分配到key,确定上传的bucket)
|
||
// uploadKey = "creative"
|
||
// wm_key true string 水印key，新业务需要提前向bfs申请（未上线)
|
||
wmKey = "creative"
|
||
// wm_text true string 水印文字，限制不超过20个字符
|
||
// wm_vertical true bool 水印图片文字，是否水平排列
|
||
wmVertical = "true"
|
||
// wm_scale true float 水印文字宽度和水印图片宽度比值(0,1]，默认1
|
||
wmScale = 1
|
||
//水印图片到文字的距离. 默认是1
|
||
distance = 1
|
||
)
|
||
|
||
// GenWm set wm by api.
|
||
func (d *Dao) GenWm(c context.Context, mid int64, wmText, ip string) (gm *watermark.GenWatermark, err error) {
|
||
params := url.Values{}
|
||
params.Set("wm_key", wmKey)
|
||
params.Set("wm_text", wmText)
|
||
params.Set("wm_vertical", wmVertical)
|
||
params.Set("wm_scale", strconv.Itoa(wmScale))
|
||
params.Set("distance", strconv.Itoa(distance))
|
||
params.Set("mid", strconv.FormatInt(mid, 10))
|
||
params.Set("appkey", conf.Conf.App.Key)
|
||
params.Set("appsecret", conf.Conf.App.Secret)
|
||
params.Set("ts", strconv.FormatInt(time.Now().Unix(), 10))
|
||
var (
|
||
req *http.Request
|
||
res struct {
|
||
Code int `json:"code"`
|
||
Data watermark.GenWatermark `json:"data"`
|
||
}
|
||
query, _ = tool.Sign(params)
|
||
genWmURL = d.genWm + "?" + query
|
||
)
|
||
log.Info("genWm url(%v)", genWmURL)
|
||
if req, err = http.NewRequest("POST", genWmURL, nil); err != nil {
|
||
log.Error("genWm url(%s) error(%v)", genWmURL, err)
|
||
return
|
||
}
|
||
req.Header.Set("X-BACKEND-BILI-REAL-IP", ip)
|
||
if err = d.client.Do(c, req, &res); err != nil {
|
||
log.Error("genWm url(%s) response(%v) error(%v)", genWmURL, res, err)
|
||
return
|
||
}
|
||
if res.Code != 0 {
|
||
log.Error("genWm url(%s) res(%v)", genWmURL, res)
|
||
err = ecode.Int(res.Code)
|
||
return
|
||
}
|
||
gm = &watermark.GenWatermark{
|
||
Location: res.Data.Location,
|
||
MD5: res.Data.MD5,
|
||
Width: res.Data.Width,
|
||
Height: res.Data.Height,
|
||
}
|
||
return
|
||
}
|