67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package conf
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/bluenviron/gortsplib/v3/pkg/headers"
|
|
)
|
|
|
|
// AuthMethods is the authMethods parameter.
|
|
type AuthMethods []headers.AuthMethod
|
|
|
|
// MarshalJSON implements json.Marshaler.
|
|
func (d AuthMethods) MarshalJSON() ([]byte, error) {
|
|
out := make([]string, len(d))
|
|
|
|
for i, v := range d {
|
|
switch v {
|
|
case headers.AuthBasic:
|
|
out[i] = "basic"
|
|
|
|
case headers.AuthDigest:
|
|
out[i] = "digest"
|
|
|
|
default:
|
|
return nil, fmt.Errorf("invalid authentication method: %v", v)
|
|
}
|
|
}
|
|
|
|
sort.Strings(out)
|
|
|
|
return json.Marshal(out)
|
|
}
|
|
|
|
// UnmarshalJSON implements json.Unmarshaler.
|
|
func (d *AuthMethods) UnmarshalJSON(b []byte) error {
|
|
var in []string
|
|
if err := json.Unmarshal(b, &in); err != nil {
|
|
return err
|
|
}
|
|
|
|
*d = nil
|
|
|
|
for _, v := range in {
|
|
switch v {
|
|
case "basic":
|
|
*d = append(*d, headers.AuthBasic)
|
|
|
|
case "digest":
|
|
*d = append(*d, headers.AuthDigest)
|
|
|
|
default:
|
|
return fmt.Errorf("invalid authentication method: '%s'", v)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnmarshalEnv implements envUnmarshaler.
|
|
func (d *AuthMethods) UnmarshalEnv(s string) error {
|
|
byts, _ := json.Marshal(strings.Split(s, ","))
|
|
return d.UnmarshalJSON(byts)
|
|
}
|