File: encoder.rb

package info (click to toggle)
ruby-perfect-toml 0.9.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 224 kB
  • sloc: ruby: 1,929; makefile: 4
file content (47 lines) | stat: -rwxr-xr-x 1,429 bytes parent folder | download | duplicates (2)
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
#!/usr/bin/env ruby

require_relative "../lib/perfect_toml"
require "json"
require "time"

def convert(json)
  return json.map {|v| convert(v) } if Array === json
  if json.key?("type")
    type, val = json["type"], json["value"]
    case type
    when "integer" then val.to_i
    when "float"
      case val
      when "inf", "+inf" then Float::INFINITY
      when "-inf" then -Float::INFINITY
      when "nan" then -Float::NAN
      else
        val.to_f
      end
    when "string" then val
    when "bool" then val == "true"
    when "datetime" then Time.iso8601(val)
    when "datetime-local"
      raise if val !~ /\A(-?\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2}(?:\.\d+)?)\z/
      PerfectTOML::LocalDateTime.new($1, $2, $3, $4, $5, $6)
    when "date-local"
      raise if val !~ /\A(-?\d{4})-(\d{2})-(\d{2})\z/
      PerfectTOML::LocalDate.new($1, $2, $3)
    when "time-local"
      raise if val !~ /\A(\d{2}):(\d{2}):(\d{2}(?:\.\d+)?)\z/
      PerfectTOML::LocalTime.new($1, $2, $3)
    else
      raise "unknown type: %p" % type
    end
  else
    json.to_h {|k, v| [k, convert(v)] }
  end
end

opts = {
  use_dot: ENV["TOML_ENCODER_USE_DOT"] == "1",
  sort_keys: ENV["TOML_ENCODER_SORT_KEYS"] == "1",
  use_literal_string: ENV["TOML_ENCODER_USE_LITERAL_STRING"] == "1",
  use_multiline_string: ENV["TOML_ENCODER_USE_MULTILINE_STRING"] == "1",
}
puts PerfectTOML.generate(convert(JSON.parse($stdin.read)), **opts)