auth/internal/domain/email.go

82 lines
1.6 KiB
Go

package domain
import (
"fmt"
"strconv"
"strings"
"testing"
)
// Email represent email identifier.
type Email struct {
user string
host string
subAddress string
}
const DefaultEmailPartsLength int = 2
var ErrEmailInvalid error = NewError(ErrorCodeInvalidRequest, "cannot parse email", "")
// ParseEmail parse strings to email identifier.
func ParseEmail(src string) (*Email, error) {
parts := strings.Split(strings.TrimPrefix(src, "mailto:"), "@")
if len(parts) != DefaultEmailPartsLength {
return nil, ErrEmailInvalid
}
result := &Email{
user: parts[0],
host: parts[1],
subAddress: "",
}
if userParts := strings.SplitN(parts[0], `+`, DefaultEmailPartsLength); len(userParts) > 1 {
result.user = userParts[0]
result.subAddress = userParts[1]
}
return result, nil
}
// UnmarshalJSON implements custom unmarshler for JSON.
func (e *Email) UnmarshalJSON(v []byte) error {
src, err := strconv.Unquote(string(v))
if err != nil {
return fmt.Errorf("Email: UnmarshalJSON: %w", err)
}
email, err := ParseEmail(src)
if err != nil {
return fmt.Errorf("Email: UnmarshalJSON: %w", err)
}
*e = *email
return nil
}
func (e Email) MarshalJSON() ([]byte, error) {
return []byte(strconv.Quote(e.String())), nil
}
// TestEmail returns valid random generated email identifier.
func TestEmail(tb testing.TB) *Email {
tb.Helper()
return &Email{
user: "user",
subAddress: "",
host: "example.com",
}
}
// String returns string representation of email identifier.
func (e Email) String() string {
if e.subAddress == "" {
return e.user + "@" + e.host
}
return e.user + "+" + e.subAddress + "@" + e.host
}