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
71
72
73
74
75
76
77
78
79
80
|
# Licensed to Elasticsearch B.V. under one or more contributor
# license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright
# ownership. Elasticsearch B.V. licenses this file to you under
# the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
module Elasticsearch
module Extensions
module ANSI
# Wrapper for the Elasticsearch response body, which adds a {#to_ansi} method
#
class ResponseBody < DelegateClass(Hash)
def initialize(body)
super(body)
end
# Return a [colorized and formatted](http://en.wikipedia.org/wiki/ANSI_escape_code)
# representation of the Elasticsearch response for:
#
# * Search results (hits and highlights)
# * Facets (terms, statistical, histogram, date_histogram)
# * Analyze API output
# * Shard allocation
#
# @example Display formatted search results
#
# require 'elasticsearch/extensions/ansi'
# puts Elasticsearch::Client.new.search.to_ansi
#
# @todo Add all facets and handlers for remaining response parts / types
#
def to_ansi(options={})
output = Actions.public_methods.select do |m|
m.to_s =~ /^display_/
end.map do |m|
Actions.send(m, self, options)
end
unless output.compact.empty?
output.compact.join("\n")
else
self.respond_to?(:awesome_inspect) ? self.awesome_inspect : self.inspect
end
end
end
end
end
end
module Elasticsearch
module Transport
module Transport
class Response
# Wrap the response body in the {Extensions::ANSI::ResponseBody} class
#
def body_to_ansi
Extensions::ANSI::ResponseBody.new @body
end
alias_method :body_original, :body
alias_method :body, :body_to_ansi
end
end
end
end
|
