grim/convey
Clone
Summary
Browse
Changes
Graph
Missed a bunch of updates on unit tests for the options update
2018-01-21, Gary Kramlich
2909e17deafe
Missed a bunch of updates on unit tests for the options update
// Convey
// Copyright 2016-2018 Gary Kramlich <grim@reaperworld.com>
//
// 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
(
"github.com/go-yaml/yaml"
)
// 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
}