torrserver/server/tgbot/utils.go

package tgbot

import (
	"fmt"
	"strconv"
	"strings"
	"unicode"

	tele "gopkg.in/telebot.v4"

	"server/settings"
	"server/tgbot/config"
	"server/web"
)

func chatMsgKey(chatID int64, msgID int) string {
	return fmt.Sprintf("%d_%d", chatID, msgID)
}

// escapeHtml escapes <, >, &, " for Telegram HTML parse mode to prevent "Unsupported start tag" errors
func escapeHtml(s string) string {
	s = strings.ReplaceAll(s, "&", "&amp;")
	s = strings.ReplaceAll(s, "<", "&lt;")
	s = strings.ReplaceAll(s, ">", "&gt;")
	s = strings.ReplaceAll(s, "\"", "&quot;")
	return s
}

// logSafeStr truncates by runes, strips emojis/symbols for clean laconic logs
func logSafeStr(s string, maxRunes int) string {
	var b strings.Builder
	n := 0
	lastSpace := true
	for _, r := range s {
		if n >= maxRunes {
			break
		}
		switch {
		case r == '\n' || r == '\r' || r == '\t':
			if !lastSpace {
				b.WriteRune(' ')
				n++
				lastSpace = true
			}
		case r < 32 || r == 127:
		case logIsEmojiOrSymbol(r):
		case unicode.IsLetter(r) || unicode.IsNumber(r) || r == '/' || r == '-' || r == '_' || r == '|' || r == ':' || r == '.' || r == ',' || r == ' ' || r == '?' || r == '!' || r == '@':
			b.WriteRune(r)
			n++
			lastSpace = (r == ' ')
		default:
			b.WriteRune(r)
			n++
			lastSpace = false
		}
	}
	return strings.TrimSpace(b.String())
}

func logIsEmojiOrSymbol(r rune) bool {
	if unicode.IsSymbol(r) {
		return true
	}
	u := uint32(r)
	return (u >= 0x1F300 && u <= 0x1F9FF) || (u >= 0x2600 && u <= 0x26FF) ||
		(u >= 0x2700 && u <= 0x27BF) || (u >= 0x1F600 && u <= 0x1F64F) ||
		(u >= 0x1F680 && u <= 0x1F6FF) || (u >= 0x1F1E0 && u <= 0x1F1FF) ||
		(u >= 0xFE00 && u <= 0xFE0F) || (u >= 0x1F000 && u <= 0x1F02F)
}

// logUser formats uid and optional username for logs
func logUser(u *tele.User) string {
	if u == nil {
		return "uid=?"
	}
	return logUserID(u.ID) + logUsername(u.Username)
}

// logUserID formats uid for logs when User is not available
func logUserID(uid int64) string {
	return "uid=" + strconv.FormatInt(uid, 10)
}

func logUsername(username string) string {
	if username == "" {
		return ""
	}
	return " @" + username
}

// logHashOrTruncate returns hash for logging if link is hash or magnet with btih, else truncated link
func logHashOrTruncate(link string) string {
	if isHash(link) {
		return link
	}
	if idx := strings.Index(link, "btih:"); idx >= 0 && idx+45 <= len(link) {
		if h := link[idx+5 : idx+45]; isHash(h) {
			return h
		}
	}
	if strings.HasPrefix(strings.ToLower(link), "torrs://") && len(link) >= 48 {
		if h := link[8:48]; isHash(h) {
			return h
		}
	}
	if len(link) > 64 {
		return link[:64] + "..."
	}
	return link
}

// getHost returns the base URL for stream/play links (e.g. http://192.168.1.1:8090)
func getHost() string {
	host := config.Cfg.HostWeb
	if host == "" {
		host = settings.PubIPv4
		if host == "" {
			ips := web.GetLocalIps()
			if len(ips) == 0 {
				host = "127.0.0.1"
			} else {
				host = ips[0]
			}
		}
	}
	if !strings.Contains(host, ":") {
		if settings.Ssl {
			host += ":" + settings.SslPort
		} else {
			host += ":" + settings.Port
		}
	}
	if !strings.HasPrefix(host, "http") {
		if settings.Ssl {
			host = "https://" + host
		} else {
			host = "http://" + host
		}
	}
	return host
}