1Panel/backend/utils/nginx/dumper.go
2022-12-04 18:20:57 +08:00

101 lines
2.2 KiB
Go

package nginx
import (
"bytes"
"fmt"
"github.com/1Panel-dev/1Panel/backend/utils/nginx/components"
"io/ioutil"
"strings"
)
var (
IndentedStyle = &Style{
SpaceBeforeBlocks: false,
StartIndent: 0,
Indent: 4,
}
)
type Style struct {
SpaceBeforeBlocks bool
StartIndent int
Indent int
}
func (s *Style) Iterate() *Style {
newStyle := &Style{
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(), d.GetBlock().GetLine()))
buf.WriteString(fmt.Sprintf("\n%s}", strings.Repeat(" ", style.StartIndent)))
}
return buf.String()
}
func DumpBlock(b components.IBlock, style *Style, startLine int) string {
var buf bytes.Buffer
line := startLine
if b.GetLine() > startLine {
for i := 0; i < b.GetLine()-startLine; i++ {
buf.WriteString("\n")
}
line = b.GetLine()
}
directives := b.GetDirectives()
for i, directive := range directives {
if directive.GetLine() > line {
for i := 0; i < b.GetLine()-line; i++ {
buf.WriteString("\n")
}
line = b.GetLine()
}
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, 1)
}
func WriteConfig(c *components.Config, style *Style) error {
return ioutil.WriteFile(c.FilePath, []byte(DumpConfig(c, style)), 0644)
}