package models
  
  import (

  	"fmt"

  	"pro2d/components/db"

  	"pro2d/protos/pb"
  )
  
  type RoleModel struct {

  	*db.MgoColl
  	Role *pb.Role
  	Heros HeroMap
  	Teams *pb.Team
  	Equip *pb.Equipment
  	Prop *pb.Prop

  }
  

  func RoleExistByUid(uid int64) (bool, *RoleModel){

  	m := NewRole(uid)

  	if err := m.Load(); err != nil {
  		return false, m
  	}
  	return true, m
  }
  

  func NewRole(uid int64) *RoleModel {
  	r := &pb.Role{Uid: uid}

  	m := &RoleModel{

  		MgoColl: db.NewMongoColl("role", db.GetBsonM("uid", r.Uid), r),

  		Role: r,

  		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
  }