package models

import (
	"pro2d/protos/pb"
	"pro2d/src/components/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
}