150 lines
4.0 KiB
Go
150 lines
4.0 KiB
Go
package screens
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"example.com/m/bot"
|
||
"github.com/NicoNex/echotron/v3"
|
||
"github.com/rs/zerolog/log"
|
||
)
|
||
|
||
type SelectWorkspace struct {
|
||
ChannelID string
|
||
BackState bot.State
|
||
}
|
||
|
||
func (s *SelectWorkspace) Enter(b *bot.Bot, mode bot.RenderMode) {
|
||
jwt, err := s.ensureJWT(b)
|
||
if err != nil {
|
||
log.Error().Err(err).Msg("Failed to get JWT")
|
||
b.SendNew("❌ Ошибка авторизации. Попробуйте /start", bot.Keyboard())
|
||
return
|
||
}
|
||
|
||
workspaces, err := b.Backend.GetWorkspaces(context.Background(), jwt)
|
||
if err != nil {
|
||
log.Error().Err(err).Msg("Failed to get workspaces")
|
||
b.SendNew("❌ Не удалось загрузить список workspace'ов", bot.Keyboard())
|
||
return
|
||
}
|
||
|
||
if len(workspaces) == 0 {
|
||
b.SendNew("У вас пока нет workspace'ов", bot.Keyboard(
|
||
bot.Row(bot.Button("🏠 Главное меню", "back")),
|
||
))
|
||
return
|
||
}
|
||
|
||
text := "📁 Выбор workspace для канала\n\n"
|
||
text += "В какой workspace добавить канал?"
|
||
|
||
var buttons [][]echotron.InlineKeyboardButton
|
||
|
||
// Создаем кнопки для каждого workspace
|
||
for _, ws := range workspaces {
|
||
buttons = append(buttons, bot.Row(
|
||
bot.Button(ws.Name, fmt.Sprintf("select_workspace:%s", ws.ID)),
|
||
))
|
||
}
|
||
|
||
// Кнопка отмены
|
||
buttons = append(buttons, bot.Row(bot.Button("❌ Отмена", "cancel")))
|
||
|
||
keyboard := bot.Keyboard(buttons...)
|
||
b.Render(text, keyboard, mode)
|
||
}
|
||
|
||
func (s *SelectWorkspace) ensureJWT(b *bot.Bot) (string, error) {
|
||
// Проверяем, есть ли JWT в сессии
|
||
if b.Session.JWT != "" {
|
||
return b.Session.JWT, nil
|
||
}
|
||
|
||
// Если нет, запрашиваем новый
|
||
jwt, err := b.Backend.GetJWTByTelegramID(context.Background(), b.ChatID)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
// Сохраняем в сессию
|
||
b.Session.JWT = jwt
|
||
return jwt, nil
|
||
}
|
||
|
||
func (s *SelectWorkspace) HandleCallback(b *bot.Bot, u *echotron.Update) {
|
||
if u.CallbackQuery == nil || u.CallbackQuery.Data == "" {
|
||
return
|
||
}
|
||
|
||
data := u.CallbackQuery.Data
|
||
|
||
switch {
|
||
case data == "cancel", data == "back":
|
||
if s.BackState != nil {
|
||
b.SetState(s.BackState, bot.NewMessage)
|
||
}
|
||
|
||
case strings.HasPrefix(data, "select_workspace:"):
|
||
// Извлекаем workspace_id
|
||
parts := strings.Split(data, ":")
|
||
if len(parts) != 2 {
|
||
log.Error().Str("callback_data", data).Msg("Invalid callback data format")
|
||
return
|
||
}
|
||
|
||
workspaceID := parts[1]
|
||
|
||
// Показываем индикатор загрузки
|
||
b.Edit("⏳ Добавляем канал в workspace...", bot.Keyboard())
|
||
|
||
// Вызываем API для привязки канала к workspace
|
||
project, err := b.Backend.AttachChannelToWorkspace(
|
||
context.Background(),
|
||
s.ChannelID,
|
||
workspaceID,
|
||
b.ChatID,
|
||
)
|
||
if err != nil {
|
||
log.Error().Err(err).Str("channel_id", s.ChannelID).Str("workspace_id", workspaceID).Msg("Failed to attach channel to workspace")
|
||
b.Edit("❌ Не удалось добавить канал в workspace", bot.Keyboard(
|
||
bot.Row(bot.Button("← Назад", "back_to_select")),
|
||
))
|
||
return
|
||
}
|
||
|
||
// Показываем успешное сообщение
|
||
successText := fmt.Sprintf("✅ Канал успешно добавлен!\n\n"+
|
||
"📊 Проект ID: %s\n"+
|
||
"📝 Название: %s\n"+
|
||
"🏷 Статус: %s",
|
||
project.ID,
|
||
project.Title,
|
||
project.Status,
|
||
)
|
||
|
||
b.Edit(successText, bot.Keyboard(
|
||
bot.Row(bot.Button("📋 Мои проекты", "my_projects")),
|
||
bot.Row(bot.Button("🏠 Главное меню", "main_menu")),
|
||
))
|
||
|
||
case data == "back_to_select":
|
||
s.Enter(b, bot.EditMessage)
|
||
|
||
case data == "my_projects":
|
||
b.SetState(&MyProjects{}, bot.NewMessage)
|
||
|
||
case data == "main_menu":
|
||
b.SetState(&MainMenu{}, bot.NewMessage)
|
||
|
||
default:
|
||
s.Enter(b, bot.NewMessage)
|
||
}
|
||
return
|
||
}
|
||
|
||
func (s *SelectWorkspace) HandleMessage(_ *bot.Bot, _ *echotron.Update) { return }
|
||
|
||
func (s *SelectWorkspace) Handle(_ *bot.Bot, _ *echotron.Update) { return }
|