frp-panel/models/client.go

package models

import (
	"encoding/json"
	"time"

	"github.com/VaalaCat/frp-panel/utils"
	v1 "github.com/fatedier/frp/pkg/config/v1"
	"github.com/samber/lo"
	"gorm.io/gorm"
)

type Client struct {
	*ClientEntity
}

type ClientEntity struct {
	ClientID      string `json:"client_id" gorm:"uniqueIndex;not null;primaryKey"`
	TenantID      int    `json:"tenant_id" gorm:"not null"`
	UserID        int    `json:"user_id" gorm:"not null"`
	ConfigContent []byte `json:"config_content"`
	ConnectSecret string `json:"connect_secret" gorm:"not null"`
	CreatedAt     time.Time
	UpdatedAt     time.Time
	DeletedAt     gorm.DeletedAt `gorm:"index"`
}

func (*Client) TableName() string {
	return "clients"
}

func (c *ClientEntity) SetConfigContent(cfg v1.ClientConfig) error {
	newCfg := struct {
		v1.ClientCommonConfig
		Proxies  []v1.ProxyConfigurer   `json:"proxies,omitempty"`
		Visitors []v1.VisitorBaseConfig `json:"visitors,omitempty"`
	}{
		ClientCommonConfig: cfg.ClientCommonConfig,
		Proxies: lo.Map(cfg.Proxies, func(item v1.TypedProxyConfig, _ int) v1.ProxyConfigurer {
			return item.ProxyConfigurer
		}),
		Visitors: lo.Map(cfg.Visitors, func(item v1.TypedVisitorConfig, _ int) v1.VisitorBaseConfig {
			return *item.GetBaseConfig()
		}),
	}
	raw, err := json.Marshal(newCfg)
	if err != nil {
		return err
	}
	c.ConfigContent = raw
	return nil
}

func (c *ClientEntity) GetConfigContent() (*v1.ClientConfig, error) {
	cliCfg, err := utils.LoadClientConfigNormal(c.ConfigContent, true)
	if err != nil {
		return nil, err
	}
	return cliCfg, err
}