146 lines
2.7 KiB
Go
146 lines
2.7 KiB
Go
package soju
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Database interface {
|
|
Close() error
|
|
|
|
ListUsers() ([]User, error)
|
|
GetUser(username string) (*User, error)
|
|
StoreUser(user *User) error
|
|
DeleteUser(id int64) error
|
|
|
|
ListNetworks(userID int64) ([]Network, error)
|
|
StoreNetwork(userID int64, network *Network) error
|
|
DeleteNetwork(id int64) error
|
|
ListChannels(networkID int64) ([]Channel, error)
|
|
StoreChannel(networKID int64, ch *Channel) error
|
|
DeleteChannel(id int64) error
|
|
|
|
ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error)
|
|
StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) error
|
|
}
|
|
|
|
type User struct {
|
|
ID int64
|
|
Username string
|
|
Password string // hashed
|
|
Admin bool
|
|
}
|
|
|
|
type SASL struct {
|
|
Mechanism string
|
|
|
|
Plain struct {
|
|
Username string
|
|
Password string
|
|
}
|
|
|
|
// TLS client certificate authentication.
|
|
External struct {
|
|
// X.509 certificate in DER form.
|
|
CertBlob []byte
|
|
// PKCS#8 private key in DER form.
|
|
PrivKeyBlob []byte
|
|
}
|
|
}
|
|
|
|
type Network struct {
|
|
ID int64
|
|
Name string
|
|
Addr string
|
|
Nick string
|
|
Username string
|
|
Realname string
|
|
Pass string
|
|
ConnectCommands []string
|
|
SASL SASL
|
|
Enabled bool
|
|
}
|
|
|
|
func (net *Network) GetName() string {
|
|
if net.Name != "" {
|
|
return net.Name
|
|
}
|
|
return net.Addr
|
|
}
|
|
|
|
func (net *Network) URL() (*url.URL, error) {
|
|
s := net.Addr
|
|
if !strings.Contains(s, "://") {
|
|
// This is a raw domain name, make it an URL with the default scheme
|
|
s = "ircs://" + s
|
|
}
|
|
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse upstream server URL: %v", err)
|
|
}
|
|
|
|
return u, nil
|
|
}
|
|
|
|
func (net *Network) GetUsername() string {
|
|
if net.Username != "" {
|
|
return net.Username
|
|
}
|
|
return net.Nick
|
|
}
|
|
|
|
func (net *Network) GetRealname() string {
|
|
if net.Realname != "" {
|
|
return net.Realname
|
|
}
|
|
return net.Nick
|
|
}
|
|
|
|
type MessageFilter int
|
|
|
|
const (
|
|
// TODO: use customizable user defaults for FilterDefault
|
|
FilterDefault MessageFilter = iota
|
|
FilterNone
|
|
FilterHighlight
|
|
FilterMessage
|
|
)
|
|
|
|
func parseFilter(filter string) (MessageFilter, error) {
|
|
switch filter {
|
|
case "default":
|
|
return FilterDefault, nil
|
|
case "none":
|
|
return FilterNone, nil
|
|
case "highlight":
|
|
return FilterHighlight, nil
|
|
case "message":
|
|
return FilterMessage, nil
|
|
}
|
|
return 0, fmt.Errorf("unknown filter: %q", filter)
|
|
}
|
|
|
|
type Channel struct {
|
|
ID int64
|
|
Name string
|
|
Key string
|
|
|
|
Detached bool
|
|
DetachedInternalMsgID string
|
|
|
|
RelayDetached MessageFilter
|
|
ReattachOn MessageFilter
|
|
DetachAfter time.Duration
|
|
DetachOn MessageFilter
|
|
}
|
|
|
|
type DeliveryReceipt struct {
|
|
ID int64
|
|
Target string // channel or nick
|
|
Client string
|
|
InternalMsgID string
|
|
}
|