tgex-backend/tg_bot/screens/placements.go

package screens

import (
	"context"
	"fmt"

	"github.com/NicoNex/echotron/v3"
	"github.com/TelegramExchange/tgex-backend/tg_bot/bot"
)

type Placements struct {
	ProjectID         string
	ProjectTitle      string
	ProjectTelegramID int64
	ProjectUsername   string
	ProjectStatus     string
	BackState         bot.State
}

const msgPlacements = `
<b>Размещения</b>

Управление размещениями для проекта.
`

func (s *Placements) Enter(b *bot.Bot, mode bot.RenderMode) {
	var rows [][]echotron.InlineKeyboardButton

	rows = append(rows, Row(Button("＋ Создать размещение", "add_purchase")))
	rows = append(rows, Row(Button("Размещения", "placements_list")))
	rows = append(rows, Row(URLButton("План закупов", "https://example.com")))
	rows = append(rows, Row(Button("← Назад", "back")))

	keyboard := Keyboard(rows...)
	b.Render(msgPlacements, keyboard, mode)

	updateProjectHeaderMedia(b, b.LastMessageID, msgPlacements, keyboard, s.ProjectTelegramID, s.ProjectTitle, s.ProjectUsername, s.ProjectStatus)
}

func (s *Placements) HandleCallback(b *bot.Bot, u *echotron.Update) {
	if u.CallbackQuery == nil || u.CallbackQuery.Data == "" {
		return
	}

	data := u.CallbackQuery.Data

	switch data {
	case "back":
		if s.BackState != nil {
			b.SetState(s.BackState, bot.EditMessage)
		}

	case "add_purchase":
		b.SetState(&AddPurchase{
			ProjectID:         s.ProjectID,
			ProjectTitle:      s.ProjectTitle,
			ProjectTelegramID: s.ProjectTelegramID,
			ProjectUsername:   s.ProjectUsername,
			ProjectStatus:     s.ProjectStatus,
			ActivePicker:      "",
			BackState:         s,
		}, bot.EditMessage)

	case "placements_list":
		b.SetState(&PlacementsList{
			ProjectID:         s.ProjectID,
			ProjectTitle:      s.ProjectTitle,
			ProjectTelegramID: s.ProjectTelegramID,
			ProjectUsername:   s.ProjectUsername,
			ProjectStatus:     s.ProjectStatus,
			BackState:         s,
		}, bot.EditMessage)

	default:
		s.Enter(b, bot.NewMessage)
	}
}

func (s *Placements) HandleMessage(_ *bot.Bot, _ *echotron.Update) { return }

func (s *Placements) Handle(_ *bot.Bot, _ *echotron.Update) { return }

func (s *Placements) Exit() {}

// PlacementsList - экран со списком размещений
type PlacementsList struct {
	ProjectID         string
	ProjectTitle      string
	ProjectTelegramID int64
	ProjectUsername   string
	ProjectStatus     string
	BackState         bot.State
}

func (s *PlacementsList) Enter(b *bot.Bot, mode bot.RenderMode) {
	s.render(b, mode)
}

func (s *PlacementsList) render(b *bot.Bot, mode bot.RenderMode) {
	resp, err := b.Backend.GetPlacements(context.Background(), b.Session.JWT, b.Session.WorkspaceID, s.ProjectID)
	if err != nil {
		b.SendNew("❌ Не удалось загрузить размещения", Keyboard())
		return
	}

	text := "<b>Список размещений</b>\n\n"

	var buttons [][]echotron.InlineKeyboardButton

	if len(resp.Placements) == 0 {
		text += `Список размещений пуст.

<i>Размещение — это план публикации рекламы в канале:</i>
  ‣ Выбор креатива
  ‣ Канал размещения
  ‣ Стоимость и формат
  ‣ Статус выполнения

Создайте первое размещение`
	} else {
		for i, placement := range resp.Placements {
			statusSymbol := "●"
			if placement.Status == "archived" {
				statusSymbol = "■"
			}

			channelName := "Без названия"
			if placement.Channel.Title != nil && *placement.Channel.Title != "" {
				channelName = *placement.Channel.Title
			} else if placement.Channel.Username != nil && *placement.Channel.Username != "" {
				channelName = "@" + *placement.Channel.Username
			}

			buttonText := fmt.Sprintf("%s #%d %s", statusSymbol, i+1, channelName)

			buttons = append(buttons, Row(
				Button(buttonText, fmt.Sprintf("placement:%s", placement.ID)),
			))
		}
	}

	// Кнопка назад
	buttons = append(buttons, Row(
		Button("← Назад", "back"),
	))

	keyboard := Keyboard(buttons...)
	b.Render(text, keyboard, mode)
}

func (s *PlacementsList) HandleCallback(b *bot.Bot, u *echotron.Update) {
	if u.CallbackQuery == nil || u.CallbackQuery.Data == "" {
		return
	}

	data := u.CallbackQuery.Data

	switch data {
	case "back":
		if s.BackState != nil {
			b.SetState(s.BackState, bot.EditMessage)
		}

	default:
		// Проверяем, не нажали ли на конкретное размещение
		if len(data) > 10 && data[:10] == "placement:" {
			placementID := data[10:]
			b.SetState(&PlacementDetails{
				ProjectID:   s.ProjectID,
				PlacementID: placementID,
				BackState:   s,
			}, bot.EditMessage)
		} else {
			s.Enter(b, bot.NewMessage)
		}
	}
}

func (s *PlacementsList) HandleMessage(_ *bot.Bot, _ *echotron.Update) { return }

func (s *PlacementsList) Handle(_ *bot.Bot, _ *echotron.Update) { return }

func (s *PlacementsList) Exit() {}