mirror of
https://github.com/FJurmanovic/wallet-go-api.git
synced 2026-02-06 14:18:12 +00:00
98 lines
3.6 KiB
Go
98 lines
3.6 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
)
|
|
|
|
type Subscription struct {
|
|
tableName struct{} `pg:"subscriptions,alias:subscriptions"`
|
|
BaseModel
|
|
Description string `json:"description" pg:"description"`
|
|
StartDate time.Time `json:"startDate" pg:"start_date", type:timestamptz`
|
|
EndDate time.Time `json:"endDate" pg:"end_date", type:timestamptz`
|
|
HasEnd bool `json:"hasEnd" pg:"hasEnd"`
|
|
SubscriptionTypeID string `json:"subscriptionTypeId" pg:"subscription_type_id"`
|
|
SubscriptionType *SubscriptionType `json:"subscriptionType", pg:"rel:has-one, fk:subscription_type_id"`
|
|
CustomRange int `json:"customRange", pg:"custom_range,default:1"`
|
|
WalletID string `json:"walletId", pg:"wallet_id"`
|
|
Wallet *Wallet `json:"wallet" pg:"rel:has-one, fk:wallet_id"`
|
|
TransactionTypeID string `json:"transactionTypeId", pg:"transaction_type_id"`
|
|
TransactionType *TransactionType `json:"transactionType", pg:"rel:has-one, fk:transaction_type_id"`
|
|
LastTransactionDate time.Time `json:"lastTransactionDate", pg:"last_transaction_date", type:timestamptz`
|
|
Amount float32 `json:"amount", pg:"amount,default:0"`
|
|
}
|
|
|
|
type SubscriptionEdit struct {
|
|
tableName struct{} `pg:"subscriptions,alias:subscriptions"`
|
|
Id string `json:"id" form:"id"`
|
|
Description string `json:"description" form:"description"`
|
|
EndDate time.Time `json:"endDate" form:"endDate" `
|
|
HasEnd bool `json:"hasEnd" form:"hasEnd"`
|
|
WalletID string `json:"walletId" form:"walletId"`
|
|
Amount json.Number `json:"amount" form:"amount"`
|
|
}
|
|
|
|
type NewSubscriptionBody struct {
|
|
WalletID string `json:"walletId" form:"walletId"`
|
|
TransactionTypeID string `json:"transactionTypeId" form:"transactionTypeId"`
|
|
SubscriptionTypeID string `json:"subscriptionTypeId" pg:"subscription_type_id"`
|
|
CustomRange json.Number `json:"customRange", pg:"custom_range"`
|
|
StartDate time.Time `json:"startDate" pg:"start_date"`
|
|
EndDate time.Time `json:"endDate" pg:"end_date"`
|
|
HasEnd bool `json:"hasEnd" pg:"hasEnd"`
|
|
Description string `json:"description" form:"description"`
|
|
Amount json.Number `json:"amount" form:"amount"`
|
|
}
|
|
|
|
type SubscriptionEnd struct {
|
|
Id string `json:"id" form:"id"`
|
|
}
|
|
|
|
func (cm *Subscription) ToTrans() *Transaction {
|
|
trans := new(Transaction)
|
|
trans.Init()
|
|
trans.Amount = cm.Amount
|
|
trans.Description = cm.Description
|
|
trans.WalletID = cm.WalletID
|
|
trans.Wallet = cm.Wallet
|
|
trans.TransactionTypeID = cm.TransactionTypeID
|
|
trans.TransactionType = cm.TransactionType
|
|
trans.DateCreated = cm.DateCreated
|
|
trans.SubscriptionID = cm.Id
|
|
return trans
|
|
}
|
|
|
|
func (cm *Subscription) HasNew() bool {
|
|
trans := cm.TransactionType
|
|
if trans != nil {
|
|
switch trans.Type {
|
|
case "monthly":
|
|
lastDate := time.Now().AddDate(0, -cm.CustomRange, 0)
|
|
if cm.LastTransactionDate.Before(lastDate) {
|
|
return true
|
|
}
|
|
return false
|
|
case "weekly":
|
|
lastDate := time.Now().AddDate(0, 0, -(7 * cm.CustomRange))
|
|
if cm.LastTransactionDate.Before(lastDate) {
|
|
return true
|
|
}
|
|
return false
|
|
case "daily":
|
|
lastDate := time.Now().AddDate(0, 0, -cm.CustomRange)
|
|
if cm.LastTransactionDate.Before(lastDate) {
|
|
return true
|
|
}
|
|
return false
|
|
default:
|
|
lastDate := time.Now().AddDate(-cm.CustomRange, 0, 0)
|
|
if cm.LastTransactionDate.Before(lastDate) {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|