HGKeeper

// Convey

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

// Package yaml contains some custom yaml unmarshallers.

package yaml

import (

"gopkg.in/yaml.v2"

)

// StringOrSlice is a []string type alias that has a custom UnmarshalYAML

// implementation.

type StringOrSlice []string

// UnmarshalYAML will return a string slice for a single string or a slice of

// strings.

func (s *StringOrSlice) UnmarshalYAML(unmarshal func(interface{}) error) error {

// check to see if we were just provided a string

str := ""

err := unmarshal(&str)

if err == nil {

*s = StringOrSlice{str}

return nil

}

// if we reach this point, the user supplied a list

if _, ok := err.(*yaml.TypeError); ok {

type rawStringOrSlice StringOrSlice

raw := rawStringOrSlice{}

if err = unmarshal(&raw); err != nil {

return err

}

*s = StringOrSlice(raw)

return nil

}

return err

}