some fixes
This commit is contained in:
@@ -22,6 +22,8 @@ func RegisterRP(bot *laniakea.Bot) {
|
||||
rp = rp.Command(rpPresetsList, "rpplist")
|
||||
rp = rp.Command(rpPresetSet, "rppset")
|
||||
rp = rp.Command(newChat, "newchat")
|
||||
rp = rp.Command(rpUserPromptGet, "rpuserpget")
|
||||
rp = rp.Command(rpUserPromptSet, "rpuserpset")
|
||||
rp = rp.Command(generate, "g", "gen", "г")
|
||||
|
||||
bot.AddPlugins(rp.Build())
|
||||
@@ -50,7 +52,10 @@ func rpPresetsList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
}
|
||||
out := make([]string, len(presets))
|
||||
for i, preset := range presets {
|
||||
out[i] = fmt.Sprintf("%s) *%s*\n%s", preset.ID, preset.Name, preset.Description)
|
||||
out[i] = fmt.Sprintf(
|
||||
"%s) *%s*\n%s",
|
||||
preset.ID, laniakea.EscapeMarkdown(preset.Name), preset.Description,
|
||||
)
|
||||
}
|
||||
ctx.Answer(strings.Join(out, "\n"))
|
||||
}
|
||||
@@ -102,6 +107,38 @@ func newChat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
}
|
||||
ctx.Answer("Был создан новый чат.")
|
||||
}
|
||||
func rpUserPromptGet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
rep := psql.NewRPRepository(db.PostgresSQL)
|
||||
user, err := rep.GetOrCreateUser(int64(ctx.FromID))
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
if user.UserPrompt == "" {
|
||||
ctx.Answer("У тебя нет описания")
|
||||
return
|
||||
}
|
||||
ctx.Answerf("Вот твое РП описание пользователя\n%s", user.UserPrompt)
|
||||
}
|
||||
func rpUserPromptSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
if len(ctx.Args) == 0 || ctx.Args[0] == "" {
|
||||
return
|
||||
}
|
||||
prompt := strings.Join(ctx.Args[1:], " ")
|
||||
rep := psql.NewRPRepository(db.PostgresSQL)
|
||||
user, err := rep.GetOrCreateUser(int64(ctx.FromID))
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
user.UserPrompt = prompt
|
||||
err = rep.UpdateUser(user)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
return
|
||||
}
|
||||
ctx.Answer("Описание пользователя было обновлено.")
|
||||
}
|
||||
|
||||
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
||||
waifuId := red.RPGetSelectedWaifu(db, ctx.FromID)
|
||||
|
||||
Reference in New Issue
Block a user