bin/common.rb.in


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

# Copyright 2016-2017 Luke Shumaker <lukeshu@sbcglobal.net>.
#
# This is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of
# the License, or (at your option) any later version.
#
# This software 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public
# License along with this manual; if not, see
# <http://www.gnu.org/licenses/>.

require 'set'
require 'yaml'

def cfg
  if @cfg.nil?
    # allow in-tree usage and tests
    this_dir         = File.dirname(__FILE__)
    is_in_build_tree = File::file? "#{this_dir}/common.rb.in"
    cfg_file         = (is_in_build_tree) ? "#{this_dir}/../parabola-hackers.yml" : '@conf_file@'

    @cfg = YAML::load(open(cfg_file))
    if ENV['PARABOLA_HACKERS_YAMLDIR']
      @cfg["yamldir"] = ENV['PARABOLA_HACKERS_YAMLDIR']
    end
  end
  return @cfg
end

def add_group(group_set, groupname)
  return if group_set.include?(groupname)
  group_set.add(groupname)
  (cfg["groupgroups"][groupname] || []).each do |subgroupname|
    add_group(group_set, subgroupname)
  end
end

def load_user_yaml(filename)
  user = YAML::load(open(filename))
  groups = Set.new
  (user["groups"] || []).each{|groupname| add_group(groups, groupname)}
  user["groups"] = groups.sort
  return user
end

def load_all_users
  users = {}
  Dir.glob("#{cfg["yamldir"]}/*.yml").map{|filename|
    uid = File.basename(filename).sub(/^([0-9]*)\.yml$/, "\\1").to_i
    user = load_user_yaml(filename)
    users[uid] = user
  }
  return users
end