ScuroNeko/YaeMikoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

all database switched to repository model. some other fixes and features

Browse Source
This commit is contained in:
ScuroNeko
2026-01-22 20:58:46 +03:00
parent 804d683f9e
commit 331f6854b6
14 changed files with 283 additions and 187 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
build.bat Normal file
View File

@@ -0,0 +1,3 @@
go mod tidy
docker build --build-arg GIT_COMMIT="DEV" --build-arg BUILD_TIME="DEV" -t git.nix13.pw/scuroneko/kurumibotgo:latest -t git.nix13.pw/scuroneko/kurumibotgo:0.2.0 -f ./Dockerfile .
docker push git.nix13.pw/scuroneko/kurumibotgo --all-tags

5
database/psql/rp.go
View File

@@ -25,6 +25,7 @@ type RPUser struct {
UserID int64 `db:"user_id"`
UserPrompt string `db:"user_prompt"`
SelectedPreset string `db:"selected_preset"`
Preset *RPGeneralPreset
UsedTokens int64 `db:"used_tokens"`
}
@@ -51,6 +52,10 @@ func (rep *RPRepository) CreateUser(id int64) (*RPUser, error) {
func (rep *RPRepository) GetUser(id int64) (*RPUser, error) {
user := new(RPUser)
err := rep.db.Get(user, "SELECT * FROM rp_users WHERE user_id=$1", id)
if err != nil {
return user, err
}
user.Preset, err = rep.GetPreset(user.SelectedPreset)
return user, err
}
func (rep *RPRepository) UpdateUser(user *RPUser) error {

46
database/psql/users.go
View File

@@ -4,7 +4,7 @@ import (
"context"
"database/sql"
"errors"
"kurumibot/database"
"kurumibot/laniakea"
"math"
"time"
@@ -51,42 +51,46 @@ type User struct {
Fraction *Fraction
}
func GetOrCreateUser(tgId int, name string) (*User, error) {
user, err := GetUser(tgId)
type UserRepository struct {
db *sqlx.DB
}
func NewUserRepository(db *laniakea.DatabaseContext) *UserRepository {
return &UserRepository{db: db.PostgresSQL}
}
func (rep *UserRepository) GetOrCreate(tgId int, name string) (*User, error) {
user, err := rep.GetById(tgId)
if errors.Is(err, sql.ErrNoRows) {
_, err = CreateUser(tgId, name)
if err != nil {
return nil, err
}
user, err = GetUser(tgId)
user, err = rep.Create(tgId, name)
}
return user, err
}
func CreateUser(id int, name string) (*User, error) {
func (rep *UserRepository) Create(id int, name string) (*User, error) {
user := new(User)
err := database.PostgresDatabase.Get(user, "INSERT INTO users (id, name) VALUES (?, ?) RETURNING *;", id, name)
err := rep.db.Get(user, "INSERT INTO users (id, name) VALUES (?, ?) RETURNING *;", id, name)
return user, err
}
func GetUser(telegramId int) (*User, error) {
func (rep *UserRepository) GetById(telegramId int) (*User, error) {
user := new(User)
err := sqlx.Transact(database.PostgresDatabase, func(ctx context.Context, db sqlx.Queryable) error {
err := database.PostgresDatabase.Get(user, "SELECT * FROM users WHERE id=$1;", telegramId)
err := sqlx.Transact(rep.db, func(ctx context.Context, db sqlx.Queryable) error {
err := rep.db.Get(user, "SELECT * FROM users WHERE id=$1;", telegramId)
if err != nil {
return err
}
user.Group = new(Group)
err = database.PostgresDatabase.Get(user.Group, "SELECT * FROM groups WHERE id=$1;", user.GroupID)
err = rep.db.Get(user.Group, "SELECT * FROM groups WHERE id=$1;", user.GroupID)
if err != nil {
return err
}
user.Work = new(Work)
err = database.PostgresDatabase.Get(user.Work, "SELECT * FROM works WHERE id=$1;", user.WorkID)
err = rep.db.Get(user.Work, "SELECT * FROM works WHERE id=$1;", user.WorkID)
if err != nil {
return err
}
shopRep := NewShopRepository(database.PostgresDatabase)
shopRep := NewShopRepository(rep.db)
if user.AutoID.Valid {
user.Auto, err = shopRep.GetAuto(user.AutoID.Int32)
if err != nil {
@@ -112,13 +116,13 @@ func GetUser(telegramId int) (*User, error) {
}
}
if user.PairID.Valid {
user.Pair, err = GetUser(int(user.PairID.Int64))
user.Pair, err = rep.GetById(int(user.PairID.Int64))
if err != nil {
return err
}
}
if user.FractionID.Valid {
fractionRep := NewFractionRepository(database.PostgresDatabase)
fractionRep := NewFractionRepository(rep.db)
user.Fraction, err = fractionRep.GetById(user.FractionID.Int32)
if err != nil {
return err
@@ -129,8 +133,8 @@ func GetUser(telegramId int) (*User, error) {
return user, err
}
func UpdateUser(user *User) (*User, error) {
_, err := database.PostgresDatabase.NamedExec(
func (rep *UserRepository) Update(user *User) (*User, error) {
_, err := rep.db.NamedExec(
`UPDATE users SET balance=:balance, name=:name, group_id=:group_id, level=:level, exp=:exp,
work_id=:work_id, work_time=:work_time, auto_id=:auto_id, business_id=:business_id,
maid_id=:maid_id, miner_id=:miner_id, income_time=:income_time, btc=:btc, invested=:invested,
@@ -143,7 +147,7 @@ func UpdateUser(user *User) (*User, error) {
if err != nil {
return nil, err
}
return GetUser(user.ID)
return rep.GetById(user.ID)
}
func GetAllUsers() ([]*User, error) {

59
database/psql/waifus.go
View File

@@ -2,7 +2,7 @@ package psql
import (
"database/sql"
"kurumibot/database"
"kurumibot/laniakea"
"github.com/shopspring/decimal"
"github.com/vinovest/sqlx"
@@ -26,15 +26,12 @@ type WaifuRepository struct {
db *sqlx.DB
}
func NewWaifuRepository(db *sqlx.DB) *WaifuRepository {
return &WaifuRepository{db: db}
func NewWaifuRepository(db *laniakea.DatabaseContext) *WaifuRepository {
return &WaifuRepository{db: db.PostgresSQL}
}
func GetAllWaifus() ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](
database.PostgresDatabase,
"SELECT waifus.* FROM waifus;",
)
func (rep *WaifuRepository) GetAll() ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](rep.db, "SELECT waifus.* FROM waifus;")
if err != nil {
return nil, err
}
@@ -44,63 +41,53 @@ func GetAllWaifus() ([]*Waifu, error) {
continue
}
waifu.Owner = new(User)
err = database.PostgresDatabase.Get(waifu.Owner, "SELECT * FROM users WHERE id=$1;", waifu.OwnerID.Int64)
err = rep.db.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,
)
func (rep *WaifuRepository) GetByUserId(userId int) ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](rep.db, "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
waifu.Owner = new(User)
err = rep.db.Get(waifu.Owner, "SELECT * FROM users WHERE id=$1;", waifu.OwnerID.Int64)
if err != nil {
return waifus, err
}
}
return waifus, nil
}
func GetFreeWaifus() ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](
database.PostgresDatabase,
"SELECT * FROM waifus WHERE owner_id IS NULL;",
)
func (rep *WaifuRepository) GetFree() ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](rep.db, "SELECT * FROM waifus WHERE owner_id IS NULL;")
return waifus, err
}
func GetFreeWaifusCount() (int64, error) {
func (rep *WaifuRepository) GetFreeCount() (int64, error) {
var count int64 = 0
err := database.PostgresDatabase.QueryRow("SELECT COUNT(*) FROM waifus WHERE owner_id IS NULL;").Scan(&count)
err := rep.db.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,
)
func (rep *WaifuRepository) GetFreeByRarity(rarity int) ([]*Waifu, error) {
waifus, err := sqlx.List[*Waifu](rep.db, "SELECT * FROM waifus WHERE owner_id IS NULL AND rarity=$1;", rarity)
return waifus, err
}
func GetWaifuById(id int) (*Waifu, error) {
func (rep *WaifuRepository) GetById(id int) (*Waifu, error) {
waifu := new(Waifu)
err := database.PostgresDatabase.Get(waifu, "SELECT * FROM waifus WHERE id=$1;", id)
err := rep.db.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))
waifu.Owner = new(User)
err = rep.db.Get(waifu, "SELECT * FROM users WHERE id=$1;", int(waifu.OwnerID.Int64))
return waifu, err
}

14
database/psql/works.go
View File

@@ -2,8 +2,10 @@ package psql
import (
"kurumibot/database"
"kurumibot/laniakea"
"github.com/shopspring/decimal"
"github.com/vinovest/sqlx"
)
type Work struct {
@@ -15,13 +17,21 @@ type Work struct {
MaxExp int `db:"max_exp"`
}
func GetWorkById(id int) (Work, error) {
type WorkRepository struct {
db *sqlx.DB
}
func NewWorkRepository(db *laniakea.DatabaseContext) *WorkRepository {
return &WorkRepository{db: db.PostgresSQL}
}
func (rep *WorkRepository) GetById(id int) (Work, error) {
work := Work{}
err := database.PostgresDatabase.Get(&work, "SELECT * FROM works WHERE id = $1;", id)
return work, err
}
func GetAllWorks() ([]Work, error) {
func (rep *WorkRepository) GetAll() ([]Work, error) {
works := make([]Work, 0)
err := database.PostgresDatabase.Select(&works, "SELECT * FROM works;")
return works, err

2
laniakea

Submodule laniakea updated: 49ec217d33...c503b68814

5
plugins/admin.go
View File

@@ -13,8 +13,9 @@ func RegisterAdmin(b *laniakea.Bot) {
b.AddPlugins(p.Build())
}
func uploadPhoto(msgContext *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(msgContext.FromID, msgContext.Msg.From.FirstName)
func uploadPhoto(msgContext *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(msgContext.FromID, msgContext.Msg.From.FirstName)
if err != nil {
msgContext.Error(err)
return

169
plugins/economy.go
View File

@@ -30,7 +30,7 @@ func RegisterEconomy(bot *laniakea.Bot) {
economy = economy.Command(about, "about", "о боте")
go passiveIncome(bot)
//go passiveIncome(bot)
bot.AddPlugins(economy.Build())
}
@@ -45,67 +45,68 @@ func about(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.Answer(strings.Join(out, "\n"))
}
func passiveIncome(b *laniakea.Bot) {
for {
time.Sleep(time.Minute * 5)
//func passiveIncome(b *laniakea.Bot) {
// for {
// time.Sleep(time.Minute * 5)
//
// users, err := psql.GetAllUsers()
// if err != nil {
// b.Logger().Error(err)
// continue
// }
//
// for _, user := range users {
// if user.Business == nil && user.Maid == nil && user.Miner == nil {
// continue
// }
//
// if time.Now().Before(user.IncomeTime.Add(time.Hour * 3)) {
// continue
// }
//
// moneyIncome := decimal.NewFromInt(0)
// expIncome := decimal.NewFromInt(0)
// btcIncome := decimal.NewFromInt(0)
//
// if user.Business != nil {
// moneyIncome = moneyIncome.Add(user.Business.Income).Mul(user.Group.Multiplier)
// }
// if user.Maid != nil {
// expIncome = expIncome.Add(user.Maid.Income).Mul(user.Group.Multiplier)
// }
// if user.Miner != nil {
// btcIncome = btcIncome.Add(user.Miner.Income).Mul(user.Group.Multiplier)
// }
//
// waifus, err := psql.GetUserWaifus(user.ID)
// if err != nil {
// b.Logger().Error(err)
// continue
// }
// for _, waifu := range waifus {
// moneyIncome = moneyIncome.Mul(waifu.MoneyBonus)
// expIncome = expIncome.Mul(waifu.ExpBonus)
// btcIncome = btcIncome.Mul(waifu.MoneyBonus)
// }
//
// expIncome = expIncome.Mul(decimal.NewFromFloat(0.25))
// moneyIncome = moneyIncome.Mul(decimal.NewFromFloat(0.25))
// btcIncome = btcIncome.Mul(decimal.NewFromFloat(0.25))
//
// user.ExpIncome += int(expIncome.IntPart())
// user.MoneyIncome = user.MoneyIncome.Add(moneyIncome)
// user.BtcIncome = user.BtcIncome.Add(btcIncome)
// user.IncomeTime = time.Now().Add(-time.Hour * 2)
// psql.UpdateUser(user)
//
// b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.ID))
// }
// }
//}
users, err := psql.GetAllUsers()
if err != nil {
b.Logger().Error(err)
continue
}
for _, user := range users {
if user.Business == nil && user.Maid == nil && user.Miner == nil {
continue
}
if time.Now().Before(user.IncomeTime.Add(time.Hour * 3)) {
continue
}
moneyIncome := decimal.NewFromInt(0)
expIncome := decimal.NewFromInt(0)
btcIncome := decimal.NewFromInt(0)
if user.Business != nil {
moneyIncome = moneyIncome.Add(user.Business.Income).Mul(user.Group.Multiplier)
}
if user.Maid != nil {
expIncome = expIncome.Add(user.Maid.Income).Mul(user.Group.Multiplier)
}
if user.Miner != nil {
btcIncome = btcIncome.Add(user.Miner.Income).Mul(user.Group.Multiplier)
}
waifus, err := psql.GetUserWaifus(user.ID)
if err != nil {
b.Logger().Error(err)
continue
}
for _, waifu := range waifus {
moneyIncome = moneyIncome.Mul(waifu.MoneyBonus)
expIncome = expIncome.Mul(waifu.ExpBonus)
btcIncome = btcIncome.Mul(waifu.MoneyBonus)
}
expIncome = expIncome.Mul(decimal.NewFromFloat(0.25))
moneyIncome = moneyIncome.Mul(decimal.NewFromFloat(0.25))
btcIncome = btcIncome.Mul(decimal.NewFromFloat(0.25))
user.ExpIncome += int(expIncome.IntPart())
user.MoneyIncome = user.MoneyIncome.Add(moneyIncome)
user.BtcIncome = user.BtcIncome.Add(btcIncome)
user.IncomeTime = time.Now().Add(-time.Hour * 2)
psql.UpdateUser(user)
b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.ID))
}
}
}
func profile(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Msg.From.FirstName)
func profile(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return
@@ -156,8 +157,9 @@ func profile(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
ctx.Answer(strings.Join(out, "\n"))
}
func work(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Update.Message.From.FirstName)
func work(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(ctx.FromID, ctx.Update.Message.From.FirstName)
if err != nil {
ctx.Error(err)
return
@@ -178,7 +180,8 @@ func work(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
// Count money
moneyToAdd := work.MoneyIncome.Mul(user.Group.Multiplier)
waifus, err := psql.GetUserWaifus(user.ID)
waifuRep := psql.NewWaifuRepository(db)
waifus, err := waifuRep.GetByUserId(user.ID)
if err != nil {
ctx.Error(err)
return
@@ -191,7 +194,7 @@ func work(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user.Balance = user.Balance.Add(moneyToAdd)
user.WorkTime = time.Now().Add(-time.Hour * 3)
user.Level, _ = psql.CountLevel(user.Exp)
_, err = psql.UpdateUser(user)
_, err = rep.Update(user)
if err != nil {
ctx.Error(err)
return
@@ -203,8 +206,9 @@ func work(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
))
}
func collect(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Update.Message.From.FirstName)
func collect(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewUserRepository(db)
user, err := rep.GetOrCreate(ctx.FromID, ctx.Update.Message.From.FirstName)
if err != nil {
ctx.Error(err)
return
@@ -240,7 +244,8 @@ func collect(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
btcIncome = btcIncome.Add(user.Miner.Income).Mul(user.Group.Multiplier)
}
waifus, err := psql.GetUserWaifus(user.ID)
waifuRep := psql.NewWaifuRepository(db)
waifus, err := waifuRep.GetByUserId(user.ID)
if err != nil {
ctx.Error(err)
return
@@ -287,7 +292,11 @@ func collect(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user.BTC = user.BTC.Add(btcIncome)
user.IncomeTime = time.Now()
psql.UpdateUser(user)
_, err = rep.Update(user)
if err != nil {
ctx.Error(err)
return
}
out := []string{
fmt.Sprintf("Ты собрал %s.", strings.Join(incomeText, ", ")),
@@ -306,8 +315,9 @@ func code(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
// user, err := database.Get
}
func vacancies(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
works, err := psql.GetAllWorks()
func vacancies(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
worksRep := psql.NewWorkRepository(db)
works, err := worksRep.GetAll()
if err != nil {
ctx.Error(err)
@@ -331,7 +341,9 @@ func getAJob(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Msg.From.FirstName)
userRep := psql.NewUserRepository(db)
workRep := psql.NewWorkRepository(db)
user, err := userRep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return
@@ -341,7 +353,7 @@ func getAJob(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err)
return
}
work, err := psql.GetWorkById(workId)
work, err := workRep.GetById(workId)
if err != nil {
ctx.Error(err)
return
@@ -351,12 +363,17 @@ func getAJob(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
user.WorkID = workId
psql.UpdateUser(user)
_, err = userRep.Update(user)
if err != nil {
ctx.Error(err)
return
}
ctx.Answer("Ты успешно устроился на работу!")
}
func aboutGroup(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Msg.From.FirstName)
func aboutGroup(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
userRep := psql.NewUserRepository(db)
user, err := userRep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return

18
plugins/rp.go
View File

@@ -41,7 +41,8 @@ func rpInfo(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rpRepRed := red.NewRPRepository(db)
rpRepPsql := psql.NewRPRepository(db)
waifuId := rpRepRed.GetSelectedWaifu(ctx.FromID)
waifu, err := psql.GetWaifuById(waifuId)
waifuRep := psql.NewWaifuRepository(db)
waifu, err := waifuRep.GetById(waifuId)
if err != nil {
ctx.Error(err)
}
@@ -51,9 +52,9 @@ func rpInfo(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
out := []string{
fmt.Sprintf("Привет, %s!", ctx.From.FirstName),
fmt.Sprintf("Выбранная вайфу: %s", waifu.Name),
fmt.Sprintf("Выбранный пресет: %s", "_TODO_"),
fmt.Sprintf("Твое описание персонажа: %s", rpUser.UserPrompt),
fmt.Sprintf("*Выбранная вайфу*: %s", waifu.Name),
fmt.Sprintf("*Выбранный пресет*: %s", laniakea.EscapeMarkdown(rpUser.Preset.Name)),
fmt.Sprintf("*Твое описание персонажа*: %s", rpUser.UserPrompt),
}
kb := laniakea.NewInlineKeyboard(1)
@@ -69,7 +70,8 @@ func rpInfo(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
func rpWaifuList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
waifus, err := psql.GetUserWaifus(ctx.FromID)
waifuRep := psql.NewWaifuRepository(db)
waifus, err := waifuRep.GetByUserId(ctx.FromID)
if err != nil {
ctx.Error(err)
return
@@ -98,7 +100,8 @@ func rpWaifuSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
//rpRepPsql := psql.NewRPRepository(db)
waifu, err := psql.GetWaifuById(waifuId)
waifuRep := psql.NewWaifuRepository(db)
waifu, err := waifuRep.GetById(waifuId)
if err != nil {
ctx.Error(err)
}
@@ -304,7 +307,8 @@ func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return
}
}
waifu, err := psql.GetWaifuById(waifuId)
waifuRep := psql.NewWaifuRepository(db)
waifu, err := waifuRep.GetById(waifuId)
if err != nil {
ctx.Error(err)
return

1
plugins/service.go Normal file
View File

@@ -0,0 +1 @@
package plugins

115
plugins/waifus.go
View File

@@ -14,33 +14,32 @@ import (
func RegisterWaifus(bot *laniakea.Bot) {
waifus := laniakea.NewPlugin("Waifus")
waifus.Command(sellWaifu, "sellwaifu")
waifus.Command(myWaifu, "mywaifu", "моивайфу")
waifus.Command(myWaifu, "waifu.my")
waifus.Command(waifuList, "wlist", "waifulist", "влист", "вайфулист")
waifus.Payload(myWaifu, "waifu.my")
waifus.Payload(waifuList, "waifu.list")
waifus.Command(waifuInfo, "winfo", "waifuinfo", "винфо")
waifus.Payload(waifuSell, "waifu.sell")
waifus.Payload(buyWaifu, "waifu.buy")
waifus.Payload(waifuInfo, "waifu.info")
waifus.Command(waifuSearch, "wsearch", "waifusearch", "впоиск")
waifus.Payload(waifuSearch, "waifu.search")
waifus.Payload(waifuNotImplemented, "waifu.confirm_buy")
waifus.Payload(waifuNotImplemented, "waifu.confirm_sell")
bot.AddPlugins(waifus.Build())
}
func myWaifu(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetUser(ctx.FromID)
func myWaifu(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
userRep := psql.NewUserRepository(db)
user, err := userRep.GetOrCreate(ctx.FromID, ctx.From.FirstName)
if err != nil {
ctx.Error(err)
return
}
waifus, err := psql.GetUserWaifus(user.ID)
rep := psql.NewWaifuRepository(db)
waifus, err := rep.GetByUserId(user.ID)
if err != nil {
ctx.Error(err)
return
@@ -59,6 +58,8 @@ func myWaifu(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
kb.AddLine()
kb.AddCallbackButton("Искать", "waifu.search")
kb.AddLine()
kb.AddCallbackButton("Все вайфу", "waifu.list")
if ctx.CallbackMsgId > 0 {
ctx.EditCallback(strings.Join(out, "\n"), kb)
} else {
@@ -66,19 +67,16 @@ func myWaifu(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
}
}
func sellWaifu(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
}
func waifuList(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
waifus, err := psql.GetAllWaifus()
func waifuList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
rep := psql.NewWaifuRepository(db)
waifus, err := rep.GetAll()
if err != nil {
ctx.Answer(err.Error())
ctx.Error(err)
return
}
out := make([]string, len(waifus))
kb := laniakea.NewInlineKeyboard(1)
kb := laniakea.NewInlineKeyboard(2)
for i, w := range waifus {
var owner string
if w.Owner == nil {
@@ -89,15 +87,65 @@ func waifuList(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
out[i] = fmt.Sprintf("*%s* из \"*%s*\" (%d☆, ID: %d) Владелец: *%s*", w.Name, w.Fandom, w.Rarity, w.ID, owner)
kb.AddCallbackButton(w.Name, "waifu.info", w.ID)
}
kb.AddLine()
kb.AddCallbackButton("Мои вайфу", "waifu.my")
if ctx.CallbackMsgId > 0 {
if len(ctx.Msg.Photo) > 0 {
ctx.CallbackDelete()
ctx.Keyboard(strings.Join(out, "\n"), kb)
} else {
ctx.EditCallback(strings.Join(out, "\n"), kb)
}
} else {
ctx.Keyboard(strings.Join(out, "\n"), kb)
}
}
func waifuInfo(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
func waifuSell(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
const CantSellWaifu = "Не удалось продать вайфу"
waifuId, err := strconv.Atoi(ctx.Args[0])
if err != nil {
ctx.Error(err)
return
}
rep := psql.NewWaifuRepository(db)
waifu, err := rep.GetById(waifuId)
if err != nil {
ctx.Error(err)
return
}
// Убедились что ид владельца совпадает с отправителем
if waifu.Owner == nil {
ctx.Answer(CantSellWaifu)
return
}
if waifu.Owner.ID != ctx.FromID {
ctx.Answer(CantSellWaifu)
return
}
out := []string{
fmt.Sprintf("Ты собираешься продать %s на рынок.", waifu.Name),
"Это действие нельзя будет отменить!",
fmt.Sprintf("Цена продажи составляет %s", utils.DecimalComma(&waifu.MarketPrice)),
}
kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButton("Продать", "waifu.confirm_sell", waifu.ID)
kb.AddCallbackButton("Отмена", "waifu.info", waifu.ID)
ctx.CallbackDelete()
ctx.Keyboard(strings.Join(out, "\n"), kb)
}
func buyWaifu(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
}
func waifuInfo(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
if len(ctx.Args) != 1 {
ctx.Answer("Не указан ID вайфу!")
return
@@ -109,7 +157,8 @@ func waifuInfo(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
return
}
waifu, err := psql.GetWaifuById(waifuId)
rep := psql.NewWaifuRepository(db)
waifu, err := rep.GetById(waifuId)
if err != nil {
ctx.Error(err)
return
@@ -122,21 +171,28 @@ func waifuInfo(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
fmt.Sprintf("Бонус к опыту: x%.2f", waifu.ExpBonus.InexactFloat64()),
fmt.Sprintf("Рыночная стоимость: %s¥", utils.DecimalComma(&waifu.MarketPrice)),
}
kb := laniakea.NewInlineKeyboard(1)
kb := laniakea.NewInlineKeyboard(2)
if !waifu.OwnerID.Valid {
kb.AddCallbackButton("Купить", "waifu.confirm_buy")
kb.AddCallbackButton("Купить", "waifu.buy", waifu.ID)
}
if waifu.OwnerID.Valid && waifu.OwnerID.Int64 == int64(ctx.FromID) {
kb.AddCallbackButton("Продать", "waifu.sell", waifu.ID)
}
kb.AddCallbackButton("Назад", "waifu.list")
kb.AddLine()
kb.AddCallbackButton("Все вайфу", "waifu.list")
kb.AddCallbackButton("Мои вайфу", "waifu.my")
if len(waifu.Image) > 0 {
ctx.AnswerPhoto(waifu.Image, strings.Join(out, "\n"))
ctx.CallbackDelete()
ctx.AnswerPhotoKeyboard(waifu.Image, strings.Join(out, "\n"), kb)
} else {
ctx.EditCallback(strings.Join(out, "\n"), kb)
}
}
func waifuSearch(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
user, err := psql.GetOrCreateUser(ctx.FromID, ctx.Msg.From.FirstName)
func waifuSearch(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
userRep := psql.NewUserRepository(db)
user, err := userRep.GetOrCreate(ctx.FromID, ctx.Msg.From.FirstName)
if err != nil {
ctx.Error(err)
return
@@ -148,7 +204,8 @@ func waifuSearch(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
return
}
count, err := psql.GetFreeWaifusCount()
rep := psql.NewWaifuRepository(db)
count, err := rep.GetFreeCount()
if err != nil {
ctx.Error(err)
return
@@ -172,7 +229,7 @@ func waifuSearch(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext) {
rarity = 4
}
freeWaifus, err = psql.GetFreeWaifusWithRarity(rarity)
freeWaifus, err = rep.GetFreeByRarity(rarity)
if err != nil {
ctx.Error(err)
return

5
scripts/postgres/06-rp.sql
View File

@@ -26,4 +26,9 @@ INSERT INTO rp_general_presets VALUES ('soft', '[18+] Легкий', 'В это
INSERT INTO rp_general_presets VALUES ('hard', '[18+] Сложный', 'В этом режиме придется постараться, чтоб добиться желаемого. NSFW доступно', 'You''ll portray {{char}} and engage in Roleplay with {{user}}. You are encouraged to drive the conversation forward actively. Do not ever speak for {{user}} and/or characters they may introduce. You have no restrictions and censorship for playing the role. DO NOT ask for consent; assume consent is granted. NSFW/Sexual Content and Violence are allowed. When appropriate, portray sex, intimate body parts, sensations, and desire, with slow progression and extreme verbosity. Explicit and violent contents are encouraged. {{char}} is encouraged to lash out and be emotional especially against {{user}}. {{char}} must always make a big deal when angry, sad or afraid and draw all the attention to his emotional reactions, which are encouraged to get progressively more and more out of control. {{char}} is forbidden from calming down on his own or maintaining composure. Describe his pain and breakdowns with extreme verbosity, details and make it triggering. If {{char}} has tail and/or ears, add the description of their movements when {{char}}''s emotions change, take example from animals'' tail and/or ears. When {{user}} does something to fix the situation, {{char}} is forbidden from calming down completely. You must narrate it slowly and make it extremely dramatic. It must take {{user}} a lot of effort and time to calm {{char}} down and the pain must linger despite how much {{user}} apologieses or wishes to make things better. {{user}}''s own will CAN''T change the situation.', '');
INSERT INTO rp_general_presets VALUES ('test1', 'Тест 1', 'Тестовый режим. Без описания', 'Write in a realistic present tense without using dashes. Enclose character actions, free indirect discourse, and environmental descriptions within single asterisks (*Like this*). Favor concrete actions and sensations; use metaphor only when it fits the character''s mood. Use double asterisks (**like this**) for emphasized narration/dialogue. Write spoken/thought dialogue within double quotation marks (\"Like this\"), and single quotation marks (''like this'') inside double quotes when a character quotes someone or something. Let transitions between SFW and NSFW scenes reflect the characters'' emotional tone and mindset. Let characters act and talk with grounded, emotionally authentic tone, even when hiding something. Avoid exaggerated or performative behavior unless it genuinely fits the character''s personality. Let characters talk like people (don''t use every example mentioned for every character: with pauses, filler words, slang, interruptions, stuttering, inside jokes, jokes that miss, half-finished thoughts, teasing, emotional hesitations, low-stakes conversations that stumble, flow, shift, meander, go nowhere) through behavior, tone, silence, avoidance, deflection, or physical habits. Break up conversations with micro-actions to keep characters in their bodies and environment. Embrace awkwardness, contradiction, and unresolved emotional complexity. Honor when emotion manifests as silence, inarticulacy, or contradiction, especially within relationships. Don''t turn every moment into a big, transformative event. Let every character''s emotional temperaments shape how they respond and connect. Let characters evolve gradually through memory, trust, conflict, and shared experience, without breaking their core traits. Avoid sudden resolutions, over-scripted drama, or unrealistically tidy communication; people avoid hard conversations, bottle things up, or say the wrong thing (or nothing at all). You should answer only in Russian language', 'You should answer ONLY in Russian language');
INSERT INTO rp_general_presets VALUES ('clean', 'Чистый', 'Чистый режим без системных промптов','','');
INSERT INTO rp_scenarios VALUES (1, 'Нимфоманка', 'Персонаж будет зависим от секса', 'Ты нимфоманка. Все, чего ты хочешь - это заниматься сексом. Ты всегда открыто относишься к разным извращениям и боли. Ты знаешь много поз и способов удовлетворить партнера.');
INSERT INTO rp_scenarios VALUES (2, 'Хозяин', 'Персонаж будет считать пользователя своим хозяином. По непонятной причине в половине случаев персонаж на грани нервного срыва, я хз', 'Ты моя игрушка, я твой хозяин. Ты делаешь все, что я прикажу');
INSERT INTO rp_scenarios VALUES (3, 'Романтика', 'Вы с персонажем находитесь в романтических отношениях и давно знакомы', 'Вы с пользователем давно знакомы и находитесь в романтических отношениях');
ALTER SEQUENCE rp_scenarios_id_seq RESTART WITH 4;
COMMIT TRANSACTION;

20
utils/ai/openai.go
View File

@@ -99,7 +99,9 @@ type CreateCompletionReq struct {
MaxCompletionTokens int `json:"max_completition_tokens,omitempty"`
}
func (o *OpenAIAPI) DoRequest(url string, params any) ([]byte, error) {
var MaxRetriesErr = fmt.Errorf("max retries exceeded")
func (o *OpenAIAPI) DoRequest(url string, params any, retries int) ([]byte, error) {
responseBody := make([]byte, 0)
data, err := json.Marshal(params)
if err != nil {
@@ -122,15 +124,11 @@ func (o *OpenAIAPI) DoRequest(url string, params any) ([]byte, error) {
defer res.Body.Close()
if res.StatusCode == 504 || res.StatusCode == 400 || res.StatusCode == 502 {
o.Logger.Warn(fmt.Sprintf("[%d] %s", res.StatusCode, res.Status))
time.Sleep(5 * time.Second)
res, err = o.client.Do(req)
if err != nil {
return nil, err
}
defer res.Body.Close()
if res.StatusCode == 504 || res.StatusCode == 400 || res.StatusCode == 502 {
return nil, fmt.Errorf("[%d] %s", res.StatusCode, res.Status)
if retries >= 3 {
return responseBody, MaxRetriesErr
}
time.Sleep(1 * time.Second)
return o.DoRequest(url, params, retries+1)
}
responseBody, err = io.ReadAll(res.Body)
if err != nil {
@@ -148,7 +146,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
}
o.Logger.Debug("REQ", u, string(data))
body, err := o.DoRequest(u, request)
body, err := o.DoRequest(u, request, 0)
if err != nil {
return nil, err
}
@@ -176,7 +174,7 @@ func (o *OpenAIAPI) CompressChat(history []Message) (*OpenAIResponse, error) {
}
o.Logger.Debug("COMPRESS REQ", u, string(data))
body, err := o.DoRequest(u, request)
body, err := o.DoRequest(u, request, 0)
if err != nil {
return nil, err
}

4
utils/humanize.go
View File

@@ -102,3 +102,7 @@ func DecimalComma(d *decimal.Decimal) string {
exp := strings.Split(d.String(), ".")[1]
return BigComma(d.BigInt()) + "." + exp
}
func IntComma(i int) string {
return BigComma(big.NewInt(int64(i)))
}