135 lines
3.5 KiB
Go
135 lines
3.5 KiB
Go
package screens
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/NicoNex/echotron/v3"
|
||
"github.com/TelegramExchange/tgex-backend/tg_bot/bot"
|
||
"github.com/TelegramExchange/tgex-backend/tg_bot/screens/ui"
|
||
"github.com/rs/zerolog/log"
|
||
)
|
||
|
||
type WorkspaceMenu struct {
|
||
CurrentPage int
|
||
BackState bot.State
|
||
}
|
||
|
||
const msgWorkspaceEmpty = `
|
||
У вас пока нет рабочих пространств.
|
||
`
|
||
|
||
const workspacesPerPage = 6
|
||
|
||
func (s *WorkspaceMenu) Enter(b *bot.Bot, mode bot.RenderMode) {
|
||
workspaces, err := b.Backend.GetWorkspaces(context.Background(), b.Session.JWT)
|
||
if err != nil {
|
||
log.Error().Err(err).Msg("Failed to get workspaces for workspace menu")
|
||
b.SendNew("❌ Не удалось загрузить рабочие пространства", Keyboard(Row(Button("← Назад", "back"))))
|
||
return
|
||
}
|
||
|
||
if len(workspaces) == 0 {
|
||
kb := Keyboard(Row(Button("← Назад", "back")))
|
||
b.Render(msgWorkspaceEmpty, kb, mode)
|
||
return
|
||
}
|
||
|
||
var allButtons []echotron.InlineKeyboardButton
|
||
for _, ws := range workspaces {
|
||
label := ws.Name
|
||
if ws.ID == b.Session.WorkspaceID {
|
||
label = "● " + label
|
||
}
|
||
allButtons = append(allButtons, Button(label, fmt.Sprintf("workspace_select:%s", ws.ID)))
|
||
}
|
||
|
||
elementRows := ui.BuildElementRows(ui.ElementLayoutConfig{
|
||
CurrentPage: s.CurrentPage,
|
||
ItemsPerPage: workspacesPerPage,
|
||
ItemsPerRow: 2,
|
||
}, allButtons)
|
||
|
||
var buttons [][]echotron.InlineKeyboardButton
|
||
buttons = append(buttons, elementRows...)
|
||
|
||
if navRow := ui.BuildNavigationRow(ui.PaginationConfig{
|
||
CurrentPage: s.CurrentPage,
|
||
TotalItems: len(allButtons),
|
||
ItemsPerPage: workspacesPerPage,
|
||
}); navRow != nil {
|
||
buttons = append(buttons, navRow)
|
||
}
|
||
|
||
buttons = append(buttons, Row(Button("← Назад", "back")))
|
||
|
||
totalPages := ui.CalculatePages(len(allButtons), workspacesPerPage)
|
||
text := fmt.Sprintf(`<b>Рабочие пространства</b>%s
|
||
|
||
Выберите текущее рабочее пространство.`, ui.FormatPageInfo(s.CurrentPage, totalPages))
|
||
|
||
b.Render(text, Keyboard(buttons...), mode)
|
||
}
|
||
|
||
func (s *WorkspaceMenu) HandleCallback(b *bot.Bot, u *echotron.Update) {
|
||
if u.CallbackQuery == nil || u.CallbackQuery.Data == "" {
|
||
return
|
||
}
|
||
|
||
data := u.CallbackQuery.Data
|
||
|
||
switch {
|
||
case data == "prev":
|
||
if s.CurrentPage > 0 {
|
||
s.CurrentPage--
|
||
}
|
||
s.Enter(b, bot.EditMessage)
|
||
|
||
case data == "next":
|
||
s.CurrentPage++
|
||
s.Enter(b, bot.EditMessage)
|
||
|
||
case data == "back":
|
||
if s.BackState != nil {
|
||
b.SetState(s.BackState, bot.EditMessage)
|
||
return
|
||
}
|
||
b.SetState(&MainMenu{}, bot.EditMessage)
|
||
|
||
case strings.HasPrefix(data, "workspace_select:"):
|
||
parts := strings.Split(data, ":")
|
||
if len(parts) != 2 {
|
||
return
|
||
}
|
||
workspaceID := parts[1]
|
||
workspaces, err := b.Backend.GetWorkspaces(context.Background(), b.Session.JWT)
|
||
if err != nil {
|
||
log.Error().Err(err).Msg("Failed to get workspaces for selection")
|
||
b.SendNew("❌ Не удалось загрузить рабочие пространства", Keyboard(Row(Button("← Назад", "back"))))
|
||
return
|
||
}
|
||
for _, ws := range workspaces {
|
||
if ws.ID == workspaceID {
|
||
b.Session.WorkspaceID = ws.ID
|
||
break
|
||
}
|
||
}
|
||
|
||
if s.BackState != nil {
|
||
b.SetState(s.BackState, bot.EditMessage)
|
||
return
|
||
}
|
||
b.SetState(&MainMenu{}, bot.EditMessage)
|
||
|
||
default:
|
||
s.Enter(b, bot.NewMessage)
|
||
}
|
||
}
|
||
|
||
func (s *WorkspaceMenu) HandleMessage(_ *bot.Bot, _ *echotron.Update) {}
|
||
|
||
func (s *WorkspaceMenu) Handle(_ *bot.Bot, _ *echotron.Update) {}
|
||
|
||
func (s *WorkspaceMenu) Exit() {}
|