Check if payment required

client.go

@@ -10,9 +10,10 @@ import (
 )
 
 type Client struct {
-	BaseUrl string
+	BaseUrl  string
-	ApiUrl  string
+	ApiUrl   string
-	ApiKey  string
+	ApiKey   string
+	MediaUrl string
 }
 
 func NewClient(options ...func(*Client)) *Client {
@@ -22,12 +23,22 @@ func NewClient(options ...func(*Client)) *Client {
 	}
 
 	// set defaults
+	var ok bool
 	if c.BaseUrl == "" {
-		c.BaseUrl = "https://stacker.news"
+		c.BaseUrl, ok = os.LookupEnv("SN_BASE_URL")
+		if !ok {
+			c.BaseUrl = "https://stacker.news"
+		}
 	}
 	if c.ApiKey == "" {
 		c.ApiKey = os.Getenv("SN_API_KEY")
 	}
+	if c.MediaUrl == "" {
+		c.MediaUrl, ok = os.LookupEnv("SN_MEDIA_URL")
+		if !ok {
+			c.MediaUrl = "https://m.stacker.news"
+		}
+	}
 	c.ApiUrl = fmt.Sprintf("%s/api/graphql", c.BaseUrl)
 
 	return c
@@ -45,6 +56,12 @@ func WithBaseUrl(baseUrl string) func(*Client) {
 	}
 }
 
+func WithMediaUrl(mediaUrl string) func(*Client) {
+	return func(c *Client) {
+		c.MediaUrl = mediaUrl
+	}
+}
+
 type GqlBody struct {
 	Query     string                 `json:"query"`
 	Variables map[string]interface{} `json:"variables,omitempty"`

items.go

@@ -226,7 +226,17 @@ func (c *Client) PostDiscussion(title string, text string, sub string) (int, err
 		Query: `
 		mutation upsertDiscussion($title: String!, $text: String, $sub: String) {
 			upsertDiscussion(title: $title, text: $text, sub: $sub) {
-				result { id }
+				result {
+					id
+				}
+				invoice {
+					id
+					hash
+					bolt11
+					satsRequested
+					expiresAt
+				}
+				paymentMethod
 			}
 		}`,
 		Variables: map[string]interface{}{
@@ -254,6 +264,11 @@ func (c *Client) PostDiscussion(title string, text string, sub string) (int, err
 		return -1, err
 	}
 
+	inv := respBody.Data.UpsertDiscussion.Invoice
+	if inv.Id != 0 {
+		return -1, fmt.Errorf("mutation requires %d sats as payment", inv.SatsRequested)
+	}
+
 	return respBody.Data.UpsertDiscussion.Result.Id, nil
 }
 
@@ -262,7 +277,17 @@ func (c *Client) PostLink(url string, title string, text string, sub string) (in
 		Query: `
 		mutation upsertLink($url: String!, $title: String!, $text: String, $sub: String!) {
 			upsertLink(url: $url, title: $title, text: $text, sub: $sub) {
-				result { id }
+				result {
+					id
+				}
+				invoice {
+					id
+					hash
+					bolt11
+					satsRequested
+					expiresAt
+				}
+				paymentMethod
 			}
 		}`,
 		Variables: map[string]interface{}{
@@ -291,6 +316,11 @@ func (c *Client) PostLink(url string, title string, text string, sub string) (in
 		return -1, err
 	}
 
+	inv := respBody.Data.UpsertLink.Invoice
+	if inv.Id != 0 {
+		return -1, fmt.Errorf("mutation requires %d sats as payment", inv.SatsRequested)
+	}
+
 	return respBody.Data.UpsertLink.Result.Id, nil
 }
 
@@ -299,7 +329,17 @@ func (c *Client) CreateComment(parentId int, text string) (int, error) {
 		Query: `
 		mutation upsertComment($parentId: ID!, $text: String!) {
 			upsertComment(parentId: $parentId, text: $text) {
-				result { id }
+				result {
+					id
+				}
+				invoice {
+					id
+					hash
+					bolt11
+					satsRequested
+					expiresAt
+				}
+				paymentMethod
 			}
 		}`,
 		Variables: map[string]interface{}{
@@ -326,6 +366,11 @@ func (c *Client) CreateComment(parentId int, text string) (int, error) {
 		return -1, err
 	}
 
+	inv := respBody.Data.UpsertComment.Invoice
+	if inv.Id != 0 {
+		return -1, fmt.Errorf("mutation requires %d sats as payment", inv.SatsRequested)
+	}
+
 	return respBody.Data.UpsertComment.Result.Id, nil
 }
 

notifications.go

@@ -0,0 +1,125 @@
+package sn
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+)
+
+type Notification struct {
+	Id   int    `json:"id,string"`
+	Type string `json:"__typename"`
+	Item Item   `json:"item"`
+}
+
+type NotificationsCursor struct {
+	LastChecked   time.Time      `json:"lastChecked"`
+	Cursor        string         `json:"cursor"`
+	Notifications []Notification `json:"notifications"`
+}
+
+type NotificationsResponse struct {
+	Errors []GqlError `json:"errors"`
+	Data   struct {
+		Notifications NotificationsCursor `json:"notifications"`
+	} `json:"data"`
+}
+
+func (c *Client) Notifications() (*NotificationsCursor, error) {
+	body := GqlBody{
+		Query: `
+		fragment ItemFields on Item {
+			id
+			user {
+				id
+				name
+			}
+			parentId
+			createdAt
+			deletedAt
+			title
+			text
+		}
+		query notifications {
+			notifications {
+				lastChecked
+				cursor
+				notifications {
+					__typename
+					... on Reply {
+						id
+						item {
+							...ItemFields
+						}
+					}
+					... on Mention {
+						id
+						item {
+							...ItemFields
+						}
+					}
+				}
+			}
+		}
+		`,
+		Variables: map[string]interface{}{},
+	}
+
+	resp, err := c.callApi(body)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var respBody NotificationsResponse
+	err = json.NewDecoder(resp.Body).Decode(&respBody)
+	if err != nil {
+		err = fmt.Errorf("error decoding notifications: %w", err)
+		return nil, err
+	}
+
+	err = c.checkForErrors(respBody.Errors)
+	if err != nil {
+		return nil, err
+	}
+	return &respBody.Data.Notifications, nil
+}
+
+func (c *Client) Mentions() ([]Notification, error) {
+	return c.filterNotifications(
+		func(n Notification) bool {
+			return n.Type == "Mention"
+		},
+	)
+}
+
+func (c *Client) Replies() ([]Notification, error) {
+	return c.filterNotifications(
+		func(n Notification) bool {
+			return n.Type == "Reply"
+		},
+	)
+}
+
+func (c *Client) filterNotifications(f func(Notification) bool) ([]Notification, error) {
+	var (
+		n   *NotificationsCursor
+		err error
+	)
+
+	if n, err = c.Notifications(); err != nil {
+		return nil, err
+	}
+
+	return filter(n.Notifications, f), nil
+}
+
+func filter[T any](s []T, f func(T) bool) []T {
+	var r []T
+	for _, v := range s {
+		if f(v) {
+			r = append(r, v)
+		}
+	}
+	return r
+}

upload.go

@@ -0,0 +1,131 @@
+package sn
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"image"
+	"image/png"
+	"io"
+	"mime/multipart"
+	"net/http"
+)
+
+type GetSignedPOST struct {
+	Url    string            `json:"url"`
+	Fields map[string]string `json:"fields"`
+}
+
+type GetSignedPOSTResponse struct {
+	Errors []GqlError `json:"errors"`
+	Data   struct {
+		GetSignedPOST GetSignedPOST `json:"getSignedPOST"`
+	} `json:"data"`
+}
+
+func (c *Client) UploadImage(img *image.RGBA) (string, error) {
+	var (
+		b      = img.Bounds()
+		width  = b.Dx()
+		height = b.Dy()
+		type_  = "image/png"
+		size   int
+	)
+
+	var imgBuf bytes.Buffer
+	if err := png.Encode(&imgBuf, img); err != nil {
+		return "", err
+	}
+	size = imgBuf.Len()
+
+	// get signed URL for S3 upload
+	body := GqlBody{
+		Query: `
+		mutation getSignedPOST($type: String!, $size: Int!, $width: Int!, $height: Int!, $avatar: Boolean) {
+			getSignedPOST(type: $type, size: $size, width: $width, height: $height, avatar: $avatar) {
+				url
+				fields
+			}
+		}`,
+		Variables: map[string]interface{}{
+			"type":   type_,
+			"size":   size,
+			"width":  width,
+			"height": height,
+			"avatar": false,
+		},
+	}
+
+	resp, err := c.callApi(body)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	var respBody GetSignedPOSTResponse
+	err = json.NewDecoder(resp.Body).Decode(&respBody)
+	if err != nil {
+		err = fmt.Errorf("error decoding getSignedPOST: %w", err)
+		return "", err
+	}
+
+	err = c.checkForErrors(respBody.Errors)
+	if err != nil {
+		return "", err
+	}
+
+	s3Url := respBody.Data.GetSignedPOST.Url
+	fields := respBody.Data.GetSignedPOST.Fields
+
+	// create multipart form
+	var (
+		buf bytes.Buffer
+		w   = multipart.NewWriter(&buf)
+		fw  io.Writer
+	)
+
+	for k, v := range fields {
+		if fw, err = w.CreateFormField(k); err != nil {
+			return "", err
+		}
+		fw.Write([]byte(v))
+	}
+
+	for k, v := range map[string]string{
+		"Content-Type":  type_,
+		"Cache-Control": "max-age=31536000",
+		"acl":           "public-read",
+	} {
+		if fw, err = w.CreateFormField(k); err != nil {
+			return "", err
+		}
+		fw.Write([]byte(v))
+	}
+
+	if fw, err = w.CreateFormFile("file", "image.png"); err != nil {
+		return "", err
+	}
+	fw.Write(imgBuf.Bytes())
+
+	if err = w.Close(); err != nil {
+		return "", err
+	}
+
+	// upload to S3
+	var req *http.Request
+	if req, err = http.NewRequest("POST", s3Url, &buf); err != nil {
+		return "", err
+	}
+	req.Header.Set("Content-Type", w.FormDataContentType())
+
+	client := http.DefaultClient
+	if resp, err = client.Do(req); err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	imgId := respBody.Data.GetSignedPOST.Fields["key"]
+	imgUrl := fmt.Sprintf("%s/%s", c.MediaUrl, imgId)
+
+	return imgUrl, nil
+}

user.go

@@ -1,6 +1,57 @@
 package sn
 
+import (
+	"encoding/json"
+	"fmt"
+)
+
 type User struct {
-	Id   int    `json:"id,string"`
+	Id       int          `json:"id,string"`
-	Name string `json:"name"`
+	Name     string       `json:"name"`
+	Privates UserPrivates `json:"privates"`
+}
+
+type UserPrivates struct {
+	Sats int `json:"sats"`
+}
+
+type MeResponse struct {
+	Errors []GqlError `json:"errors"`
+	Data   struct {
+		Me User `json:"me"`
+	} `json:"data"`
+}
+
+func (c *Client) Me() (*User, error) {
+	body := GqlBody{
+		Query: `
+		query me {
+			me {
+				id
+				name
+				privates {
+					sats
+				}
+			}
+		}`,
+	}
+
+	resp, err := c.callApi(body)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var respBody MeResponse
+	err = json.NewDecoder(resp.Body).Decode(&respBody)
+	if err != nil {
+		err = fmt.Errorf("error decoding me: %w", err)
+		return nil, err
+	}
+
+	err = c.checkForErrors(respBody.Errors)
+	if err != nil {
+		return nil, err
+	}
+	return &respBody.Data.Me, nil
 }