144 lines
3.5 KiB
Go
144 lines
3.5 KiB
Go
package revoltgo
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/sacOO7/gowebsocket"
|
|
)
|
|
|
|
const (
|
|
WS_URL = "wss://ws.revolt.chat"
|
|
API_URL = "https://api.revolt.chat"
|
|
)
|
|
|
|
// Client struct
|
|
type Client struct {
|
|
Token string
|
|
Socket gowebsocket.Socket
|
|
HTTP *http.Client
|
|
|
|
// Event Functions
|
|
OnReadyFunctions []func()
|
|
OnMessageFunctions []func(message *Message)
|
|
OnMessageUpdateFunctions []func(channel_id, message_id string, payload map[string]interface{})
|
|
OnMessageDeleteFunctions []func(channel_id, message_id string)
|
|
OnChannelCreateFunctions []func(channel *Channel)
|
|
OnChannelUpdateFunctions []func(channel_id, clear string, payload map[string]interface{})
|
|
OnChannelDeleteFunctions []func(channel_id string)
|
|
}
|
|
|
|
// On ready event will run when websocket connection is started and bot is ready to work.
|
|
func (c *Client) OnReady(fn func()) {
|
|
c.OnReadyFunctions = append(c.OnReadyFunctions, fn)
|
|
}
|
|
|
|
// On message event will run when someone sends a message.
|
|
func (c *Client) OnMessage(fn func(message *Message)) {
|
|
c.OnMessageFunctions = append(c.OnMessageFunctions, fn)
|
|
}
|
|
|
|
// On message update event will run when someone updates a message.
|
|
func (c *Client) OnMessageUpdate(fn func(channel_id, message_id string, payload map[string]interface{})) {
|
|
c.OnMessageUpdateFunctions = append(c.OnMessageUpdateFunctions, fn)
|
|
}
|
|
|
|
// On message delete event will run when someone deletes a message.
|
|
func (c *Client) OnMessageDelete(fn func(channel_id, message_id string)) {
|
|
c.OnMessageDeleteFunctions = append(c.OnMessageDeleteFunctions, fn)
|
|
}
|
|
|
|
// On channel create event will run when someone creates a channel.
|
|
func (c *Client) OnChannelCreate(fn func(channel *Channel)) {
|
|
c.OnChannelCreateFunctions = append(c.OnChannelCreateFunctions, fn)
|
|
}
|
|
|
|
// On channel update event will run when someone updates a channel.
|
|
func (c *Client) OnChannelUpdate(fn func(channel_id, clear string, payload map[string]interface{})) {
|
|
c.OnChannelUpdateFunctions = append(c.OnChannelUpdateFunctions, fn)
|
|
}
|
|
|
|
// On channel delete event will run when someone deletes a channel.
|
|
func (c *Client) OnChannelDelete(fn func(channel_id string)) {
|
|
c.OnChannelDeleteFunctions = append(c.OnChannelDeleteFunctions, fn)
|
|
}
|
|
|
|
// Fetch a channel by Id.
|
|
func (c *Client) FetchChannel(id string) (*Channel, error) {
|
|
channel := &Channel{}
|
|
channel.Client = c
|
|
|
|
data, err := c.Request("GET", "/channels/"+id, []byte{})
|
|
|
|
if err != nil {
|
|
return channel, err
|
|
}
|
|
|
|
err = json.Unmarshal(data, channel)
|
|
|
|
if err != nil {
|
|
return channel, err
|
|
}
|
|
return channel, nil
|
|
}
|
|
|
|
// Fetch an user by Id.
|
|
func (c *Client) FetchUser(id string) (*User, error) {
|
|
user := &User{}
|
|
user.Client = c
|
|
|
|
data, err := c.Request("GET", "/users/"+id, []byte{})
|
|
|
|
if err != nil {
|
|
return user, err
|
|
}
|
|
|
|
err = json.Unmarshal(data, user)
|
|
|
|
if err != nil {
|
|
return user, err
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
// Fetch a server by Id.
|
|
func (c *Client) FetchServer(id string) (*Server, error) {
|
|
server := &Server{}
|
|
server.Client = c
|
|
|
|
data, err := c.Request("GET", "/servers/"+id, []byte{})
|
|
|
|
if err != nil {
|
|
return server, err
|
|
}
|
|
|
|
err = json.Unmarshal(data, server)
|
|
|
|
if err != nil {
|
|
return server, err
|
|
}
|
|
|
|
return server, nil
|
|
}
|
|
|
|
// Create a server.
|
|
func (c *Client) CreateServer(name, description string) (*Server, error) {
|
|
server := &Server{}
|
|
server.Client = c
|
|
|
|
data, err := c.Request("POST", "/servers/create", []byte("{\"name\": \""+name+"\", \"description\": \""+description+"\", \"nonce\": \""+genULID()+"\"}"))
|
|
|
|
if err != nil {
|
|
return server, err
|
|
}
|
|
|
|
err = json.Unmarshal(data, server)
|
|
|
|
if err != nil {
|
|
return server, err
|
|
}
|
|
|
|
return server, nil
|
|
}
|