@@ -15,7 +15,7 @@
|
||||
package coredata
|
||||
|
||||
import (
|
||||
"database/sql/driver"
|
||||
"encoding"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
@@ -27,6 +27,12 @@ const (
|
||||
CookieSourceHTTP CookieSource = "HTTP"
|
||||
)
|
||||
|
||||
var (
|
||||
_ fmt.Stringer = CookieSource("")
|
||||
_ encoding.TextMarshaler = CookieSource("")
|
||||
_ encoding.TextUnmarshaler = (*CookieSource)(nil)
|
||||
)
|
||||
|
||||
func CookieSources() []CookieSource {
|
||||
return []CookieSource{
|
||||
CookieSourceScript,
|
||||
@@ -35,43 +41,33 @@ func CookieSources() []CookieSource {
|
||||
}
|
||||
}
|
||||
|
||||
func (s CookieSource) String() string {
|
||||
return string(s)
|
||||
func (v CookieSource) IsValid() bool {
|
||||
switch v {
|
||||
case
|
||||
CookieSourceScript,
|
||||
CookieSourcePreExisting,
|
||||
CookieSourceHTTP:
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func (s *CookieSource) Scan(value any) error {
|
||||
var v string
|
||||
func (v CookieSource) String() string {
|
||||
return string(v)
|
||||
}
|
||||
|
||||
switch val := value.(type) {
|
||||
case string:
|
||||
v = val
|
||||
case []byte:
|
||||
v = string(val)
|
||||
default:
|
||||
return fmt.Errorf("unsupported type for CookieSource: %T", value)
|
||||
func (v CookieSource) MarshalText() ([]byte, error) {
|
||||
return []byte(v.String()), nil
|
||||
}
|
||||
|
||||
func (v *CookieSource) UnmarshalText(text []byte) error {
|
||||
val := CookieSource(text)
|
||||
if !val.IsValid() {
|
||||
return fmt.Errorf("invalid CookieSource value: %q", string(text))
|
||||
}
|
||||
|
||||
switch CookieSource(v) {
|
||||
case CookieSourceScript:
|
||||
*s = CookieSourceScript
|
||||
case CookieSourcePreExisting:
|
||||
*s = CookieSourcePreExisting
|
||||
case CookieSourceHTTP:
|
||||
*s = CookieSourceHTTP
|
||||
default:
|
||||
return fmt.Errorf("invalid CookieSource value: %q", v)
|
||||
}
|
||||
*v = val
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s CookieSource) Value() (driver.Value, error) {
|
||||
switch s {
|
||||
case CookieSourceScript,
|
||||
CookieSourcePreExisting,
|
||||
CookieSourceHTTP:
|
||||
return string(s), nil
|
||||
default:
|
||||
return nil, fmt.Errorf("invalid CookieSource: %s", s)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user