120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package models
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
type Receipt struct {
|
|
bun.BaseModel `bun:"table:receipts"`
|
|
ReceiptID int64 `bun:",pk,autoincrement" json:"receipt_id" yaml:"receipt_id"`
|
|
UserID int64 `json:"user_id" yaml:"user_id"`
|
|
GroupID int64 `json:"group_id" yaml:"group_id"`
|
|
MerchantID int64 `json:"merchant_id" yaml:"merchant_id"`
|
|
CreditCardID int64 `json:"credit_card_id" yaml:"credit_card_id"`
|
|
CreditCard *CreditCard `bun:"credit_card,rel:belongs-to,join:credit_card_id=credit_card_id" json:"credit_card,omitempty" yaml:"credit_card,omitempty"`
|
|
Merchant *Merchant `bun:"merchant,rel:belongs-to,join:merchant_id=merchant_id" json:"merchant" yaml:"merchant"`
|
|
URL string `bun:"url" json:"url" yaml:"url"`
|
|
Type string `bun:"type" json:"type" yaml:"type"`
|
|
Category string `bun:"category" json:"category" yaml:"category"`
|
|
Items Items `bun:"rel:has-many,join:receipt_id=item_id" json:"items" yaml:"items"`
|
|
Tax float64 `bun:"tax" json:"tax,omitempty" yaml:"tax,omitempty"`
|
|
CCFee float64 `bun:"cc_fee" json:"cc_fee,omitempty" yaml:"cc_fee,omitempty"`
|
|
Tips float64 `bun:"tips" json:"tips,omitempty" yaml:"tips,omitempty"`
|
|
Total float64 `bun:"total" json:"total" yaml:"total"`
|
|
PaidWith string `bun:"paid_with" json:"paid_with" yaml:"paid_with"`
|
|
PaidAt string `bun:"paid_at" json:"paid_at" yaml:"paid_at"`
|
|
Created time.Time `bun:",nullzero,notnull,default:current_timestamp" json:"created" yaml:"created"`
|
|
Updated time.Time `json:"updated" yaml:"updated"`
|
|
}
|
|
|
|
type Receipts []*Receipt
|
|
|
|
var _ bun.AfterCreateTableHook = (*Receipt)(nil)
|
|
|
|
// AfterCreateTable .
|
|
func (*Receipt) AfterCreateTable(ctx context.Context, query *bun.CreateTableQuery) error {
|
|
_, err := query.DB().NewCreateIndex().Model((*Receipt)(nil)).Index("idx_receipts_category").Column("category").Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// NewReceipt .
|
|
func NewReceipt(j []byte) (*Receipt, error) {
|
|
var r *Receipt
|
|
if err := json.Unmarshal([]byte(j), &r); err != nil {
|
|
panic(err)
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
// Add .
|
|
func (r *Receipt) Add(db *bun.DB, ctx context.Context) error {
|
|
|
|
err := r.Merchant.Add(db, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = r.CreditCard.Add(db, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = db.NewInsert().Model(r).Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(r.Items) > 0 {
|
|
for _, item := range r.Items {
|
|
item.ReceiptID = r.ReceiptID
|
|
item.Title = strings.Title(strings.ToLower(item.Title))
|
|
}
|
|
err = r.Items.Add(db, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Update .
|
|
func (r *Receipt) Update(db *bun.DB, ctx context.Context) error {
|
|
_, err := db.NewUpdate().Model(r).WherePK().Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Remove .
|
|
func (r *Receipt) Remove(db *bun.DB, ctx context.Context) error {
|
|
err := r.Items.Remove(db, ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.NewDelete().Where("receipt_id = ?", r.ReceiptID).Exec(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Select .
|
|
func (r *Receipt) Select(db *bun.DB, ctx context.Context, id int64) error {
|
|
err := db.NewSelect().Model(r).Where("receipt_id = ?", id).Scan(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|