package models
  
  import (

  	"fmt"

  	"pro2d/protos/pb"

  	"pro2d/src/components/db"
  	"pro2d/src/components/logger"

  	"strconv"

  )
  
  type RoleModel struct {

  	*db.MgoColl

  	Role  *pb.Role

  	Heros HeroMap

  	Teams *TeamModel

  	Equip *pb.Equipment

  	Prop  *pb.Prop

  }
  

  func RoleExistByUid(uid string) *RoleModel {

  	data := &pb.Role{Uid: uid}
  
  	if err := db.FindOne(db.GetBsonM("uid", uid), data); err != nil {

  		logger.Error("Role exist err: %v", err)

  		return nil

  	}

  	return &RoleModel{

  		MgoColl: db.NewMongoColl(strconv.Itoa(int(data.Id)), data),

  		Role:    data,

  	}

  }
  

  func NewRole(id int64) *RoleModel {
  	data := &pb.Role{Id: id}

  	m := &RoleModel{

  		MgoColl: db.NewMongoColl(strconv.Itoa(int(id)), data),

  		Role:    data,
  		Heros:   make(HeroMap),

  	}

  	return m
  }
  
  func (m *RoleModel) LoadAll()  {
  }

  
  func (m *RoleModel) LoadHero()  {
  }
  
  func (m *RoleModel) AddHero(hero *pb.Hero) {

  	h := NewHero(hero.Id)
  	h.Hero = hero

  	h.Create()
  	m.Heros[fmt.Sprintf("%d%d", m.Role.Id, h.Hero.Id)] = h
  }