new telegram bot
This commit is contained in:
Artem Tsyrulnikov
222
telegram_bot/screens/my_projects.go
Normal file
222
telegram_bot/screens/my_projects.go
Normal file
@@ -0,0 +1,222 @@
|
||||
package screens
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"example.com/m/bot"
|
||||
"github.com/NicoNex/echotron/v3"
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
const projectsPerPage = 6
|
||||
|
||||
type MyProjects struct {
|
||||
CurrentPage int
|
||||
WorkspaceID string
|
||||
BackState bot.State
|
||||
}
|
||||
|
||||
func (s *MyProjects) Enter(b *bot.Bot, mode bot.RenderMode) {
|
||||
// Получаем или запрашиваем JWT
|
||||
jwt, err := s.ensureJWT(b)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("Failed to get JWT")
|
||||
b.SendNew("❌ Ошибка авторизации. Попробуйте /start login", bot.Keyboard())
|
||||
return
|
||||
}
|
||||
|
||||
// Если workspace еще не выбран, получаем список workspace'ов
|
||||
if s.WorkspaceID == "" {
|
||||
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())
|
||||
return
|
||||
}
|
||||
|
||||
// Используем первый workspace по умолчанию
|
||||
s.WorkspaceID = workspaces[0].ID
|
||||
}
|
||||
|
||||
s.renderProjects(b, mode, jwt)
|
||||
}
|
||||
|
||||
func (s *MyProjects) renderProjects(b *bot.Bot, mode bot.RenderMode, jwt string) {
|
||||
page, err := b.Backend.GetProjects(context.Background(), jwt, s.WorkspaceID, s.CurrentPage+1, projectsPerPage)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Msg("Failed to get projects")
|
||||
b.SendNew("❌ Не удалось загрузить проекты", bot.Keyboard())
|
||||
return
|
||||
}
|
||||
|
||||
text := "Страница с проектами • Управление каналами\n\n"
|
||||
|
||||
var buttons [][]echotron.InlineKeyboardButton
|
||||
|
||||
if len(page.Items) == 0 {
|
||||
text += "У вас пока нет проектов"
|
||||
} else {
|
||||
text += fmt.Sprintf("Выберите проект для управления (стр. %d из %d):", s.CurrentPage+1, page.Pages)
|
||||
|
||||
// Проекты по 2 в ряду
|
||||
for i := 0; i < len(page.Items); i += 2 {
|
||||
var row []echotron.InlineKeyboardButton
|
||||
|
||||
// Первый проект в ряду
|
||||
project := page.Items[i]
|
||||
row = append(row, bot.Button(project.Title, fmt.Sprintf("project:%s", project.ID)))
|
||||
|
||||
// Второй проект в ряду (если есть)
|
||||
if i+1 < len(page.Items) {
|
||||
project = page.Items[i+1]
|
||||
row = append(row, bot.Button(project.Title, fmt.Sprintf("project:%s", project.ID)))
|
||||
} else {
|
||||
// Добавляем пустую кнопку для выравнивания
|
||||
row = append(row, bot.Button(" ", "empty"))
|
||||
}
|
||||
|
||||
buttons = append(buttons, row)
|
||||
}
|
||||
|
||||
// Кнопки пагинации с номером страницы посередине (только если есть проекты)
|
||||
var navRow []echotron.InlineKeyboardButton
|
||||
|
||||
// Левая стрелка (активная или неактивная)
|
||||
if s.CurrentPage > 0 {
|
||||
navRow = append(navRow, bot.Button("←", "prev"))
|
||||
} else {
|
||||
navRow = append(navRow, bot.Button("·", "empty"))
|
||||
}
|
||||
|
||||
// Кнопка с номером страницы (неактивная)
|
||||
pageButton := bot.Button(fmt.Sprintf("• %d/%d •", s.CurrentPage+1, page.Pages), "page_info")
|
||||
navRow = append(navRow, pageButton)
|
||||
|
||||
// Правая стрелка (активная или неактивная)
|
||||
if s.CurrentPage+1 < page.Pages {
|
||||
navRow = append(navRow, bot.Button("→", "next"))
|
||||
} else {
|
||||
navRow = append(navRow, bot.Button("·", "empty"))
|
||||
}
|
||||
|
||||
buttons = append(buttons, navRow)
|
||||
}
|
||||
|
||||
// Кнопки управления
|
||||
buttons = append(buttons, bot.Row(
|
||||
bot.Button("Архив", "archive"),
|
||||
bot.Button("Добавить проект", "add_project"),
|
||||
))
|
||||
|
||||
// Кнопка возврата в главное меню
|
||||
buttons = append(buttons, bot.Row(bot.Button("🏠 Главное меню", "menu")))
|
||||
|
||||
keyboard := bot.Keyboard(buttons...)
|
||||
b.Render(text, keyboard, mode)
|
||||
}
|
||||
|
||||
func (s *MyProjects) 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 *MyProjects) 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 == "menu":
|
||||
b.SetState(&MainMenu{}, bot.EditMessage)
|
||||
|
||||
case data == "page_info", data == "empty":
|
||||
// Неактивные кнопки - ничего не делаем
|
||||
return
|
||||
|
||||
case data == "archive":
|
||||
// TODO: реализовать экран архива
|
||||
b.Edit("📦 Архив проектов\n\nЭта функция в разработке", bot.Keyboard(
|
||||
bot.Row(bot.Button("« Назад", "back_to_projects")),
|
||||
))
|
||||
|
||||
case data == "add_project":
|
||||
b.SetState(&AddProject{
|
||||
BackState: &MyProjects{WorkspaceID: s.WorkspaceID},
|
||||
}, bot.EditMessage)
|
||||
|
||||
case data == "back_to_projects":
|
||||
s.Enter(b, bot.EditMessage)
|
||||
|
||||
case strings.HasPrefix(data, "project:"):
|
||||
// Открытие деталей проекта
|
||||
parts := strings.Split(data, ":")
|
||||
if len(parts) == 2 {
|
||||
projectID := parts[1]
|
||||
|
||||
// Получаем проект напрямую по ID
|
||||
project, err := b.Backend.GetProject(
|
||||
context.Background(),
|
||||
s.WorkspaceID,
|
||||
projectID,
|
||||
)
|
||||
if err != nil {
|
||||
log.Error().Err(err).Str("project_id", projectID).Msg("Failed to get project")
|
||||
return
|
||||
}
|
||||
|
||||
b.SetState(&ProjectDetails{
|
||||
Project: project,
|
||||
WorkspaceID: s.WorkspaceID,
|
||||
BackState: &MyProjects{WorkspaceID: s.WorkspaceID},
|
||||
}, bot.EditMessage)
|
||||
}
|
||||
|
||||
case strings.HasPrefix(data, "workspace:"):
|
||||
// Позволяет переключаться между workspace'ами в будущем
|
||||
parts := strings.Split(data, ":")
|
||||
if len(parts) == 2 {
|
||||
s.WorkspaceID = parts[1]
|
||||
s.CurrentPage = 0
|
||||
s.Enter(b, bot.EditMessage)
|
||||
}
|
||||
|
||||
default:
|
||||
s.Enter(b, bot.NewMessage)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (s *MyProjects) HandleMessage(_ *bot.Bot, _ *echotron.Update) { return }
|
||||
|
||||
func (s *MyProjects) Handle(_ *bot.Bot, _ *echotron.Update) { return }
|
||||
Reference in New Issue
Block a user