types/types.go

@@ -1,7 +1,6 @@
 package types
 
 import (
-	"encoding/json"
 	"fmt"
 	"net/http"
 )
@@ -28,6 +27,6 @@ func (le LemmyError) Error() string {
 }
 
 type LemmyWebSocketMsg struct {
-	Op   UserOperation   `json:"op"`
-	Data json.RawMessage `json:"data"`
+	Op   UserOperation `json:"op"`
+	Data any           `json:"data"`
 }

types/websocket.go

@@ -1,37 +0,0 @@
-package types
-
-type UserJoin struct {
-	Auth string `json:"auth"`
-}
-
-type UserJoinResponse struct {
-	Joined bool `json:"joined"`
-	LemmyResponse
-}
-
-type CommunityJoin struct {
-	CommunityID int `json:"community_id"`
-}
-
-type CommunityJoinResponse struct {
-	Joined bool `json:"joined"`
-	LemmyResponse
-}
-
-type ModJoin struct {
-	CommunityID int `json:"community_id"`
-}
-
-type ModJoinResponse struct {
-	Joined bool `json:"joined"`
-	LemmyResponse
-}
-
-type PostJoin struct {
-	PostID int `json:"post_id"`
-}
-
-type PostJoinResponse struct {
-	Joined bool `json:"joined"`
-	LemmyResponse
-}

websocket.go

@@ -1,27 +1,18 @@
 package lemmy
 
 import (
-	"context"
-	"encoding/json"
-	"net/http"
 	"net/url"
-	"reflect"
 	"time"
 
+	"github.com/mitchellh/mapstructure"
 	"github.com/recws-org/recws"
 	"go.arsenm.dev/go-lemmy/types"
 )
 
-type authData struct {
-	Auth string `json:"auth"`
-}
-
 type WSClient struct {
-	conn    *recws.RecConn
-	baseURL *url.URL
-	respCh  chan types.LemmyWebSocketMsg
-	errCh   chan error
-	token   string
+	conn   *recws.RecConn
+	respCh chan types.LemmyWebSocketMsg
+	errCh  chan error
 }
 
 func NewWebSocket(baseURL string) (*WSClient, error) {
@@ -35,13 +26,12 @@ func NewWebSocket(baseURL string) (*WSClient, error) {
 	}
 	u = u.JoinPath("/api/v3")
 
-	ws.Dial(u.JoinPath("ws").String(), nil)
+	ws.Dial(u.String(), nil)
 
 	out := &WSClient{
-		conn:    ws,
-		baseURL: u,
-		respCh:  make(chan types.LemmyWebSocketMsg, 10),
-		errCh:   make(chan error, 10),
+		conn:   ws,
+		respCh: make(chan types.LemmyWebSocketMsg, 10),
+		errCh:  make(chan error, 10),
 	}
 
 	go func() {
@@ -59,40 +49,10 @@ func NewWebSocket(baseURL string) (*WSClient, error) {
 	return out, nil
 }
 
-func (c *WSClient) Login(ctx context.Context, l types.Login) error {
-	u := &url.URL{}
-	*u = *c.baseURL
-
-	if u.Scheme == "ws" {
-		u.Scheme = "http"
-	} else if u.Scheme == "wss" {
-		u.Scheme = "https"
-	}
-
-	hc := &Client{baseURL: u, client: http.DefaultClient}
-	err := hc.Login(ctx, l)
-	if err != nil {
-		return err
-	}
-	c.token = hc.token
-	return nil
-}
-
 func (c *WSClient) Request(op types.UserOperation, data any) error {
-	if data == nil {
-		data = authData{}
-	}
-
-	data = c.setAuth(data)
-
-	d, err := json.Marshal(data)
-	if err != nil {
-		return err
-	}
-
 	return c.conn.WriteJSON(types.LemmyWebSocketMsg{
 		Op:   op,
-		Data: d,
+		Data: data,
 	})
 }
 
@@ -104,29 +64,13 @@ func (c *WSClient) Errors() <-chan error {
 	return c.errCh
 }
 
-func (c *WSClient) setAuth(data any) any {
-	val := reflect.New(reflect.TypeOf(data))
-	val.Elem().Set(reflect.ValueOf(data))
-
-	authField := val.Elem().FieldByName("Auth")
-	if !authField.IsValid() {
-		return data
+func DecodeResponse(data, out any) error {
+	dec, err := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
+		TagName: "json",
+		Result:  out,
+	})
+	if err != nil {
+		return err
 	}
-
-	switch authField.Type().String() {
-	case "string":
-		authField.SetString(c.token)
-	case "types.Optional[string]":
-		setMtd := authField.MethodByName("Set")
-		out := setMtd.Call([]reflect.Value{reflect.ValueOf(c.token)})
-		authField.Set(out[0])
-	default:
-		return data
-	}
-
-	return val.Elem().Interface()
-}
-
-func DecodeResponse(data json.RawMessage, out any) error {
-	return json.Unmarshal(data, out)
+	return dec.Decode(data)
 }