package models
  
  import (

  	"pro2d/pb"

  	"pro2d/utils/db"

  )
  
  type HeroModel struct {

  	*db.Schema

  	Hero *pb.Hero
  }
  type HeroMap map[string]*HeroModel
  
  func GetHeros(hm HeroMap) map[string]*pb.Hero  {
  	h := make(map[string]*pb.Hero)
  	for k, v := range hm {
  		h[k] = v.Hero
  	}
  	return h
  }
  

  func NewHero(id string) *HeroModel {

  	h := &pb.Hero{
  		Id:         id,
  	}

  	m := &HeroModel{

  		Schema: db.NewSchema(id, h),

  		Hero:   h,

  	}

  	return m
  }