mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 19:36:36 +08:00
split license handler helpers
This commit is contained in:
ShukeBta
@@ -1,18 +1,10 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/ShukeBta/MediaStationGo/internal/service"
|
||||
)
|
||||
@@ -22,10 +14,6 @@ const (
|
||||
licenseHMACSecretSetting = "license.hmac_secret" // #nosec G101 -- setting key name, not the HMAC secret value.
|
||||
licenseDeviceIDSetting = "license.device_id"
|
||||
licenseDeviceNameSetting = "license.device_name"
|
||||
|
||||
licenseHeartbeatInterval = 12 * time.Hour
|
||||
licenseHeartbeatCheckInterval = 30 * time.Minute
|
||||
licenseHeartbeatStartupDelay = 2 * time.Minute
|
||||
)
|
||||
|
||||
type licenseActivateReq struct {
|
||||
@@ -177,353 +165,3 @@ func licenseHeartbeatHandler(svc *service.Container) gin.HandlerFunc {
|
||||
c.JSON(http.StatusOK, licenseActivationView(state))
|
||||
}
|
||||
}
|
||||
|
||||
func refreshLicenseCapacityBestEffort(ctx context.Context, svc *service.Container) {
|
||||
if svc == nil || svc.Repo == nil || svc.Repo.Setting == nil {
|
||||
return
|
||||
}
|
||||
_, _, _ = maybeSendLicenseHeartbeat(ctx, svc, 0)
|
||||
}
|
||||
|
||||
// RunLicenseHeartbeatLoop keeps the license server aware of active deployments.
|
||||
// The loop checks periodically, but only sends when the last stored heartbeat is
|
||||
// older than licenseHeartbeatInterval.
|
||||
func RunLicenseHeartbeatLoop(ctx context.Context, svc *service.Container) {
|
||||
if svc == nil {
|
||||
return
|
||||
}
|
||||
run := func(interval time.Duration) {
|
||||
state, sent, err := maybeSendLicenseHeartbeat(ctx, svc, interval)
|
||||
logLicenseHeartbeatResult(svc, state, sent, err)
|
||||
}
|
||||
runStartup := func() {
|
||||
state, sent, err := maybeSendStartupLicenseHeartbeat(ctx, svc)
|
||||
logLicenseHeartbeatResult(svc, state, sent, err)
|
||||
}
|
||||
|
||||
timer := time.NewTimer(licenseHeartbeatStartupDelay)
|
||||
defer timer.Stop()
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-timer.C:
|
||||
runStartup()
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(licenseHeartbeatCheckInterval)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
run(licenseHeartbeatInterval)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func logLicenseHeartbeatResult(svc *service.Container, state service.LicenseActivationState, sent bool, err error) {
|
||||
if svc == nil || svc.Log == nil {
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
svc.Log.Warn("license heartbeat failed", zap.Error(err))
|
||||
return
|
||||
}
|
||||
if sent {
|
||||
svc.Log.Info("license heartbeat sent", zap.String("device_id", state.DeviceID))
|
||||
}
|
||||
}
|
||||
|
||||
func maybeSendStartupLicenseHeartbeat(ctx context.Context, svc *service.Container) (service.LicenseActivationState, bool, error) {
|
||||
state, err := loadLicenseState(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
if strings.TrimSpace(state.LicenseKey) == "" {
|
||||
return state, false, nil
|
||||
}
|
||||
return maybeSendLicenseHeartbeat(ctx, svc, 0)
|
||||
}
|
||||
|
||||
func maybeSendLicenseHeartbeat(ctx context.Context, svc *service.Container, interval time.Duration) (service.LicenseActivationState, bool, error) {
|
||||
state, err := loadLicenseState(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
if !licenseHeartbeatEligible(state) {
|
||||
return state, false, nil
|
||||
}
|
||||
if !licenseHeartbeatDue(state, interval) {
|
||||
client, clientErr := newLicenseClient(ctx, svc)
|
||||
if clientErr != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
deviceID, idErr := ensureLicenseDeviceID(ctx, svc, state.DeviceID)
|
||||
if idErr != nil {
|
||||
return state, false, idErr
|
||||
}
|
||||
refreshed, ok, requested, refreshErr := refreshLicenseServerStatus(ctx, client, state, deviceID)
|
||||
if refreshErr == nil && ok {
|
||||
state = refreshed
|
||||
_ = persistLicenseState(ctx, svc, state)
|
||||
}
|
||||
if !requested {
|
||||
return state, false, nil
|
||||
}
|
||||
}
|
||||
next, err := sendLicenseHeartbeat(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, err
|
||||
}
|
||||
return next, true, nil
|
||||
}
|
||||
|
||||
func licenseHeartbeatEligible(state service.LicenseActivationState) bool {
|
||||
return strings.TrimSpace(state.LicenseKey) != ""
|
||||
}
|
||||
|
||||
func licenseHeartbeatDue(state service.LicenseActivationState, interval time.Duration) bool {
|
||||
if interval <= 0 {
|
||||
return true
|
||||
}
|
||||
updatedAt := strings.TrimSpace(state.UpdatedAt)
|
||||
if updatedAt == "" {
|
||||
return true
|
||||
}
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02 15:04:05", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, updatedAt); err == nil {
|
||||
return time.Since(t) >= interval
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func sendLicenseHeartbeat(ctx context.Context, svc *service.Container) (service.LicenseActivationState, error) {
|
||||
client, err := newLicenseClient(ctx, svc)
|
||||
if err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
oldState, _ := loadLicenseState(ctx, svc)
|
||||
deviceID, err := ensureLicenseDeviceID(ctx, svc, oldState.DeviceID)
|
||||
if err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
deviceName, _ := svc.Repo.Setting.Get(ctx, licenseDeviceNameSetting)
|
||||
if strings.TrimSpace(deviceName) == "" {
|
||||
deviceName = defaultLicenseDeviceName()
|
||||
_ = svc.Repo.Setting.Set(ctx, licenseDeviceNameSetting, deviceName)
|
||||
}
|
||||
var upstream licenseServerSignedResp
|
||||
if err := client.post(ctx, "/api/v1/heartbeat", licenseHeartbeatPayload(oldState, deviceID, deviceName), &upstream); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
if err := client.verifySigned(&upstream); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
state := licenseStateFromSigned(upstream, deviceID, deviceName)
|
||||
state.LicenseKey = oldState.LicenseKey
|
||||
if refreshed, ok, _, refreshErr := refreshLicenseServerStatus(ctx, client, state, deviceID); refreshErr == nil && ok {
|
||||
refreshed.LicenseKey = state.LicenseKey
|
||||
state = refreshed
|
||||
}
|
||||
if err := persistLicenseState(ctx, svc, state); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
return state, nil
|
||||
}
|
||||
|
||||
func refreshLicenseServerStatus(ctx context.Context, client *licenseClient, state service.LicenseActivationState, deviceID string) (service.LicenseActivationState, bool, bool, error) {
|
||||
var upstream licenseServerStatusResp
|
||||
if err := client.get(ctx, "/api/v1/status/"+url.PathEscape(deviceID), &upstream); err != nil {
|
||||
return state, false, false, err
|
||||
}
|
||||
if !upstream.Valid {
|
||||
state.Valid = false
|
||||
state.UpdatedAt = time.Now().Format(time.RFC3339)
|
||||
return state, false, upstream.HeartbeatRequested, nil
|
||||
}
|
||||
applyLicenseStatus(&state, upstream, deviceID)
|
||||
return state, true, upstream.HeartbeatRequested, nil
|
||||
}
|
||||
|
||||
func ensureLicenseDeviceID(ctx context.Context, svc *service.Container, candidate string) (string, error) {
|
||||
if strings.TrimSpace(candidate) != "" {
|
||||
return strings.TrimSpace(candidate), svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, strings.TrimSpace(candidate))
|
||||
}
|
||||
existing, err := svc.Repo.Setting.Get(ctx, licenseDeviceIDSetting)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if strings.TrimSpace(existing) != "" {
|
||||
return strings.TrimSpace(existing), nil
|
||||
}
|
||||
var buf [16]byte
|
||||
if _, err := rand.Read(buf[:]); err != nil {
|
||||
return "", err
|
||||
}
|
||||
id := "msgo-" + hex.EncodeToString(buf[:])
|
||||
return id, svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, id)
|
||||
}
|
||||
|
||||
func defaultLicenseDeviceName() string {
|
||||
host, _ := os.Hostname()
|
||||
if strings.TrimSpace(host) == "" {
|
||||
return "MediaStationGo Server"
|
||||
}
|
||||
return "MediaStationGo - " + host
|
||||
}
|
||||
|
||||
func licenseStateFromSigned(resp licenseServerSignedResp, deviceID, deviceName string) service.LicenseActivationState {
|
||||
expiry := ""
|
||||
if resp.ExpiryDate != nil {
|
||||
expiry = *resp.ExpiryDate
|
||||
}
|
||||
return service.LicenseActivationState{
|
||||
Valid: resp.Valid,
|
||||
LicenseType: resp.LicenseType,
|
||||
ExpiryDate: expiry,
|
||||
MaxDevices: resp.MaxDevices,
|
||||
MaxUsers: resp.MaxUsers,
|
||||
UnlimitedUsers: !resp.LegacySignature && resp.MaxUsers == nil,
|
||||
DaysRemaining: resp.DaysRemaining,
|
||||
NextHeartbeat: resp.NextHeartbeat,
|
||||
DeviceID: deviceID,
|
||||
DeviceName: deviceName,
|
||||
UpdatedAt: time.Now().Format(time.RFC3339),
|
||||
}
|
||||
}
|
||||
|
||||
func licenseHeartbeatPayload(state service.LicenseActivationState, deviceID, deviceName string) map[string]any {
|
||||
payload := map[string]any{
|
||||
"fingerprint": deviceID,
|
||||
"instance_id": deviceID,
|
||||
"device_name": deviceName,
|
||||
}
|
||||
if key := strings.TrimSpace(state.LicenseKey); key != "" {
|
||||
payload["key"] = key
|
||||
}
|
||||
return payload
|
||||
}
|
||||
|
||||
func licenseStatusMaxUsers(state service.LicenseActivationState) any {
|
||||
active := state.Valid && !licenseStateExpired(state.ExpiryDate)
|
||||
if active {
|
||||
if state.UnlimitedUsers {
|
||||
return nil
|
||||
}
|
||||
if state.MaxUsers != nil && *state.MaxUsers > 0 {
|
||||
return *state.MaxUsers
|
||||
}
|
||||
return service.LicensedUserLimit
|
||||
}
|
||||
return service.OpenSourceUserLimit
|
||||
}
|
||||
|
||||
func applyLicenseStatus(state *service.LicenseActivationState, upstream licenseServerStatusResp, deviceID string) {
|
||||
state.Valid = upstream.Valid
|
||||
if upstream.LicenseType != nil {
|
||||
state.LicenseType = *upstream.LicenseType
|
||||
}
|
||||
if upstream.ExpiryDate != nil {
|
||||
state.ExpiryDate = *upstream.ExpiryDate
|
||||
} else {
|
||||
state.ExpiryDate = ""
|
||||
}
|
||||
if upstream.MaxDevices > 0 {
|
||||
state.MaxDevices = upstream.MaxDevices
|
||||
}
|
||||
state.MaxUsers = upstream.MaxUsers
|
||||
state.UnlimitedUsers = upstream.UnlimitedUsers
|
||||
state.DaysRemaining = upstream.DaysRemaining
|
||||
if upstream.DeviceName != "" {
|
||||
state.DeviceName = upstream.DeviceName
|
||||
}
|
||||
state.DeviceID = deviceID
|
||||
state.UpdatedAt = time.Now().Format(time.RFC3339)
|
||||
}
|
||||
|
||||
func persistLicenseState(ctx context.Context, svc *service.Container, state service.LicenseActivationState) error {
|
||||
data, err := json.Marshal(state)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if strings.TrimSpace(state.DeviceID) != "" {
|
||||
_ = svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, strings.TrimSpace(state.DeviceID))
|
||||
}
|
||||
return svc.Repo.Setting.Set(ctx, service.LicenseSettingActivation, string(data))
|
||||
}
|
||||
|
||||
func loadLicenseState(ctx context.Context, svc *service.Container) (service.LicenseActivationState, error) {
|
||||
raw, err := svc.Repo.Setting.Get(ctx, service.LicenseSettingActivation)
|
||||
if err != nil || raw == "" {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
var state service.LicenseActivationState
|
||||
if err := json.Unmarshal([]byte(raw), &state); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
return state, nil
|
||||
}
|
||||
|
||||
func licenseActivationView(state service.LicenseActivationState) gin.H {
|
||||
updatedAt := state.UpdatedAt
|
||||
if strings.TrimSpace(updatedAt) == "" {
|
||||
updatedAt = time.Now().Format(time.RFC3339)
|
||||
}
|
||||
return gin.H{
|
||||
"id": state.DeviceID,
|
||||
"key_id": state.LicenseType,
|
||||
"key": maskLicenseKey(state.LicenseKey),
|
||||
"device_id": state.DeviceID,
|
||||
"device_name": state.DeviceName,
|
||||
"plan": state.LicenseType,
|
||||
"max_activations": state.MaxDevices,
|
||||
"max_users": state.MaxUsers,
|
||||
"unlimited_users": state.UnlimitedUsers,
|
||||
"expires_at": emptyAsNil(state.ExpiryDate),
|
||||
"valid": state.Valid && !licenseStateExpired(state.ExpiryDate),
|
||||
"heartbeat_at": updatedAt,
|
||||
"created_at": updatedAt,
|
||||
}
|
||||
}
|
||||
|
||||
func maskLicenseKey(key string) string {
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
return ""
|
||||
}
|
||||
if len(key) <= 8 {
|
||||
return key
|
||||
}
|
||||
return key[:5] + "..." + key[len(key)-4:]
|
||||
}
|
||||
|
||||
func licenseStatusMessage(active bool, clientErr error) string {
|
||||
if active {
|
||||
return "已激活"
|
||||
}
|
||||
if clientErr != nil && !strings.Contains(clientErr.Error(), "not configured") {
|
||||
return clientErr.Error()
|
||||
}
|
||||
return "开源版：最多 20 个用户"
|
||||
}
|
||||
|
||||
func licenseStateExpired(expiry string) bool {
|
||||
if expiry == "" {
|
||||
return false
|
||||
}
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02 15:04:05", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, expiry); err == nil {
|
||||
return time.Now().After(t)
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func emptyAsNil(v string) any {
|
||||
if strings.TrimSpace(v) == "" {
|
||||
return nil
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
@@ -0,0 +1,172 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/ShukeBta/MediaStationGo/internal/service"
|
||||
)
|
||||
|
||||
const (
|
||||
licenseHeartbeatInterval = 12 * time.Hour
|
||||
licenseHeartbeatCheckInterval = 30 * time.Minute
|
||||
licenseHeartbeatStartupDelay = 2 * time.Minute
|
||||
)
|
||||
|
||||
func refreshLicenseCapacityBestEffort(ctx context.Context, svc *service.Container) {
|
||||
if svc == nil || svc.Repo == nil || svc.Repo.Setting == nil {
|
||||
return
|
||||
}
|
||||
_, _, _ = maybeSendLicenseHeartbeat(ctx, svc, 0)
|
||||
}
|
||||
|
||||
// RunLicenseHeartbeatLoop keeps the license server aware of active deployments.
|
||||
// The loop checks periodically, but only sends when the last stored heartbeat is
|
||||
// older than licenseHeartbeatInterval.
|
||||
func RunLicenseHeartbeatLoop(ctx context.Context, svc *service.Container) {
|
||||
if svc == nil {
|
||||
return
|
||||
}
|
||||
run := func(interval time.Duration) {
|
||||
state, sent, err := maybeSendLicenseHeartbeat(ctx, svc, interval)
|
||||
logLicenseHeartbeatResult(svc, state, sent, err)
|
||||
}
|
||||
runStartup := func() {
|
||||
state, sent, err := maybeSendStartupLicenseHeartbeat(ctx, svc)
|
||||
logLicenseHeartbeatResult(svc, state, sent, err)
|
||||
}
|
||||
|
||||
timer := time.NewTimer(licenseHeartbeatStartupDelay)
|
||||
defer timer.Stop()
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-timer.C:
|
||||
runStartup()
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(licenseHeartbeatCheckInterval)
|
||||
defer ticker.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
run(licenseHeartbeatInterval)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func logLicenseHeartbeatResult(svc *service.Container, state service.LicenseActivationState, sent bool, err error) {
|
||||
if svc == nil || svc.Log == nil {
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
svc.Log.Warn("license heartbeat failed", zap.Error(err))
|
||||
return
|
||||
}
|
||||
if sent {
|
||||
svc.Log.Info("license heartbeat sent", zap.String("device_id", state.DeviceID))
|
||||
}
|
||||
}
|
||||
|
||||
func maybeSendStartupLicenseHeartbeat(ctx context.Context, svc *service.Container) (service.LicenseActivationState, bool, error) {
|
||||
state, err := loadLicenseState(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
if strings.TrimSpace(state.LicenseKey) == "" {
|
||||
return state, false, nil
|
||||
}
|
||||
return maybeSendLicenseHeartbeat(ctx, svc, 0)
|
||||
}
|
||||
|
||||
func maybeSendLicenseHeartbeat(ctx context.Context, svc *service.Container, interval time.Duration) (service.LicenseActivationState, bool, error) {
|
||||
state, err := loadLicenseState(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
if !licenseHeartbeatEligible(state) {
|
||||
return state, false, nil
|
||||
}
|
||||
if !licenseHeartbeatDue(state, interval) {
|
||||
client, clientErr := newLicenseClient(ctx, svc)
|
||||
if clientErr != nil {
|
||||
return state, false, nil
|
||||
}
|
||||
deviceID, idErr := ensureLicenseDeviceID(ctx, svc, state.DeviceID)
|
||||
if idErr != nil {
|
||||
return state, false, idErr
|
||||
}
|
||||
refreshed, ok, requested, refreshErr := refreshLicenseServerStatus(ctx, client, state, deviceID)
|
||||
if refreshErr == nil && ok {
|
||||
state = refreshed
|
||||
_ = persistLicenseState(ctx, svc, state)
|
||||
}
|
||||
if !requested {
|
||||
return state, false, nil
|
||||
}
|
||||
}
|
||||
next, err := sendLicenseHeartbeat(ctx, svc)
|
||||
if err != nil {
|
||||
return state, false, err
|
||||
}
|
||||
return next, true, nil
|
||||
}
|
||||
|
||||
func licenseHeartbeatEligible(state service.LicenseActivationState) bool {
|
||||
return strings.TrimSpace(state.LicenseKey) != ""
|
||||
}
|
||||
|
||||
func licenseHeartbeatDue(state service.LicenseActivationState, interval time.Duration) bool {
|
||||
if interval <= 0 {
|
||||
return true
|
||||
}
|
||||
updatedAt := strings.TrimSpace(state.UpdatedAt)
|
||||
if updatedAt == "" {
|
||||
return true
|
||||
}
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02 15:04:05", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, updatedAt); err == nil {
|
||||
return time.Since(t) >= interval
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func sendLicenseHeartbeat(ctx context.Context, svc *service.Container) (service.LicenseActivationState, error) {
|
||||
client, err := newLicenseClient(ctx, svc)
|
||||
if err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
oldState, _ := loadLicenseState(ctx, svc)
|
||||
deviceID, err := ensureLicenseDeviceID(ctx, svc, oldState.DeviceID)
|
||||
if err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
deviceName, _ := svc.Repo.Setting.Get(ctx, licenseDeviceNameSetting)
|
||||
if strings.TrimSpace(deviceName) == "" {
|
||||
deviceName = defaultLicenseDeviceName()
|
||||
_ = svc.Repo.Setting.Set(ctx, licenseDeviceNameSetting, deviceName)
|
||||
}
|
||||
var upstream licenseServerSignedResp
|
||||
if err := client.post(ctx, "/api/v1/heartbeat", licenseHeartbeatPayload(oldState, deviceID, deviceName), &upstream); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
if err := client.verifySigned(&upstream); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
state := licenseStateFromSigned(upstream, deviceID, deviceName)
|
||||
state.LicenseKey = oldState.LicenseKey
|
||||
if refreshed, ok, _, refreshErr := refreshLicenseServerStatus(ctx, client, state, deviceID); refreshErr == nil && ok {
|
||||
refreshed.LicenseKey = state.LicenseKey
|
||||
state = refreshed
|
||||
}
|
||||
if err := persistLicenseState(ctx, svc, state); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
return state, nil
|
||||
}
|
||||
@@ -0,0 +1,211 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/ShukeBta/MediaStationGo/internal/service"
|
||||
)
|
||||
|
||||
func refreshLicenseServerStatus(ctx context.Context, client *licenseClient, state service.LicenseActivationState, deviceID string) (service.LicenseActivationState, bool, bool, error) {
|
||||
var upstream licenseServerStatusResp
|
||||
if err := client.get(ctx, "/api/v1/status/"+url.PathEscape(deviceID), &upstream); err != nil {
|
||||
return state, false, false, err
|
||||
}
|
||||
if !upstream.Valid {
|
||||
state.Valid = false
|
||||
state.UpdatedAt = time.Now().Format(time.RFC3339)
|
||||
return state, false, upstream.HeartbeatRequested, nil
|
||||
}
|
||||
applyLicenseStatus(&state, upstream, deviceID)
|
||||
return state, true, upstream.HeartbeatRequested, nil
|
||||
}
|
||||
|
||||
func ensureLicenseDeviceID(ctx context.Context, svc *service.Container, candidate string) (string, error) {
|
||||
if strings.TrimSpace(candidate) != "" {
|
||||
return strings.TrimSpace(candidate), svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, strings.TrimSpace(candidate))
|
||||
}
|
||||
existing, err := svc.Repo.Setting.Get(ctx, licenseDeviceIDSetting)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if strings.TrimSpace(existing) != "" {
|
||||
return strings.TrimSpace(existing), nil
|
||||
}
|
||||
var buf [16]byte
|
||||
if _, err := rand.Read(buf[:]); err != nil {
|
||||
return "", err
|
||||
}
|
||||
id := "msgo-" + hex.EncodeToString(buf[:])
|
||||
return id, svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, id)
|
||||
}
|
||||
|
||||
func defaultLicenseDeviceName() string {
|
||||
host, _ := os.Hostname()
|
||||
if strings.TrimSpace(host) == "" {
|
||||
return "MediaStationGo Server"
|
||||
}
|
||||
return "MediaStationGo - " + host
|
||||
}
|
||||
|
||||
func licenseStateFromSigned(resp licenseServerSignedResp, deviceID, deviceName string) service.LicenseActivationState {
|
||||
expiry := ""
|
||||
if resp.ExpiryDate != nil {
|
||||
expiry = *resp.ExpiryDate
|
||||
}
|
||||
return service.LicenseActivationState{
|
||||
Valid: resp.Valid,
|
||||
LicenseType: resp.LicenseType,
|
||||
ExpiryDate: expiry,
|
||||
MaxDevices: resp.MaxDevices,
|
||||
MaxUsers: resp.MaxUsers,
|
||||
UnlimitedUsers: !resp.LegacySignature && resp.MaxUsers == nil,
|
||||
DaysRemaining: resp.DaysRemaining,
|
||||
NextHeartbeat: resp.NextHeartbeat,
|
||||
DeviceID: deviceID,
|
||||
DeviceName: deviceName,
|
||||
UpdatedAt: time.Now().Format(time.RFC3339),
|
||||
}
|
||||
}
|
||||
|
||||
func licenseHeartbeatPayload(state service.LicenseActivationState, deviceID, deviceName string) map[string]any {
|
||||
payload := map[string]any{
|
||||
"fingerprint": deviceID,
|
||||
"instance_id": deviceID,
|
||||
"device_name": deviceName,
|
||||
}
|
||||
if key := strings.TrimSpace(state.LicenseKey); key != "" {
|
||||
payload["key"] = key
|
||||
}
|
||||
return payload
|
||||
}
|
||||
|
||||
func licenseStatusMaxUsers(state service.LicenseActivationState) any {
|
||||
active := state.Valid && !licenseStateExpired(state.ExpiryDate)
|
||||
if active {
|
||||
if state.UnlimitedUsers {
|
||||
return nil
|
||||
}
|
||||
if state.MaxUsers != nil && *state.MaxUsers > 0 {
|
||||
return *state.MaxUsers
|
||||
}
|
||||
return service.LicensedUserLimit
|
||||
}
|
||||
return service.OpenSourceUserLimit
|
||||
}
|
||||
|
||||
func applyLicenseStatus(state *service.LicenseActivationState, upstream licenseServerStatusResp, deviceID string) {
|
||||
state.Valid = upstream.Valid
|
||||
if upstream.LicenseType != nil {
|
||||
state.LicenseType = *upstream.LicenseType
|
||||
}
|
||||
if upstream.ExpiryDate != nil {
|
||||
state.ExpiryDate = *upstream.ExpiryDate
|
||||
} else {
|
||||
state.ExpiryDate = ""
|
||||
}
|
||||
if upstream.MaxDevices > 0 {
|
||||
state.MaxDevices = upstream.MaxDevices
|
||||
}
|
||||
state.MaxUsers = upstream.MaxUsers
|
||||
state.UnlimitedUsers = upstream.UnlimitedUsers
|
||||
state.DaysRemaining = upstream.DaysRemaining
|
||||
if upstream.DeviceName != "" {
|
||||
state.DeviceName = upstream.DeviceName
|
||||
}
|
||||
state.DeviceID = deviceID
|
||||
state.UpdatedAt = time.Now().Format(time.RFC3339)
|
||||
}
|
||||
|
||||
func persistLicenseState(ctx context.Context, svc *service.Container, state service.LicenseActivationState) error {
|
||||
data, err := json.Marshal(state)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if strings.TrimSpace(state.DeviceID) != "" {
|
||||
_ = svc.Repo.Setting.Set(ctx, licenseDeviceIDSetting, strings.TrimSpace(state.DeviceID))
|
||||
}
|
||||
return svc.Repo.Setting.Set(ctx, service.LicenseSettingActivation, string(data))
|
||||
}
|
||||
|
||||
func loadLicenseState(ctx context.Context, svc *service.Container) (service.LicenseActivationState, error) {
|
||||
raw, err := svc.Repo.Setting.Get(ctx, service.LicenseSettingActivation)
|
||||
if err != nil || raw == "" {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
var state service.LicenseActivationState
|
||||
if err := json.Unmarshal([]byte(raw), &state); err != nil {
|
||||
return service.LicenseActivationState{}, err
|
||||
}
|
||||
return state, nil
|
||||
}
|
||||
|
||||
func licenseActivationView(state service.LicenseActivationState) gin.H {
|
||||
updatedAt := state.UpdatedAt
|
||||
if strings.TrimSpace(updatedAt) == "" {
|
||||
updatedAt = time.Now().Format(time.RFC3339)
|
||||
}
|
||||
return gin.H{
|
||||
"id": state.DeviceID,
|
||||
"key_id": state.LicenseType,
|
||||
"key": maskLicenseKey(state.LicenseKey),
|
||||
"device_id": state.DeviceID,
|
||||
"device_name": state.DeviceName,
|
||||
"plan": state.LicenseType,
|
||||
"max_activations": state.MaxDevices,
|
||||
"max_users": state.MaxUsers,
|
||||
"unlimited_users": state.UnlimitedUsers,
|
||||
"expires_at": emptyAsNil(state.ExpiryDate),
|
||||
"valid": state.Valid && !licenseStateExpired(state.ExpiryDate),
|
||||
"heartbeat_at": updatedAt,
|
||||
"created_at": updatedAt,
|
||||
}
|
||||
}
|
||||
|
||||
func maskLicenseKey(key string) string {
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
return ""
|
||||
}
|
||||
if len(key) <= 8 {
|
||||
return key
|
||||
}
|
||||
return key[:5] + "..." + key[len(key)-4:]
|
||||
}
|
||||
|
||||
func licenseStatusMessage(active bool, clientErr error) string {
|
||||
if active {
|
||||
return "已激活"
|
||||
}
|
||||
if clientErr != nil && !strings.Contains(clientErr.Error(), "not configured") {
|
||||
return clientErr.Error()
|
||||
}
|
||||
return "开源版：最多 20 个用户"
|
||||
}
|
||||
|
||||
func licenseStateExpired(expiry string) bool {
|
||||
if expiry == "" {
|
||||
return false
|
||||
}
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02 15:04:05", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, expiry); err == nil {
|
||||
return time.Now().After(t)
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func emptyAsNil(v string) any {
|
||||
if strings.TrimSpace(v) == "" {
|
||||
return nil
|
||||
}
|
||||
return v
|
||||
}
|
||||
Reference in New Issue
Block a user