Runtime assertions for Ruby
literal.fun
ruby
Simplify Rails type registration
+7
-13
+7
-13
lib/literal/railtie.rb
+7
-13
lib/literal/railtie.rb
···
2
2
3
3
class Literal::Railtie < Rails::Railtie
4
4
initializer "literal.register_literal_enum_type" do
5
-
ActiveRecord::Type.register(:literal_enum) do |name, type:|
6
-
Literal::Rails::EnumType.new(type)
7
-
end
8
-
9
-
ActiveRecord::Type.register(:literal_flags) do |name, type:|
10
-
Literal::Rails::FlagsType.new(type)
11
-
end
12
-
13
-
ActiveModel::Type.register(:literal_enum) do |name, type:|
14
-
Literal::Rails::EnumType.new(type)
15
-
end
5
+
[ActiveRecord::Type, ActiveModel::Type].each do |registry|
6
+
registry.register(:literal_enum) do |name, type:|
7
+
Literal::Rails::EnumType.new(type)
8
+
end
16
9
17
-
ActiveModel::Type.register(:literal_flags) do |name, type:|
18
-
Literal::Rails::FlagsType.new(type)
10
+
registry.register(:literal_flags) do |name, type:|
11
+
Literal::Rails::FlagsType.new(type)
12
+
end
19
13
end
20
14
end
21
15
end