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

import (

"github.com/go-yaml/yaml"

)

type StringOrSlice []string

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

}