107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package psql
|
|
|
|
import (
|
|
"database/sql"
|
|
"kurumibot/database"
|
|
|
|
"github.com/shopspring/decimal"
|
|
"github.com/vinovest/sqlx"
|
|
)
|
|
|
|
type Waifu struct {
|
|
ID int
|
|
OwnerID sql.NullInt64 `db:"owner_id"`
|
|
Name string
|
|
Rarity int
|
|
ExpBonus decimal.Decimal `db:"exp_bonus"`
|
|
MoneyBonus decimal.Decimal `db:"money_bonus"`
|
|
MarketPrice decimal.Decimal `db:"market_price"`
|
|
Fandom string
|
|
Image string
|
|
RpPrompt string `db:"rp_prompt"`
|
|
Owner *User
|
|
}
|
|
|
|
type WaifuRepository struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func NewWaifuRepository(db *sqlx.DB) *WaifuRepository {
|
|
return &WaifuRepository{db: db}
|
|
}
|
|
|
|
func GetAllWaifus() ([]*Waifu, error) {
|
|
waifus, err := sqlx.List[*Waifu](
|
|
database.PostgresDatabase,
|
|
"SELECT waifus.* FROM waifus;",
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, waifu := range waifus {
|
|
if !waifu.OwnerID.Valid {
|
|
continue
|
|
}
|
|
waifu.Owner = new(User)
|
|
err = database.PostgresDatabase.Get(waifu.Owner, "SELECT * FROM users WHERE id=$1;", waifu.OwnerID.Int64)
|
|
}
|
|
return waifus, err
|
|
}
|
|
|
|
func GetUserWaifus(userId int) ([]*Waifu, error) {
|
|
waifus, err := sqlx.List[*Waifu](
|
|
database.PostgresDatabase,
|
|
"SELECT waifus.* FROM waifus WHERE owner_id=$1;",
|
|
userId,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
user, err := GetUser(userId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, waifu := range waifus {
|
|
waifu.Owner = user
|
|
}
|
|
return waifus, nil
|
|
}
|
|
|
|
func GetFreeWaifus() ([]*Waifu, error) {
|
|
waifus, err := sqlx.List[*Waifu](
|
|
database.PostgresDatabase,
|
|
"SELECT * FROM waifus WHERE owner_id IS NULL;",
|
|
)
|
|
return waifus, err
|
|
}
|
|
|
|
func GetFreeWaifusCount() (int64, error) {
|
|
var count int64 = 0
|
|
err := database.PostgresDatabase.QueryRow("SELECT COUNT(*) FROM waifus WHERE owner_id IS NULL;").Scan(&count)
|
|
return count, err
|
|
}
|
|
|
|
func GetFreeWaifusWithRarity(rarity int) ([]*Waifu, error) {
|
|
waifus, err := sqlx.List[*Waifu](
|
|
database.PostgresDatabase,
|
|
"SELECT * FROM waifus WHERE owner_id IS NULL AND rarity=$1;",
|
|
rarity,
|
|
)
|
|
return waifus, err
|
|
}
|
|
|
|
func GetWaifuById(id int) (*Waifu, error) {
|
|
waifu := new(Waifu)
|
|
err := database.PostgresDatabase.Get(waifu, "SELECT * FROM waifus WHERE id=$1;", id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !waifu.OwnerID.Valid {
|
|
return waifu, err
|
|
}
|
|
waifu.Owner, err = GetUser(int(waifu.OwnerID.Int64))
|
|
return waifu, err
|
|
}
|