58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package psql
|
|
|
|
import (
|
|
"kurumibot/database"
|
|
|
|
"github.com/shopspring/decimal"
|
|
)
|
|
|
|
type Waifu struct {
|
|
ID int
|
|
OwnerID int
|
|
Owner *User
|
|
Name string
|
|
Rarity int
|
|
ExpBonus decimal.Decimal
|
|
MoneyBonus decimal.Decimal
|
|
MarketPrice decimal.Decimal
|
|
Fandom string
|
|
Image string
|
|
RpPrompt string
|
|
}
|
|
|
|
func GetAllWaifus() ([]*Waifu, error) {
|
|
waifus := make([]*Waifu, 0)
|
|
tx := database.PostgresDatabase.Joins("Owner").Find(&waifus).Order("id")
|
|
return waifus, tx.Error
|
|
}
|
|
|
|
func GetUserWaifus(userId int) ([]*Waifu, error) {
|
|
waifus := make([]*Waifu, 0)
|
|
tx := database.PostgresDatabase.Find(&waifus, "owner_id = ?", userId).Order("id")
|
|
return waifus, tx.Error
|
|
}
|
|
|
|
func GetFreeWaifus() ([]*Waifu, error) {
|
|
waifus := make([]*Waifu, 0)
|
|
tx := database.PostgresDatabase.Find(&waifus, "owner_id is null").Order("id")
|
|
return waifus, tx.Error
|
|
}
|
|
|
|
func GetFreeWaifusCount() (int64, error) {
|
|
var count int64 = 0
|
|
tx := database.PostgresDatabase.Model(&Waifu{}).Where("owner_id is null").Count(&count)
|
|
return count, tx.Error
|
|
}
|
|
|
|
func GetFreeWaifusWithRarity(rarity int) ([]*Waifu, error) {
|
|
waifus := make([]*Waifu, 0)
|
|
tx := database.PostgresDatabase.Find(&waifus, "owner_id is null and rarity = ?", rarity)
|
|
return waifus, tx.Error
|
|
}
|
|
|
|
func GetWaifuById(id int) (*Waifu, error) {
|
|
waifu := new(Waifu)
|
|
tx := database.PostgresDatabase.Joins("Owner").Find(waifu, id)
|
|
return waifu, tx.Error
|
|
}
|