openbilibili/app/interface/main/videoup/service/check_staff.go

package service

import (
	"context"
	"fmt"
	"go-common/app/interface/main/videoup/model/archive"
	accapi "go-common/app/service/main/account/api"
	"go-common/library/ecode"
	"go-common/library/log"
	"go-common/library/sync/errgroup"
	"strings"
	"sync"
	"unicode/utf8"
)

// checkAddStaff 新增稿件：检查联合投稿参数
func (s *Service) checkAddStaff(c context.Context, ap *archive.ArcParam, mid int64, ip string) (err error) {
	var (
		_logs  []string
		_logs2 []string
	)
	defer func() {
		logStr := strings.Join(_logs, "\n")
		if err != nil {
			log.Error("s.checkAddStaff ap: %+v \nmid: %d \nlogs\n:%s", ap, mid, logStr)
		} else {
			log.Info("s.checkAddStaff ap: %+v \nmid: %d \nlogs\n%s", ap, mid, logStr)
		}
	}()
	ap.HandleStaff = false
	//新增稿件时，如果没填写Staffs，则不是联合投稿，跳过检查
	if len(ap.Staffs) == 0 {
		ap.Staffs = make([]*archive.Staff, 0)
		_logs = append(_logs, "INFO:非联合投稿，忽略")
		return
	}
	//如果不是自制稿件，不允许传Staffs
	if ap.Copyright != archive.CopyrightOriginal && len(ap.Staffs) != 0 {
		_logs = append(_logs, "ERR：非自制稿件，不允许传Staffs。")
		err = ecode.VideoupStaffCopyright
		return
	}
	//非全量的情况下，检查当前UP主是在白名单中
	_logs = append(_logs, fmt.Sprintf("INFO：灰度开关 %v。分区配置 %v 注意：这里打出的数据可能不是最新的，有可能后面重新Load了", s.staffGary, s.staffTypeCache))
	if err = s.checkStaffGray(ap.TypeID, mid); err != nil {
		_logs = append(_logs, fmt.Sprintf("INFO：UP主(%d)不在灰度名单中。", mid))
		return
	}
	_logs2, err = s.checkStaffData(c, ap.TypeID, ap.Staffs, []*archive.Staff{}, ip)
	_logs = append(_logs, _logs2...)
	if err != nil {
		return
	}
	//检查拉黑情况
	_logs2, _, err = s.checkStaffRelation(c, mid, ap.Staffs, ip)
	_logs = append(_logs, _logs2...)
	if err != nil {
		return
	}

	ap.HandleStaff = true
	_logs = append(_logs, "INFO:通过")
	return
}

// checkEditStaff 稿件编辑：检查联合投稿相关参数
func (s *Service) checkEditStaff(c context.Context, ap *archive.ArcParam, mid int64, a *archive.Archive, ip string) (err error) {
	var (
		cStaffs    []*archive.Staff //当前稿件的联合投稿人
		diffStaffs []*archive.Staff //变更的联合投稿人
		_logs      []string
		_logs2     []string
	)
	defer func() {
		logStr := strings.Join(_logs, "\n")
		if err != nil {
			log.Error("s.checkEditStaff ap: %+v \nmid: %d \narchive:%+v \nlogs:\n%v", ap, mid, a, logStr)
		} else {
			log.Info("s.checkEditStaff ap: %+v \nmid: %d \narchive:%+v \nlogs\n%v", ap, mid, a, logStr)
		}
	}()
	ap.HandleStaff = false
	//如果UP主不在白名单中，不修改Staff数据
	_logs = append(_logs, fmt.Sprintf("INFO：灰度开关 %v，分区配置 %v 注意：这里打出的数据可能不是最新的，有可能后面重新Load了", s.staffGary, s.staffTypeCache))
	if err = s.checkStaffGray(ap.TypeID, mid); err != nil {
		_logs = append(_logs, fmt.Sprintf("INFO:UP主(%d)不在灰度中", mid))
		//如果UP主不在白名单中，则将staff置空，不修改staff
		ap.Staffs = []*archive.Staff{}
		ap.HandleStaff = false
		err = nil
		return
	}
	//获取当前稿件的Staff信息
	if cStaffs, err = s.arc.ApplyStaffs(c, a.Aid, ip); err != nil {
		_logs = append(_logs, fmt.Sprintf("ERR:获取线上Staffs失败，err:%v", err))
		return
	}
	if len(ap.Staffs) == 0 { //删除Staffs
		_logs2, _, err = s.checkStaffRelation(c, mid, cStaffs, ip)
		_logs = append(_logs, _logs2...)
		if err != nil {
			return
		}
		_logs = append(_logs, "INFO:UP主删除所有Staff")
		ap.HandleStaff = true
		return
	}
	//preEdit()的时候已经加上了以下"移区"、"换类型"的判断逻辑，这里冗余一下
	//联合投稿不允许自制稿件改成非自制
	if a.Copyright == archive.CopyrightOriginal && ap.Copyright != archive.CopyrightOriginal && a.AttrVal(archive.AttrBitStaff) == archive.AttrYes {
		_logs = append(_logs, "ERROR:联合投稿不允许自制稿件改成非自制")
		err = ecode.VideoupStaffChangeCopyright
		return
	}
	_logs2, err = s.checkStaffData(c, ap.TypeID, ap.Staffs, cStaffs, ip)
	_logs = append(_logs, _logs2...)
	if err != nil {
		return
	}
	//只验证修改过的Staff信息
	changes, _logs2 := s.getStaffChanges(cStaffs, ap.Staffs)
	_logs = append(_logs, _logs2...)
	_logs = append(_logs, fmt.Sprintf("INFO:当前Staffs：%+v 提交Staffs：%+v 修改的Staffs：%+v", cStaffs, ap.Staffs, changes))
	for _, v := range changes {
		for _, m := range v {
			diffStaffs = append(diffStaffs, m)
		}
	}
	_logs2, _, err = s.checkStaffRelation(c, mid, diffStaffs, ip)
	_logs = append(_logs, _logs2...)
	if err != nil {
		return
	}
	ap.HandleStaff = true
	_logs = append(_logs, "INFO:通过")
	return
}

// checkStaffGray
func (s *Service) checkStaffGray(typeid int16, mid int64) (err error) {
	if s.staffGary {
		if _, ok := s.staffUps[mid]; !ok {
			log.Error("当前Up主(%d)不在联合投稿白名单中。", mid)
			err = ecode.VideoupStaffAuth
			return
		}
		var ok1, ok2 bool
		_, ok1 = s.staffTypeCache[typeid]
		_, ok2 = s.staffTypeCache[0]
		if ok1 && s.staffTypeCache[typeid].MaxStaff <= 0 {
			log.Error("当前分区(%d)在联合投稿黑名单中。", typeid)
			err = ecode.VideoupStaffTypeNotExists
			return
		}
		if !ok1 && !ok2 {
			log.Error("当前分区(%d)不在联合投稿白名单中。", typeid)
			err = ecode.VideoupStaffTypeNotExists
			return
		}
	}
	return
}

// checkStaffData 检查联合投稿人的数据格式（全量验证，不管UP主有没有编辑，都会走这个验证）
func (s *Service) checkStaffData(c context.Context, typeid int16, staffs, cStaffs []*archive.Staff, ip string) (_logs []string, err error) {
	var (
		titles              []string
		cards               map[int64]*accapi.Card
		staffMids           []int64
		staffMap, cStaffMap map[int64]string
		maxStaff            int
	)
	staffMap = make(map[int64]string)
	cStaffMap = make(map[int64]string)
	if len(cStaffs) != 0 {
		for _, v := range cStaffs {
			cStaffMap[v.Mid] = v.Title
		}
	}
	//检查分区的配置
	if conf, ok := s.staffTypeCache[typeid]; ok {
		maxStaff = conf.MaxStaff
	} else if conf, ok := s.staffTypeCache[0]; ok {
		maxStaff = conf.MaxStaff
	} else {
		_logs = append(_logs, fmt.Sprintf("ERR：分区%d不在配置里。配置：%v", typeid, s.staffTypeCache))
		err = ecode.VideoupStaffTypeNotExists
		return
	}
	if maxStaff == 0 {
		_logs = append(_logs, fmt.Sprintf("ERR：分区%d是黑名单。配置：%v", typeid, s.staffTypeCache))
		err = ecode.VideoupStaffTypeNotExists
		return
	}
	//检查Staff数量
	if len(staffs) > maxStaff {
		_logs = append(_logs, fmt.Sprintf("ERR：Staff数量超限。最多：%d；传递：%d", maxStaff, len(staffs)))
		err = ecode.VideoupStaffCountLimit
		return
	}
	//检查Staff职能、Mid
	for i, v := range staffs {
		staffs[i].Title = strings.TrimSpace(v.Title)
		v.Title = staffs[i].Title
		if v.Mid == 0 {
			_logs = append(_logs, "ERR：Staff Mid为0。")
			err = ecode.VideoupStaffMidInvalid
			return
		}
		tl := utf8.RuneCountInString(v.Title)
		if tl < 2 {
			_logs = append(_logs, fmt.Sprintf("ERR：职能(%v)长度不合法，长度：%d。", v.Title, tl))
			err = ecode.VideoupStaffTitleShort
			return
		}
		if tl > 4 {
			_logs = append(_logs, fmt.Sprintf("ERR：职能(%v)长度不合法，长度：%d。", v.Title, tl))
			err = ecode.VideoupStaffTitleLength
			return
		}
		if !_staffNameReg.MatchString(v.Title) {
			_logs = append(_logs, fmt.Sprintf("ERR：职能(%v)字符不合法。", v.Title))
			err = ecode.VideoupStaffTitleChar
			return
		}
		//不校验未修改的职能
		if cTitle, ok := cStaffMap[v.Mid]; !ok || cTitle != v.Title {
			titles = append(titles, v.Title)
		}
		staffMap[v.Mid] = v.Title
		staffMids = append(staffMids, v.Mid)
	}
	if len(staffMap) != len(staffs) {
		_logs = append(_logs, fmt.Sprintf("ERR：Staff存在重复。传递：%v；去重后：%v", staffs, staffMap))
		err = ecode.VideoupStaffMidRepeat
		return
	}
	//职能名称敏感词
	_, hit, err := s.filter.VideoMultiFilter(c, titles, ip)
	if err != nil {
		_logs = append(_logs, fmt.Sprintf("ERR：职能敏感词接口失败。error：%v", err))
		return
	}
	if len(hit) > 0 {
		_logs = append(_logs, fmt.Sprintf("ERR：职能存在敏感词。敏感词：%v", hit))
		err = ecode.VideoupStaffTitleFilter
		return
	}
	//Staff Mid合法性检查
	if cards, err = s.acc.Cards(c, staffMids, ip); err != nil {
		_logs = append(_logs, fmt.Sprintf("ERR：Staff账号信息获取失败。error：%v", err))
		return
	}
	for _, v := range staffMids {
		if _, ok := cards[v]; !ok {
			_logs = append(_logs, fmt.Sprintf("ERR：Staff Mid(%d)不存在。", v))
			err = ecode.VideoupStaffMidInvalid
			return
		}
	}
	return
}

// checkStaffRelation 检查联合投稿人拉黑关系
func (s *Service) checkStaffRelation(c context.Context, mid int64, staffs []*archive.Staff, ip string) (_logs []string, blocked []*archive.Staff, err error) {
	var (
		mids     []int64
		rels     map[int64]int //relations
		pass     = true
		staffMap map[int64]*archive.Staff
		cards    map[int64]*accapi.Card
	)
	blocked = make([]*archive.Staff, 0)
	staffMap = make(map[int64]*archive.Staff)
	for _, v := range staffs {
		mids = append(mids, v.Mid)
		staffMap[v.Mid] = v
	}
	if rels, err = s.FRelations(c, mid, mids, ip); err != nil {
		_logs = append(_logs, fmt.Sprintf("ERR：获取拉黑信息失败，error:%v", err))
		return
	}
	for k, v := range rels {
		if v >= 128 {
			_logs = append(_logs, fmt.Sprintf("ERR：Staff(%d)在黑名单中", k))
			pass = false
			blocked = append(blocked, staffMap[k])
			continue
		}
	}
	cards, err = s.acc.Cards(c, mids, ip)
	if !pass {
		if err != nil {
			_logs = append(_logs, fmt.Sprintf("ERR：账号(%v)信息获取失败 error:%v", mids, err))
			err = ecode.Errorf(ecode.VideoupStaffBlocked, ecode.VideoupStaffBlocked.Message(), "")
			return
		}
		var bNames []string
		for _, v := range blocked {
			if _, ok := cards[v.Mid]; ok {
				bNames = append(bNames, cards[v.Mid].Name)
			} else {
				_logs = append(_logs, fmt.Sprintf("ERR：账号(%d)信息获取失败 error:%v", v.Mid, err))
			}
		}
		err = ecode.Errorf(ecode.VideoupStaffBlocked, ecode.VideoupStaffBlocked.Message(), strings.Join(bNames, "、"))
	}
	for _, staff := range staffs {
		if _, ok := cards[staff.Mid]; !ok {
			_logs = append(_logs, fmt.Sprintf("ERR：账号(%d)信息获取失败 error:%v", staff.Mid, err))
			err = ecode.Errorf(ecode.CreativeAccServiceErr, "参与者(%d)信息获取失败", staff.Mid)
			return
		}
		if cards[staff.Mid].Silence == 1 {
			_logs = append(_logs, fmt.Sprintf("ERR：Staff Mid(%d)被封禁。", staff.Mid))
			err = ecode.Errorf(ecode.VideoupStaffUpSilence, ecode.VideoupStaffUpSilence.Message(), cards[staff.Mid].Name)
			return
		}
	}

	return
}

// getStaffChanges 获取联合投稿人的变更
func (s *Service) getStaffChanges(oS, nS []*archive.Staff) (changes map[string]map[int64]*archive.Staff, _logs []string) {
	var (
		allS = make([]*archive.Staff, 0)
		oMap = make(map[int64]*archive.Staff)
		nMap = make(map[int64]*archive.Staff)
	)
	str := ""
	for _, v := range oS {
		str += fmt.Sprintf("%d %s;", v.Mid, v.Title)
		oMap[v.Mid] = v
	}
	_logs = append(_logs, " 原Staffs："+str)
	str = ""
	for _, v := range nS {
		str += fmt.Sprintf("%d %s;", v.Mid, v.Title)
		nMap[v.Mid] = v
	}
	_logs = append(_logs, " 提交Staffs："+str)
	changes = make(map[string]map[int64]*archive.Staff)
	changes["add"] = make(map[int64]*archive.Staff)
	changes["edit"] = make(map[int64]*archive.Staff)
	changes["del"] = make(map[int64]*archive.Staff)
	allS = append(allS, oS...)
	allS = append(allS, nS...)
	for _, v := range allS {
		if _, ok := oMap[v.Mid]; !ok {
			changes["add"][v.Mid] = v
		} else if _, ok := nMap[v.Mid]; !ok {
			changes["del"][v.Mid] = v
		} else if oMap[v.Mid].Title != nMap[v.Mid].Title {
			changes["edit"][v.Mid] = v
		}
	}
	return
}

// FRelations 获取用户与mid的关系（Relations的反向）
func (s *Service) FRelations(c context.Context, mid int64, fids []int64, ip string) (res map[int64]int, err error) {
	var (
		g, ctx = errgroup.WithContext(c)
		sm     sync.RWMutex
	)

	res = make(map[int64]int)
	for _, v := range fids {
		g.Go(func() error {
			var r map[int64]int
			if r, err = s.acc.Relations(ctx, v, []int64{mid}, ip); err != nil {
				return err
			}
			sm.Lock()
			res[v] = r[mid]
			sm.Unlock()
			return nil
		})
	}
	if err = g.Wait(); err != nil {
		log.Error("s.FRelations(%d,%v) error(%v)", mid, fids, err)
	}
	return
}

// checkStaffMoveType 联合投稿不允许移区和转载类型（有申请的Staff时，都不允许移区）
func (s *Service) checkStaffMoveType(c context.Context, ap *archive.ArcParam, a *archive.Archive, ip string) (err error) {
	var (
		_logs []string
	)
	defer func() {
		if err != nil {
			log.Error("s.checkStaffMoveType ap: %+v, archive:%+v logs(%v)", ap, a, _logs)
		} else {
			log.Info("s.checkStaffMoveType ap: %+v, archive:%+v logs(%v)", ap, a, _logs)
		}
	}()
	//如果没发生移区和修改转载类型，则直接通过
	if ap.TypeID == a.TypeID && a.Copyright == ap.Copyright {
		_logs = append(_logs, "INFO:没有修改分区和转载类型")
		return
	}
	var (
		cStaffs []*archive.Staff //当前稿件的联合投稿人
	)
	if cStaffs, err = s.arc.ApplyStaffs(c, a.Aid, ip); err != nil {
		_logs = append(_logs, fmt.Sprintf("ERR:获取Staff失败。error:%v", err))
		log.Error("checkStaffMoveType() 获取线上Staffs失败，err:%v", err)
		return
	}
	if len(cStaffs) != 0 {
		_logs = append(_logs, fmt.Sprintf("ERR: 不允许操作。当前Staffs(%v)", cStaffs))
		err = ecode.VideoupStaffChangeTypeCopyright
		return
	}
	_logs = append(_logs, "INFO:通过")
	return
}