95 lines
1.6 KiB
Go
95 lines
1.6 KiB
Go
package domain
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strconv"
|
|
"testing"
|
|
|
|
http "github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// URL describe any valid HTTP URL.
|
|
type URL struct {
|
|
*http.URI
|
|
}
|
|
|
|
// ParseURL parse string as URL.
|
|
func ParseURL(src string) (*URL, error) {
|
|
u := http.AcquireURI()
|
|
if err := u.Parse(nil, []byte(src)); err != nil {
|
|
return nil, fmt.Errorf("cannot parse URL: %w", err)
|
|
}
|
|
|
|
return &URL{URI: u}, nil
|
|
}
|
|
|
|
// MustParseURL parse string as URL or panic.
|
|
func MustParseURL(src string) *URL {
|
|
uri, err := ParseURL(src)
|
|
if err != nil {
|
|
panic("MustParseURL: " + err.Error())
|
|
}
|
|
|
|
return uri
|
|
}
|
|
|
|
// TestURL returns URL of provided input for tests.
|
|
func TestURL(tb testing.TB, src string) *URL {
|
|
tb.Helper()
|
|
|
|
u := http.AcquireURI()
|
|
u.Update(src)
|
|
|
|
return &URL{
|
|
URI: u,
|
|
}
|
|
}
|
|
|
|
// UnmarshalForm implements custom unmarshler for form values.
|
|
func (u *URL) UnmarshalForm(v []byte) error {
|
|
url, err := ParseURL(string(v))
|
|
if err != nil {
|
|
return fmt.Errorf("URL: UnmarshalForm: %w", err)
|
|
}
|
|
|
|
*u = *url
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnmarshalJSON implements custom unmarshler for JSON.
|
|
func (u *URL) UnmarshalJSON(v []byte) error {
|
|
src, err := strconv.Unquote(string(v))
|
|
if err != nil {
|
|
return fmt.Errorf("URL: UnmarshalJSON: %w", err)
|
|
}
|
|
|
|
url, err := ParseURL(src)
|
|
if err != nil {
|
|
return fmt.Errorf("URL: UnmarshalJSON: %w", err)
|
|
}
|
|
|
|
*u = *url
|
|
|
|
return nil
|
|
}
|
|
|
|
func (u URL) MarshalJSON() ([]byte, error) {
|
|
return []byte(strconv.Quote(u.String())), nil
|
|
}
|
|
|
|
// URL returns url.URL representation of URL.
|
|
func (u URL) URL() *url.URL {
|
|
if u.URI == nil {
|
|
return nil
|
|
}
|
|
|
|
result, err := url.ParseRequestURI(u.String())
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
return result
|
|
}
|