Auto-create vaults

README.md

@@ -74,6 +74,17 @@ user.preferences.vault_attributes # => ["time_zone", "datetime_format", "hotkeys
 ```
 
 
+### Auto-create vaults
+
+Vaults can be created automatically after the parent is created.
+
+```ruby
+class User < ApplicationRecord
+  vault :preferences, auto_create: true
+  # for multiple vaults
+  vaults :preferences, :settings, auto_create_all: true
+end
+```
 
 
 ## Contributing

lib/rails_vault.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "store_attribute"
+
 require "rails_vault/version"
 require "generators/vault_generator"
 require "rails_vault/engine"

lib/rails_vault/vaults.rb

@@ -3,20 +3,36 @@ module Vaults
     extend ActiveSupport::Concern
 
     class_methods do
-      def vault(association_name, class_name: nil)
+      def vault(association_name, class_name: nil, auto_create: false)
+        vault_class = class_name || "#{self}::#{association_name.to_s.camelize}"
+
         has_one(
           association_name,
           as: :resource,
-          class_name: class_name.presence || "#{self}::#{association_name.to_s.camelize}",
+          class_name: vault_class,
           dependent: :destroy
         )
+
+        if auto_create
+          after_create -> { create_vault(association_name, vault_class) }
+        end
       end
 
-      def vaults(*association_names)
+      def vaults(*association_names, auto_create_all: false)
         association_names.each do |association_name|
-          has_one(association_name, as: :resource, dependent: :destroy)
+          vault(association_name, auto_create: auto_create_all)
         end
       end
     end
+
+    included do
+      private
+
+      def create_vault(association_name, vault_class)
+        return if public_send(association_name).present?
+
+        vault_class.constantize.create!(resource: self)
+      end
+    end
   end
 end

rails_vault.gemspec

@@ -18,6 +18,7 @@ Gem::Specification.new do |spec|
   spec.files = Dir["{bin,db,lib}/**/*", "Rakefile", "README.md", "rails_vault.gemspec", "Gemfile", "Gemfile.lock"]
 
   spec.required_ruby_version = ">= 3.1.0"
+
   spec.add_dependency "rails", ">= 7.2.2"
   spec.add_dependency "store_attribute", ">= 1.3"
 end