#!/usr/bin/ruby
require 'yaml'
require 'csv'

def usage
    puts <<EOF
Usage: #{$0} dictionary-source-file coverage-file

    reads dictionary-source-file and produces YAML output
    with all visible words and their help text, flags, etc.

    if coverage-file is specified, merge test coverage data
EOF
    exit 1
end

dict = ARGV.shift || usage
File.readable?(dict) || abort("#{dict} not found!")

cov = ARGV.shift

coverage = {}
if cov && File.readable?(cov)
    coverage = YAML.load(File.read(cov))
end

input = File.read(dict)
output = Hash.new()

help = []
input.lines.each do |line|
    case line
    when /;.*noindex/
        # nothing
    when /^\s*;\s+H:\s*(.+)/
        help << $1
    when /^\s*([dh])word(q?)\s+(.+)/
        label, name, flags = CSV.parse_line($3)
        name.upcase!
        name.tr!("'", '"') if $2 == 'q'
        output[name] ||= {'label' => label, 'headerless' => ($1 == 'h')}
        output[name].merge!({'help' => help}) unless help.empty?
        if flags
            fl = flags.split(/[|\+]/)
            output[name].merge!({'flags' => fl}) unless fl.empty?
        end
        output[name].merge!({"tests" => coverage[name.downcase]}) if coverage[name.downcase]
    when /^\s*eword/
        help = []
    end
end

puts output.to_yaml