SimpleCloudNotifier/server/api/models/user.go

package models

import (
	"database/sql"
	"github.com/blockloop/scan"
	"time"
)

type User struct {
	UserID            int64
	Username          *string
	SendKey           string
	ReadKey           string
	AdminKey          string
	TimestampCreated  time.Time
	TimestampLastRead *time.Time
	TimestampLastSent *time.Time
	MessagesSent      int
	QuotaToday        int
	QuotaDay          *string
	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,
		QuotaToday:        u.QuotaToday,
		QuotaDay:          u.QuotaDay,
		IsPro:             u.IsPro,
	}
}

type UserJSON struct {
	UserID            int64   `json:"user_id"`
	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_last_read"`
	TimestampLastSent *string `json:"timestamp_last_sent"`
	MessagesSent      int     `json:"messages_sent"`
	QuotaToday        int     `json:"quota_today"`
	QuotaDay          *string `json:"quota_day"`
	IsPro             bool    `json:"is_pro"`
}

type UserDB struct {
	UserID            int64   `db:"user_id"`
	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"`
	QuotaToday        int     `db:"quota_today"`
	QuotaDay          *string `db:"quota_day"`
	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,
		QuotaToday:        u.QuotaToday,
		QuotaDay:          u.QuotaDay,
		IsPro:             u.IsPro,
	}
}

func DecodeUser(r *sql.Rows) (User, error) {
	var udb UserDB
	err := scan.RowStrict(&udb, r)
	if err != nil {
		return User{}, err
	}
	return udb.Model(), nil
}