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
|
# GlusterFS module by James
# Copyright (C) 2010-2013+ James Shubin
# Written by James Shubin <james@shubin.ca>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
require 'facter'
# regexp to match a brick pattern, eg: annex1.example.com:/storage1a
regexp = /^[a-zA-Z]{1}[a-zA-Z0-9\.\-]{0,}:\/[a-zA-Z0-9]{1}[a-zA-Z0-9\/\.\-]{0,}$/ # TODO: is this right ?
# find the module_vardir
dir = Facter.value('puppet_vardirtmp') # nil if missing
if dir.nil? # let puppet decide if present!
dir = Facter.value('puppet_vardir')
if dir.nil?
var = nil
else
var = dir.gsub(/\/$/, '')+'/'+'tmp/' # ensure trailing slash
end
else
var = dir.gsub(/\/$/, '')+'/'
end
if var.nil?
# if we can't get a valid vardirtmp, then we can't continue
valid_brickdir = nil
else
module_vardir = var+'gluster/'
valid_brickdir = module_vardir.gsub(/\/$/, '')+'/brick/'
end
found = {}
result = {}
if not(valid_brickdir.nil?) and File.directory?(valid_brickdir)
Dir.glob(valid_brickdir+'*.*').each do |f|
b = File.basename(f)
g = b.split('.') # $name.group
group = g.pop() # pop off suffix (the group name)
if not found.key?(group)
found[group] = [] # initialize
end
if g.length >= 1
x = g.join('.') # in case value had dots in it.
brick = File.open(f, 'r').read.strip # read into str
# eg: annex1.example.com:/storage1a
split = brick.split(':') # do some $name parsing
host = split[0] # host fqdn
# NOTE: technically $path should be everything BUT split[0]. This
# lets our $path include colons if for some reason they're needed.
#path = split[1] # brick mount or storage path
path = brick.slice(host.length+1, brick.length-host.length-1)
if brick.length > 0 and regexp.match(brick)
found[group].push({'host' => host, 'path' => path})
# TODO: print warning on else...
end
end
end
end
# transform to strings
found.keys.each do |group|
# build final result
result[group] = found[group].collect {|x| x['host']+':'+x['path'] }
end
# build the correctly sorted brick list...
result.keys.each do |x|
Facter.add('gluster_brick_group_'+x) do
#confine :operatingsystem => %w{CentOS, RedHat, Fedora}
setcode {
# don't reuse single variable to avoid bug #:
# http://projects.puppetlabs.com/issues/22455
# TODO: facter should support native list types :)
result[x].join(',')
}
end
end
# list of generated gluster_ports_volume's
Facter.add('gluster_brick_group_facts') do
#confine :operatingsystem => %w{CentOS, RedHat, Fedora}
setcode {
result.keys.collect {|x| 'gluster_brick_group_'+x }.join(',')
}
end
# vim: ts=8
|
