Netpulse_SasS/server/internal/alerting/telegram_callback.go

package alerting

import (
	"encoding/json"
	"fmt"
	"strings"
	"time"

	"github.com/netpulse/netpulse/server/internal/store"
)

// Розбір і перевірка натискання кнопки Telegram.
//
// Усе, що тут лежить, навмисно не знає ні про мережу, ні про базу:
// «кому належить цей чат» і «що взагалі натиснули» — рішення, які
// мають перевірятись тестом, а не стендом із живим ботом.

// ---------------------------------------------------------------------
// Те, що приходить від Bot API
// ---------------------------------------------------------------------

type tgUser struct {
	ID        int64  `json:"id"`
	Username  string `json:"username"`
	FirstName string `json:"first_name"`
	LastName  string `json:"last_name"`
	IsBot     bool   `json:"is_bot"`
}

type tgChat struct {
	ID   int64  `json:"id"`
	Type string `json:"type"`
}

type tgMessage struct {
	MessageID int64   `json:"message_id"`
	Chat      tgChat  `json:"chat"`
	From      *tgUser `json:"from"`
	Text      string  `json:"text"`
	ThreadID  int     `json:"message_thread_id"`
}

type tgCallbackQuery struct {
	ID      string     `json:"id"`
	From    tgUser     `json:"from"`
	Message *tgMessage `json:"message"`
	Data    string     `json:"data"`
}

type tgUpdate struct {
	UpdateID      int64            `json:"update_id"`
	Message       *tgMessage       `json:"message"`
	CallbackQuery *tgCallbackQuery `json:"callback_query"`
}

// ---------------------------------------------------------------------
// callback_data
// ---------------------------------------------------------------------

// callbackAction — розібране натискання.
type callbackAction struct {
	// Kind — "ack" або "mute". Перелік закритий: невідома дія — це не
	// «поки не реалізовано», а або наша стара збірка, або чужа спроба.
	Kind string
	// ID — алерт для ack, хост для mute. Завжди uuid.
	ID string
}

// maxCallbackData — стеля Bot API на callback_data (1–64 байти).
//
// Перевіряється явно, хоч Telegram і не пропустив би довшого: дані
// приходять із мережі, і «нам їх точно вкоротить той бік» — це довіра
// до чужого сервера там, де вона нічого не коштує.
const maxCallbackData = 64

// parseCallbackData розбирає вміст кнопки.
//
// У callback_data лежить рівно те, що ми туди поклали (див. notify.go),
// але прийти воно може від будь-кого, хто бачив формат: сам рядок ніяк
// не підписаний і не зашифрований. Тому цей розбір НІЧОГО не
// авторизує — він лише каже, що саме просять зробити. Право на дію
// перевіряється далі й іншим: чатом, з якого прийшло натискання, і
// прив'язаним до нього користувачем.
func parseCallbackData(raw string) (callbackAction, error) {
	if raw == "" || len(raw) > maxCallbackData {
		return callbackAction{}, fmt.Errorf("некоректна довжина callback_data")
	}
	kind, id, ok := strings.Cut(raw, ":")
	if !ok {
		return callbackAction{}, fmt.Errorf("немає роздільника")
	}
	switch kind {
	case "ack", "mute":
	default:
		return callbackAction{}, fmt.Errorf("невідома дія %q", kind)
	}
	if !isUUID(id) {
		return callbackAction{}, fmt.Errorf("ідентифікатор не uuid")
	}
	return callbackAction{Kind: kind, ID: id}, nil
}

// isUUID перевіряє канонічну форму 8-4-4-4-12.
//
// Не для краси: далі цей рядок іде в запит як ::uuid, і кривий текст
// повернувся б помилкою розбору з глибини SQL — тобто «внутрішня
// помилка» замість «кнопка застаріла».
func isUUID(s string) bool {
	if len(s) != 36 {
		return false
	}
	for i := range len(s) {
		c := s[i]
		if i == 8 || i == 13 || i == 18 || i == 23 {
			if c != '-' {
				return false
			}
			continue
		}
		switch {
		case c >= '0' && c <= '9', c >= 'a' && c <= 'f', c >= 'A' && c <= 'F':
		default:
			return false
		}
	}
	return true
}

// ---------------------------------------------------------------------
// Чий це чат
// ---------------------------------------------------------------------

// matchChannel шукає канал, якому належить чат натискання.
//
// Це головна перевірка всього приймача. Кабінет береться НЕ з
// callback_data — там немає нічого, чому можна вірити, — а з того, у
// який чат прийшло повідомлення з кнопкою. Чат налаштований у каналі
// руками адміністратора кабінету, тож збіг chat_id означає рівно те,
// що потрібно: кнопку натиснули в чаті цього кабінету.
//
// Один бот може обслуговувати кілька каналів (різні чати, а в
// self-hosted — і різні кабінети), тому перебір іде по всій групі
// каналів цього бота. Коли на той самий чат заведено кілька каналів,
// перевага в того, у якого збігається ще й гілка: канал із
// message_thread_id описує чат точніше, ніж канал без нього.
func matchChannel(chans []store.Channel, chatID int64, threadID int) (store.Channel, bool) {
	var loose store.Channel
	var found bool
	for _, c := range chans {
		var cfg telegramConfig
		if err := json.Unmarshal(c.Config, &cfg); err != nil {
			continue
		}
		if cfg.ChatID.String() != fmt.Sprint(chatID) {
			continue
		}
		if cfg.ThreadID != 0 && cfg.ThreadID == threadID {
			return c, true
		}
		if !found {
			loose, found = c, true
		}
	}
	return loose, found
}

// ---------------------------------------------------------------------
// Що дописати в повідомлення після дії
// ---------------------------------------------------------------------

// tgLocation — часовий пояс, у якому показуються години в Telegram.
//
// Київ, а не UTC: повідомлення читає черговий, і «підтверджено о 01:12»
// має збігатися з тим, що показує йому годинник на телефоні. Якщо бази
// поясів у системі немає (мінімальний образ без tzdata), лишається UTC —
// зсунута на кілька годин позначка гірша за правильну, але незрівнянно
// краща за відсутність рядка взагалі.
var tgLocation = func() *time.Location {
	if l, err := time.LoadLocation("Europe/Kyiv"); err == nil {
		return l
	}
	return time.UTC
}()

// ackLine — рядок, який дописується в повідомлення після підтвердження.
//
// Ім'я тут обов'язкове. «Підтверджено» без автора не відповідає на
// єдине питання, заради якого решта команди дивиться в чат: чи вже
// хтось узявся, чи всі думають, що взявся хтось інший.
func ackLine(who string, at time.Time) string {
	return fmt.Sprintf("✅ Підтверджено: %s, %s", who, at.In(tgLocation).Format("02.01 15:04"))
}

// muteLine — те саме для заглушення.
func muteLine(who string, until time.Time) string {
	return fmt.Sprintf("🔕 Заглушено до %s (%s)",
		until.In(tgLocation).Format("02.01 15:04"), who)
}

// maxMessageText — стеля Bot API на текст повідомлення.
const maxMessageText = 4096

// withStatus дописує підсумок до тексту повідомлення.
//
// Повторний рядок не додається: друге натискання має привести
// повідомлення в той самий вигляд, що й перше, а не відростити другий
// «Підтверджено». Це та сама ідемпотентність, що й у самій дії, тільки
// видима — саме її людина й перевіряє очима.
func withStatus(text, line string) string {
	if strings.Contains(text, line) {
		return text
	}
	out := strings.TrimRight(text, "\n") + "\n\n" + line
	if len(out) > maxMessageText {
		// Обрізати доводиться з голови, а не з хвоста: підсумок —
		// найновіше й найпотрібніше в повідомленні, а початок алерту
		// вже прочитали, коли він прийшов.
		out = "…" + out[len(out)-maxMessageText+3:]
	}
	return out
}

// tgDisplayName — як назвати того, хто натиснув, коли прив'язки немає.
//
// Використовується лише в текстах відмов: у самому підтвердженні
// автором стоїть користувач NetPulse, а не telegram-акаунт.
func tgDisplayName(u tgUser) string {
	if u.Username != "" {
		return "@" + u.Username
	}
	name := strings.TrimSpace(u.FirstName + " " + u.LastName)
	if name != "" {
		return name
	}
	return fmt.Sprint(u.ID)
}

// ---------------------------------------------------------------------
// Команда прив'язки
// ---------------------------------------------------------------------

// parseLinkCommand дістає код із повідомлення боту.
//
// Приймається і «/link КОД», і «/link@netpulse_bot КОД»: у групових
// чатах Telegram дописує ім'я бота до команди сам, і без цього
// прив'язка працювала б лише в особистих повідомленнях.
//
// Регістр коду не має значення — абетка кодів (див. міграцію 0061) уся
// у верхньому регістрі, а телефон охоче виправляє введене на нижній.
func parseLinkCommand(text string) (string, bool) {
	fields := strings.Fields(strings.TrimSpace(text))
	if len(fields) == 0 {
		return "", false
	}
	cmd, _, _ := strings.Cut(fields[0], "@")
	if cmd != "/link" {
		return "", false
	}
	if len(fields) < 2 {
		return "", true
	}
	return strings.ToUpper(fields[1]), true
}