package models

import (
	"pro2d/pb"
)

type HeroModel struct {
	*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: NewSchema(id, h),
		Hero:   h,
	}
	return m
}