1Panel/backend/utils/nginx/dumper.go
2022-11-17 21:20:33 +08:00

126 lines
2.7 KiB
Go

package nginx
import (
"bytes"
"fmt"
components "github.com/1Panel-dev/1Panel/backend/utils/nginx/components"
"io/ioutil"
"sort"
"strings"
)
var (
//NoIndentStyle default style
NoIndentStyle = &Style{
SortDirectives: false,
StartIndent: 0,
Indent: 0,
}
//IndentedStyle default style
IndentedStyle = &Style{
SortDirectives: false,
StartIndent: 0,
Indent: 4,
}
//NoIndentSortedStyle default style
NoIndentSortedStyle = &Style{
SortDirectives: true,
StartIndent: 0,
Indent: 0,
}
//NoIndentSortedSpaceStyle default style
NoIndentSortedSpaceStyle = &Style{
SortDirectives: true,
SpaceBeforeBlocks: true,
StartIndent: 0,
Indent: 0,
}
)
type Style struct {
SortDirectives bool
SpaceBeforeBlocks bool
StartIndent int
Indent int
}
func NewStyle() *Style {
style := &Style{
SortDirectives: false,
StartIndent: 0,
Indent: 4,
}
return style
}
func (s *Style) Iterate() *Style {
newStyle := &Style{
SortDirectives: s.SortDirectives,
SpaceBeforeBlocks: s.SpaceBeforeBlocks,
StartIndent: s.StartIndent + s.Indent,
Indent: s.Indent,
}
return newStyle
}
func DumpDirective(d components.IDirective, style *Style) string {
var buf bytes.Buffer
if style.SpaceBeforeBlocks && d.GetBlock() != nil {
buf.WriteString("\n")
}
buf.WriteString(fmt.Sprintf("%s%s", strings.Repeat(" ", style.StartIndent), d.GetName()))
if len(d.GetParameters()) > 0 {
buf.WriteString(fmt.Sprintf(" %s", strings.Join(d.GetParameters(), " ")))
}
if d.GetBlock() == nil {
if d.GetName() != "" {
buf.WriteRune(';')
buf.WriteString(" ")
}
if d.GetComment() != "" {
buf.WriteString(d.GetComment())
}
} else {
buf.WriteString(" {")
if d.GetComment() != "" {
buf.WriteString(" ")
buf.WriteString(d.GetComment())
}
buf.WriteString("\n")
buf.WriteString(DumpBlock(d.GetBlock(), style.Iterate()))
buf.WriteString(fmt.Sprintf("\n%s}", strings.Repeat(" ", style.StartIndent)))
}
return buf.String()
}
func DumpBlock(b components.IBlock, style *Style) string {
var buf bytes.Buffer
directives := b.GetDirectives()
if style.SortDirectives {
sort.SliceStable(directives, func(i, j int) bool {
return directives[i].GetName() < directives[j].GetName()
})
}
for i, directive := range directives {
buf.WriteString(DumpDirective(directive, style))
if i != len(directives)-1 {
buf.WriteString("\n")
}
}
return buf.String()
}
func DumpConfig(c *components.Config, style *Style) string {
return DumpBlock(c.Block, style)
}
func WriteConfig(c *components.Config, style *Style) error {
return ioutil.WriteFile(c.FilePath, []byte(DumpConfig(c, style)), 0644)
}