openbilibili/app/interface/main/app-interface/dao/bangumi/dao.go

package bangumi

import (
	"context"
	"encoding/json"
	"net/url"
	"strconv"

	"go-common/app/interface/main/app-interface/conf"
	"go-common/app/interface/main/app-interface/model/bangumi"
	seasongrpc "go-common/app/service/openplatform/pgc-season/api/grpc/season/v1"
	"go-common/library/ecode"
	"go-common/library/log"
	httpx "go-common/library/net/http/blademaster"
	"go-common/library/net/metadata"
	"go-common/library/xstr"

	"github.com/pkg/errors"
)

const (
	_season     = "/api/inner/season"
	_movie      = "/internal_api/movie_aid_info"
	_bp         = "/sponsor/inner/xAjaxGetBP"
	_concern    = "/api/get_concerned_season"
	_hasFollows = "/follow/internal_api/has_follows"
	_card       = "/pgc/internal/season/search/card"
	_favDisplay = "/pgc/internal/follow/app/tab/view"
)

// Dao is bangumi dao
type Dao struct {
	client     *httpx.Client
	season     string
	movie      string
	bp         string
	concern    string
	hasFollows string
	card       string
	favDisplay string
	// grpc
	rpcClient seasongrpc.SeasonClient
}

// New bangumi dao
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		client:     httpx.NewClient(c.HTTPBangumi),
		season:     c.Host.Bangumi + _season,
		movie:      c.Host.Bangumi + _movie,
		bp:         c.Host.Bangumi + _bp,
		concern:    c.Host.Bangumi + _concern,
		hasFollows: c.Host.Bangumi + _hasFollows,
		card:       c.Host.APICo + _card,
		favDisplay: c.Host.APICo + _favDisplay,
	}
	var err error
	if d.rpcClient, err = seasongrpc.NewClient(c.PGCRPC); err != nil {
		log.Error("seasongrpc NewClientt error(%v)", err)
	}
	return
}

// Season bangumi Season.
func (d *Dao) Season(c context.Context, aid, mid int64, ip string) (s *bangumi.Season, err error) {
	params := url.Values{}
	params.Set("aid", strconv.FormatInt(aid, 10))
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("type", "av")
	params.Set("build", "app-interface")
	params.Set("platform", "Golang")
	var res struct {
		Code   int             `json:"code"`
		Result *bangumi.Season `json:"result"`
	}
	if err = d.client.Get(c, d.season, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.season+"?"+params.Encode())
		return
	}
	s = res.Result
	return
}

// BPInfo get bp info data.
func (d *Dao) BPInfo(c context.Context, aid, mid int64, ip string) (data json.RawMessage, err error) {
	params := url.Values{}
	params.Set("aid", strconv.FormatInt(aid, 10))
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("build", "app-interface")
	params.Set("platform", "Golang")
	var res struct {
		Code int             `json:"code"`
		Data json.RawMessage `json:"data"`
	}
	if err = d.client.Get(c, d.bp, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.bp+"?"+params.Encode())
		return
	}
	data = res.Data
	return
}

// Movie bangumi Movie
func (d *Dao) Movie(c context.Context, aid, mid int64, build int, mobiApp, device, ip string) (m *bangumi.Movie, err error) {
	params := url.Values{}
	params.Set("id", strconv.FormatInt(aid, 10))
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("build", strconv.Itoa(build))
	params.Set("device", device)
	params.Set("mobi_app", mobiApp)
	params.Set("platform", "Golang")
	var res struct {
		Code   int            `json:"code"`
		Result *bangumi.Movie `json:"result"`
	}
	if err = d.client.Get(c, d.movie, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.movie+"?"+params.Encode())
		return
	}
	m = res.Result
	return
}

// Concern get concern data from api.
func (d *Dao) Concern(c context.Context, mid, vmid int64, pn, ps int) (ss []*bangumi.Season, total int, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("taid", strconv.FormatInt(vmid, 10))
	params.Set("page", strconv.Itoa(pn))
	params.Set("pagesize", strconv.Itoa(ps))
	params.Set("build", "app-interface")
	params.Set("platform", "Golang")
	var res struct {
		Code   int               `json:"code"`
		Total  string            `json:"count"`
		Result []*bangumi.Season `json:"result"`
	}
	if err = d.client.Get(c, d.concern, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.concern+"?"+params.Encode())
		return
	}
	ss = res.Result
	total, _ = strconv.Atoi(res.Total)
	return
}

// HasFollows get bngumi tab.
func (d *Dao) HasFollows(c context.Context, mid int64) (has bool, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("type", "2,3,5")
	var res struct {
		Code   int `json:"code"`
		Result int `json:"result"`
	}
	if err = d.client.Get(c, d.hasFollows, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.hasFollows+"?"+params.Encode())
		return
	}
	if res.Result == 1 {
		has = true
	}
	return
}

// Card bangumi card.
func (d *Dao) Card(c context.Context, mid int64, sids []int64) (s map[string]*bangumi.Card, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("season_ids", xstr.JoinInts(sids))
	var res struct {
		Code   int                      `json:"code"`
		Result map[string]*bangumi.Card `json:"result"`
	}
	if err = d.client.Get(c, d.card, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.card+"?"+params.Encode())
		return
	}
	s = res.Result
	return
}

// FavDisplay fav tab display or not.
func (d *Dao) FavDisplay(c context.Context, mid int64) (bangumi, cinema int, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	var res struct {
		Code   int `json:"code"`
		Result struct {
			Bangumi int `json:"bangumi"`
			Cinema  int `json:"cinema"`
		} `json:"result"`
	}
	if err = d.client.Get(c, d.favDisplay, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.favDisplay+"?"+params.Encode())
		return
	}
	bangumi = res.Result.Bangumi
	cinema = res.Result.Cinema
	return
}