package material

type RoomType struct {
	ID         int    `json:"id"`
	Name       string `json:"name"`
	ShortName  string `json:"short_name"`
	MaxRoomNum int    `json:"max_room_num"`
}

var Params = struct {
	RoomType []RoomType `json:"room_type"`
}{
	RoomType: []RoomType{
		{0, "开工", "", 1},
		{1, "拆改", "", 10},
		{2, "水电", "", 10},
		{3, "泥工", "", 1},
		{4, "防水", "", 10},
		{5, "木工", "", 10},
		{6, "刮瓷", "", 10},
		{7, "工程款", "", 10},
		{8, "安全", "", 10},
		{9, "主材验收", "", 10},
		{10, "交房", "", 10},
	},
}

func GetRoomTypeMap(shortName bool) map[int]string {
	roomTypeMap := make(map[int]string)
	for _, v := range Params.RoomType {
		if shortName {
			roomTypeMap[v.ID] = v.ShortName
		} else {
			roomTypeMap[v.ID] = v.Name
		}
	}
	return roomTypeMap
}