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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
module Graphlient
class Query
SCALAR_TYPES = {
int: 'Int',
float: 'Float',
string: 'String',
boolean: 'Boolean'
}.freeze
ROOT_NODES = %w[query mutation subscription].freeze
attr_accessor :query_str
def initialize(&block)
@indents = 0
@query_str = ''
@variables = []
instance_eval(&block)
end
def method_missing(method_name, *args, &block)
append_node(method_name, args, &block)
end
ROOT_NODES.each do |root_node|
define_method(root_node) do |*args, &block|
@variables = args.first unless args.empty?
append_node(root_node, args, arg_processor: ->(k, v) { "$#{k}: #{variable_string(v)}" }, &block)
end
end
def respond_to_missing?(method_name, include_private = false)
super
end
def to_s
query_str.strip
end
private
def append_node(node, args, arg_processor: nil, &block)
# add field
@query_str << "\n#{indent}#{node}"
# add filter
hash_arguments = hash_arg(args)
@query_str << "(#{args_str(hash_arguments, arg_processor: arg_processor)})" if hash_arguments
if block_given?
@indents += 1
@query_str << '{'
instance_eval(&block)
@query_str << '}'
@indents -= 1
end
@query_str << "\n#{indent}"
end
def indent
' ' * @indents
end
def hash_arg(args)
args.detect { |arg| arg.is_a? Hash }
end
def args_str(hash_args, arg_processor: nil)
hash_args.map do |k, v|
arg_processor ? arg_processor.call(k, v) : argument_string(k, v)
end.join(', ')
end
def argument_string(key, val)
"#{key}: #{argument_value_string(val)}"
end
def variable_string(val)
case val
when :id, :id!
val.to_s.upcase
when ->(v) { SCALAR_TYPES.key?(v.to_s.delete('!').to_sym) }
# scalar types
val.to_s.camelize
when Array
"[#{variable_string(val.first)}]"
else
val.to_s
end
end
def argument_value_string(value)
case value
when String
"\"#{value}\""
when Numeric
value.to_s
when Array
"[#{value.map { |v| argument_value_string(v) }.join(', ')}]"
when Hash
"{ #{value.map { |k, v| "#{k}: #{argument_value_string(v)}" }.join(', ')} }"
when Symbol
@variables.key?(value) ? "$#{value}" : value.to_s.camelize(:lower)
else
value
end
end
end
end
|
