Class: Org

Inherits:
Object
  • Object
show all
Defined in:
lib/terraorg/model/org.rb

Constant Summary collapse

MAX_SQUADS_PER_PERSON = 
2
SCHEMA_VERSION = 
'v1'.freeze

Instance Method Summary collapse

Constructor Details

#initialize(parsed_data, platoons, squads, people, gsuite_domain) ⇒ Org

Returns a new instance of Org.



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
 
# File 'lib/terraorg/model/org.rb', line 7

def initialize(parsed_data, platoons, squads, people, gsuite_domain)
  @people = people
  @id = parsed_data.fetch('id')
  @name = parsed_data.fetch('name')
  @metadata = parsed_data.fetch('metadata', {})
  @manager = @people.get_or_create!(parsed_data.fetch('manager'))
  @manager_location = parsed_data.fetch('manager_location')
  # @member_exception_people is a Hash of Squad::Teams, like the teams attribute in a Squad.
  @member_exception_people = Hash[parsed_data.fetch('exception_people').map { |p|
    [p.fetch('location'), Squad::Team.new(p, @people)]
  }]
  @member_exception_squad_names = []
  @member_exception_squads = parsed_data.fetch('exception_squads').map do |s|
    @member_exception_squad_names.push(s)
    squads.lookup!(s)
  end
  @member_platoon_names = []
  @member_platoons = parsed_data.fetch('platoons').map do |p|
    @member_platoon_names.push(p)
    platoons.lookup!(p)
  end

  # used to generate a group
  @gsuite_domain = gsuite_domain

  # used for validate!
  @platoons = platoons
  @squads = squads
end

Instance Method Details

#generate_tf ⇒ Object 



153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
 
# File 'lib/terraorg/model/org.rb', line 153

def generate_tf
  tf = @member_platoons.map { |p| p.generate_tf(@id) }.join("\n")
  File.write('auto.platoons.tf', tf)

  tf = @member_exception_squads.map { |s| s.generate_tf(@id) }.join("\n")
  File.write('auto.exception_squads.tf', tf)

  # Roll all platoons and exception squads into the org.
  roll_up_to_org = \
    @member_exception_squads.map { |s| s.unique_name(@id, nil) } + \
    @member_platoons.map { |p| p.unique_name(@id) }

  # Generate the org, which contains:
  # - exception people (added manually to group)
  # - the org manager (added manually to group)
  # - all the platoons (via rule)
  # - all exception squads (via rule)
  org_condition = roll_up_to_org.map {
    |n| "\\\"${okta_group.#{n}.id}\\\""
  }.join(',')

  description = "#{@name} organization worldwide members (terraorg)"
  tf = "# Okta for Organization: \#{@name}\nresource \"okta_group\" \"\#{unique_name}\" {\nname = \"\#{unique_name}\"\ndescription = \"\#{description}\"\nusers = \#{Util.persons_tf(members)}\n}\n\n\#{Util.gsuite_group_tf(unique_name, @gsuite_domain, members, description)}\n"

  # Generate a special group for all org members grouped by country
  all_squads = (@member_platoons.map(&:member_squads) + @member_exception_squads).flatten
  all_locations = {}
  (all_squads.map(&:teams).flatten + [@member_exception_people]).each do |team|
    team.each do |location, subteam|
      all_locations[location] = all_locations.fetch(location, Set.new).merge(subteam.members)
    end
  end

  # Manually add the manager to a specific location
  all_locations[@manager_location] = all_locations.fetch(@manager_location, Set.new).add(@manager)

  all_locations.each do |l, m|
    name = "#{unique_name}-#{l.downcase}"
    tf += "resource \"okta_group\" \"\#{name}\" {\nname = \"\#{name}\"\ndescription = \"\#{@name} organization members based in \#{l} (terraorg)\"\nusers = \#{Util.persons_tf(m)}\n}\n"
  end

  File.write('auto.org.tf', tf)
end

#get_acl_groups(attr: :id) ⇒ Object 



97
98
99
100
101
102
 
# File 'lib/terraorg/model/org.rb', line 97

def get_acl_groups(attr: :id)
  # Return a LIST_NAME => [MEMBER1, MEMBER2...] hash of ACL groups
  { unique_name => members.map(&attr).sort }.
    merge(get_platoon_acl_groups(attr: attr)).
    merge(get_exception_squad_acl_groups(attr: attr))
end

#get_exception_squad_acl_groups(attr: :id) ⇒ Object 



108
109
110
 
# File 'lib/terraorg/model/org.rb', line 108

def get_exception_squad_acl_groups(attr: :id)
  @member_exception_squads.map { |p| p.get_acl_groups(@id) }.map(&attr).reduce({}, :merge)
end

#get_platoon_acl_groups(attr: :id) ⇒ Object 



104
105
106
 
# File 'lib/terraorg/model/org.rb', line 104

def get_platoon_acl_groups(attr: :id)
  @member_platoons.map { |p| p.get_acl_groups(@id, attr: attr) }.reduce({}, :merge)
end

#get_platoons_md ⇒ Object 



116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/terraorg/model/org.rb', line 116

def get_platoons_md
  # 90 degree rotated version of the Platoons page of the legacy Engineering Squads sheet
  # Format:
  # Platoon,Total,Members
  # [ORG_HUMAN_NAME],[COUNT]
  # [PLAT1_HUMAN_NAME],[PLAT1_COUNT],[PLAT1_MEMBER1],[PLAT1_MEMBER2],...
  # [PLAT2_HUMAN_NAME],[PLAT2_COUNT],[PLAT2_MEMBER1],[PLAT2_MEMBER2],...
  md_lines = [
    '# Engineering Platoons List',
    '',
    '|Platoon|Manager|Total|Members|',
    '|---|---|---|---|',
    "|_#{@name} Total_|#{@manager.name}|#{members.size}|",
  ]
  md_lines += @member_platoons.map(&:get_platoons_psv_row)

  raise "Users have left the company: #{@people.inactive.map(&:id).join(', ')}" unless @people.inactive.empty?
  md_lines.join("\n")
end

#get_squads_md ⇒ Object 



136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
 
# File 'lib/terraorg/model/org.rb', line 136

def get_squads_md
  # Format:
  # platoon name, squad name, PM, email list, SME, slack, # people, squad manager, eng product owner, members
  md_lines = [
    '# Engineering Squads List',
    '',
    '|Platoon|Squad|PM|Mailing list|TS SME|Slack|# Engineers|Squad Manager|Eng Product Owner|Members|',
    '|---|---|---|---|---|---|---|---|---|---|',
  ]
  md_lines += @member_platoons.map { |s| s.get_squads_psv_rows(@id) }
  md_lines += @member_exception_squads.map { |s| s.to_md('_No Platoon_', @id) }
  md_lines.push "|_No Platoon_|_No Squad_|||||#{@member_exception_people.size}|||#{@member_exception_people.values.map(&:to_md).join(' / ')}|"

  raise "Users have left the company: #{@people.inactive.map(&:id).join(', ')}" unless @people.inactive.empty?
  md_lines.join("\n")
end

#members ⇒ Object 



93
94
95
 
# File 'lib/terraorg/model/org.rb', line 93

def members
  Set.new([@manager] + (@member_platoons + @member_exception_squads).map(&:members).flatten + @member_exception_people.map { |_, t| t.members }.flatten).to_a
end

#to_h ⇒ Object

Output a canonical (sorted, formatted) version of this organization.

  • Sort the platoon names lexically

  • Sort the exception squad ids lexically

  • Sort the exception people ids lexically



216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
 
# File 'lib/terraorg/model/org.rb', line 216

def to_h
  obj = { 'version' => SCHEMA_VERSION, 'id' => @id, 'name' => @name, 'manager_location' => @manager_location, 'manager' => @manager.id }
  obj['platoons'] = @platoons.all.map(&:id).sort
  if @metadata
    obj['metadata'] = @metadata
  end
  if @member_exception_people
    obj['exception_people'] = @member_exception_people.values.sort_by(&:location).map(&:to_h)
  end
  if @member_exception_squads
    obj['exception_squads'] = @member_exception_squads.map(&:id).sort
  end

  obj
end

#unique_name ⇒ Object 



112
113
114
 
# File 'lib/terraorg/model/org.rb', line 112

def unique_name
  "#{@id}-all"
end

#validate! ⇒ Object 



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
 
# File 'lib/terraorg/model/org.rb', line 37

def validate!
  # Do not allow the JSON files to contain any people who have left.
  raise "Users have left the company: #{@people.inactive.map(&:id).join(', ')}" unless @people.inactive.empty?

  # Do not allow the org to be totally empty.
  raise 'Org has no platoons or exception squads' if @member_platoons.size + @member_exception_squads.size == 0

  # Require all platoons to be part of the org.
  platoon_diff = Set.new(@platoons.all_names) - Set.new(@member_platoon_names)
  unless platoon_diff.empty?
    raise "Platoons are not used in the org: #{platoon_diff.to_a.sort}"
  end

  # Require all squads to be used in the org.
  squad_diff = Set.new(@squads.all_names) - Set.new(@platoons.all_squad_names) - Set.new(@member_exception_squad_names)
  unless squad_diff.empty?
    raise "Squad(s) are not used in the org: #{squad_diff.to_a.sort}"
  end

  all_squads = (@member_platoons.map(&:member_squads) + @member_exception_squads).flatten
  seen_squads = {}

  # Validate that a squad is not part of more than one platoon
  all_squads.map(&:id).each do |id|
    seen_squads[id] = seen_squads.fetch(id, 0) + 1
  end
  more_than_one_platoon = seen_squads.select do |squad, count|
    count > 1
  end
  if !more_than_one_platoon.empty?
    raise "Squads are part of more than one platoon: #{more_than_one_platoon}"
  end

  # Validate that a squad member belongs to at most two squads in the entire org
  squad_count = {}
  all_squads.map(&:teams).flatten.map(&:values).flatten.map(&:members).flatten.each do |member|
    squad_count[member.id] = squad_count.fetch(member.id, 0) + 1
  end
  more_than_max_squads = squad_count.select do |member, count|
    count > MAX_SQUADS_PER_PERSON
  end
  if !more_than_max_squads.empty?
    # TODO(joshk): Enforce after consulting with Sean
    $stderr.puts "WARNING: Members are part of more than #{MAX_SQUADS_PER_PERSON} squads: #{more_than_max_squads}"
  end

  # Validate that a squad member is not also an org exception
  exceptions = Set.new(@member_exception_people.map { |_, t| t.members }.flatten.map(&:id))
  exception_and_squad_member = squad_count.keys.select do |member|
    exceptions.member? member
  end
  if !exception_and_squad_member.empty?
    raise "Exception members are also squad members: #{exception_and_squad_member}"
  end
end