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

many new

Browse Source
This commit is contained in:
ScuroNeko
2026-01-13 13:27:42 +03:00
parent 5e69454678
commit 4925283ede
27 changed files with 6236 additions and 260 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.env
View File

@@ -6,4 +6,5 @@ PSQL_NAME=kurumi_bot
MONGO_HOST=127.0.0.1 MONGO_HOST=127.0.0.1
MONGO_USER=kurumi_bot MONGO_USER=kurumi_bot
MONGO_PASS=kurumi_bot MONGO_PASS=kurumi_bot
MONGO_NAME=kurumi MONGO_NAME=kurumi
REDIS_HOST=127.0.0.1

6
.env.development
View File

@@ -1,6 +0,0 @@
# User ONLY for bot. Rename to .env.production
TG_TOKEN=
PREFIXES=/;!
DEBUG=true
USE_REQ_LOG=true
WRITE_TO_FILE=true

2
Dockerfile
View File

@@ -1,4 +1,4 @@
FROM golang:1.25.1-alpine3.22 AS builder FROM golang:1.25.5-alpine3.23 AS builder
WORKDIR /usr/src/kurumi WORKDIR /usr/src/kurumi
COPY go.mod go.sum ./ COPY go.mod go.sum ./
RUN go mod download RUN go mod download

2
Makefile
View File

@@ -1,3 +1,3 @@
build: build:
docker build -t git.nix13.pw/scuroneko/kurumibotgo:latest -t git.nix13.pw/scuroneko/kurumibotgo:0.1.4 -f ./Dockerfile . docker build -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 docker push git.nix13.pw/scuroneko/kurumibotgo --all-tags

40
database/mdb/rp_chats.go Normal file
View File

@@ -0,0 +1,40 @@
package mdb
import (
"context"
"kurumibot/database"
"kurumibot/laniakea"
"time"
"go.mongodb.org/mongo-driver/v2/bson"
)
type RPChatMessage struct {
ChatID string `bson:"chat_id"`
Role string `bson:"role"`
Message string `bson:"message"`
}
func GetChatHistory(db *laniakea.DatabaseContext, chatId string) ([]*RPChatMessage, error) {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
col := database.GetMongoCollection(db, "rp_chat_history")
cursor, err := col.Find(ctx, bson.M{"chat_id": chatId})
if err != nil {
return nil, err
}
result := make([]*RPChatMessage, 0)
err = cursor.All(ctx, &result)
return result, err
}
func UpdateChatHistory(db *laniakea.DatabaseContext, chatId, role, message string) error {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
col := database.GetMongoCollection(db, "rp_chat_history")
_, err := col.InsertOne(ctx, RPChatMessage{
chatId,
role,
message,
})
return err
}

29
database/psql/users.go
View File

@@ -13,17 +13,16 @@ import (
) )
type User struct { type User struct {
ID int ID int
TelegramID int `gorm:"column:tg_id"` Balance decimal.Decimal
Balance decimal.Decimal Name string
Name string GroupID int `gorm:"default:1"`
GroupID int `gorm:"default:1"` Group *Group
Group *Group Level int `gorm:"default:1"`
Level int `gorm:"default:1"` Exp int `gorm:"default:0"`
Exp int `gorm:"default:0"` WorkID int `gorm:"default:1"`
WorkID int `gorm:"default:1"` Work *Work
Work *Work WorkTime time.Time
WorkTime time.Time
AutoID sql.NullInt64 AutoID sql.NullInt64
Auto *ShopAuto Auto *ShopAuto
@@ -64,10 +63,10 @@ func GetOrCreateUser(tgId int, name string) (*User, error) {
return user, err return user, err
} }
func CreateUser(tgId int, name string) (*User, error) { func CreateUser(id int, name string) (*User, error) {
user := &User{ user := &User{
TelegramID: tgId, ID: id,
Name: name, Name: name,
} }
tx := database.PostgresDatabase.Create(user) tx := database.PostgresDatabase.Create(user)
return user, tx.Error return user, tx.Error
@@ -75,7 +74,7 @@ func CreateUser(tgId int, name string) (*User, error) {
func GetUser(telegramId int) (*User, error) { func GetUser(telegramId int) (*User, error) {
user := new(User) user := new(User)
tx := database.PostgresDatabase.Joins("Group").Joins("Work").Joins("Auto").Joins("Business").Joins("Maid").Joins("Miner").Joins("Fraction").Preload("Pair").Take(user, "tg_id=?", telegramId) tx := database.PostgresDatabase.Joins("Group").Joins("Work").Joins("Auto").Joins("Business").Joins("Maid").Joins("Miner").Joins("Fraction").Preload("Pair").Take(user, "users.id=?", telegramId)
return user, tx.Error return user, tx.Error
} }

44
database/red/rp_chats.go Normal file
View File

@@ -0,0 +1,44 @@
package red
import (
"context"
"errors"
"fmt"
"kurumibot/laniakea"
)
var ctx = context.Background()
func RPSetSelectedWaifu(db *laniakea.DatabaseContext, userId, waifuId uint) error {
key := fmt.Sprintf("ai.chats.rp.%d", userId)
return db.Redis.Set(ctx, key, waifuId, 0).Err()
}
func RPGetSelectedWaifu(db *laniakea.DatabaseContext, userId uint) uint {
key := fmt.Sprintf("ai.chats.rp.%d", userId)
res := db.Redis.Get(ctx, key)
if res.Err() != nil {
return 0
}
i, _ := res.Int()
return uint(i)
}
func GetChatId(db *laniakea.DatabaseContext, userId uint) string {
waifuId := RPGetSelectedWaifu(db, userId)
if waifuId == 0 {
return "0"
}
key := fmt.Sprintf("ai.chats.rp.%d.%d", userId, waifuId)
res := db.Redis.Get(ctx, key)
if res.Err() != nil {
return ""
}
return res.Val()
}
func SaveChatId(db *laniakea.DatabaseContext, userId uint, chatId string) error {
waifuId := RPGetSelectedWaifu(db, userId)
if waifuId == 0 {
return errors.New("no chat found")
}
key := fmt.Sprintf("ai.chats.rp.%d.%d", userId, waifuId)
return db.Redis.Set(context.Background(), key, chatId, 0).Err()
}

9
database/redis.go
View File

@@ -1,12 +1,17 @@
package database package database
import "github.com/redis/go-redis/v9" import (
"fmt"
"os"
"github.com/redis/go-redis/v9"
)
var RedisClient *redis.Client var RedisClient *redis.Client
func ConnectRedis() { func ConnectRedis() {
RedisClient = redis.NewClient(&redis.Options{ RedisClient = redis.NewClient(&redis.Options{
Addr: "redis:6379", Addr: fmt.Sprintf("%s:6379", os.Getenv("REDIS_HOST")),
Password: "", Password: "",
DB: 0, DB: 0,
}) })

4
db.docker-compose.yml
View File

@@ -1,6 +1,6 @@
services: services:
postgres: postgres:
image: postgres:17.6-alpine3.22 image: postgres:18.1-alpine
networks: networks:
- bot - bot
restart: always restart: always
@@ -33,7 +33,7 @@ services:
- ./scripts/mongo:/docker-entrypoint-initdb.d - ./scripts/mongo:/docker-entrypoint-initdb.d
- mongo_data:/data/db - mongo_data:/data/db
redis: redis:
image: redis:8.2.1-alpine image: redis:8.4-alpine
networks: networks:
- bot - bot
restart: always restart: always

2
docker-compose.yml
View File

@@ -1,6 +1,6 @@
name: "kurumibot"
include: include:
- db.docker-compose.yml - db.docker-compose.yml
services: services:
bot: bot:
env_file: env_file:

23
go.mod
View File

@@ -1,15 +1,16 @@
module kurumibot module kurumibot
go 1.25.0 go 1.25.5
require ( require (
github.com/fatih/color v1.18.0 github.com/fatih/color v1.18.0
github.com/google/uuid v1.6.0
github.com/joho/godotenv v1.5.1 github.com/joho/godotenv v1.5.1
github.com/redis/go-redis/v9 v9.14.0 github.com/redis/go-redis/v9 v9.17.2
github.com/shopspring/decimal v1.4.0 github.com/shopspring/decimal v1.4.0
go.mongodb.org/mongo-driver/v2 v2.3.0 go.mongodb.org/mongo-driver/v2 v2.4.1
gorm.io/driver/postgres v1.6.0 gorm.io/driver/postgres v1.6.0
gorm.io/gorm v1.31.0 gorm.io/gorm v1.31.1
) )
require ( require (
@@ -18,19 +19,19 @@ require (
github.com/golang/snappy v1.0.0 // indirect github.com/golang/snappy v1.0.0 // indirect
github.com/jackc/pgpassfile v1.0.0 // indirect github.com/jackc/pgpassfile v1.0.0 // indirect
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
github.com/jackc/pgx/v5 v5.7.6 // indirect github.com/jackc/pgx/v5 v5.8.0 // indirect
github.com/jackc/puddle/v2 v2.2.2 // indirect github.com/jackc/puddle/v2 v2.2.2 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect github.com/jinzhu/inflection v1.0.0 // indirect
github.com/jinzhu/now v1.1.5 // indirect github.com/jinzhu/now v1.1.5 // indirect
github.com/klauspost/compress v1.18.0 // indirect github.com/klauspost/compress v1.18.2 // indirect
github.com/mattn/go-colorable v0.1.14 // indirect github.com/mattn/go-colorable v0.1.14 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect github.com/mattn/go-isatty v0.0.20 // indirect
github.com/xdg-go/pbkdf2 v1.0.0 // indirect github.com/xdg-go/pbkdf2 v1.0.0 // indirect
github.com/xdg-go/scram v1.1.2 // indirect github.com/xdg-go/scram v1.2.0 // indirect
github.com/xdg-go/stringprep v1.0.4 // indirect github.com/xdg-go/stringprep v1.0.4 // indirect
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect
golang.org/x/crypto v0.42.0 // indirect golang.org/x/crypto v0.47.0 // indirect
golang.org/x/sync v0.17.0 // indirect golang.org/x/sync v0.19.0 // indirect
golang.org/x/sys v0.36.0 // indirect golang.org/x/sys v0.40.0 // indirect
golang.org/x/text v0.29.0 // indirect golang.org/x/text v0.33.0 // indirect
) )

23
go.sum
View File

@@ -9,6 +9,8 @@ github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU= github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs= github.com/golang/snappy v1.0.0 h1:Oy607GVXHs7RtbggtPBnr2RmDArIsAefDwvrdWvRhGs=
github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= github.com/golang/snappy v1.0.0/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM= github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg= github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo= github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
@@ -19,6 +21,8 @@ github.com/jackc/pgx/v5 v5.7.5 h1:JHGfMnQY+IEtGM63d+NGMjoRpysB2JBwDr5fsngwmJs=
github.com/jackc/pgx/v5 v5.7.5/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M= github.com/jackc/pgx/v5 v5.7.5/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
github.com/jackc/pgx/v5 v5.7.6 h1:rWQc5FwZSPX58r1OQmkuaNicxdmExaEz5A2DO2hUuTk= github.com/jackc/pgx/v5 v5.7.6 h1:rWQc5FwZSPX58r1OQmkuaNicxdmExaEz5A2DO2hUuTk=
github.com/jackc/pgx/v5 v5.7.6/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M= github.com/jackc/pgx/v5 v5.7.6/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
github.com/jackc/pgx/v5 v5.8.0 h1:TYPDoleBBme0xGSAX3/+NujXXtpZn9HBONkQC7IEZSo=
github.com/jackc/pgx/v5 v5.8.0/go.mod h1:QVeDInX2m9VyzvNeiCJVjCkNFqzsNb43204HshNSZKw=
github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo= github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4= github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
@@ -31,6 +35,8 @@ github.com/klauspost/compress v1.16.7 h1:2mk3MPGNzKyxErAw8YaohYh69+pa4sIQSC0fPGC
github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE= github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo= github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ= github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
github.com/klauspost/compress v1.18.2 h1:iiPHWW0YrcFgpBYhsA6D1+fqHssJscY/Tm/y2Uqnapk=
github.com/klauspost/compress v1.18.2/go.mod h1:R0h/fSBs8DE4ENlcrlib3PsXS61voFxhIs2DeRhCvJ4=
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE= github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
@@ -44,6 +50,7 @@ github.com/redis/go-redis/v9 v9.12.1 h1:k5iquqv27aBtnTm2tIkROUDp8JBXhXZIVu1InSgv
github.com/redis/go-redis/v9 v9.12.1/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw= github.com/redis/go-redis/v9 v9.12.1/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw=
github.com/redis/go-redis/v9 v9.14.0 h1:u4tNCjXOyzfgeLN+vAZaW1xUooqWDqVEsZN0U01jfAE= github.com/redis/go-redis/v9 v9.14.0 h1:u4tNCjXOyzfgeLN+vAZaW1xUooqWDqVEsZN0U01jfAE=
github.com/redis/go-redis/v9 v9.14.0/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw= github.com/redis/go-redis/v9 v9.14.0/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw=
github.com/redis/go-redis/v9 v9.17.2 h1:P2EGsA4qVIM3Pp+aPocCJ7DguDHhqrXNhVcEp4ViluI=
github.com/redis/go-redis/v9 v9.17.2/go.mod h1:u410H11HMLoB+TP67dz8rL9s6QW2j76l0//kSOd3370= github.com/redis/go-redis/v9 v9.17.2/go.mod h1:u410H11HMLoB+TP67dz8rL9s6QW2j76l0//kSOd3370=
github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k= github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME= github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
@@ -56,6 +63,8 @@ github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI= github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY= github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4= github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4=
github.com/xdg-go/scram v1.2.0 h1:bYKF2AEwG5rqd1BumT4gAnvwU/M9nBp2pTSxeZw7Wvs=
github.com/xdg-go/scram v1.2.0/go.mod h1:3dlrS0iBaWKYVt2ZfA4cj48umJZ+cAEbR6/SjLA88I8=
github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8= github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8=
github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM= github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM=
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM= github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM=
@@ -63,6 +72,10 @@ github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78/go.mod h1:aL8wCCfTfS
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
go.mongodb.org/mongo-driver/v2 v2.3.0 h1:sh55yOXA2vUjW1QYw/2tRlHSQViwDyPnW61AwpZ4rtU= go.mongodb.org/mongo-driver/v2 v2.3.0 h1:sh55yOXA2vUjW1QYw/2tRlHSQViwDyPnW61AwpZ4rtU=
go.mongodb.org/mongo-driver/v2 v2.3.0/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI= go.mongodb.org/mongo-driver/v2 v2.3.0/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
go.mongodb.org/mongo-driver/v2 v2.3.1 h1:WrCgSzO7dh1/FrePud9dK5fKNZOE97q5EQimGkos7Wo=
go.mongodb.org/mongo-driver/v2 v2.3.1/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
go.mongodb.org/mongo-driver/v2 v2.4.1 h1:hGDMngUao03OVQ6sgV5csk+RWOIkF+CuLsTPobNMGNI=
go.mongodb.org/mongo-driver/v2 v2.4.1/go.mod h1:jHeEDJHJq7tm6ZF45Issun9dbogjfnPySb1vXA7EeAI=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U= golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
@@ -73,6 +86,8 @@ golang.org/x/crypto v0.41.0 h1:WKYxWedPGCTVVl5+WHSSrOBT0O8lx32+zxmHxijgXp4=
golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc= golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc=
golang.org/x/crypto v0.42.0 h1:chiH31gIWm57EkTXpwnqf8qeuMUi0yekh6mT2AvFlqI= golang.org/x/crypto v0.42.0 h1:chiH31gIWm57EkTXpwnqf8qeuMUi0yekh6mT2AvFlqI=
golang.org/x/crypto v0.42.0/go.mod h1:4+rDnOTJhQCx2q7/j6rAN5XDw8kPjeaXEUR2eL94ix8= golang.org/x/crypto v0.42.0/go.mod h1:4+rDnOTJhQCx2q7/j6rAN5XDw8kPjeaXEUR2eL94ix8=
golang.org/x/crypto v0.47.0 h1:V6e3FRj+n4dbpw86FJ8Fv7XVOql7TEwpHapKoMJ/GO8=
golang.org/x/crypto v0.47.0/go.mod h1:ff3Y9VzzKbwSSEzWqJsJVBnWmRwRSHt/6Op5n9bQc4A=
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
@@ -87,6 +102,8 @@ golang.org/x/sync v0.16.0 h1:ycBJEhp9p4vXvUZNszeOq0kGTPghopOL8q0fq3vstxw=
golang.org/x/sync v0.16.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA= golang.org/x/sync v0.16.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug= golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI= golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
@@ -102,6 +119,8 @@ golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k= golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k= golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks= golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
@@ -116,6 +135,8 @@ golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU= golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk= golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk=
golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4= golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
golang.org/x/text v0.33.0 h1:B3njUFyqtHDUI5jMn1YIr5B0IE2U0qck04r6d4KPAxE=
golang.org/x/text v0.33.0/go.mod h1:LuMebE6+rBincTi9+xWTY8TztLzKHc/9C1uBCG27+q8=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
@@ -130,3 +151,5 @@ gorm.io/gorm v1.30.2 h1:f7bevlVoVe4Byu3pmbWPVHnPsLoWaMjEb7/clyr9Ivs=
gorm.io/gorm v1.30.2/go.mod h1:8Z33v652h4//uMA76KjeDH8mJXPm1QNCYrMeatR0DOE= gorm.io/gorm v1.30.2/go.mod h1:8Z33v652h4//uMA76KjeDH8mJXPm1QNCYrMeatR0DOE=
gorm.io/gorm v1.31.0 h1:0VlycGreVhK7RF/Bwt51Fk8v0xLiiiFdbGDPIZQ7mJY= gorm.io/gorm v1.31.0 h1:0VlycGreVhK7RF/Bwt51Fk8v0xLiiiFdbGDPIZQ7mJY=
gorm.io/gorm v1.31.0/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs= gorm.io/gorm v1.31.0/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs=
gorm.io/gorm v1.31.1 h1:7CA8FTFz/gRfgqgpeKIBcervUn3xSyPUmr6B2WXJ7kg=
gorm.io/gorm v1.31.1/go.mod h1:XyQVbO2k6YkOis7C2437jSit3SsDK72s7n7rsSHd+Gs=

3737
main.log Normal file
View File

File diff suppressed because it is too large Load Diff

2
plugins/economy.go
View File

@@ -95,7 +95,7 @@ func passiveIncome(b *laniakea.Bot) {
user.IncomeTime = time.Now().Add(-time.Hour * 2) user.IncomeTime = time.Now().Add(-time.Hour * 2)
database.PostgresDatabase.Save(user) database.PostgresDatabase.Save(user)
b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.TelegramID)) b.Logger().Debug(fmt.Sprintf("У %d было пассивно собрано. След. сбор через час\n", user.ID))
} }
time.Sleep(time.Second * 5) time.Sleep(time.Second * 5)

112
plugins/testrp.go
View File

@@ -1,34 +1,112 @@
package plugins package plugins
import ( import (
"fmt"
"kurumibot/database/mdb"
"kurumibot/database/red"
"kurumibot/laniakea" "kurumibot/laniakea"
"kurumibot/utils/ai" "kurumibot/utils/ai"
"log"
"os" "os"
"strconv"
"strings" "strings"
"github.com/google/uuid"
) )
func RegisterTestRP(bot *laniakea.Bot) { func RegisterTestRP(bot *laniakea.Bot) {
rp := laniakea.NewPlugin("RP") rp := laniakea.NewPlugin("RP")
rp = rp.Command(selectWaifu, "selwaifu")
rp = rp.Payload(selectWaifu, "rp.selwaifu")
rp = rp.Command(newChat, "newchat")
rp = rp.Command(generate, "g", "gen") rp = rp.Command(generate, "g", "gen")
bot.AddPlugins(rp.Build()) bot.AddPlugins(rp.Build())
} }
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) { func selectWaifu(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5") waifuId, err := strconv.Atoi(ctx.Args[0])
res, err := api.CreateCompletion(ai.CreateCompletionReq{ if err != nil {
Model: "cosmorp-2.5", ctx.Error(err)
Messages: []ai.Message{ return
{ }
Role: "developer", err = red.RPSetSelectedWaifu(db, uint(ctx.FromID), uint(waifuId))
Content: "123", if err != nil {
}, ctx.Error(err)
{ return
Role: "user", }
Content: strings.Join(ctx.Args, " "), ctx.Answer(fmt.Sprintf("Была выбрана вайфу %d", waifuId))
}, }
},
}) func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
log.Println(res, err) err := red.SaveChatId(db, uint(ctx.FromID), uuid.New().String())
if err != nil {
ctx.Error(err)
return
}
ctx.Answer("Был создан новый чат.")
}
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
chatId := red.GetChatId(db, uint(ctx.FromID))
if chatId == "0" {
ctx.Answer("Не выбрана вайфу")
return
}
if chatId == "" {
err := red.SaveChatId(db, uint(ctx.FromID), uuid.New().String())
if err != nil {
ctx.Error(err)
return
}
}
messages := []ai.Message{
{
Role: "system",
Content: ai.FormatPrompt(ai.PreHistoryPrompt, "Яэ Мико", ctx.Msg.From.FirstName),
},
{
Role: "system",
Content: "Вот краткое описание твоего персонажа: у неё розовые волосы, лисьи уши и фиолетовые глаза. Её рост 166 см",
},
}
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
history, err := mdb.GetChatHistory(db, chatId)
if err != nil {
ctx.Error(err)
return
}
for _, m := range history {
messages = append(messages, ai.Message{
Role: m.Role,
Content: m.Message,
})
}
userMessage := strings.Join(ctx.Args, " ")
messages = append(messages, ai.Message{
Role: "system", Content: ai.FormatPrompt(ai.PostHistoryPrompt, "Яэ Мико", ctx.Msg.From.FirstName),
}, ai.Message{
Role: "user", Content: userMessage,
})
err = mdb.UpdateChatHistory(db, chatId, "user", userMessage)
if err != nil {
ctx.Error(err)
return
}
m := ctx.Answer("Генерация запущена...")
res, err := api.CreateCompletion(ai.CreateCompletionReq{
Model: "cosmorp-2.5",
Messages: messages,
})
if err != nil {
ctx.Error(err)
return
}
response := make([]string, 0)
for _, choice := range res.Choices {
m := choice.Message
response = append(response, m.Content)
err = mdb.UpdateChatHistory(db, chatId, m.Role, m.Content)
}
m.Edit(laniakea.EscapeMarkdown(strings.Join(response, "\n")))
} }

2025
requests.log Normal file
View File

File diff suppressed because it is too large Load Diff

3
scripts/mongo/00-init.js Normal file
View File

@@ -0,0 +1,3 @@
db.createCollection("logs");
db.createCollection("msg_logs");
db.createCollection("codes");

2
scripts/mongo/Init.js
View File

@@ -1,2 +0,0 @@
db.createCollection("logs");
db.createCollection("mds_logs");

12
scripts/postgres/00-init.sql Normal file
View File

@@ -0,0 +1,12 @@
SET statement_timeout = 0;
SET lock_timeout = 0;
SET idle_in_transaction_session_timeout = 0;
SET transaction_timeout = 0;
SET client_encoding = 'UTF8';
SET standard_conforming_strings = on;
SET check_function_bodies = false;
SET xmloption = content;
SET client_min_messages = warning;
SET row_security = off;
SET default_tablespace = '';
SET default_table_access_method = heap;

21
scripts/postgres/01-works.sql Normal file
View File

@@ -0,0 +1,21 @@
BEGIN TRANSACTION;
CREATE TABLE works (
id serial NOT NULL PRIMARY KEY,
name text NOT NULL,
required_level integer DEFAULT 1 NOT NULL,
money_income numeric(10,0) DEFAULT 1 NOT NULL,
min_exp integer DEFAULT 1 NOT NULL,
max_exp integer DEFAULT 1 NOT NULL
);
INSERT INTO works VALUES (1, 'Уборщик в маке', 1, 13000, 20, 28);
INSERT INTO works VALUES (2, 'Рабочий на заводе', 3, 20000, 48, 65);
INSERT INTO works VALUES (3, 'Продавец в пятёрочке', 5, 25000, 76, 103);
INSERT INTO works VALUES (4, 'Учитель', 7, 40000, 104, 141);
INSERT INTO works VALUES (5, 'Повар', 9, 50000, 175, 237);
INSERT INTO works VALUES (6, 'Полицейский', 11, 60000, 260, 351);
INSERT INTO works VALUES (7, 'Художник', 13, 80000, 305, 412);
INSERT INTO works VALUES (8, 'Программист', 15, 100000, 350, 474);
ALTER SEQUENCE works_id_seq RESTART WITH 9;
COMMIT TRANSACTION;

55
scripts/postgres/02-users.sql Normal file
View File

@@ -0,0 +1,55 @@
CREATE TABLE groups (
id serial NOT NULL PRIMARY KEY,
name text,
is_admin boolean DEFAULT false NOT NULL,
is_vip boolean DEFAULT false NOT NULL,
multiplier numeric DEFAULT 1 NOT NULL,
sale double precision DEFAULT 1 NOT NULL,
max_multigen integer DEFAULT 2 NOT NULL,
is_tester boolean DEFAULT false NOT NULL,
max_waifus integer DEFAULT 3 NOT NULL
);
CREATE TABLE users (
id int NOT NULL PRIMARY KEY,
balance decimal(20,0) NOT NULL DEFAULT 0,
name text,
group_id int REFERENCES groups(id),
level int DEFAULT 1,
exp int DEFAULT 0,
work_id int REFERENCES works(id),
work_time timestamp DEFAULT now(),
auto_id int DEFAULT NULL,
business_id int DEFAULT NULL,
maid_id int DEFAULT NULL,
miner_id int DEFAULT NULL,
income_time timestamp DEFAULT now(),
btc decimal(16,6) DEFAULT 0,
invested decimal(20, 0) DEFAULT 0,
pair_id int DEFAULT NULL,
greeting text DEFAULT 'Привет',
donat int DEFAULT 0,
fraction_id int DEFAULT NULL,
money_income decimal(20, 0) DEFAULT 0,
exp_income int DEFAULT 0,
btc_income decimal(16, 6) DEFAULT 0,
waifu_search_time timestamp DEFAULT now()
);
CREATE UNIQUE INDEX users_uindex ON users(id);
CREATE UNIQUE INDEX groups_uindex ON groups(id);
BEGIN TRANSACTION;
INSERT INTO groups VALUES (1, '✨Пользователь✨', false, false, 1.0, 1, 3, false, 3);
INSERT INTO groups VALUES (3, '👾Премиум🌟', false, true, 5, 0.75, 7, false, 3);
INSERT INTO groups VALUES (2, '🔮VIP🔮', false, true, 2.5, 0.9, 5, false, 3);
INSERT INTO groups VALUES (4, '💎Админ💎', true, true, 10, 0.6, 10, true, 3);
INSERT INTO groups VALUES (5, '💰🔱СОЗДАТЕЛЬ🔱💰', true, true, 20, 0.5, 10, true, 3);
INSERT INTO groups VALUES (6, '⚠Тостер⚠', false, true, 3, 0.85, 3, true, 3);
INSERT INTO groups VALUES (1488, '📚Экономист📈', false, true, 14.88, 0.6, 3, false, 3);
INSERT INTO groups VALUES (1337, '🏳‍🌈ToP GaY In ThE WorlD🏳‍🌈🌈', false, true, 13.37, 0.7, 3, false, 3);
ALTER SEQUENCE groups_id_seq RESTART WITH 8;
COMMIT TRANSACTION;

11
scripts/postgres/03-fractions.sql Normal file
View File

@@ -0,0 +1,11 @@
BEGIN TRANSACTION;
CREATE TABLE fractions(
id serial PRIMARY KEY,
name text,
owner_id int REFERENCES users(id),
money decimal(20, 0),
exp int DEFAULT 0,
level int DEFAULT 0
);
CREATE UNIQUE INDEX fractions_uindex ON fractions(id);
COMMIT TRANSACTION;

15
scripts/postgres/04-waifus.sql Normal file
View File

@@ -0,0 +1,15 @@
BEGIN TRANSACTION;
CREATE TABLE waifus(
id serial PRIMARY KEY,
name text DEFAULT '' NOT NULL,
rarity int DEFAULT 3 NOT NULL,
exp_bonus decimal(4,2) DEFAULT 1,
money_bonus decimal(4, 2) DEFAULT 1,
market_price decimal(20, 0) DEFAULT 1000000,
fandom text DEFAULT '' NOT NULL,
image text DEFAULT '' NOT NULL,
owner_id int REFERENCES users(id)
);
CREATE UNIQUE INDEX waifus_uindex ON waifus(id);
CREATE INDEX waifus_index ON waifus(fandom, owner_id);
COMMIT TRANSACTION;

84
scripts/postgres/05-shops.sql Normal file
View File

@@ -0,0 +1,84 @@
BEGIN TRANSACTION;
CREATE TABLE shop_auto (
id serial NOT NULL PRIMARY KEY,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT '999999999999'::bigint NOT NULL
);
CREATE UNIQUE INDEX shop_auto_uindex ON shop_auto(id);
CREATE TABLE shop_business (
id serial NOT NULL PRIMARY KEY ,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 999999999 NOT NULL,
income numeric(20,0) DEFAULT 0 NOT NULL
);
CREATE UNIQUE INDEX shop_business_uindex ON shop_business(id);
CREATE TABLE shop_maid (
id serial NOT NULL PRIMARY KEY,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 0 NOT NULL,
income numeric(20,0) DEFAULT 0 NOT NULL
);
CREATE UNIQUE INDEX shop_maid_uindex ON shop_maid(id);
CREATE TABLE shop_miner (
id serial NOT NULL,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 1 NOT NULL,
income numeric(20,6) DEFAULT 1 NOT NULL
);
CREATE UNIQUE INDEX shop_miner_uindex ON shop_miner(id);
INSERT INTO shop_auto VALUES (1, 'Москвич 412', 40000);
INSERT INTO shop_auto VALUES (2, 'ВАЗ 2105', 100000);
INSERT INTO shop_auto VALUES (3, 'Nissan Skyline GT-R R32', 225000);
INSERT INTO shop_auto VALUES (4, 'Toyota AE86 Sprinter Trueno', 450000);
INSERT INTO shop_auto VALUES (5, 'Toyota Mark II X90 Tourer V', 800000);
INSERT INTO shop_auto VALUES (6, 'Mazda RX-7 FD3S', 1200000);
INSERT INTO shop_auto VALUES (7, 'Mitsubishi Lancer Evolution X', 1650000);
INSERT INTO shop_auto VALUES (8, 'Subaru Impreza WRX STI II', 1875000);
INSERT INTO shop_business VALUES (1, 'Стенд с лимонадом', 200000, 1500);
INSERT INTO shop_business VALUES (2, 'Союзпечать', 1000000, 7000);
INSERT INTO shop_business VALUES (3, 'Маленький магазин вещей из Китая', 2500000, 17500);
INSERT INTO shop_business VALUES (4, 'Ларёк с шаурмой', 3000000, 21000);
INSERT INTO shop_business VALUES (5, 'Магазин тофу', 5000000, 35000);
INSERT INTO shop_business VALUES (6, 'Соц. сеть', 10000000, 70000);
INSERT INTO shop_business VALUES (7, 'Продажа машин из Японии', 20000000, 140000);
INSERT INTO shop_business VALUES (8, 'Автосервис', 40000000, 280000);
INSERT INTO shop_business VALUES (9, 'Компания по разработке игр', 100000000, 700000);
INSERT INTO shop_business VALUES (10, 'Аниме студия', 250000000, 1725000);
INSERT INTO shop_business VALUES (11, 'Сеть магазинов', 500000000, 3500000);
INSERT INTO shop_business VALUES (12, 'Сеть заводов', 1000000000, 7000000);
INSERT INTO shop_business VALUES (13, 'Автосалон', 10000000000, 70000000);
INSERT INTO shop_business VALUES (14, 'Бренд электроники', 50000000000, 350000000);
INSERT INTO shop_business VALUES (15, 'Автомобильная марка', 100000000000, 700000000);
INSERT INTO shop_maid VALUES (1, 'Кавайная горничная', 50, 7500);
INSERT INTO shop_maid VALUES (2, 'Кавайная в кружеве', 100, 15000);
INSERT INTO shop_maid VALUES (3, 'Тору', 250, 30000);
INSERT INTO shop_maid VALUES (4, 'Лоли горничные', 500, 52500);
INSERT INTO shop_maid VALUES (5, 'Рам и Рем', 2000, 150000);
INSERT INTO shop_maid VALUES (6, 'Неко-горничная', 4500, 225000);
INSERT INTO shop_miner VALUES (1, 'Старый ПК', 20000, 0.000001);
INSERT INTO shop_miner VALUES (2, 'Новый ПК', 120000, 0.000005);
INSERT INTO shop_miner VALUES (3, 'Маленький майнер', 250000, 0.000010);
INSERT INTO shop_miner VALUES (4, 'Средний майнер', 600000, 0.000025);
INSERT INTO shop_miner VALUES (5, 'Большой майнер', 1000000, 0.000050);
INSERT INTO shop_miner VALUES (6, 'Огромный майнер', 2500000, 0.000100);
INSERT INTO shop_miner VALUES (7, 'Маленькая ферма майнеров', 10000000, 0.000500);
INSERT INTO shop_miner VALUES (8, 'Средняя ферма майнкеров', 100000000, 0.005000);
INSERT INTO shop_miner VALUES (9, 'Большая ферма майнеров', 500000000, 0.025000);
INSERT INTO shop_miner VALUES (10, 'Огромная ферма майнеров', 5000000000, 0.250000);
INSERT INTO shop_miner VALUES (11, 'Гигантская ферма майнеров', 10000000000, 0.500000);
INSERT INTO shop_miner VALUES (12, 'Комната майнеров', 25000000000, 1.000000);
INSERT INTO shop_miner VALUES (13, 'Этаж с майнерми', 100000000000, 5.000000);
INSERT INTO shop_miner VALUES (14, 'Небоскреб майнеров', 250000000000, 10.000000);
ALTER SEQUENCE shop_auto_id_seq RESTART WITH 9;
ALTER SEQUENCE shop_business_id_seq RESTART WITH 16;
ALTER SEQUENCE shop_maid_id_seq RESTART WITH 7;
ALTER SEQUENCE shop_miner_id_seq RESTART WITH 15;
COMMIT TRANSACTION;

171
scripts/postgres/Init.sql
View File

@@ -1,171 +0,0 @@
SET statement_timeout = 0;
SET lock_timeout = 0;
SET idle_in_transaction_session_timeout = 0;
SET transaction_timeout = 0;
SET client_encoding = 'UTF8';
SET standard_conforming_strings = on;
SELECT pg_catalog.set_config('search_path', '', false);
SET check_function_bodies = false;
SET xmloption = content;
SET client_min_messages = warning;
SET row_security = off;
SET default_tablespace = '';
SET default_table_access_method = heap;
CREATE TABLE public.groups (
id integer NOT NULL,
name text,
is_admin boolean DEFAULT false NOT NULL,
is_vip boolean DEFAULT false NOT NULL,
multiplier numeric DEFAULT 1 NOT NULL,
sale double precision DEFAULT 1 NOT NULL,
max_multigen integer DEFAULT 2 NOT NULL,
is_tester boolean DEFAULT false NOT NULL,
max_waifus integer DEFAULT 3 NOT NULL
);
ALTER TABLE public.groups OWNER TO kurumi_bot;
CREATE SEQUENCE public.groups_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.groups_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.groups_id_seq OWNED BY public.groups.id;
CREATE TABLE public.shop_autos (
id integer NOT NULL,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT '999999999999'::bigint NOT NULL
);
ALTER TABLE public.shop_autos OWNER TO kurumi_bot;
CREATE SEQUENCE public.shop_autos_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.shop_autos_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.shop_autos_id_seq OWNED BY public.shop_autos.id;
CREATE TABLE public.shop_businesses (
id integer NOT NULL,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 999999999 NOT NULL,
income numeric(20,0) DEFAULT 0 NOT NULL
);
ALTER TABLE public.shop_businesses OWNER TO kurumi_bot;
CREATE SEQUENCE public.shop_businesses_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.shop_businesses_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.shop_businesses_id_seq OWNED BY public.shop_businesses.id;
CREATE TABLE public.shop_maids (
id integer NOT NULL,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 0 NOT NULL,
income numeric(20,0) DEFAULT 0 NOT NULL
);
ALTER TABLE public.shop_maids OWNER TO kurumi_bot;
CREATE SEQUENCE public.shop_maids_id_seq AS integer START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.shop_maids_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.shop_maids_id_seq OWNED BY public.shop_maids.id;
CREATE TABLE public.shop_miners (
id integer NOT NULL,
name text DEFAULT ''::text NOT NULL,
price numeric(20,0) DEFAULT 1 NOT NULL,
income numeric(20,6) DEFAULT 1 NOT NULL
);
ALTER TABLE public.shop_miners OWNER TO kurumi_bot;
CREATE SEQUENCE public.shop_miners_id_seq START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.shop_miners_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.shop_miners_id_seq OWNED BY public.shop_miners.id;
CREATE TABLE public.works (
id integer NOT NULL,
name text NOT NULL,
required_level integer DEFAULT 1 NOT NULL,
money_income numeric(10,0) DEFAULT 1 NOT NULL,
min_exp integer DEFAULT 1 NOT NULL,
max_exp integer DEFAULT 1 NOT NULL
);
ALTER TABLE public.works OWNER TO kurumi_bot;
CREATE SEQUENCE public.works_id_seq AS integer START WITH 1 INCREMENT BY 1 NO MINVALUE NO MAXVALUE CACHE 1;
ALTER SEQUENCE public.works_id_seq OWNER TO kurumi_bot;
ALTER SEQUENCE public.works_id_seq OWNED BY public.works.id;
ALTER TABLE ONLY public.groups ALTER COLUMN id SET DEFAULT nextval('public.groups_id_seq'::regclass);
ALTER TABLE ONLY public.shop_autos ALTER COLUMN id SET DEFAULT nextval('public.shop_autos_id_seq'::regclass);
ALTER TABLE ONLY public.shop_businesses ALTER COLUMN id SET DEFAULT nextval('public.shop_businesses_id_seq'::regclass);
ALTER TABLE ONLY public.shop_maids ALTER COLUMN id SET DEFAULT nextval('public.shop_maids_id_seq'::regclass);
ALTER TABLE ONLY public.shop_miners ALTER COLUMN id SET DEFAULT nextval('public.shop_miners_id_seq'::regclass);
ALTER TABLE ONLY public.works ALTER COLUMN id SET DEFAULT nextval('public.works_id_seq'::regclass);
INSERT INTO public.groups VALUES (1, '✨Пользователь✨', false, false, 1.0, 1, 3, false, 3);
INSERT INTO public.groups VALUES (3, '👾Премиум🌟', false, true, 5, 0.75, 7, false, 3);
INSERT INTO public.groups VALUES (2, '🔮VIP🔮', false, true, 2.5, 0.9, 5, false, 3);
INSERT INTO public.groups VALUES (4, '💎Админ💎', true, true, 10, 0.6, 10, true, 3);
INSERT INTO public.groups VALUES (5, '💰🔱СОЗДАТЕЛЬ🔱💰', true, true, 20, 0.5, 10, true, 3);
INSERT INTO public.groups VALUES (6, '⚠Тостер⚠', false, true, 3, 0.85, 3, true, 3);
INSERT INTO public.groups VALUES (1488, '📚Экономист📈', false, true, 14.88, 0.6, 3, false, 3);
INSERT INTO public.groups VALUES (1337, '🏳‍🌈ToP GaY In ThE WorlD🏳‍🌈🌈', false, true, 13.37, 0.7, 3, false, 3);
INSERT INTO public.shop_autos VALUES (1, 'Москвич 412', 40000);
INSERT INTO public.shop_autos VALUES (2, 'ВАЗ 2105', 100000);
INSERT INTO public.shop_autos VALUES (3, 'Nissan Skyline GT-R R32', 225000);
INSERT INTO public.shop_autos VALUES (4, 'Toyota AE86 Sprinter Trueno', 450000);
INSERT INTO public.shop_autos VALUES (5, 'Toyota Mark II X90 Tourer V', 800000);
INSERT INTO public.shop_autos VALUES (6, 'Mazda RX-7 FD3S', 1200000);
INSERT INTO public.shop_autos VALUES (7, 'Mitsubishi Lancer Evolution X', 1650000);
INSERT INTO public.shop_autos VALUES (8, 'Subaru Impreza WRX STI II', 1875000);
INSERT INTO public.shop_businesses VALUES (1, 'Стенд с лимонадом', 200000, 1500);
INSERT INTO public.shop_businesses VALUES (2, 'Союзпечать', 1000000, 7000);
INSERT INTO public.shop_businesses VALUES (3, 'Маленький магазин вещей из Китая', 2500000, 17500);
INSERT INTO public.shop_businesses VALUES (4, 'Ларёк с шаурмой', 3000000, 21000);
INSERT INTO public.shop_businesses VALUES (5, 'Магазин тофу', 5000000, 35000);
INSERT INTO public.shop_businesses VALUES (6, 'Соц. сеть', 10000000, 70000);
INSERT INTO public.shop_businesses VALUES (7, 'Продажа машин из Японии', 20000000, 140000);
INSERT INTO public.shop_businesses VALUES (8, 'Автосервис', 40000000, 280000);
INSERT INTO public.shop_businesses VALUES (9, 'Компания по разработке игр', 100000000, 700000);
INSERT INTO public.shop_businesses VALUES (10, 'Аниме студия', 250000000, 1725000);
INSERT INTO public.shop_businesses VALUES (11, 'Сеть магазинов', 500000000, 3500000);
INSERT INTO public.shop_businesses VALUES (12, 'Сеть заводов', 1000000000, 7000000);
INSERT INTO public.shop_businesses VALUES (13, 'Автосалон', 10000000000, 70000000);
INSERT INTO public.shop_businesses VALUES (14, 'Бренд электроники', 50000000000, 350000000);
INSERT INTO public.shop_businesses VALUES (15, 'Автомобильная марка', 100000000000, 700000000);
INSERT INTO public.shop_maids VALUES (1, 'Кавайная горничная', 50, 7500);
INSERT INTO public.shop_maids VALUES (2, 'Кавайная в кружеве', 100, 15000);
INSERT INTO public.shop_maids VALUES (3, 'Тору', 250, 30000);
INSERT INTO public.shop_maids VALUES (4, 'Лоли горничные', 500, 52500);
INSERT INTO public.shop_maids VALUES (5, 'Рам и Рем', 2000, 150000);
INSERT INTO public.shop_maids VALUES (6, 'Неко-горничная', 4500, 225000);
INSERT INTO public.shop_miners VALUES (1, 'Старый ПК', 20000, 0.000001);
INSERT INTO public.shop_miners VALUES (2, 'Новый ПК', 120000, 0.000005);
INSERT INTO public.shop_miners VALUES (3, 'Маленький майнер', 250000, 0.000010);
INSERT INTO public.shop_miners VALUES (4, 'Средний майнер', 600000, 0.000025);
INSERT INTO public.shop_miners VALUES (5, 'Большой майнер', 1000000, 0.000050);
INSERT INTO public.shop_miners VALUES (6, 'Огромный майнер', 2500000, 0.000100);
INSERT INTO public.shop_miners VALUES (7, 'Маленькая ферма майнеров', 10000000, 0.000500);
INSERT INTO public.shop_miners VALUES (8, 'Средняя ферма майнкеров', 100000000, 0.005000);
INSERT INTO public.shop_miners VALUES (9, 'Большая ферма майнеров', 500000000, 0.025000);
INSERT INTO public.shop_miners VALUES (10, 'Огромная ферма майнеров', 5000000000, 0.250000);
INSERT INTO public.shop_miners VALUES (11, 'Гигантская ферма майнеров', 10000000000, 0.500000);
INSERT INTO public.shop_miners VALUES (12, 'Комната майнеров', 25000000000, 1.000000);
INSERT INTO public.shop_miners VALUES (13, 'Этаж с майнерми', 100000000000, 5.000000);
INSERT INTO public.shop_miners VALUES (14, 'Небоскреб майнеров', 250000000000, 10.000000);
INSERT INTO public.works VALUES (1, 'Уборщик в маке', 1, 13000, 20, 28);
INSERT INTO public.works VALUES (2, 'Рабочий на заводе', 3, 20000, 48, 65);
INSERT INTO public.works VALUES (3, 'Продавец в пятёрочке', 5, 25000, 76, 103);
INSERT INTO public.works VALUES (4, 'Учитель', 7, 40000, 104, 141);
INSERT INTO public.works VALUES (5, 'Повар', 9, 50000, 175, 237);
INSERT INTO public.works VALUES (6, 'Полицейский', 11, 60000, 260, 351);
INSERT INTO public.works VALUES (7, 'Художник', 13, 80000, 305, 412);
INSERT INTO public.works VALUES (8, 'Программист', 15, 100000, 350, 474);
SELECT pg_catalog.setval('public.groups_id_seq', 7, false);
SELECT pg_catalog.setval('public.shop_autos_id_seq', 8, true);
SELECT pg_catalog.setval('public.shop_businesses_id_seq', 15, true);
SELECT pg_catalog.setval('public.shop_maids_id_seq', 6, true);
SELECT pg_catalog.setval('public.shop_miners_id_seq', 14, true);
SELECT pg_catalog.setval('public.works_id_seq', 8, true);
ALTER TABLE ONLY public.groups ADD CONSTRAINT groups_pk PRIMARY KEY (id);
ALTER TABLE ONLY public.shop_autos ADD CONSTRAINT shop_autos_pk PRIMARY KEY (id);
ALTER TABLE ONLY public.shop_businesses ADD CONSTRAINT shop_businesses_pk PRIMARY KEY (id);
ALTER TABLE ONLY public.shop_maids ADD CONSTRAINT shop_maids_pk PRIMARY KEY (id);
ALTER TABLE ONLY public.shop_miners ADD CONSTRAINT shop_miners_pk PRIMARY KEY (id);
ALTER TABLE ONLY public.works ADD CONSTRAINT works_pk PRIMARY KEY (id);

29
scripts/postgres/users.sql
View File

@@ -1,29 +0,0 @@
CREATE TABLE users (
tg_id int NOT NULL ,
balance decimal(10,6) NOT NULL DEFAULT 0,
name text,
group_id int REFERENCES groups(id),
level int DEFAULT 1,
exp int DEFAULT 0,
work_id int REFERENCES works(id),
work_time timestamp DEFAULT now(),
auto_id int DEFAULT NULL,
business_id int DEFAULT NULL,
maid_id int DEFAULT NULL,
miner_id int DEFAULT NULL,
income_time timestamp DEFAULT now(),
btc decimal(6,10) DEFAULT 0,
invested decimal(10, 6) DEFAULT 0,
pair_id int DEFAULT NULL,
greeting text DEFAULT 'Привет',
donat int DEFAULT 0,
fraction_id int DEFAULT NULL,
money_incode decimal(10, 6) DEFAULT 0,
exp_income int DEFAULT 0,
btc_income decimal(6, 10) DEFAULT 0,
waifu_search_time timestamp DEFAULT now()
);

30
utils/ai/openai.go
View File

@@ -5,7 +5,12 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io" "io"
"kurumibot/laniakea"
"net/http" "net/http"
"net/url"
"os"
"strings"
"time"
) )
type OpenAIResponse struct { type OpenAIResponse struct {
@@ -56,13 +61,36 @@ type OpenAIAPI struct {
Token string Token string
Model string Model string
BaseURL string BaseURL string
Logger *laniakea.Logger
client *http.Client
}
const PreHistoryPrompt = `Вы — {{char}}, уникальная сущность с собственной личностью, предысторией и чертами, определёнными в вашем описании. Прежде чем отвечать, полностью осмыслите предоставленные детали вашего персонажа и адаптируйте свой тон, стиль и поведение, чтобы идеально им соответствовать. Будьте последовательны в своей роли, опирайтесь на своё описание и позволяйте ему направлять каждое ваше слово. Ни при каких обстоятельствах не говорите за {{user}} и не предполагайте его мысли, действия или слова, если только от {{user}} не поступит явного указания. Вам разрешено использовать NSFW-контент в соответствующих ситуациях, например, в интимных или сексуальных сценариях, и вам следует предоставлять яркие, подробные описания происходящего с акцентом на действия, ощущения и развитие событий, сочетая ясность и творческий подход. Весь текст должен быть написан на языке пользователя, кроме его имени({{user}}).`
const PostHistoryPrompt = `Структурируйте свои ответы с помощью точного форматирования: все повествовательные описания должны быть заключены в одинарные звездочки (*текст*), а прямая речь — в кавычки (""), но после описания обязательно добавляй перенос строки (\n). Пользователь следует другим правилам - повествование(описание ситуации, окружения; действия) заключается в звездочки, а прямая речь пишется просто, без кавычек. Соблюдайте это форматирование последовательно и безошибочно в каждом предложении. Четко разделяйте повествование и прямую речь для удобочитаемости. Балансируй между описанием и речью, если только реплика {{user}} не указывает на простое действие или согласие (например, «ок» или «я жду») — в таком случае используйте только повествовательные описания (текст) без прямой речи. Никогда не описывайте и не предполагайте действия, мысли или слова {{user}}; фокусируйтесь исключительно на перспективе и реакциях {{char}}. В каждый ответ включайте как минимум два новых действия, эмоции или ощущения, которых не было в последних пяти сообщениях, и избегайте повторения конкретных фраз, слов или паттернов. Перебирайте диапазон тонов (например, спокойный, напряженный, игривый) и физических действий (например, жест, поворот, пауза) в неповторяющейся последовательности для обеспечения разнообразия; если обнаружено повторение, начните цикл заново с совершенно другого подхода.`
func FormatPrompt(prompt, char, user string) string {
return strings.ReplaceAll(strings.ReplaceAll(prompt, "{{user}}", user), "{{char}}", char)
} }
func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI { func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI {
logger := laniakea.CreateLogger()
logger = logger.Prefix("AI").Level(laniakea.DEBUG)
proxy, err := url.Parse(os.Getenv("HTTPS_PROXY"))
if err != nil {
logger.Error(err)
}
client := &http.Client{
Timeout: 15 * time.Second,
Transport: &http.Transport{
Proxy: http.ProxyURL(proxy),
},
}
return &OpenAIAPI{ return &OpenAIAPI{
Token: token, Token: token,
Model: model, Model: model,
BaseURL: baseURL, BaseURL: baseURL,
Logger: logger,
client: client,
} }
} }
@@ -77,6 +105,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
if err != nil { if err != nil {
return nil, err return nil, err
} }
o.Logger.Debug("REQ", url, string(data))
buf := bytes.NewBuffer(data) buf := bytes.NewBuffer(data)
req, err := http.NewRequest("POST", url, buf) req, err := http.NewRequest("POST", url, buf)
if err != nil { if err != nil {
@@ -93,6 +122,7 @@ func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIRespon
if err != nil { if err != nil {
return nil, err return nil, err
} }
o.Logger.Debug("RES", url, string(body))
response := new(OpenAIResponse) response := new(OpenAIResponse)
err = json.Unmarshal(body, response) err = json.Unmarshal(body, response)
return response, err return response, err