mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-14 16:49:34 +08:00
241 lines
6.2 KiB
Go
241 lines
6.2 KiB
Go
package components
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type Location struct {
|
|
Modifier string
|
|
Match string
|
|
Cache bool
|
|
ProxyPass string
|
|
Host string
|
|
CacheTime int
|
|
CacheUint string
|
|
Comment string
|
|
Directives []IDirective
|
|
Line int
|
|
Parameters []string
|
|
Replaces map[string]string
|
|
}
|
|
|
|
func NewLocation(directive IDirective) *Location {
|
|
location := &Location{
|
|
Modifier: "",
|
|
Match: "",
|
|
}
|
|
directives := make([]IDirective, 0)
|
|
if len(directive.GetParameters()) == 0 {
|
|
panic("no enough parameter for location")
|
|
}
|
|
for _, dir := range directive.GetBlock().GetDirectives() {
|
|
directives = append(directives, dir)
|
|
params := dir.GetParameters()
|
|
switch dir.GetName() {
|
|
case "proxy_pass":
|
|
location.ProxyPass = params[0]
|
|
case "proxy_set_header":
|
|
if params[0] == "Host" {
|
|
location.Host = params[1]
|
|
}
|
|
case "proxy_cache":
|
|
location.Cache = true
|
|
case "if":
|
|
if params[0] == "(" && params[1] == "$uri" && params[2] == "~*" {
|
|
dirs := dir.GetBlock().GetDirectives()
|
|
for _, di := range dirs {
|
|
if di.GetName() == "expires" {
|
|
re := regexp.MustCompile(`^(\d+)(\w+)$`)
|
|
matches := re.FindStringSubmatch(di.GetParameters()[0])
|
|
if matches == nil {
|
|
continue
|
|
}
|
|
cacheTime, err := strconv.Atoi(matches[1])
|
|
if err != nil {
|
|
continue
|
|
}
|
|
unit := matches[2]
|
|
location.CacheUint = unit
|
|
location.CacheTime = cacheTime
|
|
}
|
|
}
|
|
}
|
|
case "sub_filter":
|
|
if location.Replaces == nil {
|
|
location.Replaces = make(map[string]string, 0)
|
|
}
|
|
location.Replaces[strings.Trim(params[0], "\"")] = strings.Trim(params[1], "\"")
|
|
}
|
|
}
|
|
|
|
params := directive.GetParameters()
|
|
if len(params) == 1 {
|
|
location.Match = params[0]
|
|
} else if len(params) == 2 {
|
|
location.Match = params[1]
|
|
location.Modifier = params[0]
|
|
}
|
|
location.Parameters = directive.GetParameters()
|
|
location.Line = directive.GetLine()
|
|
location.Comment = directive.GetComment()
|
|
location.Directives = directives
|
|
return location
|
|
}
|
|
|
|
func (l *Location) GetName() string {
|
|
return "location"
|
|
}
|
|
|
|
func (l *Location) GetParameters() []string {
|
|
return l.Parameters
|
|
}
|
|
|
|
func (l *Location) GetBlock() IBlock {
|
|
return l
|
|
}
|
|
|
|
func (l *Location) GetComment() string {
|
|
return l.Comment
|
|
}
|
|
|
|
func (l *Location) GetLine() int {
|
|
return l.Line
|
|
}
|
|
|
|
func (l *Location) GetDirectives() []IDirective {
|
|
return l.Directives
|
|
}
|
|
|
|
func (l *Location) FindDirectives(directiveName string) []IDirective {
|
|
directives := make([]IDirective, 0)
|
|
for _, directive := range l.Directives {
|
|
if directive.GetName() == directiveName {
|
|
directives = append(directives, directive)
|
|
}
|
|
if directive.GetBlock() != nil {
|
|
directives = append(directives, directive.GetBlock().FindDirectives(directiveName)...)
|
|
}
|
|
}
|
|
return directives
|
|
}
|
|
|
|
func (l *Location) UpdateDirective(key string, params []string) {
|
|
if key == "" || len(params) == 0 {
|
|
return
|
|
}
|
|
directives := l.Directives
|
|
index := -1
|
|
for i, dir := range directives {
|
|
if dir.GetName() == key {
|
|
if IsRepeatKey(key) {
|
|
oldParams := dir.GetParameters()
|
|
if !(len(oldParams) > 0 && oldParams[0] == params[0]) {
|
|
continue
|
|
}
|
|
}
|
|
index = i
|
|
break
|
|
}
|
|
}
|
|
newDirective := &Directive{
|
|
Name: key,
|
|
Parameters: params,
|
|
}
|
|
if index > -1 {
|
|
directives[index] = newDirective
|
|
} else {
|
|
directives = append(directives, newDirective)
|
|
}
|
|
l.Directives = directives
|
|
}
|
|
|
|
func (l *Location) RemoveDirective(key string, params []string) {
|
|
directives := l.Directives
|
|
var newDirectives []IDirective
|
|
for _, dir := range directives {
|
|
if dir.GetName() == key {
|
|
if len(params) > 0 {
|
|
oldParams := dir.GetParameters()
|
|
if oldParams[0] == params[0] {
|
|
continue
|
|
}
|
|
} else {
|
|
continue
|
|
}
|
|
}
|
|
newDirectives = append(newDirectives, dir)
|
|
}
|
|
l.Directives = newDirectives
|
|
}
|
|
|
|
func (l *Location) ChangePath(Modifier string, Match string) {
|
|
if Match != "" && Modifier != "" {
|
|
l.Parameters = []string{Modifier, Match}
|
|
}
|
|
if Match != "" && Modifier == "" {
|
|
l.Parameters = []string{Match}
|
|
}
|
|
l.Modifier = Modifier
|
|
l.Match = Match
|
|
}
|
|
|
|
func (l *Location) AddCache(cacheTime int, cacheUint string) {
|
|
l.RemoveDirective("add_header", []string{"Cache-Control", "no-cache"})
|
|
l.RemoveDirective("if", []string{"(", "$uri", "~*", `"\.(gif|png|jpg|css|js|woff|woff2)$"`, ")"})
|
|
directives := l.GetDirectives()
|
|
newDir := &Directive{
|
|
Name: "if",
|
|
Parameters: []string{"(", "$uri", "~*", `"\.(gif|png|jpg|css|js|woff|woff2)$"`, ")"},
|
|
Block: &Block{},
|
|
}
|
|
block := &Block{}
|
|
block.Directives = append(block.Directives, &Directive{
|
|
Name: "expires",
|
|
Parameters: []string{strconv.Itoa(cacheTime) + cacheUint},
|
|
})
|
|
newDir.Block = block
|
|
directives = append(directives, newDir)
|
|
l.Directives = directives
|
|
l.UpdateDirective("proxy_ignore_headers", []string{"Set-Cookie", "Cache-Control", "expires"})
|
|
l.UpdateDirective("proxy_cache", []string{"proxy_cache_panel"})
|
|
l.UpdateDirective("proxy_cache_key", []string{"$host$uri$is_args$args"})
|
|
l.UpdateDirective("proxy_cache_valid", []string{"200", "304", "301", "302", "10m"})
|
|
l.Cache = true
|
|
l.CacheTime = cacheTime
|
|
l.CacheUint = cacheUint
|
|
}
|
|
|
|
func (l *Location) RemoveCache() {
|
|
l.RemoveDirective("if", []string{"(", "$uri", "~*", `"\.(gif|png|jpg|css|js|woff|woff2)$"`, ")"})
|
|
l.RemoveDirective("proxy_ignore_headers", []string{"Set-Cookie"})
|
|
l.RemoveDirective("proxy_cache", []string{"proxy_cache_panel"})
|
|
l.RemoveDirective("proxy_cache_key", []string{"$host$uri$is_args$args"})
|
|
l.RemoveDirective("proxy_cache_valid", []string{"200"})
|
|
|
|
l.UpdateDirective("add_header", []string{"Cache-Control", "no-cache"})
|
|
|
|
l.CacheTime = 0
|
|
l.CacheUint = ""
|
|
l.Cache = false
|
|
}
|
|
|
|
func (l *Location) AddSubFilter(subFilters map[string]string) {
|
|
l.RemoveDirective("sub_filter", []string{})
|
|
l.Replaces = subFilters
|
|
for k, v := range subFilters {
|
|
l.UpdateDirective("sub_filter", []string{fmt.Sprintf(`"%s"`, k), fmt.Sprintf(`"%s"`, v)})
|
|
}
|
|
l.UpdateDirective("proxy_set_header", []string{"Accept-Encoding", `""`})
|
|
l.UpdateDirective("sub_filter_once", []string{"off"})
|
|
}
|
|
|
|
func (l *Location) RemoveSubFilter() {
|
|
l.RemoveDirective("sub_filter", []string{})
|
|
l.RemoveDirective("proxy_set_header", []string{"Accept-Encoding", `""`})
|
|
l.RemoveDirective("sub_filter_once", []string{"off"})
|
|
l.Replaces = nil
|
|
}
|