map.go

package wen

import "strings"

type Map map[string]interface{}

func (m Map) M(s string) Map {
	if !m.Have(s) {
		return nil
	}
	if t, ok := m[s].(map[string]interface{}); ok {
		return t
	} else {
		return m[s].(Map)
	}
}

func (m Map) S(s string) string {
	if !m.Have(s) {
		return ""
	}
	return m[s].(string)
}

func (m Map) Int(s string) int {
	if !m.Have(s) {
		return 0
	}
	return m[s].(int)
}
func (m Map) Bool(s string) bool {
	if !m.Have(s) {
		return false
	}
	return m[s].(bool)
}
func (m Map) I(s string) interface{} {
	if !m.Have(s) {
		return nil
	}
	return m[s]
}
func (m Map) Have(s string) bool {
	_, ok := m[s]
	return ok
}

func (m Map) Get(s string) interface{} {
	splits := strings.Split(s, ".")
	if len(splits) == 1 {
		return m.I(s)
	}
	var r = m
	for index, split := range splits {
		if index == len(splits)-1 {
			return r.I(split)
		} else {
			r = r.M(split)
		}
	}
	return nil
}
func (m Map) Set(s string, v interface{}) {
	splits := strings.Split(s, ".")
	if len(splits) == 1 {
		m[s] = v
		return
	}
	var r = m
	for index, split := range splits {
		if index == len(splits)-1 {
			r.Set(split, v)
		} else {
			if !r.Have(split) {
				r.Set(split, Map{})
			}
			r = r.M(split)
		}
	}
}