openbilibili/app/interface/main/app-view/dao/report/dao.go

package report

import (
	"context"
	"crypto/hmac"
	"crypto/sha1"
	"encoding/base64"
	"fmt"
	"hash"
	"io"
	"net/http"
	"net/url"
	"strconv"
	"time"

	"go-common/app/interface/main/app-view/conf"
	"go-common/library/ecode"
	httpx "go-common/library/net/http/blademaster"
	"go-common/library/net/metadata"

	"github.com/pkg/errors"
)

const (
	_add      = "/videoup/archive/report"
	_timeout  = 800 * time.Millisecond
	_bucket   = "archive"
	_url      = "http://bfs.bilibili.co/bfs/archive/"
	_template = "%s\n%s\n\n%d\n"
	_method   = "PUT"
	_key      = "8d4e593ba7555502"
	_secret   = "0bdbd4c7caeeddf587c3c4daec0475"
)

// Dao is report dao
type Dao struct {
	client    *httpx.Client
	bfsClient *http.Client
	add       string
}

// New is appeal inital func .
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		client:    httpx.NewClient(c.HTTPWrite),
		bfsClient: http.DefaultClient,
		add:       c.Host.Archive + _add,
	}
	return
}

// AddAppeal add appeal .
func (d *Dao) AddReport(c context.Context, mid, aid int64, mold int, ak, reason, pics string) (err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("access_key", ak)
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("aid", strconv.FormatInt(aid, 10))
	params.Set("type", strconv.Itoa(mold))
	params.Set("reason", reason)
	params.Set("pics", pics)
	var res struct {
		Code int `json:"code"`
	}
	if err = d.client.Post(c, d.add, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.add+"?"+params.Encode())
	}
	return
}

// Upload imgage upload .
func (d *Dao) Upload(c context.Context, fileType string, body io.Reader) (location string, err error) {
	req, err := http.NewRequest(_method, _url, body)
	if err != nil {
		return
	}
	expire := time.Now().Unix()
	authorization := authorize(_key, _secret, _method, _bucket, expire)
	req.Header.Set("Host", _url)
	req.Header.Add("Date", fmt.Sprint(expire))
	req.Header.Add("Authorization", authorization)
	req.Header.Add("Content-Type", fileType)
	// timeout
	c, cancel := context.WithTimeout(c, _timeout)
	req = req.WithContext(c)
	defer cancel()
	resp, err := d.bfsClient.Do(req)
	if err != nil {
		return
	}
	if resp.StatusCode != http.StatusOK {
		err = errors.Wrap(ecode.Int(resp.StatusCode), req.URL.String())
		return
	}
	code := resp.Header.Get("Code")
	if code != strconv.Itoa(http.StatusOK) {
		err = errors.Wrap(ecode.String(code), req.URL.String())
		return
	}
	location = resp.Header.Get("Location")
	return
}

// authorize returns authorization for upload file to bfs
func authorize(key, secret, method, bucket string, expire int64) (authorization string) {
	var (
		content   string
		mac       hash.Hash
		signature string
	)
	content = fmt.Sprintf(_template, method, bucket, expire)
	mac = hmac.New(sha1.New, []byte(secret))
	mac.Write([]byte(content))
	signature = base64.StdEncoding.EncodeToString(mac.Sum(nil))
	authorization = fmt.Sprintf("%s:%s:%d", key, signature, expire)
	return
}