T3MP3ST public launch build

2026-07-03 10:57:20 -04:00
1 parent 972247b881
commit 184c7d4e59
@@ -0,0 +1,87 @@
+package model
+
+import (
+	"log"
+	"slices"
+	"time"
+
+	"github.com/goccy/go-json"
+	"gorm.io/gorm"
+
+	pb "github.com/nezhahq/nezha/proto"
+)
+
+type Server struct {
+	Common
+
+	Name                   string `json:"name"`
+	UUID                   string `json:"uuid,omitempty" gorm:"unique"`
+	Note                   string `json:"note,omitempty"`           // 管理员可见备注
+	PublicNote             string `json:"public_note,omitempty"`    // 公开备注
+	DisplayIndex           int    `json:"display_index"`            // 展示排序，越大越靠前
+	HideForGuest           bool   `json:"hide_for_guest,omitempty"` // 对游客隐藏
+	EnableDDNS             bool   `json:"enable_ddns,omitempty"`    // 启用DDNS
+	DDNSProfilesRaw        string `gorm:"default:'[]';column:ddns_profiles_raw" json:"-"`
+	OverrideDDNSDomainsRaw string `gorm:"default:'{}';column:override_ddns_domains_raw" json:"-"`
+
+	DDNSProfiles        []uint64            `gorm:"-" json:"ddns_profiles,omitempty" validate:"optional"` // DDNS配置
+	OverrideDDNSDomains map[uint64][]string `gorm:"-" json:"override_ddns_domains,omitempty" validate:"optional"`
+
+	Host       *Host      `gorm:"-" json:"host,omitempty"`
+	State      *HostState `gorm:"-" json:"state,omitempty"`
+	GeoIP      *GeoIP     `gorm:"-" json:"geoip,omitempty"`
+	LastActive time.Time  `gorm:"-" json:"last_active,omitempty"`
+
+	TaskStream  pb.NezhaService_RequestTaskServer `gorm:"-" json:"-"`
+	ConfigCache chan any                          `gorm:"-" json:"-"`
+
+	PrevTransferInSnapshot  uint64 `gorm:"-" json:"-"` // 上次数据点时的入站使用量
+	PrevTransferOutSnapshot uint64 `gorm:"-" json:"-"` // 上次数据点时的出站使用量
+}
+
+func InitServer(s *Server) {
+	s.Host = &Host{}
+	s.State = &HostState{}
+	s.GeoIP = &GeoIP{}
+	s.ConfigCache = make(chan any, 1)
+}
+
+func (s *Server) CopyFromRunningServer(old *Server) {
+	s.Host = old.Host
+	s.State = old.State
+	s.GeoIP = old.GeoIP
+	s.LastActive = old.LastActive
+	s.TaskStream = old.TaskStream
+	s.ConfigCache = old.ConfigCache
+	s.PrevTransferInSnapshot = old.PrevTransferInSnapshot
+	s.PrevTransferOutSnapshot = old.PrevTransferOutSnapshot
+}
+
+func (s *Server) AfterFind(tx *gorm.DB) error {
+	if s.DDNSProfilesRaw != "" {
+		if err := json.Unmarshal([]byte(s.DDNSProfilesRaw), &s.DDNSProfiles); err != nil {
+			log.Println("NEZHA>> Server.AfterFind:", err)
+			return nil
+		}
+	}
+	if s.OverrideDDNSDomainsRaw != "" {
+		if err := json.Unmarshal([]byte(s.OverrideDDNSDomainsRaw), &s.OverrideDDNSDomains); err != nil {
+			log.Println("NEZHA>> Server.AfterFind:", err)
+			return nil
+		}
+	}
+	return nil
+}
+
+func (s *Server) SplitList(x []*Server) ([]*Server, []*Server) {
+	pri := func(s *Server) bool {
+		return s.DisplayIndex == 0
+	}
+
+	i := slices.IndexFunc(x, pri)
+	if i == -1 {
+		return nil, x
+	}
+
+	return x[:i], x[i:]
+}