- 增加资产附加属性功能

pkg/api/asset.go

@@ -106,6 +106,31 @@ func AssetUpdateEndpoint(c echo.Context) error {
 	return Success(c, nil)
 }
 
+func AssetGetAttributeEndpoint(c echo.Context) error {
+
+	assetId := c.Param("id")
+	attributeMap, err := model.FindAssetAttrMapByAssetId(assetId)
+	if err != nil {
+		return err
+	}
+	return Success(c, attributeMap)
+}
+
+func AssetUpdateAttributeEndpoint(c echo.Context) error {
+	m := echo.Map{}
+	if err := c.Bind(&m); err != nil {
+		return err
+	}
+
+	assetId := c.Param("id")
+	protocol := c.QueryParam("protocol")
+	err := model.UpdateAssetAttributes(assetId, protocol, m)
+	if err != nil {
+		return err
+	}
+	return Success(c, "")
+}
+
 func AssetDeleteEndpoint(c echo.Context) error {
 	id := c.Param("id")
 	split := strings.Split(id, ",")

pkg/api/routes.go

@@ -74,6 +74,8 @@ func SetupRoutes() *echo.Echo {
 		assets.GET("/paging", AssetPagingEndpoint)
 		assets.POST("/:id/tcping", AssetTcpingEndpoint)
 		assets.PUT("/:id", AssetUpdateEndpoint)
+		assets.GET("/:id/attributes", AssetGetAttributeEndpoint)
+		assets.PUT("/:id/attributes", AssetUpdateAttributeEndpoint)
 		assets.DELETE("/:id", AssetDeleteEndpoint)
 		assets.GET("/:id", AssetGetEndpoint)
 		assets.POST("/:id/change-owner", Admin(AssetChangeOwnerEndpoint))

pkg/api/tunnel.go

@@ -124,6 +124,15 @@ func TunEndpoint(c echo.Context) error {
 
 		configuration.SetParameter("hostname", session.IP)
 		configuration.SetParameter("port", strconv.Itoa(session.Port))
+
+		// 加载资产配置的属性，优先级比全局配置的高，因此最后加载，覆盖掉全局配置
+		attributes, _ := model.FindAssetAttributeByAssetId(session.AssetId)
+		if len(attributes) > 0 {
+			for i := range attributes {
+				attribute := attributes[i]
+				configuration.SetParameter(attribute.Name, attribute.Value)
+			}
+		}
 	}
 	for name := range configuration.Parameters {
 		// 替换数据库空格字符串占位符为真正的空格

pkg/guacd/guacd.go

@@ -15,9 +15,11 @@ const (
 	RecordingPath       = "recording-path"
 	CreateRecordingPath = "create-recording-path"
 
-	FontName    = "font-name"
-	FontSize    = "font-size"
-	ColorScheme = "color-scheme"
+	FontName     = "font-name"
+	FontSize     = "font-size"
+	ColorScheme  = "color-scheme"
+	Backspace    = "backspace"
+	TerminalType = "terminal-type"
 
 	EnableDrive              = "enable-drive"
 	DriveName                = "drive-name"
@@ -31,6 +33,17 @@ const (
 	DisableBitmapCaching     = "disable-bitmap-caching"
 	DisableOffscreenCaching  = "disable-offscreen-caching"
 	DisableGlyphCaching      = "disable-glyph-caching"
+
+	ColorDepth  = "color-depth"
+	Cursor      = "cursor"
+	SwapRedBlue = "swap-red-blue"
+	DestHost    = "dest-host"
+	DestPort    = "dest-port"
+
+	UsernameRegex     = "username-regex"
+	PasswordRegex     = "password-regex"
+	LoginSuccessRegex = "login-success-regex"
+	LoginFailureRegex = "login-failure-regex"
 )
 
 const Delimiter = ';'

pkg/model/asset-attribute.go

@@ -1 +1,108 @@
 package model
+
+import (
+	"fmt"
+	"github.com/labstack/echo/v4"
+	"gorm.io/gorm"
+	"next-terminal/pkg/global"
+	"next-terminal/pkg/guacd"
+	"next-terminal/pkg/utils"
+)
+
+type AssetAttribute struct {
+	Id      string `gorm:"index" json:"id"`
+	AssetId string `gorm:"index" json:"assetId"`
+	Name    string `gorm:"index" json:"name"`
+	Value   string `json:"value"`
+}
+
+func (r *AssetAttribute) TableName() string {
+	return "asset_attributes"
+}
+
+var SSHParameterNames = []string{guacd.FontName, guacd.FontSize, guacd.ColorScheme, guacd.Backspace, guacd.TerminalType}
+var RDPParameterNames = []string{guacd.EnableWallpaper, guacd.EnableTheming, guacd.EnableFontSmoothing, guacd.EnableFullWindowDrag, guacd.EnableDesktopComposition, guacd.EnableMenuAnimations, guacd.DisableBitmapCaching, guacd.DisableOffscreenCaching, guacd.DisableGlyphCaching}
+var VNCParameterNames = []string{guacd.ColorDepth, guacd.Cursor, guacd.SwapRedBlue, guacd.DestHost, guacd.DestPort}
+var TelnetParameterNames = []string{guacd.FontName, guacd.FontSize, guacd.ColorScheme, guacd.Backspace, guacd.TerminalType, guacd.UsernameRegex, guacd.PasswordRegex, guacd.LoginSuccessRegex, guacd.LoginFailureRegex}
+
+func UpdateAssetAttributes(assetId, protocol string, m echo.Map) error {
+	var data []AssetAttribute
+	var parameterNames []string
+	switch protocol {
+	case "ssh":
+		parameterNames = SSHParameterNames
+	case "rdp":
+		parameterNames = RDPParameterNames
+	case "vnc":
+		parameterNames = VNCParameterNames
+	case "telnet":
+		parameterNames = TelnetParameterNames
+	}
+
+	for i := range parameterNames {
+		name := parameterNames[i]
+		if m[name] != nil && m[name] != "" {
+			data = append(data, genAttribute(assetId, name, m))
+		}
+	}
+
+	return global.DB.Transaction(func(tx *gorm.DB) error {
+		err := tx.Where("asset_id = ?", assetId).Delete(&AssetAttribute{}).Error
+		if err != nil {
+			return err
+		}
+		return tx.CreateInBatches(&data, len(data)).Error
+	})
+}
+
+func genAttribute(assetId, name string, m echo.Map) AssetAttribute {
+	value := fmt.Sprintf("%v", m[name])
+	attribute := AssetAttribute{
+		Id:      utils.Sign([]string{assetId, name}),
+		AssetId: assetId,
+		Name:    name,
+		Value:   value,
+	}
+	return attribute
+}
+
+func FindAssetAttributeByAssetId(assetId string) (o []AssetAttribute, err error) {
+	err = global.DB.Where("asset_id = ?", assetId).Find(&o).Error
+	if o == nil {
+		o = make([]AssetAttribute, 0)
+	}
+	return o, err
+}
+
+func FindAssetAttrMapByAssetId(assetId string) (map[string]interface{}, error) {
+	asset, err := FindAssetById(assetId)
+	if err != nil {
+		return nil, err
+	}
+	attributes, err := FindAssetAttributeByAssetId(assetId)
+	if err != nil {
+		return nil, err
+	}
+
+	var parameterNames []string
+	switch asset.Protocol {
+	case "ssh":
+		parameterNames = SSHParameterNames
+	case "rdp":
+		parameterNames = RDPParameterNames
+	case "vnc":
+		parameterNames = VNCParameterNames
+	}
+	propertiesMap := FindAllPropertiesMap()
+	var attributeMap = make(map[string]interface{})
+	for name := range propertiesMap {
+		if utils.Contains(parameterNames, name) {
+			attributeMap[name] = propertiesMap[name]
+		}
+	}
+
+	for i := range attributes {
+		attributeMap[attributes[i].Name] = attributes[i].Value
+	}
+	return attributeMap, nil
+}