EdgeAdmin/internal/web/actions/default/ui/components.go

package ui

import (
	"bytes"
	"crypto/md5"
	"encoding/json"
	"fmt"
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/servers/server/settings/conds/condutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
	"github.com/iwind/TeaGo/Tea"
	"github.com/iwind/TeaGo/actions"
	"github.com/iwind/TeaGo/files"
	"github.com/iwind/TeaGo/logs"
	"github.com/iwind/TeaGo/maps"
	"net/http"
)

type ComponentsAction actions.Action

var componentsData = []byte{}
var componentsDataSum string

func (this *ComponentsAction) RunGet(params struct{}) {
	this.AddHeader("Content-Type", "text/javascript; charset=utf-8")

	// etag
	var requestETag = this.Header("If-None-Match")
	if len(requestETag) > 0 && requestETag == "\""+componentsDataSum+"\"" {
		this.ResponseWriter.WriteHeader(http.StatusNotModified)
		return
	}

	if !Tea.IsTesting() && len(componentsData) > 0 {
		this.AddHeader("Last-Modified", "Fri, 06 Sep 2019 08:29:50 GMT")
		_, _ = this.Write(componentsData)
		return
	}

	var buffer = bytes.NewBuffer([]byte{})

	var webRoot string
	if Tea.IsTesting() {
		webRoot = Tea.Root + "/../web/public/js/components/"
	} else {
		webRoot = Tea.Root + "/web/public/js/components/"
	}
	f := files.NewFile(webRoot)

	f.Range(func(file *files.File) {
		if !file.IsFile() {
			return
		}
		if file.Ext() != ".js" {
			return
		}
		data, err := file.ReadAll()
		if err != nil {
			logs.Error(err)
			return
		}
		buffer.Write(data)
		buffer.Write([]byte{'\n', '\n'})
	})

	// 条件组件
	typesJSON, err := json.Marshal(condutils.ReadAllAvailableCondTypes())
	if err != nil {
		logs.Println("ComponentsAction marshal request cond types failed: " + err.Error())
	} else {
		buffer.WriteString("window.REQUEST_COND_COMPONENTS = ")
		buffer.Write(typesJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// 条件操作符
	requestOperatorsJSON, err := json.Marshal(shared.AllRequestOperators())
	if err != nil {
		logs.Println("ComponentsAction marshal request operators failed: " + err.Error())
	} else {
		buffer.WriteString("window.REQUEST_COND_OPERATORS = ")
		buffer.Write(requestOperatorsJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// 请求变量
	requestVariablesJSON, err := json.Marshal(shared.DefaultRequestVariables())
	if err != nil {
		logs.Println("ComponentsAction marshal request variables failed: " + err.Error())
	} else {
		buffer.WriteString("window.REQUEST_VARIABLES = ")
		buffer.Write(requestVariablesJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// 指标
	metricHTTPKeysJSON, err := json.Marshal(serverconfigs.FindAllMetricKeyDefinitions(serverconfigs.MetricItemCategoryHTTP))
	if err != nil {
		logs.Println("ComponentsAction marshal metric http keys failed: " + err.Error())
	} else {
		buffer.WriteString("window.METRIC_HTTP_KEYS = ")
		buffer.Write(metricHTTPKeysJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// IP地址阈值项目
	ipAddrThresholdItemsJSON, err := json.Marshal(nodeconfigs.FindAllIPAddressThresholdItems())
	if err != nil {
		logs.Println("ComponentsAction marshal ip addr threshold items failed: " + err.Error())
	} else {
		buffer.WriteString("window.IP_ADDR_THRESHOLD_ITEMS = ")
		buffer.Write(ipAddrThresholdItemsJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// IP地址阈值动作
	ipAddrThresholdActionsJSON, err := json.Marshal(nodeconfigs.FindAllIPAddressThresholdActions())
	if err != nil {
		logs.Println("ComponentsAction marshal ip addr threshold actions failed: " + err.Error())
	} else {
		buffer.WriteString("window.IP_ADDR_THRESHOLD_ACTIONS = ")
		buffer.Write(ipAddrThresholdActionsJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// WAF checkpoints
	var wafCheckpointsMaps = []maps.Map{}
	for _, checkpoint := range firewallconfigs.AllCheckpoints {
		wafCheckpointsMaps = append(wafCheckpointsMaps, maps.Map{
			"name":        checkpoint.Name,
			"prefix":      checkpoint.Prefix,
			"description": checkpoint.Description,
		})
	}
	wafCheckpointsJSON, err := json.Marshal(wafCheckpointsMaps)
	if err != nil {
		logs.Println("ComponentsAction marshal waf rule checkpoints failed: " + err.Error())
	} else {
		buffer.WriteString("window.WAF_RULE_CHECKPOINTS = ")
		buffer.Write(wafCheckpointsJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// WAF操作符
	wafOperatorsJSON, err := json.Marshal(firewallconfigs.AllRuleOperators)
	if err != nil {
		logs.Println("ComponentsAction marshal waf rule operators failed: " + err.Error())
	} else {
		buffer.WriteString("window.WAF_RULE_OPERATORS = ")
		buffer.Write(wafOperatorsJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	// WAF验证码类型
	captchaTypesJSON, err := json.Marshal(firewallconfigs.FindAllCaptchaTypes())
	if err != nil {
		logs.Println("ComponentsAction marshal captcha types failed: " + err.Error())
	} else {
		buffer.WriteString("window.WAF_CAPTCHA_TYPES = ")
		buffer.Write(captchaTypesJSON)
		buffer.Write([]byte{';', '\n', '\n'})
	}

	componentsData = buffer.Bytes()

	// ETag
	var h = md5.New()
	h.Write(buffer.Bytes())
	componentsDataSum = fmt.Sprintf("%x", h.Sum(nil))
	this.AddHeader("ETag", "\""+componentsDataSum+"\"")

	_, _ = this.Write(componentsData)
}