You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1Panel/backend/utils/nginx/components/http.go

110 lines
2.4 KiB

package components
import (
"errors"
)
type Http struct {
Comment string
Servers []*Server
Directives []IDirective
}
func (h *Http) GetComment() string {
return h.Comment
}
func NewHttp(directive IDirective) (*Http, error) {
if block := directive.GetBlock(); block != nil {
http := &Http{
Servers: []*Server{},
Directives: []IDirective{},
Comment: block.GetComment(),
}
for _, directive := range block.GetDirectives() {
if server, ok := directive.(*Server); ok {
http.Servers = append(http.Servers, server)
continue
}
http.Directives = append(http.Directives, directive)
}
return http, nil
}
return nil, errors.New("http directive must have a block")
}
func (h *Http) GetName() string {
return "http"
}
func (h *Http) GetParameters() []string {
return []string{}
}
func (h *Http) GetDirectives() []IDirective {
directives := make([]IDirective, 0)
directives = append(directives, h.Directives...)
for _, directive := range h.Servers {
directives = append(directives, directive)
}
return directives
}
func (h *Http) FindDirectives(directiveName string) []IDirective {
directives := make([]IDirective, 0)
for _, directive := range h.GetDirectives() {
if directive.GetName() == directiveName {
directives = append(directives, directive)
}
if directive.GetBlock() != nil {
directives = append(directives, directive.GetBlock().FindDirectives(directiveName)...)
}
}
return directives
}
func (h *Http) UpdateDirectives(directiveName string, directive Directive) {
directives := make([]IDirective, len(h.GetDirectives()))
index := -1
for i, dir := range h.GetDirectives() {
if dir.GetName() == directiveName {
index = i
break
}
}
if index > -1 {
directives[index] = &directive
} else {
directives = append(directives, &directive)
}
h.Directives = directives
}
func (h *Http) AddDirectives(directive Directive) {
directives := append(h.GetDirectives(), &directive)
h.Directives = directives
}
func (h *Http) RemoveDirectives(names []string) {
nameMaps := make(map[string]struct{}, len(names))
for _, name := range names {
nameMaps[name] = struct{}{}
}
directives := h.GetDirectives()
var newDirectives []IDirective
for _, dir := range directives {
if _, ok := nameMaps[dir.GetName()]; ok {
continue
}
newDirectives = append(newDirectives, dir)
}
h.Directives = newDirectives
}
func (h *Http) GetBlock() IBlock {
return h
}