1
# frozen_string_literal: true
2

3 11
require "yaml"
4 11
require "json"
5

6
##
7
# Extension based upon Sequel::Migration and Sequel::Migrator
8
#
9
# Adds the Sequel::Seed module and the Sequel::Seed::Base and Sequel::Seeder
10
# classes, which allow the user to easily group entity changes and seed/fixture
11
# the database to a newer version only (unlike migrations, seeds are not
12
# directional).
13
#
14
# To load the extension:
15
#
16
#   Sequel.extension :seed
17
#
18
# It is also important to set the environment:
19
#
20
#   Sequel::Seed.setup(:development)
21

22 11
module Sequel
23 11
  class << self
24
    ##
25
    # Creates a Seed subclass according to the given +block+.
26
    #
27
    # The +env_labels+ lists on which environments the seed should be applicable.
28
    # If the current environment is not applicable, the seed is ignored. On the
29
    # other hand, if it is applicable, it will be listed in Seed.descendants and
30
    # subject to application (if it was not applied yet).
31
    #
32
    # Expected seed call:
33
    #
34
    #   Sequel.seed(:test) do # seed is only applicable to the test environment
35
    #     def run
36
    #       Entity.create attribute: value
37
    #     end
38
    #   end
39
    #
40
    # Wildcard seed:
41
    #
42
    #   Sequel.seed do # seed is applicable to every environment, or no environment
43
    #     def run
44
    #       Entity.create attribute: value
45
    #     end
46
    #   end
47
    #
48

49 11
    def seed(*env_labels, &block)
50 11
      return if env_labels.length > 0 && !env_labels.map(&:to_sym).include?(Seed.environment)
51

52 11
      seed = Class.new(Seed::Base)
53 11
      seed.class_eval(&block) if block_given?
54 11
      Seed::Base.inherited(seed) unless Seed::Base.descendants.include?(seed)
55 11
      seed
56
    end
57
  end
58

59 11
  module Seed
60 11
    class Error < Sequel::Error
61
    end
62

63 11
    class << self
64 11
      attr_reader :environment
65

66
      ##
67
      # Sets the Sequel::Seed"s environment to +env+ over which the Seeds should be applied
68 11
      def setup(env, opts = {})
69 11
        @environment = env.to_sym
70 11
        @options ||= {}
71 11
        @options[:disable_warning] ||= opts[:disable_warning] || false
72
      end
73

74
      ##
75
      # Keep backward compatibility on how to setup the Sequel::Seed environment
76
      #
77
      # Sets the environment +env+ over which the Seeds should be applied
78 11
      def environment=(env)
79 11
        setup(env)
80
      end
81

82
      ##
83
      # Keep backward compatibility on how to get Sequel::Seed::Base class descendants
84 11
      def descendants
85 11
        Base.descendants
86
      end
87

88
      ##
89
      # Keep backward compatibility on how to append a Sequel::Seed::Base descendant class
90 11
      def inherited(base)
91 11
        Base.inherited(base)
92
      end
93
    end
94

95
    ##
96
    # Helper methods for the Sequel::Seed project.
97

98 11
    module Helpers
99 11
      class << self
100 11
        def camelize(term, uppercase_first_letter = true)
101 11
          string = term.to_s
102 11
          if uppercase_first_letter
103 11
            string.gsub(/\/(.?)/) { "::" + $1.upcase }.gsub(/(^|_)(.)/) { $2.upcase }
104
          else
105 0
            string.first + camelize(string)[1..-1]
106
          end
107
        end
108
      end
109
    end
110

111 11
    module SeedDescriptor
112 11
      def apply_seed_descriptor(seed_descriptor)
113 11
        case seed_descriptor
114
        when Hash
115 11
          apply_seed_hash(seed_descriptor)
116
        when Array
117 11
          seed_descriptor.each { |seed_hash| apply_seed_hash(seed_hash) }
118
        end
119
      end
120

121 11
      private
122

123 11
      def apply_seed_hash(seed_hash)
124 11
        return unless seed_hash.class <= Hash
125 11
        if seed_hash.has_key?("environment")
126 11
          case seed_hash["environment"]
127
          when String, Symbol
128 11
            return if seed_hash["environment"].to_sym != Seed.environment
129
          when Array
130 0
            return unless seed_hash["environment"].map(&:to_sym).include?(Seed.environment)
131
          end
132
        end
133

134 11
        keys = seed_hash.keys
135 11
        keys.delete("environment")
136 11
        keys.each do |key|
137 11
          key_hash = seed_hash[key]
138 11
          entries = nil
139 11
          class_name = if key_hash.has_key?("class")
140 11
            entries = key_hash["entries"]
141 11
            key_hash["class"]
142
          else
143 11
            Helpers.camelize(key)
144
          end
145
          # It will raise an error if the class name is not defined
146 11
          class_const = Kernel.const_get(class_name)
147 11
          if entries
148 11
            entries.each { |hash| create_model(class_const, hash) }
149
          else
150 11
            create_model(class_const, key_hash)
151
          end
152
        end
153
      end
154

155 11
      def create_model(class_const, hash)
156 11
        object_instance = class_const.new
157 11
        object_instance_attr = hash.each do |attr, value|
158 11
          object_instance.set({attr.to_sym => value})
159
        end
160 11
        raise(Error, "Attempt to create invalid model instance of #{class_name}") unless object_instance.valid?
161 11
        object_instance.save
162
      end
163
    end
164

165 11
    class Base
166 11
      class << self
167 11
        def apply
168 11
          new.run
169
        end
170

171 11
        def descendants
172 11
          @descendants ||= []
173
        end
174

175 11
        def inherited(base)
176 11
          descendants << base
177
        end
178
      end
179

180 11
      def run
181
      end
182
    end
183

184
    ##
185
    # Class resposible for applying all the seeds related to the current environment,
186
    # if and only if they were not previously applied.
187
    #
188
    # To apply the seeds/fixtures:
189
    #
190
    #   Sequel::Seeder.apply(db, directory)
191
    #
192
    # +db+ holds the Sequel database connection
193
    #
194
    # +directory+ the path to the seeds/fixtures files
195
  end
196

197 11
  class Seeder
198 11
    SEED_FILE_PATTERN = /\A(\d+)_.+\.(rb|json|yml|yaml)\z/i.freeze
199 11
    RUBY_SEED_FILE_PATTERN = /\A(\d+)_.+\.(rb)\z/i.freeze
200 11
    YAML_SEED_FILE_PATTERN = /\A(\d+)_.+\.(yml|yaml)\z/i.freeze
201 11
    JSON_SEED_FILE_PATTERN = /\A(\d+)_.+\.(json)\z/i.freeze
202 11
    SEED_SPLITTER = "_".freeze
203 11
    MINIMUM_TIMESTAMP = 20000101
204

205 11
    Error = Seed::Error
206

207 11
    def self.apply(db, directory, opts = {})
208 11
      seeder_class(directory).new(db, directory, opts).run
209
    end
210

211 11
    def self.seeder_class(directory)
212 11
      if self.equal?(Seeder)
213 11
        Dir.new(directory).each do |file|
214 11
          next unless SEED_FILE_PATTERN.match(file)
215 11
          return TimestampSeeder if file.split(SEED_SPLITTER, 2).first.to_i > MINIMUM_TIMESTAMP
216
        end
217 11
        raise(Error, "seeder not available for files; please check the configured seed directory \"#{directory}\". Also ensure seed files are in YYYYMMDD_seed_file.rb format.")
218
      else
219 0
        self
220
      end
221
    end
222

223 11
    attr_reader :column
224

225 11
    attr_reader :db
226

227 11
    attr_reader :directory
228

229 11
    attr_reader :ds
230

231 11
    attr_reader :files
232

233 11
    attr_reader :table
234

235 11
    def initialize(db, directory, opts = {})
236 11
      raise(Error, "Must supply a valid seed path") unless File.directory?(directory)
237 11
      @db = db
238 11
      @directory = directory
239 11
      @allow_missing_seed_files = opts[:allow_missing_seed_files]
240 11
      @files = get_seed_files
241 11
      schema, table = @db.send(:schema_and_table, opts[:table]  || self.class.const_get(:DEFAULT_SCHEMA_TABLE))
242 11
      @table = schema ? Sequel::SQL::QualifiedIdentifier.new(schema, table) : table
243 11
      @column = opts[:column] || self.class.const_get(:DEFAULT_SCHEMA_COLUMN)
244 11
      @ds = schema_dataset
245 11
      @use_transactions = opts[:use_transactions]
246
    end
247

248 11
    private
249

250 11
    def checked_transaction(seed, &block)
251 11
      use_trans = if @use_transactions.nil?
252 11
        @db.supports_transactional_ddl?
253
      else
254 0
        @use_transactions
255
      end
256

257 11
      if use_trans
258 11
        db.transaction(&block)
259
      else
260 11
        yield
261
      end
262
    end
263

264 11
    def remove_seed_classes
265 11
      Seed::Base.descendants.each do |c|
266 11
        Object.send(:remove_const, c.to_s) rescue nil
267
      end
268 11
      Seed::Base.descendants.clear
269
    end
270

271 11
    def seed_version_from_file(filename)
272 0
      filename.split(SEED_SPLITTER, 2).first.to_i
273
    end
274
  end
275

276
  ##
277
  # A Seeder subclass to apply timestamped seeds/fixtures files.
278
  # It follows the same syntax & semantics for the Seeder superclass.
279
  #
280
  # To apply the seeds/fixtures:
281
  #
282
  #   Sequel::TimestampSeeder.apply(db, directory)
283
  #
284
  # +db+ holds the Sequel database connection
285
  #
286
  # +directory+ the path to the seeds/fixtures files
287

288 11
  class TimestampSeeder < Seeder
289 11
    DEFAULT_SCHEMA_COLUMN = :filename
290 11
    DEFAULT_SCHEMA_TABLE = :schema_seeds
291

292 11
    Error = Seed::Error
293

294 11
    attr_reader :applied_seeds
295

296 11
    attr_reader :seed_tuples
297

298 11
    def initialize(db, directory, opts = {})
299 11
      super
300 11
      @applied_seeds = get_applied_seeds
301 11
      @seed_tuples = get_seed_tuples
302
    end
303

304 11
    def run
305 11
      seed_tuples.each do |s, f|
306 11
        t = Time.now
307 11
        db.log_info("Applying seed file `#{f}`")
308 11
        checked_transaction(s) do
309 11
          s.apply
310 11
          fi = f.downcase
311 11
          ds.insert(column => fi)
312
        end
313 11
        db.log_info("Seed file `#{f}` applied, it took #{sprintf("%0.6f", Time.now - t)} seconds")
314
      end
315
      nil
316
    end
317

318 11
    private
319

320 11
    def get_applied_seeds
321 11
      am = ds.select_order_map(column)
322 11
      missing_seed_files = am - files.map { |f| File.basename(f).downcase }
323 11
      if missing_seed_files.length > 0 && !@allow_missing_seed_files
324 0
        raise(Error, "Seed files not in file system: #{missing_seed_files.join(", ")}")
325
      end
326 11
      am
327
    end
328

329 11
    def get_seed_files
330 11
      files = []
331 11
      Dir.new(directory).each do |file|
332 11
        next unless SEED_FILE_PATTERN.match(file)
333 11
        files << File.join(directory, file)
334
      end
335 11
      files.sort_by { |f| SEED_FILE_PATTERN.match(File.basename(f))[1].to_i }
336
    end
337

338 11
    def get_seed_tuples
339 11
      remove_seed_classes
340 11
      seeds = []
341 11
      ms = Seed::Base.descendants
342 11
      files.each do |path|
343 11
        f = File.basename(path)
344 11
        fi = f.downcase
345 11
        if !applied_seeds.include?(fi)
346
          #begin
347 11
          load(path) if RUBY_SEED_FILE_PATTERN.match(f)
348 11
          create_yaml_seed(path) if YAML_SEED_FILE_PATTERN.match(f)
349 11
          create_json_seed(path) if JSON_SEED_FILE_PATTERN.match(f)
350
          #rescue Exception => e
351
            #raise(Error, "error while processing seed file #{path}: #{e.inspect}")
352
          #end
353 11
          el = [ms.last, f]
354 11
          next if ms.last.nil?
355 11
          if ms.last < Seed::Base && !seeds.include?(el)
356 11
            seeds << el
357
          end
358
        end
359
      end
360 11
      seeds
361
    end
362

363 11
    def create_yaml_seed(path)
364 11
      seed_descriptor = YAML::load(File.open(path))
365 11
      seed = Class.new(Seed::Base)
366 11
      seed.const_set "YAML_SEED", seed_descriptor
367 11
      seed.class_eval do
368 11
        include Seed::SeedDescriptor
369

370 11
        def run
371 11
          seed_descriptor = self.class.const_get "YAML_SEED"
372 11
          raise(Error, "YAML seed improperly defined") if seed_descriptor.nil?
373 11
          self.apply_seed_descriptor(seed_descriptor)
374
        end
375
      end
376 11
      Seed::Base.inherited(seed) unless Seed::Base.descendants.include?(seed)
377 11
      seed
378
    end
379

380 11
    def create_json_seed(path)
381 11
      seed_descriptor = JSON.parse(File.read(path))
382 11
      seed = Class.new(Seed::Base)
383 11
      seed.const_set "JSON_SEED", seed_descriptor
384 11
      seed.class_eval do
385 11
        include Seed::SeedDescriptor
386

387 11
        def run
388 11
          seed_descriptor = self.class.const_get "JSON_SEED"
389 11
          raise(Error, "JSON seed improperly defined") if seed_descriptor.nil?
390 11
          self.apply_seed_descriptor(seed_descriptor)
391
        end
392
      end
393 11
      Seed::Base.inherited(seed) unless Seed::Base.descendants.include?(seed)
394 11
      seed
395
    end
396

397 11
    def schema_dataset
398 11
      c = column
399 11
      ds = db.from(table)
400 11
      if !db.table_exists?(table)
401 11
        db.create_table(table) { String c, primary_key: true }
402 9
      elsif !ds.columns.include?(c)
403 0
        raise(Error, "Seeder table \"#{table}\" does not contain column \"#{c}\"")
404
      end
405 11
      ds
406
    end
407
  end
408
end

Read our documentation on viewing source code .

Loading