add(func): spuuort virtual host

Author: PIKACHUIM

internal/conf/const.go

@@ -191,4 +191,5 @@ const (
 	PathKey
 	SharingIDKey
 	SkipHookKey
+	VirtualHostKey
 )

internal/db/db.go

@@ -12,7 +12,7 @@ var db *gorm.DB
 
 func Init(d *gorm.DB) {
 	db = d
-	err := AutoMigrate(new(model.Storage), new(model.User), new(model.Meta), new(model.SettingItem), new(model.SearchNode), new(model.TaskItem), new(model.SSHPublicKey), new(model.SharingDB))
+	err := AutoMigrate(new(model.Storage), new(model.User), new(model.Meta), new(model.SettingItem), new(model.SearchNode), new(model.TaskItem), new(model.SSHPublicKey), new(model.SharingDB), new(model.VirtualHost))
 	if err != nil {
 		log.Fatalf("failed migrate database: %s", err.Error())
 	}

internal/db/virtual_host.go

@@ -0,0 +1,45 @@
+package db
+
+import (
+	"github.com/OpenListTeam/OpenList/v4/internal/model"
+	"github.com/pkg/errors"
+)
+
+func GetVirtualHostByDomain(domain string) (*model.VirtualHost, error) {
+	var v model.VirtualHost
+	if err := db.Where("domain = ?", domain).First(&v).Error; err != nil {
+		return nil, errors.Wrapf(err, "failed select virtual host")
+	}
+	return &v, nil
+}
+
+func GetVirtualHostById(id uint) (*model.VirtualHost, error) {
+	var v model.VirtualHost
+	if err := db.First(&v, id).Error; err != nil {
+		return nil, errors.Wrapf(err, "failed get virtual host")
+	}
+	return &v, nil
+}
+
+func CreateVirtualHost(v *model.VirtualHost) error {
+	return errors.WithStack(db.Create(v).Error)
+}
+
+func UpdateVirtualHost(v *model.VirtualHost) error {
+	return errors.WithStack(db.Save(v).Error)
+}
+
+func GetVirtualHosts(pageIndex, pageSize int) (vhosts []model.VirtualHost, count int64, err error) {
+	vhostDB := db.Model(&model.VirtualHost{})
+	if err = vhostDB.Count(&count).Error; err != nil {
+		return nil, 0, errors.Wrapf(err, "failed get virtual hosts count")
+	}
+	if err = vhostDB.Order(columnName("id")).Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&vhosts).Error; err != nil {
+		return nil, 0, errors.Wrapf(err, "failed find virtual hosts")
+	}
+	return vhosts, count, nil
+}
+
+func DeleteVirtualHostById(id uint) error {
+	return errors.WithStack(db.Delete(&model.VirtualHost{}, id).Error)
+}

internal/model/virtual_host.go

@@ -0,0 +1,9 @@
+package model
+
+type VirtualHost struct {
+	ID         uint   `json:"id" gorm:"primaryKey"`
+	Enabled    bool   `json:"enabled"`
+	Domain     string `json:"domain" gorm:"unique" binding:"required"`
+	Path       string `json:"path" binding:"required"`
+	WebHosting bool   `json:"web_hosting"`
+}

server/handles/virtual_host.go

@@ -0,0 +1,83 @@
+package handles
+
+import (
+	"strconv"
+
+	"github.com/OpenListTeam/OpenList/v4/internal/model"
+	"github.com/OpenListTeam/OpenList/v4/internal/op"
+	"github.com/OpenListTeam/OpenList/v4/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+func ListVirtualHosts(c *gin.Context) {
+	var req model.PageReq
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	req.Validate()
+	vhosts, total, err := op.GetVirtualHosts(req.Page, req.PerPage)
+	if err != nil {
+		common.ErrorResp(c, err, 500, true)
+		return
+	}
+	common.SuccessResp(c, common.PageResp{
+		Content: vhosts,
+		Total:   total,
+	})
+}
+
+func GetVirtualHost(c *gin.Context) {
+	idStr := c.Query("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	vhost, err := op.GetVirtualHostById(uint(id))
+	if err != nil {
+		common.ErrorResp(c, err, 500, true)
+		return
+	}
+	common.SuccessResp(c, vhost)
+}
+
+func CreateVirtualHost(c *gin.Context) {
+	var req model.VirtualHost
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	if err := op.CreateVirtualHost(&req); err != nil {
+		common.ErrorResp(c, err, 500, true)
+	} else {
+		common.SuccessResp(c)
+	}
+}
+
+func UpdateVirtualHost(c *gin.Context) {
+	var req model.VirtualHost
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	if err := op.UpdateVirtualHost(&req); err != nil {
+		common.ErrorResp(c, err, 500, true)
+	} else {
+		common.SuccessResp(c)
+	}
+}
+
+func DeleteVirtualHost(c *gin.Context) {
+	idStr := c.Query("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	if err := op.DeleteVirtualHostById(uint(id)); err != nil {
+		common.ErrorResp(c, err, 500, true)
+		return
+	}
+	common.SuccessResp(c)
+}

server/middlewares/virtual_host.go

@@ -0,0 +1,43 @@
+package middlewares
+
+import (
+	"strings"
+
+	"github.com/OpenListTeam/OpenList/v4/internal/conf"
+	"github.com/OpenListTeam/OpenList/v4/internal/op"
+	"github.com/OpenListTeam/OpenList/v4/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+// VirtualHost 虚拟主机中间件，根据请求的 Host 头匹配虚拟主机配置
+func VirtualHost(c *gin.Context) {
+	host := c.Request.Host
+	// 去掉端口号
+	domain := stripPort(host)
+	if domain == "" {
+		c.Next()
+		return
+	}
+
+	vhost, err := op.GetVirtualHostByDomain(domain)
+	if err != nil || !vhost.Enabled {
+		// 未找到匹配的虚拟主机或未启用，继续正常处理
+		c.Next()
+		return
+	}
+
+	// 将虚拟主机信息存入请求上下文
+	common.GinWithValue(c, conf.VirtualHostKey, vhost)
+	c.Next()
+}
+
+// stripPort 去掉 host 中的端口号
+func stripPort(host string) string {
+	if idx := strings.LastIndex(host, ":"); idx != -1 {
+		// 确保不是 IPv6 地址（IPv6 地址用 [] 包裹）
+		if !strings.Contains(host, "[") {
+			return host[:idx]
+		}
+	}
+	return host
+}

server/router.go

@@ -36,6 +36,7 @@ func Init(e *gin.Engine) {
 	g.GET("/i/:link_name", handles.Plist)
 	common.SecretKey = []byte(conf.Conf.JwtSecret)
 	g.Use(middlewares.StoragesLoaded)
+	g.Use(middlewares.VirtualHost)
 	if conf.Conf.MaxConnections > 0 {
 		g.Use(middlewares.MaxAllowed(conf.Conf.MaxConnections))
 	}
@@ -122,6 +123,13 @@ func admin(g *gin.RouterGroup) {
 	meta.POST("/update", handles.UpdateMeta)
 	meta.POST("/delete", handles.DeleteMeta)
 
+	vhost := g.Group("/vhost")
+	vhost.GET("/list", handles.ListVirtualHosts)
+	vhost.GET("/get", handles.GetVirtualHost)
+	vhost.POST("/create", handles.CreateVirtualHost)
+	vhost.POST("/update", handles.UpdateVirtualHost)
+	vhost.POST("/delete", handles.DeleteVirtualHost)
+
 	user := g.Group("/user")
 	user.GET("/list", handles.ListUsers)
 	user.GET("/get", handles.GetUser)