tgex-backend/telegram_bot/screens/add_creative.go

package screens

import (
	"context"
	"fmt"
	"strings"

	"example.com/m/adapter/backend"
	"example.com/m/bot"
	"github.com/NicoNex/echotron/v3"
	"github.com/rs/zerolog/log"
)

type AddCreative struct {
	CreativeEditorFields // Встраивание общих полей

	ProjectID   string
	WorkspaceID string
	BackState   bot.State
}

func (s *AddCreative) Enter(b *bot.Bot, mode bot.RenderMode) {
	text := `<i>… › Креативы › Добавление</i>

<b>＋ Добавить креатив</b>

Введите <b>название</b> креатива:

<i>Например:</i>
  ‣ Летняя акция
  ‣ Баннер 1920x1080
  ‣ Видео промо`

	buttons := [][]echotron.InlineKeyboardButton{
		bot.Row(bot.Button("« Отмена", "cancel")),
	}

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

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

	data := u.CallbackQuery.Data

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

	case data == "edit_text":
		// Редактируем панель управления - просим текст
		s.ShowControlPanel(b, `<b>✎ Введите текст креатива:</b>

⚠ <b>Важно:</b> Текст должен содержать <b>ОДНУ</b> инвайт-ссылку вашего канала

<b>Формат ссылки:</b>
<code>https://t.me/+xxx</code>

<b>Пример:</b>
<i>Присоединяйтесь к нашему каналу!
https://t.me/+AbCdEfGhIjKlMn</i>`, [][]echotron.InlineKeyboardButton{
			bot.Row(bot.Button("« Отмена", "cancel_edit")),
		})

	case data == "add_button":
		// Начинаем процесс добавления кнопки
		s.WaitingForButtonText = true

		s.ShowControlPanel(b, `<b>＋ Добавить кнопку</b>

Введите <b>текст кнопки:</b>

<i>Например:</i>
• Перейти на сайт
• Написать нам`, [][]echotron.InlineKeyboardButton{
			bot.Row(bot.Button("« Отмена", "cancel_add_button")),
		})

	case data == "add_media":
		// Начинаем процесс добавления медиа
		s.WaitingForMedia = true

		s.ShowControlPanel(b, `<b>＋ Добавить медиа</b>

Отправьте <b>фото, видео или GIF</b>

<i>Это медиа будет отображаться в креативе вместе с текстом</i>`, [][]echotron.InlineKeyboardButton{
			bot.Row(bot.Button("« Отмена", "cancel_media")),
		})

	case data == "delete_media":
		// Удаляем медиа
		s.MediaType = ""
		s.MediaFileID = ""
		s.MediaChanged = true
		s.UpdateCreativePreview(b)
		s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")

	case data == "cancel_add_button":
		// Отменяем добавление кнопки
		// Если мы уже создали кнопку с placeholder - удаляем её
		if s.WaitingForButtonURL && len(s.Buttons) > 0 {
			lastButton := s.Buttons[len(s.Buttons)-1]
			if lastButton.URL == "https://example.com" {
				s.Buttons = s.Buttons[:len(s.Buttons)-1]
				s.UpdateCreativePreview(b)
			}
		}
		s.WaitingForButtonText = false
		s.WaitingForButtonURL = false
		s.CurrentButtonText = nil
		s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")

	case data == "cancel_edit", data == "cancel_media":
		// Отменяем редактирование текста или добавление медиа
		s.WaitingForMedia = false
		s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")

	case data == "confirm_create":
		// Создаём креатив
		s.createCreative(b)

	case strings.HasPrefix(data, "delete_button:"):
		// Удаляем кнопку по индексу
		parts := strings.Split(data, ":")
		if len(parts) == 2 {
			var index int
			if _, err := fmt.Sscanf(parts[1], "%d", &index); err == nil {
				if index >= 0 && index < len(s.Buttons) {
					s.Buttons = append(s.Buttons[:index], s.Buttons[index+1:]...)
					s.UpdateCreativePreview(b)
					s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
				}
			}
		}

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

func (s *AddCreative) HandleMessage(b *bot.Bot, u *echotron.Update) {
	if u.Message == nil {
		return
	}

	// Шаг 1: Получаем название
	if s.Name == nil {
		if u.Message.Text == "" {
			return
		}
		name := u.Message.Text
		s.Name = &name

		// Отправляем креатив + панель управления
		s.sendCreativeAndPanel(b)
		return
	}

	// Обработка медиа
	if s.WaitingForMedia {
		handled := false

		if u.Message.Photo != nil && len(u.Message.Photo) > 0 {
			// Берем самое большое фото
			photo := u.Message.Photo[len(u.Message.Photo)-1]
			s.MediaType = "photo"
			s.MediaFileID = photo.FileID
			s.MediaChanged = true
			handled = true
		} else if u.Message.Video != nil {
			s.MediaType = "video"
			s.MediaFileID = u.Message.Video.FileID
			s.MediaChanged = true
			handled = true
		} else if u.Message.Animation != nil {
			s.MediaType = "animation"
			s.MediaFileID = u.Message.Animation.FileID
			s.MediaChanged = true
			handled = true
		}

		if handled {
			// Удаляем сообщение пользователя
			_, err := b.DeleteMessage(b.ChatID, u.Message.ID)
			if err != nil {
				log.Error().Err(err).Msg("Failed to delete user message")
			}

			s.WaitingForMedia = false
			s.UpdateCreativePreview(b)
			s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
			return
		}
	}

	// Если не медиа и не текст - игнорируем
	if u.Message.Text == "" {
		return
	}

	// Удаляем сообщение пользователя
	_, err := b.DeleteMessage(b.ChatID, u.Message.ID)
	if err != nil {
		log.Error().Err(err).Msg("Failed to delete user message")
	}

	// Обработка добавления кнопки - текст кнопки
	if s.WaitingForButtonText {
		buttonText := u.Message.Text
		s.CurrentButtonText = &buttonText
		s.WaitingForButtonText = false
		s.WaitingForButtonURL = true

		// Сразу создаём кнопку с placeholder URL (используем валидный URL)
		s.Buttons = append(s.Buttons, InlineButton{
			Text: buttonText,
			URL:  "https://example.com",
		})

		// Обновляем превью креатива, чтобы показать новую кнопку
		s.UpdateCreativePreview(b)

		// Редактируем панель управления - просим URL
		s.ShowControlPanel(b, `<b>✧ Введите URL для кнопки:</b>

URL должен начинаться с <code>https://</code>

<i>Например:</i>
<code>https://example.com</code>
<code>https://t.me/yourchannel</code>`, [][]echotron.InlineKeyboardButton{
			bot.Row(bot.Button("« Отмена", "cancel_add_button")),
		})
		return
	}

	// Обработка добавления кнопки - URL
	if s.WaitingForButtonURL {
		url := u.Message.Text

		// Базовая валидация URL
		if !strings.HasPrefix(url, "https://") {
			s.ShowControlPanel(b, `<b>❌ Неверный формат URL</b>

URL должен начинаться с <code>https://</code>

Попробуйте ещё раз.`, [][]echotron.InlineKeyboardButton{
				bot.Row(bot.Button("« Отмена", "cancel_add_button")),
			})
			return
		}

		// Обновляем URL последней кнопки (которую создали с placeholder)
		if len(s.Buttons) > 0 {
			s.Buttons[len(s.Buttons)-1].URL = url
		}
		s.WaitingForButtonURL = false
		s.CurrentButtonText = nil

		// Обновляем креатив и возвращаем панель управления
		s.UpdateCreativePreview(b)
		s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
		return
	}

	// Шаг 2: Получаем текст
	if s.Text == nil {
		text := s.GetFormattedText(u.Message)
		s.Text = &text

		// Обновляем креатив и возвращаем панель управления
		s.UpdateCreativePreview(b)
		s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
		return
	}

	// Если текст уже задан - это повторное редактирование
	text := s.GetFormattedText(u.Message)
	s.Text = &text

	// Обновляем креатив и возвращаем панель управления
	s.UpdateCreativePreview(b)
	s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
}

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

// sendCreativeAndPanel отправляет превью креатива и панель управления
func (s *AddCreative) sendCreativeAndPanel(b *bot.Bot) {
	// Устанавливаем навигацию
	s.Navigation = "… › Креативы › Добавление"

	// 1. Отправляем превью креатива
	s.SendCreativePreview(b)

	// 2. Отправляем панель управления
	s.ShowManagementPanel(b, "✓ Сохранить", "confirm_create", "cancel")
}

func (s *AddCreative) createCreative(b *bot.Bot) {
	// Получаем JWT
	jwt, err := s.ensureJWT(b)
	if err != nil {
		log.Error().Err(err).Msg("Failed to get JWT")
		b.SendNew("<b>❌ Ошибка авторизации</b>\n\nПопробуйте /start login", bot.Keyboard())
		return
	}

	var mediaData []byte
	if s.MediaFileID != "" {
		mediaData, err = b.DownloadFileBytes(s.MediaFileID)
		if err != nil {
			log.Error().Err(err).Msg("Failed to download creative media")
			b.SendNew("<b>❌ Не удалось загрузить медиа</b>", bot.Keyboard(
				bot.Row(bot.Button("← Назад", "cancel")),
			))
			return
		}
	}

	buttons := make([]backend.CreativeButton, 0, len(s.Buttons))
	for _, button := range s.Buttons {
		buttons = append(buttons, backend.CreativeButton{
			Text: button.Text,
			URL:  button.URL,
		})
	}

	// Создаём креатив через API
	creative, err := b.Backend.CreateCreative(
		context.Background(),
		jwt,
		s.WorkspaceID,
		s.ProjectID,
		backend.CreateCreativeInput{
			Name:        *s.Name,
			Text:        *s.Text,
			MediaType:   s.MediaType,
			MediaFileID: s.MediaFileID,
			MediaData:   mediaData,
			Buttons:     buttons,
		},
	)

	if err != nil {
		log.Error().Err(err).Msg("Failed to create creative")

		// Проверяем, это ошибка валидации инвайт-ссылки
		errMsg := err.Error()
		var userMsg string

		if strings.Contains(errMsg, "Creative text must contain one invite link") {
			userMsg = `<b>❌ Ошибка валидации</b>

Текст должен содержать <b>ОДНУ</b> инвайт-ссылку вашего канала!

<b>Формат ссылки:</b>
<code>https://t.me/+xxx</code>

<b>Пример правильного текста:</b>
<i>Присоединяйтесь к нашему каналу!
https://t.me/+AbCdEfGhIjKlMn</i>`
		} else if strings.Contains(errMsg, "Creative text must contain only one invite link") {
			userMsg = `<b>❌ Слишком много ссылок</b>

Текст должен содержать <b>ТОЛЬКО ОДНУ</b> инвайт-ссылку.
У вас их несколько. Удалите лишние.`
		} else {
			userMsg = `<b>❌ Ошибка создания</b>

Не удалось создать креатив.
Попробуйте ещё раз или вернитесь назад.`
		}

		b.SendNew(userMsg, bot.Keyboard(
			bot.Row(bot.Button("← Назад", "back_from_preview")),
		))
		return
	}

	// Успех! Показываем сообщение и возвращаемся
	successText := `<b>✅ Креатив успешно создан!</b>

📄 <b>Название:</b> ` + creative.Name

	if len(s.Buttons) > 0 {
		successText += fmt.Sprintf(`
🔘 <b>Кнопок добавлено:</b> %d`, len(s.Buttons))
	}

	b.SendNew(successText, bot.Keyboard())

	// Возвращаемся на экран креативов
	if s.BackState != nil {
		b.SetState(s.BackState, bot.NewMessage)
	}
}

func (s *AddCreative) 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
}