goapacheconf/configfile.go at main · r2dtools/goapacheconf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
package goapacheconf

import (
	"os"

	"github.com/r2dtools/goapacheconf/internal/rawparser"
)

type ConfigFile struct {
	FilePath   string
	configFile *rawparser.Config
	config     *Config
}

func (c *ConfigFile) FindDirectives(directiveName string) []Directive {
	var (
		directives []Directive
		ifModules  []string
	)

	for _, entry := range c.configFile.GetEntries() {
		directives = append(directives, c.config.findDirectivesRecursively(directiveName, c.FilePath, c.configFile, entry, ifModules, true)...)
	}

	return directives
}

func (c *ConfigFile) FindRewriteRuleDirectives() []RewriteRuleDirective {
	return findDirectives[RewriteRuleDirective](c, RewriteRule)
}

func (c *ConfigFile) FindAlliasDirectives() []AliasDirective {
	return findDirectives[AliasDirective](c, Alias)
}

func (c *ConfigFile) FindBlocks(blockName string) []Block {
	var (
		blocks    []Block
		ifModules []string
	)

	for _, entry := range c.configFile.GetEntries() {
		blocks = append(blocks, c.config.findBlocksRecursively(blockName, c.FilePath, c.configFile, entry, ifModules, true)...)
	}

	return blocks
}

func (c *ConfigFile) FindVirtualHostBlocks() []VirtualHostBlock {
	return findBlocks[VirtualHostBlock](c, VirtualHost)
}

func (c *ConfigFile) FindVirtualHostBlocksByServerName(serverName string) []VirtualHostBlock {
	return findVirtualHostBlocksByServerName(c, serverName)
}

func (c *ConfigFile) FindIfModuleBlocks() []IfModuleBlock {
	return findBlocks[IfModuleBlock](c, IfModule)
}

func (c *ConfigFile) FindIfModuleBlocksByModuleName(moduleName string) []IfModuleBlock {
	return findIfModuleBlocksByModuleName(c, moduleName)
}

func (c *ConfigFile) DeleteDirective(directive Directive) {
	deleteDirective(c.configFile, directive)
}

func (c *ConfigFile) DeleteDirectiveByName(directiveName string) {
	deleteDirectiveByName(c.configFile, directiveName)
}

func (c *ConfigFile) AppendDirective(directive Directive) Directive {
	entries := c.configFile.GetEntries()
	directive.setContainer(c.configFile)

	var prevEntry *rawparser.Entry

	if len(entries) > 0 {
		prevEntry = entries[len(entries)-1]
	}

	if prevEntry == nil || len(prevEntry.EndNewLines) == 0 {
		directive.PrependNewLine()
	}

	entries = append(entries, directive.entry)

	c.configFile.SetEntries(entries)

	return directive
}

func (c *ConfigFile) PrependDirective(directive Directive) Directive {
	entries := c.configFile.GetEntries()
	directive.setContainer(c.configFile)

	directive.PrependNewLine()
	entries = append([]*rawparser.Entry{directive.entry}, entries...)

	c.configFile.SetEntries(entries)

	return directive
}

func (c *ConfigFile) DeleteVirtualHostBlock(virtualHostBlock VirtualHostBlock) {
	deleteBlock(c.configFile, virtualHostBlock.Block)
}

func (c *ConfigFile) AddBlock(name string, parameters []string, begining bool) Block {
	return newBlock(c.configFile, c.config, name, parameters, nil, begining)
}

func (c *ConfigFile) AppendAliasDirective(aliasDirective AliasDirective) AliasDirective {
	directive := c.AppendDirective(aliasDirective.Directive)

	return AliasDirective{Directive: directive}
}

func (c *ConfigFile) PrependAliasDirective(aliasDirective AliasDirective) AliasDirective {
	directive := c.PrependDirective(aliasDirective.Directive)

	return AliasDirective{Directive: directive}
}

func (c *ConfigFile) DeleteAliasDirective(aliasDirective AliasDirective) {
	deleteDirective(c.configFile, aliasDirective.Directive)
}

func (c *ConfigFile) Dump() (string, error) {
	content, err := c.config.rawDumper.Dump(c.configFile)

	if err != nil {
		return "", err
	}

	file, err := os.OpenFile(c.FilePath, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0666)

	if err != nil {
		return "", err
	}

	defer file.Close()

	_, err = file.WriteString(content)

	if err != nil {
		return "", err
	}

	return content, nil
}