openbilibili/app/service/main/videoup/model/archive/forbid.go

package archive

// const Forbid Enum Values
const (
	// Forbid style
	ForbidRank      = "rank"
	ForbidDynamic   = "dynamic"
	ForbidRecommend = "recommend"
	ForbidShow      = "show"
	// ForbidRank style
	ForbidRankMain      = 0
	ForbidRankRecentArc = 1
	ForbidRankAllArc    = 2
	// ForbidDynamic style
	ForbidDynamicMain = 0
	// ForbidRecommend style
	ForbidRecommendMain = 0
	// ForbidShow style
	ForbidShowMain    = 0
	ForbidShowMobile  = 1
	ForbidShowWeb     = 2
	ForbidShowOversea = 3
	ForbidShowOnline  = 4
)

// ForbidAttr forbid attribute
type ForbidAttr struct {
	Aid        int64 `json:"aid"`
	RankV      int32 `json:"rank_attr"`
	DynamicV   int32 `json:"dynamic_attr"`
	RecommendV int32 `json:"recommend_attr"`
	ShowV      int32 `json:"show_attr"`
	SearchV    int32 `json:"search"`
	PushBlogV  int32 `json:"push_blog"`
	// specific
	Rank struct {
		Main      int32 `json:"main"`
		RecentArc int32 `json:"recent_arc"`
		AllArc    int32 `json:"all_arc"`
	} `json:"_"`
	Dynamic struct {
		Main int32 `json:"main"`
	} `json:"-"`
	Recommend struct {
		Main int32 `json:"main"`
	} `json:"-"`
	Show struct {
		Main    int32 `json:"main"`
		Mobile  int32 `json:"mobile"`
		Web     int32 `json:"web"`
		Oversea int32 `json:"oversea"`
		Online  int32 `json:"online"`
	} `json:"-"`
}

// Convert convert db value into attr.
func (f *ForbidAttr) Convert() {
	// rank
	f.Rank.Main = f.RankV & 1
	f.Rank.RecentArc = (f.RankV >> 1) & 1
	f.Rank.AllArc = (f.RankV >> 2) & 1
	// dynamic
	f.Dynamic.Main = f.DynamicV & 1
	// recommend
	f.Recommend.Main = f.RecommendV & 1
	// show
	f.Show.Main = f.ShowV & 1
	f.Show.Mobile = (f.ShowV >> 1) & 1
	f.Show.Web = (f.ShowV >> 2) & 1
	f.Show.Oversea = (f.ShowV >> 3) & 1
	f.Show.Online = (f.ShowV >> 4) & 1
}

// Reverse reverse attr into db value.
func (f *ForbidAttr) Reverse() {
	// rank
	f.RankV = (f.Rank.AllArc << 2) | (f.Rank.RecentArc << 1) | f.Rank.Main
	// dynamic
	f.DynamicV = f.Dynamic.Main
	// recommend
	f.RecommendV = f.Recommend.Main
	// show
	f.ShowV = (f.Show.Online << 4) | (f.Show.Oversea << 3) | (f.Show.Web << 2) | (f.Show.Mobile << 1) | f.Show.Main
}

// SetAttr set forbid attr.
func (f *ForbidAttr) SetAttr(name string, v int32, bit uint) (change bool) {
	if name == ForbidRank {
		old := f.RankV
		f.RankV = f.RankV&(^(1 << bit)) | (v << bit)
		change = old == f.RankV
	} else if name == ForbidDynamic {
		old := f.DynamicV
		f.DynamicV = f.DynamicV&(^(1 << bit)) | (v << bit)
		change = old == f.DynamicV
	} else if name == ForbidRecommend {
		old := f.RecommendV
		f.RecommendV = f.RecommendV&(^(1 << bit)) | (v << bit)
		change = old == f.RecommendV
	} else if name == ForbidShow {
		old := f.ShowV
		f.ShowV = f.ShowV&(^(1 << bit)) | (v << bit)
		change = old == f.ShowV
	}
	return
}