rentease/internal/booking/store.go

package booking

import (
	"fmt"
	"time"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type PgStore struct {
	db *gorm.DB
}

func NewPgStore(db *gorm.DB) *PgStore {
	return &PgStore{db}
}

func (ps *PgStore) All() []*Line {
	bookings := make([]*Line, 0)
	ps.db.Raw(`
    select bookings.id, customer_name, "from", "to", platform, sum(price * quantity) as total, canceled
    from bookings
      left join items on bookings.id = items.booking_id
    group by bookings.id
    order by id desc;
    `).
		Scan(&bookings)
	return bookings
}

func (ps *PgStore) Search(value string) []*Line {
	bookings := make([]*Line, 0)
	ps.db.Raw(`
    select bookings.id, customer_name, "from", "to", platform, sum(price * quantity) as total, canceled
    from bookings
      left join items on bookings.id = items.booking_id
    where customer_name ilike ?
    group by bookings.id
    order by id desc;
    `, fmt.Sprintf("%%%s%%", value)).
		Scan(&bookings)
	return bookings
}

func (ps *PgStore) List(from, to time.Time) ([]*Line, error) {
	bookings := make([]*Line, 0)
	if err := ps.db.Raw(`
    select bookings.id, customer_name, "from", "to", platform, sum(price * quantity) as total, platform_fees
    from bookings
      join items on bookings.id = items.booking_id
      where "to" between ? and ?
    group by bookings.id
    order by bookings.id;
    `,
		from.Format(time.DateOnly), to.Format(time.DateOnly)).
		Scan(&bookings).
		Error; err != nil {
		return nil, err
	}

	return bookings, nil
}

func (ps *PgStore) CardTotal(from, to time.Time) (float64, error) {
	cardTotal := 0.0
	if err := ps.db.Raw(`
    select sum(total)
    from (select sum(price * quantity) as total
      from bookings
        join items on bookings.id = items.booking_id
      where "to" between ? and ?
        and payment_method = 'Card'
      group by booking_id) as t;
    `,
		from.Format(time.DateOnly), to.Format(time.DateOnly)).
		Scan(&cardTotal).
		Error; err != nil {
		return 0.0, err
	}
	return cardTotal, nil
}

func (ps *PgStore) Get(id int) *Booking {
	b := &Booking{Id: id}
	ps.db.Preload("Items").First(b)
	return b
}

func (ps *PgStore) Create(b *Booking) error {
	return ps.db.Create(b).Error
}

func (ps *PgStore) Update(b *Booking) error {
	return ps.db.Save(b).Error
}

func (ps *PgStore) Cancel(id int) error {
	b := &Booking{Id: id}
	return ps.db.Model(&b).
		Update("canceled", true).
		Error
}

// items

func (ps *PgStore) CreateItem(i *Item) error {
	return ps.db.Create(i).Error
}

func (ps *PgStore) PayItem(id int) (*Item, error) {
	i := new(Item)
	if err := ps.db.Model(i).
		Clauses(clause.Returning{}).
		Where("id = ?", id).
		Update("payment_status", "Completed").
		Error; err != nil {
		return nil, err
	}
	return i, nil
}

func (ps *PgStore) GetItem(id int) (*Item, error) {
	i := &Item{Id: id}
	err := ps.db.First(i).Error
	return i, err
}

func (ps *PgStore) UpdateItem(id int, item string, paymentMethod string, paymentStatus string, qty int, price float64) (*Item, error) {
	i := new(Item)
	err := ps.db.Model(i).
		Clauses(clause.Returning{}).
		Where("id = ?", id).
		Updates(map[string]any{
			"item":           item,
			"payment_method": paymentMethod,
			"payment_status": paymentStatus,
			"quantity":       qty,
			"price":          price,
		}).
		Error
	return i, err
}