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
|
# 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'
require 'digest/sha1'
# TODO: the ruby uuid method can be used when newer ruby versions are used here
# require 'securerandom'
# SecureRandom.uuid
# uuid regexp
regexp = /^[a-f0-9]{8}\-[a-f0-9]{4}\-[a-f0-9]{4}\-[a-f0-9]{4}\-[a-f0-9]{12}$/
fqdn = Facter.value('fqdn') # this could be nil !
# 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
module_vardir = nil
valid_brickdir = nil
uuiddir = nil
else
module_vardir = var+'gluster/'
valid_brickdir = module_vardir.gsub(/\/$/, '')+'/brick/'
uuiddir = valid_brickdir+'fsuuid/' # safe dir that won't get purged...
end
# NOTE: module specific mkdirs, needed to ensure there is no blocking/deadlock!
if not(var.nil?) and not File.directory?(var)
Dir::mkdir(var)
end
if not(module_vardir.nil?) and not File.directory?(module_vardir)
Dir::mkdir(module_vardir)
end
if not(valid_brickdir.nil?) and not File.directory?(valid_brickdir)
Dir::mkdir(valid_brickdir)
end
found = {}
# generate uuid and parent directory if they don't already exist...
if not(valid_brickdir.nil?) and File.directory?(valid_brickdir)
if not File.directory?(uuiddir)
Dir::mkdir(uuiddir)
end
# loop through brick dir, looking for brick names to make fsuuid's for!
if not(valid_brickdir.nil?) and File.directory?(valid_brickdir) and File.directory?(uuiddir)
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 g.length >= 1
# NOTE: some of this code is unnecessary, but i
# kept it because it matches the brick parsing.
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 fqdn is nil, generate for everyone...
# (other hosts data will just be unused...)
if not(fqdn.nil?)
# otherwise, skip hosts that aren't us!
if host != fqdn
next
end
end
uuidfile = uuiddir + b
# we sha1 to prevent weird characters in facter
key = Digest::SHA1.hexdigest(host + ':' + path + '.' + group)
# create an fsuuid for each brick and store it
# in our vardir if it doesn't already exist...
if not File.exist?(uuidfile)
result = system("/usr/bin/uuidgen > '" + uuidfile + "'")
if not(result)
# TODO: print warning
end
end
# create facts from all the uuid files found...
uuid = File.open(uuidfile, 'r').read.strip.downcase # read into str
if uuid.length == 36 and regexp.match(uuid)
# avoid: http://projects.puppetlabs.com/issues/22455
found[key] = uuid
# TODO: print warning on else...
end
end
end
end
end
found.keys.each do |x|
Facter.add('gluster_brick_fsuuid_'+x) do
#confine :operatingsystem => %w{CentOS, RedHat, Fedora}
setcode {
found[x]
}
end
end
# list of generated gluster_brick_fsuuid's
Facter.add('gluster_brick_fsuuid_facts') do
#confine :operatingsystem => %w{CentOS, RedHat, Fedora}
setcode {
found.keys.collect {|x| 'gluster_brick_fsuuid_'+x }.join(',')
}
end
# vim: ts=8
|
