Add Sender module

pkg/api/models.go

@@ -3,29 +3,49 @@ package api
 import (
 	"fmt"
 	"io"
+	"net/url"
 	"strconv"
 
+	"github.com/99designs/gqlgen/graphql"
 	"github.com/oklog/ulid"
 )
 
-type ULID ulid.ULID
+func MarshalULID(u ulid.ULID) graphql.Marshaler {
+	return graphql.WriterFunc(func(w io.Writer) {
+		fmt.Fprint(w, strconv.Quote(u.String()))
+	})
+}
 
-func (u *ULID) UnmarshalGQL(v interface{}) (err error) {
-	str, ok := v.(string)
+func UnmarshalULID(v interface{}) (ulid.ULID, error) {
+	rawULID, ok := v.(string)
 	if !ok {
-		return fmt.Errorf("ulid must be a string")
+		return ulid.ULID{}, fmt.Errorf("ulid must be a string")
 	}
 
-	id, err := ulid.Parse(str)
+	u, err := ulid.Parse(rawULID)
 	if err != nil {
-		return fmt.Errorf("failed to parse ULID: %w", err)
+		return ulid.ULID{}, fmt.Errorf("failed to parse ULID: %w", err)
 	}
 
-	*u = ULID(id)
-
-	return nil
+	return u, nil
 }
 
-func (u ULID) MarshalGQL(w io.Writer) {
-	fmt.Fprint(w, strconv.Quote(ulid.ULID(u).String()))
+func MarshalURL(u *url.URL) graphql.Marshaler {
+	return graphql.WriterFunc(func(w io.Writer) {
+		fmt.Fprint(w, strconv.Quote(u.String()))
+	})
+}
+
+func UnmarshalURL(v interface{}) (*url.URL, error) {
+	rawURL, ok := v.(string)
+	if !ok {
+		return nil, fmt.Errorf("url must be a string")
+	}
+
+	u, err := url.Parse(rawURL)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse URL: %w", err)
+	}
+
+	return u, nil
 }