lib/amazing_print/ext/sequel.rb
# frozen_string_literal: true
# Copyright (c) 2010-2016 Michael Dvorkin and contributors
#
# AmazingPrint is freely distributable under the terms of MIT license.
# See LICENSE file or http://www.opensource.org/licenses/mit-license.php
#------------------------------------------------------------------------------
module AmazingPrint
module Sequel
def self.included(base)
base.send :alias_method, :cast_without_sequel, :cast
base.send :alias_method, :cast, :cast_with_sequel
end
# Add Sequel class names to the dispatcher pipeline.
#------------------------------------------------------------------------------
def cast_with_sequel(object, type)
cast = cast_without_sequel(object, type)
if defined?(::Sequel::Model) && object.is_a?(::Sequel::Model)
cast = :sequel_document
elsif defined?(::Sequel::Model) && object.is_a?(Class) && object.ancestors.include?(::Sequel::Model) && object&.name != 'Sequel::Model'
cast = :sequel_model_class
elsif defined?(::Sequel::Mysql2::Dataset) && object.class.ancestors.include?(::Sequel::Mysql2::Dataset)
cast = :sequel_dataset
end
cast
end
# Format Sequel Document object.
#------------------------------------------------------------------------------
def awesome_sequel_document(object)
data = object.values.sort_by(&:to_s).each_with_object({}) do |c, hash|
hash[c[0].to_sym] = c[1]
end
data = { errors: object.errors, values: data } unless object.errors.empty?
"#{object} #{awesome_hash(data)}"
end
# Format Sequel Dataset object.
#------------------------------------------------------------------------------
def awesome_sequel_dataset(dataset)
[awesome_array(dataset.to_a), amazing_print(dataset.sql)].join("\n")
end
# Format Sequel Model class.
#------------------------------------------------------------------------------
def awesome_sequel_model_class(object)
data = object.db_schema.inject({}) { |h, (prop, defn)| h.merge(prop => defn[:db_type]) }
name = "class #{awesome_simple(object.to_s, :class)}"
base = "< #{awesome_simple(object.superclass.to_s, :class)}"
[name, base, awesome_hash(data)].join(' ')
end
end
end
AmazingPrint::Formatter.include AmazingPrint::Sequel