SimpleCloudNotifier/scnserver/models/user.go

180 lines
4.5 KiB
Go
Raw Normal View History

2022-11-18 21:25:40 +01:00
package models
2022-11-18 23:12:37 +01:00
import (
scn "blackforestbytes.com/simplecloudnotifier"
"github.com/jmoiron/sqlx"
2022-11-19 12:47:23 +01:00
"gogs.mikescher.com/BlackForestBytes/goext/langext"
2022-12-11 03:14:42 +01:00
"gogs.mikescher.com/BlackForestBytes/goext/sq"
2022-11-18 23:12:37 +01:00
"time"
)
2022-11-18 21:25:40 +01:00
type User struct {
2022-11-20 22:18:24 +01:00
UserID UserID
2022-11-18 21:25:40 +01:00
Username *string
SendKey string
2022-11-18 23:12:37 +01:00
ReadKey string
2022-11-18 21:25:40 +01:00
AdminKey string
TimestampCreated time.Time
TimestampLastRead *time.Time
TimestampLastSent *time.Time
MessagesSent int
2022-11-19 15:13:47 +01:00
QuotaUsed int
QuotaUsedDay *string
2022-11-18 21:25:40 +01:00
IsPro bool
ProToken *string
}
func (u User) JSON() UserJSON {
return UserJSON{
UserID: u.UserID,
Username: u.Username,
ReadKey: u.ReadKey,
SendKey: u.SendKey,
AdminKey: u.AdminKey,
TimestampCreated: u.TimestampCreated.Format(time.RFC3339Nano),
TimestampLastRead: timeOptFmt(u.TimestampLastRead, time.RFC3339Nano),
TimestampLastSent: timeOptFmt(u.TimestampLastSent, time.RFC3339Nano),
MessagesSent: u.MessagesSent,
2022-12-14 17:56:03 +01:00
QuotaUsed: u.QuotaUsedToday(),
QuotaPerDay: u.QuotaPerDay(),
QuotaRemaining: u.QuotaRemainingToday(),
2022-11-18 21:25:40 +01:00
IsPro: u.IsPro,
2022-11-20 22:18:24 +01:00
DefaultChannel: u.DefaultChannel(),
2022-11-18 21:25:40 +01:00
}
}
2022-11-24 12:53:27 +01:00
func (u User) JSONWithClients(clients []Client) UserJSONWithClients {
return UserJSONWithClients{
UserJSON: u.JSON(),
Clients: langext.ArrMap(clients, func(v Client) ClientJSON { return v.JSON() }),
}
}
2022-11-19 15:13:47 +01:00
func (u User) MaxContentLength() int {
if u.IsPro {
return 16384
} else {
return 2048
}
}
2022-11-20 20:34:18 +01:00
func (u User) MaxTitleLength() int {
return 120
}
2022-11-19 15:13:47 +01:00
func (u User) QuotaPerDay() int {
if u.IsPro {
return 1000
} else {
return 50
}
}
func (u User) QuotaUsedToday() int {
now := scn.QuotaDayString()
2022-11-19 15:13:47 +01:00
if u.QuotaUsedDay != nil && *u.QuotaUsedDay == now {
return u.QuotaUsed
} else {
return 0
}
}
func (u User) QuotaRemainingToday() int {
return u.QuotaPerDay() - u.QuotaUsedToday()
}
2022-11-20 22:18:24 +01:00
func (u User) DefaultChannel() string {
return "main"
}
2022-11-30 21:39:14 +01:00
func (u User) DefaultPriority() int {
return 1
}
2022-11-21 22:52:44 +01:00
func (u User) MaxChannelNameLength() int {
return 120
}
func (u User) MaxSenderName() int {
return 120
}
func (u User) MaxUserMessageID() int {
return 64
}
func (u User) MaxTimestampDiffHours() int {
return 24
}
2022-11-18 21:25:40 +01:00
type UserJSON struct {
2022-11-20 22:18:24 +01:00
UserID UserID `json:"user_id"`
2022-11-18 21:25:40 +01:00
Username *string `json:"username"`
ReadKey string `json:"read_key"`
SendKey string `json:"send_key"`
AdminKey string `json:"admin_key"`
TimestampCreated string `json:"timestamp_created"`
TimestampLastRead *string `json:"timestamp_lastread"`
TimestampLastSent *string `json:"timestamp_lastsent"`
2022-11-18 21:25:40 +01:00
MessagesSent int `json:"messages_sent"`
2022-11-19 15:13:47 +01:00
QuotaUsed int `json:"quota_used"`
2022-12-14 17:56:03 +01:00
QuotaRemaining int `json:"quota_remaining"`
QuotaPerDay int `json:"quota_max"`
2022-11-18 21:25:40 +01:00
IsPro bool `json:"is_pro"`
2022-11-20 22:18:24 +01:00
DefaultChannel string `json:"default_channel"`
2022-11-18 21:25:40 +01:00
}
2022-11-18 23:12:37 +01:00
2022-11-24 12:53:27 +01:00
type UserJSONWithClients struct {
UserJSON
Clients []ClientJSON `json:"clients"`
}
2022-11-18 23:12:37 +01:00
type UserDB struct {
2022-11-20 22:18:24 +01:00
UserID UserID `db:"user_id"`
2022-11-18 23:12:37 +01:00
Username *string `db:"username"`
SendKey string `db:"send_key"`
ReadKey string `db:"read_key"`
AdminKey string `db:"admin_key"`
TimestampCreated int64 `db:"timestamp_created"`
TimestampLastRead *int64 `db:"timestamp_lastread"`
TimestampLastSent *int64 `db:"timestamp_lastsent"`
MessagesSent int `db:"messages_sent"`
2022-11-19 15:13:47 +01:00
QuotaUsed int `db:"quota_used"`
QuotaUsedDay *string `db:"quota_used_day"`
2022-11-18 23:12:37 +01:00
IsPro bool `db:"is_pro"`
ProToken *string `db:"pro_token"`
}
func (u UserDB) Model() User {
return User{
UserID: u.UserID,
Username: u.Username,
SendKey: u.SendKey,
ReadKey: u.ReadKey,
AdminKey: u.AdminKey,
TimestampCreated: time.UnixMilli(u.TimestampCreated),
TimestampLastRead: timeOptFromMilli(u.TimestampLastRead),
TimestampLastSent: timeOptFromMilli(u.TimestampLastSent),
MessagesSent: u.MessagesSent,
2022-11-19 15:13:47 +01:00
QuotaUsed: u.QuotaUsed,
QuotaUsedDay: u.QuotaUsedDay,
2022-11-18 23:12:37 +01:00
IsPro: u.IsPro,
}
}
func DecodeUser(r *sqlx.Rows) (User, error) {
data, err := sq.ScanSingle[UserDB](r, sq.SModeFast, sq.Safe, true)
2022-11-18 23:12:37 +01:00
if err != nil {
return User{}, err
}
2022-11-19 12:47:23 +01:00
return data.Model(), nil
}
func DecodeUsers(r *sqlx.Rows) ([]User, error) {
data, err := sq.ScanAll[UserDB](r, sq.SModeFast, sq.Safe, true)
2022-11-19 12:47:23 +01:00
if err != nil {
return nil, err
}
return langext.ArrMap(data, func(v UserDB) User { return v.Model() }), nil
2022-11-18 23:12:37 +01:00
}