package invite

import (
	"strings"
	"time"
)

// defaultIfBlank 在字符串为空时返回默认值。
func defaultIfBlank(value, fallback string) string {
	if strings.TrimSpace(value) == "" {
		return fallback
	}
	return value
}

// normalizeSysOrigin 统一系统标识格式。
func normalizeSysOrigin(value string) string {
	return strings.TrimSpace(value)
}

// normalizeTimezone 统一时区配置，缺省回退到默认时区。
func normalizeTimezone(value string) string {
	value = strings.TrimSpace(value)
	if value == "" {
		return defaultTimezone
	}
	return value
}

// monthKeyAt 计算某个时间点在指定时区下的月份标识。
func monthKeyAt(t time.Time, timezone string) string {
	location, err := time.LoadLocation(normalizeTimezone(timezone))
	if err != nil {
		location = time.FixedZone(defaultTimezone, 3*3600)
	}
	return t.In(location).Format("200601")
}

// monthResetInfoAt 返回指定时间所在活动月的结束时间和剩余秒数。
func monthResetInfoAt(t time.Time, timezone string) (time.Time, int64) {
	location, err := time.LoadLocation(normalizeTimezone(timezone))
	if err != nil {
		location = time.FixedZone(defaultTimezone, 3*3600)
	}
	localTime := t.In(location)
	periodEnd := time.Date(localTime.Year(), localTime.Month()+1, 1, 0, 0, 0, 0, location)
	seconds := int64(periodEnd.Sub(localTime).Seconds())
	if seconds < 0 {
		seconds = 0
	}
	return periodEnd, seconds
}

// digitsOnly 判断字符串是否只包含数字。
func digitsOnly(value string) bool {
	for _, ch := range value {
		if ch < '0' || ch > '9' {
			return false
		}
	}
	return true
}

// truncate 截断字符串，避免超出数据库字段长度。
func truncate(value string, size int) string {
	if len(value) <= size {
		return value
	}
	return value[:size]
}