Add default_keyword_init option

Author: Arcovion

lib/typed_struct.rb

@@ -18,9 +18,21 @@ class TypedStruct < Struct
   alias_method :__class__, :class
 
   class << self
+    def default_keyword_init
+      @@default_keyword_init
+    end
+
+    def default_keyword_init=(default)
+      @@default_keyword_init = default
+    end
+
     def new(opts = Options.new, **properties)
-      if const_defined?("RSpec") && RUBY_VERSION < "3.2" && opts[:keyword_init].nil?
-        opts[:keyword_init] = false
+      if opts[:keyword_init].nil?
+        opts[:keyword_init] = if RUBY_VERSION < "3.2"
+          default_keyword_init || false
+        else
+          default_keyword_init
+        end
       end
 
       properties.each_key do |prop|

spec/typed_struct_spec.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 RSpec.describe TypedStruct do
+  before { TypedStruct.default_keyword_init = nil }
+
   it "helps avoid primitive obsession" do
     Price = TypedStruct.new(price: Rational) do
       %i[- + / *].each do |op|
@@ -28,6 +30,31 @@
     expect { y.int = "abc" }.to raise_error TypeError
   end
 
+  it "has default_keyword_init option" do
+    attrs = RUBY_VERSION < "3.2" ?  {int: {int: 5}} : {int: 5}
+
+    TypedStruct.default_keyword_init = nil
+    expect(TypedStruct.default_keyword_init).to be nil
+    x = Struct.new(:int)
+    y = TypedStruct.new(int: Rbs("untyped"))
+    expect(x.new(int: 5)).to have_attributes attrs
+    expect(y.new(int: 5)).to have_attributes attrs
+
+    TypedStruct.default_keyword_init = true
+    expect(TypedStruct.default_keyword_init).to be true
+    x = Struct.new(:int, keyword_init: true)
+    y = TypedStruct.new(int: Rbs("untyped"))
+    expect(x.new(int: 5)).to have_attributes int: 5
+    expect(y.new(int: 5)).to have_attributes int: 5
+
+    TypedStruct.default_keyword_init = false
+    expect(TypedStruct.default_keyword_init).to be false
+    x = Struct.new(:int, keyword_init: false)
+    y = TypedStruct.new(int: Rbs("untyped"))
+    expect(x.new(int: 5)).to have_attributes int: {int: 5}
+    expect(y.new(int: 5)).to have_attributes int: {int: 5}
+  end
+
   it "has an options attribute" do
     x = TypedStruct.new(int: Integer, str: String)
     expect(x.instance_variables).to contain_exactly :@options
@@ -45,13 +72,11 @@
       y = TypedStruct.new(int: Rbs("untyped"))
       attrs = RUBY_VERSION < "3.2" ?  {int: {int: 5}} : {int: 5}
       expect(x.new(5)).to have_attributes int: 5
-      expect(x.new(int: 5)).to have_attributes attrs
       expect(y.new(5)).to have_attributes int: 5
+      expect(x.new(int: 5)).to have_attributes attrs
       expect(y.new(int: 5)).to have_attributes attrs
       x = Struct.new(:int, keyword_init: true)
       y = TypedStruct.new({ keyword_init: true }, int: Rbs("untyped"))
-      expect { x.new(5) }.to raise_error ArgumentError, "wrong number of arguments (given 1, expected 0)"
-      expect { y.new(5) }.to raise_error ArgumentError, "wrong number of arguments (given 1, expected 0)"
       expect(x.new(int: 5)).to have_attributes int: 5
       expect(y.new(int: 5)).to have_attributes int: 5
     end
@@ -71,13 +96,15 @@
 
     it "has identical error messages for presence checks" do
       x = Struct.new(:int, keyword_init: true)
-      expect { x.new(str: 5, abc: "xyz") }.to raise_error ArgumentError, "unknown keywords: str, abc"
       y = TypedStruct.new({ keyword_init: true }, int: Integer)
+      expect { x.new(5) }.to raise_error ArgumentError, "wrong number of arguments (given 1, expected 0)"
+      expect { y.new(5) }.to raise_error ArgumentError, "wrong number of arguments (given 1, expected 0)"
+      expect { x.new(str: 5, abc: "xyz") }.to raise_error ArgumentError, "unknown keywords: str, abc"
       expect { y.new(str: 5, abc: "xyz") }.to raise_error ArgumentError, "unknown keywords: str, abc"
       a = x.new(int: 5)
+      b = y.new(int: 5)
       expect { a.str = 5 }.to raise_error NoMethodError, "undefined method `str=' for #<struct int=5>"
       expect { a[:str] = 5 }.to raise_error NameError, "no member 'str' in struct"
-      b = y.new(int: 5)
       expect { b.str = 5 }.to raise_error NoMethodError, "undefined method `str=' for #<struct int=5>"
       expect { b[:str] = 5 }.to raise_error NameError, "no member 'str' in struct"
       x = Struct.new(:int)
@@ -89,7 +116,7 @@
     it "supports the same methods" do
       a = Struct.new(:str, :int)
       b = TypedStruct.new(str: String, int: Integer)
-      expect(a.public_methods).to contain_exactly *b.public_methods
+      expect(a.public_methods).to contain_exactly *b.public_methods.grep_v(:default_keyword_init).grep_v(:default_keyword_init=)
       expect(a.public_instance_methods).to contain_exactly *b.public_instance_methods.grep_v(:__class__)
       expect(a.public_instance_methods(false)).to contain_exactly *b.new("abc", 5).public_methods(false).grep_v(:[]=)
     end