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
|
#!/usr/bin/env ruby
require 'json'
require 'fileutils'
include FileUtils
# Parses the argument array _args_, according to the pattern _s_, to
# retrieve the single character command line options from it. If _s_ is
# 'xy:' an option '-x' without an option argument is searched, and an
# option '-y foo' with an option argument ('foo').
#
# An option hash is returned with all found options set to true or the
# found option argument.
def go(s, args = ARGV)
b, v = s.scan(/(.)(:?)/).inject([{},{}]) { |t,(o,a)|
t[a.empty? ? 0 : 1][o] = a.empty? ? false : nil
t
}
while a = args.shift
a !~ /\A-(.+)/ and args.unshift a and break
p = $1
until p == ''
o = p.slice!(0, 1)
if v.key?(o)
v[o] = if p == '' then args.shift or break 1 else p end
break
elsif b.key?(o)
b[o] = true
else
args.unshift a
break 1
end
end and break
end
b.merge(v)
end
opts = go 'slhi:', args = ARGV.dup
if opts['h'] || opts['l'] && opts['s']
puts <<EOT
Usage: #{File.basename($0)} [OPTION] [FILE]
If FILE is skipped, this scripts waits for input from STDIN. Otherwise
FILE is opened, read, and used as input for the prettifier.
OPTION can be
-s to output the shortest possible JSON (precludes -l)
-l to output a longer, better formatted JSON (precludes -s)
-i EXT prettifies FILE in place, saving a backup to FILE.EXT
-h this help
EOT
exit 0
end
filename = nil
str = nil
json_opts = {:max_nesting => false, :create_additions => false}
if args.empty?
str = STDIN.read
else
str = File.read(filename = args.first)
end
json = JSON[str, json_opts]
output = if opts['s']
JSON.fast_generate json, json_opts
else # default is -l
JSON.pretty_generate json, json_opts
end
if opts['i'] && filename
cp filename, "#{filename}.#{opts['i']}"
File.open(filename, 'w') { |f| f.puts output }
else
puts output
end
|