CSV to JSON

csv2json.rb

#!/usr/bin/env ruby
# encoding UTF-8

require 'csv'
require 'json'
require 'open-uri'

# Check if we're working with an URL or filename
if ARGV[0].match(URI.regexp)
  file = File.open(open(ARGV[0]))
else
  file = File.open(ARGV[0])
end

# Set Variables
case ARGV[1]
when "comma"
  col_sep =  ","
when "tab"
 col_sep =  "\t"
else
  col_sep = ","
end
output_filename = ARGV[2] || 'json.txt'

# Convert CSV to Hash
puts "Converting data"
csv_data = CSV.read(file, col_sep: col_sep)
headers = csv_data.shift.map {|i| i.to_s }
string_data = csv_data.map {|row| row.map {|cell| cell.to_s } }
csv_array_hash = string_data.map {|row| Hash[*headers.zip(row).flatten] }

# Write file
File.open(output_filename, 'w') {|f| f.write csv_array_hash.to_json}
puts "Created file with JSON at #{File.join(Dir.pwd, output_filename)}"