108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"go-site-clone/types"
|
|
"time"
|
|
)
|
|
|
|
// SiteRecord 站点数据记录（带时间戳）
|
|
type SiteRecord struct {
|
|
types.NginxSiteConfig
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// AddSite 添加站点配置
|
|
func (s *Store) AddSite(site types.NginxSiteConfig) error {
|
|
record := SiteRecord{
|
|
NginxSiteConfig: site,
|
|
CreatedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
}
|
|
return s.Set(BucketSites, site.Name, record)
|
|
}
|
|
|
|
// GetSite 获取站点配置
|
|
func (s *Store) GetSite(siteName string) (*types.NginxSiteConfig, error) {
|
|
var record SiteRecord
|
|
if err := s.Get(BucketSites, siteName, &record); err != nil {
|
|
return nil, err
|
|
}
|
|
return &record.NginxSiteConfig, nil
|
|
}
|
|
|
|
// UpdateSite 更新站点配置
|
|
func (s *Store) UpdateSite(site types.NginxSiteConfig) error {
|
|
var record SiteRecord
|
|
|
|
// 尝试获取现有记录以保留创建时间
|
|
if err := s.Get(BucketSites, site.Name, &record); err == nil {
|
|
record.NginxSiteConfig = site
|
|
record.UpdatedAt = time.Now()
|
|
} else {
|
|
// 如果不存在，创建新记录
|
|
record = SiteRecord{
|
|
NginxSiteConfig: site,
|
|
CreatedAt: time.Now(),
|
|
UpdatedAt: time.Now(),
|
|
}
|
|
}
|
|
|
|
return s.Set(BucketSites, site.Name, record)
|
|
}
|
|
|
|
// DeleteSite 删除站点配置
|
|
func (s *Store) DeleteSite(siteName string) error {
|
|
return s.Delete(BucketSites, siteName)
|
|
}
|
|
|
|
// GetAllSites 获取所有站点配置
|
|
func (s *Store) GetAllSites() ([]types.NginxSiteConfig, error) {
|
|
var records []SiteRecord
|
|
if err := s.GetAll(BucketSites, &records); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sites := make([]types.NginxSiteConfig, len(records))
|
|
for i, record := range records {
|
|
sites[i] = record.NginxSiteConfig
|
|
}
|
|
return sites, nil
|
|
}
|
|
|
|
// SiteExists 检查站点是否存在
|
|
func (s *Store) SiteExists(siteName string) bool {
|
|
return s.Exists(BucketSites, siteName)
|
|
}
|
|
|
|
// GetSitesByStatus 根据启用状态获取站点
|
|
func (s *Store) GetSitesByStatus(enabled bool) ([]types.NginxSiteConfig, error) {
|
|
allSites, err := s.GetAllSites()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var filtered []types.NginxSiteConfig
|
|
for _, site := range allSites {
|
|
if site.Enabled == enabled {
|
|
filtered = append(filtered, site)
|
|
}
|
|
}
|
|
return filtered, nil
|
|
} // UpdateSiteStatus 更新站点启用状态
|
|
func (s *Store) UpdateSiteStatus(siteName string, enabled bool) error {
|
|
site, err := s.GetSite(siteName)
|
|
if err != nil {
|
|
return fmt.Errorf("站点不存在: %v", err)
|
|
}
|
|
|
|
site.Enabled = enabled
|
|
return s.UpdateSite(*site)
|
|
}
|
|
|
|
// GetSitesCount 获取站点总数
|
|
func (s *Store) GetSitesCount() int {
|
|
return s.Count(BucketSites)
|
|
}
|