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
|
# frozen_string_literal: true
require 'ostruct'
module StarTrek
names = [
'USS Enterprise',
'USS Excelsior',
'USS Reliant',
'IKS Koraga',
'IKS Kronos One',
'IRW Khazara',
'IRW Praetus',
]
# Set up "Bases" in MongoDB
class Base
include Mongoid::Document
field :name, type: String
field :sector, type: String
field :faction_id, type: Integer
has_many :residents, class_name: 'StarTrek::Resident', inverse_of: :base
end
class Resident
include Mongoid::Document
field :name, type: String
belongs_to :base, class_name: 'StarTrek::Base'
end
Base.collection.drop
dsk7 = Base.create!(name: "Deep Space Station K-7", sector: "Mempa", faction_id: 1)
dsk7.residents.create!(name: "Shir th'Talias")
dsk7.residents.create!(name: "Lurry")
dsk7.residents.create!(name: "Mackenzie Calhoun")
r1 = Base.create!(name: "Regula I", sector: "Mutara", faction_id: 1)
r1.residents.create!(name: "V. Madison")
r1.residents.create!(name: "D. March")
r1.residents.create!(name: "C. Marcus")
Base.create!(name: "Deep Space Nine", sector: "Bajoran", faction_id: 1)
Base.create!(name: "Firebase P'ok", sector: nil, faction_id: 2)
Base.create!(name: "Ganalda Space Station", sector: "Archanis", faction_id: 2)
Base.create!(name: "Rh'Ihho Station", sector: "Rator", faction_id: 3)
class FactionRecord
attr_reader :id, :name, :ships, :bases, :bases_clone
def initialize(id:, name:, ships:, bases:, bases_clone:)
@id = id
@name = name
@ships = ships
@bases = bases
@bases_clone = bases_clone
end
end
federation = FactionRecord.new(
id: '1',
name: 'United Federation of Planets',
ships: ['1', '2', '3'],
bases: Base.where(faction_id: 1),
bases_clone: Base.where(faction_id: 1),
)
klingon = FactionRecord.new(
id: '2',
name: 'Klingon Empire',
ships: ['4', '5'],
bases: Base.where(faction_id: 2),
bases_clone: Base.where(faction_id: 2),
)
romulan = FactionRecord.new(
id: '2',
name: 'Romulan Star Empire',
ships: ['6', '7'],
bases: Base.where(faction_id: 3),
bases_clone: Base.where(faction_id: 3),
)
DATA = {
"Faction" => {
"1" => federation,
"2" => klingon,
"3" => romulan,
},
"Ship" => names.each_with_index.reduce({}) do |memo, (name, idx)|
id = (idx + 1).to_s
memo[id] = OpenStruct.new(name: name, id: id)
memo
end,
"Base" => Hash.new { |h, k| h[k] = Base.find(k) }
}
def DATA.create_ship(name, faction_id)
new_id = (self["Ship"].keys.map(&:to_i).max + 1).to_s
new_ship = OpenStruct.new(id: new_id, name: name)
self["Ship"][new_id] = new_ship
self["Faction"][faction_id].ships << new_id
new_ship
end
end
|
