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

some button styles

Browse Source
This commit is contained in:
ScuroNeko
2026-02-12 14:18:41 +03:00
parent ee6729e12a
commit a45c5735ad
1 changed files with 24 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
plugins/rp.go
View File

@@ -166,7 +166,7 @@ func rpWaifuList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
kb.AddCallbackButton(waifu.Name, "rp.waifu_set", waifu.ID) kb.AddCallbackButton(waifu.Name, "rp.waifu_set", waifu.ID)
} }
kb.AddLine() kb.AddLine()
kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleDanger, "rp.info")
ctx.EditCallback(strings.Join(out, "\n"), kb) ctx.EditCallback(strings.Join(out, "\n"), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -189,7 +189,8 @@ func rpWaifuSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err) ctx.Error(err)
} }
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("На главную", "rp.info") kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleSuccess, "rp.info")
ctx.EditCallbackf("Была выбрана вайфу %s", kb, waifu.Name) ctx.EditCallbackf("Была выбрана вайфу %s", kb, waifu.Name)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -211,7 +212,7 @@ func rpPresetsList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
kb.AddCallbackButton(preset.Name, "rp.preset_set", preset.ID) kb.AddCallbackButton(preset.Name, "rp.preset_set", preset.ID)
} }
kb.AddLine() kb.AddLine()
kb.AddCallbackButton("Назад", "rp.info") kb.AddCallbackButtonStyle("Назад", laniakea.ButtonStyleDanger, "rp.info")
ctx.EditCallback(strings.Join(out, "\n"), kb) ctx.EditCallback(strings.Join(out, "\n"), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -236,7 +237,8 @@ func rpPresetSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
return return
} }
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("На главную", "rp.info") kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStylePrimary, "rp.info")
ctx.EditCallbackf("Был выбран пресет %s", kb, preset.Name) ctx.EditCallbackf("Был выбран пресет %s", kb, preset.Name)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -256,7 +258,7 @@ func rpModelList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
kb.AddCallbackButton(model.Name, "rp.model_set", model.ID) kb.AddCallbackButton(model.Name, "rp.model_set", model.ID)
} }
kb.AddLine() kb.AddLine()
kb.AddCallbackButton("Назад", "rp.info") kb.AddCallbackButtonStyle("Назад", laniakea.ButtonStyleDanger, "rp.info")
ctx.EditCallback(strings.Join(out, "\n"), kb) ctx.EditCallback(strings.Join(out, "\n"), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -279,7 +281,8 @@ func rpModelSet(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.Error(err) ctx.Error(err)
return return
} }
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("На главную", "rp.info") kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleSuccess, "rp.info")
ctx.EditCallback(fmt.Sprintf("Была установлена модель %s", model.Name), kb) ctx.EditCallback(fmt.Sprintf("Была установлена модель %s", model.Name), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -296,7 +299,7 @@ func rpScenarioList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
for i, scenario := range scenarios { for i, scenario := range scenarios {
out[i] = fmt.Sprintf("*%s* - %s\n", scenario.Name, scenario.Description) out[i] = fmt.Sprintf("*%s* - %s\n", scenario.Name, scenario.Description)
} }
kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleDanger, "rp.info")
ctx.EditCallback("Список сценариев\n"+strings.Join(out, "\n"), kb) ctx.EditCallback("Список сценариев\n"+strings.Join(out, "\n"), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -313,7 +316,7 @@ func rpSettingList(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
out[i] = fmt.Sprintf("*%s* - %s\n", setting.Name, setting.Description) out[i] = fmt.Sprintf("*%s* - %s\n", setting.Name, setting.Description)
} }
kb := laniakea.NewInlineKeyboard(1) kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleSuccess, "rp.info")
ctx.EditCallback(strings.Join(out, "\n"), kb) ctx.EditCallback(strings.Join(out, "\n"), kb)
ctx.AnswerCbQuery() ctx.AnswerCbQuery()
} }
@@ -338,8 +341,8 @@ func chatStat(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
tokens := redisRpRep.GetChatTokens(ctx.FromID, waifuId) tokens := redisRpRep.GetChatTokens(ctx.FromID, waifuId)
kb := laniakea.NewInlineKeyboard(1) kb := laniakea.NewInlineKeyboard(1)
kb = kb.AddCallbackButton("Сжать чат", "rp.compress_chat") kb.AddCallbackButtonStyle("Сжать чат", laniakea.ButtonStyleSuccess, "rp.compress_chat")
kb = kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleDanger, "rp.info")
out := []string{ out := []string{
"Статистика чата", "Статистика чата",
fmt.Sprintf("*ID*: `%s`", chatId), fmt.Sprintf("*ID*: `%s`", chatId),
@@ -427,8 +430,10 @@ func newChatStage2(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
for _, scenario := range scenarios { for _, scenario := range scenarios {
isSelected := selectedScenariosIds.Index(scenario.ID) >= 0 isSelected := selectedScenariosIds.Index(scenario.ID) >= 0
prefix := "" prefix := ""
style := laniakea.ButtonStylePrimary
if isSelected { if isSelected {
prefix = "✅" prefix = "✅"
style = laniakea.ButtonStyleSuccess
} }
out = append(out, fmt.Sprintf("%s*%s* - %s", prefix, scenario.Name, scenario.Description)) out = append(out, fmt.Sprintf("%s*%s* - %s", prefix, scenario.Name, scenario.Description))
if isSelected { if isSelected {
@@ -436,8 +441,9 @@ func newChatStage2(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
} else { } else {
scenariosIds = selectedScenariosIds.Push(scenario.ID) scenariosIds = selectedScenariosIds.Push(scenario.ID)
} }
kb.AddCallbackButton(
fmt.Sprintf("%s%s", prefix, scenario.Name), kb.AddCallbackButtonStyle(
fmt.Sprintf("%s%s", prefix, scenario.Name), style,
"rp.new_chat_s2", utils.AppendToInt(settingId, scenariosIds)..., "rp.new_chat_s2", utils.AppendToInt(settingId, scenariosIds)...,
) )
} }
@@ -782,7 +788,7 @@ func regenerateResponse(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext)
} }
kb := laniakea.NewInlineKeyboard(1) kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButtonStyle("🔄 Перегенерировать 🔄", laniakea.ButtonStyleDanger, "rp.regenerate", count) kb.AddCallbackButtonStyle("🔄 Перегенерировать 🔄", laniakea.ButtonStyleSuccess, "rp.regenerate", count)
ctx.EditCallback(laniakea.EscapeMarkdown(res.Choices[0].Message.Content), kb) ctx.EditCallback(laniakea.EscapeMarkdown(res.Choices[0].Message.Content), kb)
} }
@@ -790,8 +796,8 @@ func compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
ctx.AnswerCbQueryText("Запущено сжатие чата…") ctx.AnswerCbQueryText("Запущено сжатие чата…")
_compress(ctx, db) _compress(ctx, db)
kb := laniakea.NewInlineKeyboard(1) kb := laniakea.NewInlineKeyboard(1)
kb = kb.AddCallbackButton("Назад", "rp.chat_stat") kb.AddCallbackButton("Назад", "rp.chat_stat")
kb = kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleDanger, "rp.info")
} }
func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) { func _compress(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
@@ -934,7 +940,7 @@ func compressSettingStage2(ctx *laniakea.MsgContext, _ *laniakea.DatabaseContext
} }
} }
kb.AddLine().AddCallbackButton("Назад", "rp.compress_setting_s1", method) kb.AddLine().AddCallbackButton("Назад", "rp.compress_setting_s1", method)
kb.AddCallbackButton("На главную", "rp.info") kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStyleDanger, "rp.info")
ctx.EditCallback(strings.Join(out, "\n"), kb) ctx.EditCallback(strings.Join(out, "\n"), kb)
} }
func compressSetting(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) { func compressSetting(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
@@ -960,6 +966,7 @@ func compressSetting(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
if err != nil { if err != nil {
ctx.Error(err) ctx.Error(err)
} }
kb := laniakea.NewInlineKeyboard(1).AddCallbackButton("На главную", "rp.info") kb := laniakea.NewInlineKeyboard(1)
kb.AddCallbackButtonStyle("На главную", laniakea.ButtonStylePrimary, "rp.info")
ctx.EditCallback("Настройки сжатия были обновлены", kb) ctx.EditCallback("Настройки сжатия были обновлены", kb)
} }