diff --git a/README.md b/README.md index 3a234c5..89d2d0e 100644 --- a/README.md +++ b/README.md @@ -49,7 +49,7 @@ This will: ### Define vault attributes ```ruby -class User::Preferences < RailsVault::Base +class User::Preferences < Vault vault_attribute :time_zone, :string, default: "UTC" vault_attribute :datetime_format, :string, default: "dd-mm-yyyy" vault_attribute :hotkeys_disabled, :boolean, default: false diff --git a/lib/generators/templates/vault.rb.tt b/lib/generators/templates/vault.rb.tt index 1d6214b..9705550 100644 --- a/lib/generators/templates/vault.rb.tt +++ b/lib/generators/templates/vault.rb.tt @@ -1,4 +1,4 @@ -class <%= class_name %> < RailsVault::Base +class <%= class_name %> < Vault <%- attributes.each do |attribute| -%> vault_attribute :<%= attribute.name %>, :<%= attribute.type %> <%- end -%> diff --git a/lib/rails_vault.rb b/lib/rails_vault.rb index 03a5b8b..67aed75 100644 --- a/lib/rails_vault.rb +++ b/lib/rails_vault.rb @@ -9,5 +9,4 @@ require "rails_vault/vaults" module RailsVault - class Error < StandardError; end end diff --git a/lib/rails_vault/base.rb b/lib/rails_vault/base.rb index 38c12b7..4337a6f 100644 --- a/lib/rails_vault/base.rb +++ b/lib/rails_vault/base.rb @@ -27,3 +27,5 @@ def vault_attribute(key, *attributes) def vault_attributes = payload.keys end end + +Vault = RailsVault::Base