dic_creater.rb

require "csv"
require 'json'

INPUT_FILE_PATH = "./input/aws_services_name_base.tsv"
OUPTPUT_DETAIL_FILE_PATH = "./output/aws_dic_detail.txt"
OUPTPUT_SIMPLE_FILE_PATH = "./output/aws_dic_simple.txt"

OUPTPUT_DETAIL_JSON_FILE_PATH = "./site/src/static/output/aws_dic_detail.json"
OUPTPUT_SIMPLE_JSON_FILE_PATH = "./site/src/static/output/aws_dic_simple.json"

class DicCreater
  attr_reader :detail_arr, :simple_arr
  def initialize
    @detail_arr = []
    @simple_arr = []
  end

  def separate_row_data(row)
    dt = detail_transcript(row)
    @detail_arr += transform(row, dt, 2)
    @simple_arr += transform(row, simple_transcript(row), 2)
  end

  def detail_transcript(row)
    !row[0].nil? ? row[0] + " " + row[1] : row[1] 
  end

  def simple_transcript(row)
    row[1]
  end

  def transform(row, transcript, start)
    arr = []
    i = start
    loop do
      raise "end" if row[i].nil? 
      arr.push([row[i], transcript, "名詞"])
      i+=1
    rescue
      return arr
    end
  end
end

dc = DicCreater.new
input = CSV.read(INPUT_FILE_PATH, col_sep: "\t", headers: true)

input.each do  |row|
  dc.separate_row_data(row)
end

CSV.open(OUPTPUT_DETAIL_FILE_PATH, "w", :col_sep => "\t") do |io|
  dc.detail_arr.each { |row| io.puts(row) }
end

CSV.open(OUPTPUT_SIMPLE_FILE_PATH, "w", :col_sep => "\t") do |io|
  dc.simple_arr.each { |row| io.puts(row) }
end

# def convert_json(arr)
#   hash = {}
#   arr.each do |cell|
#     hash[cell[0]] = cell[1]
#   end
#   hash.to_json
# end
def convert_json(arr)
  result = arr.map do |cell|
    { key: cell[0], val: cell[1] }
  end
  result.to_json
end

File.open(OUPTPUT_DETAIL_JSON_FILE_PATH, "w") do |text|
  text.puts convert_json(dc.detail_arr)
end

File.open(OUPTPUT_SIMPLE_JSON_FILE_PATH, "w") do |text|
  text.puts convert_json(dc.simple_arr)
end