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
|
Puppet::Type.type(:vpp_config).provide(:vpp) do
def file_path
'/etc/vpp/startup.conf'
end
def initialize(value={})
super(value)
settings_arr = @resource[:setting].split('/')
@section = settings_arr[0]
@real_setting = settings_arr[1]
@dev = settings_arr[2]
if @section.nil? || @real_setting.nil?
fail("#{@resource[:setting]} is not a valid setting string")
end
if @dev
@search_regex = /^\s*dev\s+#{@dev}\s*{([^}]*)}?/
else
@search_regex = /^\s*#{@real_setting}\s+(\S+)?/
end
end
def write_config(config)
if File.read(file_path) != config
File.open(file_path, 'w') do |fh|
fh.puts(config)
end
end
end
def get_sections
vpp_config = File.read(file_path)
scanner = StringScanner.new vpp_config
#first skip to section beginning
string = scanner.scan_until(/^\s*#{@section}\s*{\s*/)
#if we can't find the section, add it to the end
return vpp_config+"\n#{@section} {", "", "}\n" unless string
level = 1
before = string
after = ''
section_config = ''
while current_char = scanner.getch
case current_char
when '{'
level += 1
section_config << current_char
when '}'
level -= 1
if level == 0
after = current_char + scanner.post_match
break
else
section_config << current_char
end
else
section_config << current_char
end
end
fail("Failed to parse VPP config: #{vpp_config}") unless level == 0
return before, section_config, after
end
def add_setting(value)
before, section_config, after = get_sections
if @dev
if value.to_s.empty?
setting_string = "#{@real_setting} #{@dev}"
else
setting_string = "#{@real_setting} #{@dev} {#{value}}"
end
else
setting_string = "#{@real_setting} #{value}"
end
if section_config =~ @search_regex
section_config.sub!(@search_regex, " #{setting_string}")
else
section_config.rstrip!
section_config << "\n #{setting_string}\n"
end
write_config(before+section_config+after)
end
def create
add_setting(@resource[:value])
end
def destroy
before, section_config, after = get_sections
section_config.sub!(@search_regex, "")
write_config(before+section_config+after)
end
def exists?
before, section_config, after = get_sections
@search_regex.match(section_config)
end
def value
before, section_config, after = get_sections
@search_regex.match(section_config) { |m| m[1] }
end
def value=(value)
add_setting(value)
end
end
|
