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
|
# frozen-string-literal: true
require 'securerandom'
module Sequel
module Plugins
# The uuid plugin creates hooks that automatically create a uuid for every
# instance.
#
# Usage:
#
# # Uuid all model instances using +uuid+
# # (called before loading subclasses)
# Sequel::Model.plugin :uuid
#
# # Uuid Album instances, with custom column name
# Album.plugin :uuid, field: :my_uuid
module Uuid
# Configure the plugin by setting the available options. Note that
# if this method is run more than once, previous settings are ignored,
# and it will just use the settings given or the default settings. Options:
# :field :: The field to hold the uuid (default: :uuid)
# :force :: Whether to overwrite an existing uuid (default: false)
def self.configure(model, opts=OPTS)
model.instance_exec do
@uuid_field = opts[:field]||:uuid
@uuid_overwrite = opts[:force]||false
end
end
module ClassMethods
# The field to store the uuid
attr_reader :uuid_field
# Whether to overwrite the create uuid if it already exists
def uuid_overwrite?
@uuid_overwrite
end
Plugins.inherited_instance_variables(self, :@uuid_field=>nil, :@uuid_overwrite=>nil)
end
module InstanceMethods
# Set the uuid when creating
def before_validation
set_uuid if new?
super
end
private
# Create a new UUID. This method can be overridden to use a separate
# method for creating UUIDs.
def create_uuid
SecureRandom.uuid
end
# If the object has accessor methods for the uuid field, and the uuid
# value is nil or overwriting it is allowed, set the uuid.
def set_uuid(uuid=create_uuid)
field = model.uuid_field
meth = :"#{field}="
if respond_to?(field) && respond_to?(meth) && (model.uuid_overwrite? || get_column_value(field).nil?)
set_column_value(meth, uuid)
end
end
end
end
end
end
|
