kotovalexarian-likes-github/awesome_print
1
0
Fork 0
mirror of https://github.com/awesome-print/awesome_print synced 2023-03-27 23:22:34 -04:00
Code Releases Activity

Resolved merge conflicts

Browse source
This commit is contained in:
Mike Dvorkin 2013-09-18 13:04:45 -07:00
commit 86ff96d03c
2 changed files with 58 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
lib/awesome_print.rb
View file

@ -31,4 +31,5 @@ unless defined?(AwesomePrint::Inspector)
require File.dirname(__FILE__) + "/awesome_print/ext/nokogiri" if defined?(Nokogiri)
require File.dirname(__FILE__) + "/awesome_print/ext/no_brainer" if defined?(NoBrainer)
require File.dirname(__FILE__) + "/awesome_print/ext/ripple" if defined?(Ripple)
require File.dirname(__FILE__) + "/awesome_print/ext/sequel" if defined?(Sequel)
end

57
lib/awesome_print/ext/sequel.rb Normal file
View file

@ -0,0 +1,57 @@
# Copyright (c) 2010-2012 Michael Dvorkin
#
# Awesome Print is freely distributable under the terms of MIT license.
# See LICENSE file or http://www.opensource.org/licenses/mit-license.php
#------------------------------------------------------------------------------
module AwesomePrint
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)
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 { |key| key.to_s }.inject({}) do |hash, c|
hash[c[0].to_sym] = c[1]
hash
end
if !object.errors.empty?
data = {:errors => object.errors, :values => data}
end
"#{object} #{awesome_hash(data)}"
end
# Format Sequel Dataset object.
#------------------------------------------------------------------------------
def awesome_sequel_dataset(dataset)
[awesome_array(dataset.to_a), awesome_print(dataset.sql)].join("\n")
end
# Format Sequel Model class.
#------------------------------------------------------------------------------
def awesome_sequel_model_class(object)
data = object.db_schema.inject({}) {|h, (name,data)| h.merge(name => data[:db_type])}
"class #{object} < #{object.superclass} " << awesome_hash(data)
end
end
end
AwesomePrint::Formatter.send(:include, AwesomePrint::Sequel)