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
|
#
# Author: Martin Magr <mmagr@redhat.com>
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# Forked from https://github.com/puppetlabs/puppetlabs-inifile .
require File.expand_path('../openstackconfig/section', __FILE__)
module Puppet
module Util
class OpenStackConfig
@@SECTION_REGEX = /^\s*\[(.*)\]\s*$/
def initialize(path)
@path = path
@order = []
@sections = {}
parse_file
end
attr_reader :path
def section_names
@sections.keys
end
def get_settings(section_name)
@sections[section_name].settings
end
def get_value(section_name, setting_name)
if @sections.has_key?(section_name)
@sections[section_name].settings[setting_name]
end
end
def set_value(section_name, setting_name, value)
unless @sections.has_key?(section_name)
add_section(section_name)
end
if @sections[section_name].settings.has_key?(setting_name)
@sections[section_name].update_setting(setting_name, value)
else
@sections[section_name].add_setting(setting_name, value)
end
end
def remove_setting(section_name, setting_name, value=nil)
@sections[section_name].remove_setting(setting_name, value)
end
def save
File.open(@path, 'w') do |fh|
@order.each do |section_name|
if section_name.length > 0
fh.puts("[#{section_name}]")
end
unless @sections[section_name].lines.empty?
@sections[section_name].lines.each do |line|
fh.puts(line)
end
end
end
end
end
private
# This is mostly here because it makes testing easier
# --we don't have to try to stub any methods on File.
def self.readlines(path)
# If this type is ever used with very large files, we should
# write this in a different way, using a temp
# file; for now assuming that this type is only used on
# small-ish config files that can fit into memory without
# too much trouble.
File.file?(path) ? File.readlines(path) : []
end
def parse_file
# We always create a "global" section at the beginning of the file,
# for anything that appears before the first named section.
lines = []
current_section = ''
OpenStackConfig.readlines(@path).each do |line|
if match = @@SECTION_REGEX.match(line)
add_section(current_section, lines)
# start new section parsing
lines = []
current_section = match[1]
else
lines.push(line)
end
end
add_section(current_section, lines)
end
def add_section(section_name, lines=nil)
@order.push(section_name)
@sections[section_name] = Section.new(section_name, lines)
end
end
end
end
|
