178 lines
8.0 KiB
Go
178 lines
8.0 KiB
Go
package user
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// JSONInt64Slice is a custom type for handling []int64 as JSON in database
|
|
type JSONInt64Slice []int64
|
|
|
|
// Scan implements sql.Scanner interface
|
|
func (j *JSONInt64Slice) Scan(value interface{}) error {
|
|
if value == nil {
|
|
*j = []int64{}
|
|
return nil
|
|
}
|
|
|
|
// Handle []byte
|
|
bytes, ok := value.([]byte)
|
|
if !ok {
|
|
// Try to handle string
|
|
str, ok := value.(string)
|
|
if !ok {
|
|
*j = []int64{}
|
|
return nil
|
|
}
|
|
bytes = []byte(str)
|
|
}
|
|
|
|
if len(bytes) == 0 {
|
|
*j = []int64{}
|
|
return nil
|
|
}
|
|
|
|
// Check if it's a JSON array
|
|
if bytes[0] != '[' {
|
|
// Not a JSON array, return empty slice
|
|
*j = []int64{}
|
|
return nil
|
|
}
|
|
|
|
return json.Unmarshal(bytes, j)
|
|
}
|
|
|
|
// Value implements driver.Valuer interface
|
|
func (j JSONInt64Slice) Value() (driver.Value, error) {
|
|
if len(j) == 0 {
|
|
return "[]", nil
|
|
}
|
|
return json.Marshal(j)
|
|
}
|
|
|
|
type User struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
Password string `gorm:"type:varchar(100);not null;comment:User Password"`
|
|
Algo string `gorm:"type:varchar(20);default:'default';comment:Encryption Algorithm"`
|
|
Salt string `gorm:"type:varchar(20);default:null;comment:Password Salt"`
|
|
Avatar string `gorm:"type:MEDIUMTEXT;comment:User Avatar"`
|
|
Balance int64 `gorm:"default:0;comment:User Balance"` // User Balance Amount
|
|
ReferCode string `gorm:"type:varchar(20);default:'';comment:Referral Code"`
|
|
RefererId int64 `gorm:"index:idx_referer;comment:Referrer ID"`
|
|
Commission int64 `gorm:"default:0;comment:Commission"` // Commission Amount
|
|
ReferralPercentage uint8 `gorm:"default:0;comment:Referral"` // Referral Percentage
|
|
OnlyFirstPurchase *bool `gorm:"default:true;not null;comment:Only First Purchase"` // Only First Purchase Referral
|
|
GiftAmount int64 `gorm:"default:0;comment:User Gift Amount"`
|
|
Enable *bool `gorm:"default:true;not null;comment:Is Account Enabled"`
|
|
IsAdmin *bool `gorm:"default:false;not null;comment:Is Admin"`
|
|
EnableBalanceNotify *bool `gorm:"default:false;not null;comment:Enable Balance Change Notifications"`
|
|
EnableLoginNotify *bool `gorm:"default:false;not null;comment:Enable Login Notifications"`
|
|
EnableSubscribeNotify *bool `gorm:"default:false;not null;comment:Enable Subscription Notifications"`
|
|
EnableTradeNotify *bool `gorm:"default:false;not null;comment:Enable Trade Notifications"`
|
|
AuthMethods []AuthMethods `gorm:"foreignKey:UserId;references:Id"`
|
|
UserDevices []Device `gorm:"foreignKey:UserId;references:Id"`
|
|
Rules string `gorm:"type:TEXT;comment:User Rules"`
|
|
LastLoginTime *time.Time `gorm:"default:NULL;comment:Last Login Time"`
|
|
MemberStatus string `gorm:"type:varchar(20);default:'';comment:Member Status"`
|
|
Remark string `gorm:"type:varchar(255);default:'';comment:Remark"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
DeletedAt gorm.DeletedAt `gorm:"index;comment:Deletion Time"`
|
|
}
|
|
|
|
func (*User) TableName() string {
|
|
return "user"
|
|
}
|
|
|
|
type Subscribe struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
|
User User `gorm:"foreignKey:UserId;references:Id"`
|
|
OrderId int64 `gorm:"index:idx_order_id;not null;comment:Order ID"`
|
|
SubscribeId int64 `gorm:"index:idx_subscribe_id;not null;comment:Subscription ID"`
|
|
NodeGroupId int64 `gorm:"index:idx_node_group_id;not null;default:0;comment:Node Group ID (single ID)"`
|
|
GroupLocked *bool `gorm:"type:tinyint(1);not null;default:0;comment:Group Locked"`
|
|
StartTime time.Time `gorm:"default:CURRENT_TIMESTAMP(3);not null;comment:Subscription Start Time"`
|
|
ExpireTime time.Time `gorm:"default:NULL;comment:Subscription Expire Time"`
|
|
FinishedAt *time.Time `gorm:"default:NULL;comment:Finished Time"`
|
|
Traffic int64 `gorm:"default:0;comment:Traffic"`
|
|
Download int64 `gorm:"default:0;comment:Download Traffic"`
|
|
Upload int64 `gorm:"default:0;comment:Upload Traffic"`
|
|
ExpiredDownload int64 `gorm:"default:0;comment:Expired period download traffic (bytes)"`
|
|
ExpiredUpload int64 `gorm:"default:0;comment:Expired period upload traffic (bytes)"`
|
|
Token string `gorm:"index:idx_token;unique;type:varchar(255);default:'';comment:Token"`
|
|
UUID string `gorm:"type:varchar(255);unique;index:idx_uuid;default:'';comment:UUID"`
|
|
Status uint8 `gorm:"type:tinyint(1);default:0;comment:Subscription Status: 0: Pending 1: Active 2: Finished 3: Expired 4: Deducted 5: stopped"`
|
|
Note string `gorm:"type:varchar(500);default:'';comment:User note for subscription"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
}
|
|
|
|
func (*Subscribe) TableName() string {
|
|
return "user_subscribe"
|
|
}
|
|
|
|
type AuthMethods struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
|
AuthType string `gorm:"type:varchar(255);not null;comment:Auth Type 1: apple 2: google 3: github 4: facebook 5: telegram 6: email 7: mobile 8: device"`
|
|
AuthIdentifier string `gorm:"type:varchar(255);unique;index:idx_auth_identifier;not null;comment:Auth Identifier"`
|
|
Verified bool `gorm:"default:false;not null;comment:Is Verified"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
}
|
|
|
|
func (*AuthMethods) TableName() string {
|
|
return "user_auth_methods"
|
|
}
|
|
|
|
type Device struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
Ip string `gorm:"type:varchar(255);not null;comment:Device IP"`
|
|
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
|
UserAgent string `gorm:"default:null;comment:UserAgent."`
|
|
Identifier string `gorm:"type:varchar(255);unique;index:idx_identifier;default:'';comment:Device Identifier"`
|
|
ShortCode string `gorm:"type:varchar(255);default:'';comment:Short Code"`
|
|
Online bool `gorm:"default:false;not null;comment:Online"`
|
|
Enabled bool `gorm:"default:true;not null;comment:Enabled"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
}
|
|
|
|
func (*Device) TableName() string {
|
|
return "user_device"
|
|
}
|
|
|
|
type DeviceOnlineRecord struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
UserId int64 `gorm:"type:bigint;not null;comment:User ID"`
|
|
Identifier string `gorm:"type:varchar(255);not null;comment:Device Identifier"`
|
|
OnlineTime time.Time `gorm:"comment:Online Time"` // The time when the device goes online
|
|
OfflineTime time.Time `gorm:"comment:Offline Time"`
|
|
OnlineSeconds int64 `gorm:"comment:Offline Seconds"`
|
|
DurationDays int64 `gorm:"comment:Duration Days"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
}
|
|
|
|
func (DeviceOnlineRecord) TableName() string {
|
|
return "user_device_online_record"
|
|
}
|
|
|
|
type Withdrawal struct {
|
|
Id int64 `gorm:"primaryKey"`
|
|
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
|
Amount int64 `gorm:"not null;comment:Withdrawal Amount"`
|
|
Content string `gorm:"type:text;comment:Withdrawal Content"`
|
|
Status uint8 `gorm:"type:tinyint(1);default:0;comment:Withdrawal Status: 0: Pending 1: Approved 2: Rejected"`
|
|
Reason string `gorm:"type:varchar(500);default:'';comment:Rejection Reason"`
|
|
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
|
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
|
}
|
|
|
|
func (*Withdrawal) TableName() string {
|
|
return "user_withdrawal"
|
|
}
|