require 'rake/testtask'
require 'colorize'

Rake::TestTask.new do |t|
  t.libs << "test"
  t.test_files = FileList['test/**/*_test.rb']
  t.warning = true
  t.verbose = true
end

task :default => :test

namespace :dd do
  require 'csv'
  require 'fileutils'
  require 'rubygems'
  require 'json'

  desc 'Create Data Dictionary from repository'
  task :create do

    folder = "dd/#{ENV['VERSION'] || Spout::Application.new.version}"
    FileUtils.mkpath folder

    CSV.open("#{folder}/variables.csv", "wb") do |csv|
      keys = %w(id display_name description type units domain labels calculation)
      csv << ['folder'] + keys
      Dir.glob("variables/**/*.json").each do |file|
        if json = JSON.parse(File.read(file)) rescue false
          variable_folder = file.gsub(/variables\//, '').split('/')[0..-2].join('/')
          csv << [variable_folder] + keys.collect{|key| json[key].kind_of?(Array) ? json[key].join(';') : json[key].to_s}
        end
      end
    end
    CSV.open("#{folder}/domains.csv", "wb") do |csv|
      keys = %w(value display_name description)
      csv << ['folder', 'id'] + keys
      Dir.glob("domains/**/*.json").each do |file|
        if json = JSON.parse(File.read(file)) rescue false
          domain_folder = file.gsub(/domains\//, '').split('/')[0..-2].join('/')
          domain_name = file.gsub(/domains\//, '').split('/').last.to_s.gsub(/.json/, '')
          json.each do |hash|
            csv << [domain_folder, domain_name] + keys.collect{|key| hash[key]}
          end
        end
      end
    end

    puts "Data Dictionary Created in #{folder}"
  end

  desc 'Initialize JSON repository from a CSV file: CSV=datadictionary.csv'
  task :import do
    additional_csv_info =  "\n\nFor additional information on specifying CSV column headers before import see:\n\n    " + "https://github.com/sleepepi/spout#generate-a-new-repository-from-an-existing-csv-file".colorize( :light_cyan ) + "\n\n"

    puts ENV['CSV'].inspect
    if File.exists?(ENV['CSV'].to_s)
      CSV.parse( File.open(ENV['CSV'].to_s, 'r:iso-8859-1:utf-8'){|f| f.read}, headers: true ) do |line|
        row = line.to_hash
        if not row.keys.include?('id')
          puts "\nMissing column header `".colorize( :red ) + "id".colorize( :light_cyan ) + "` in data dictionary.".colorize( :red ) + additional_csv_info
          exit(1)
        end
        next if row['id'] == ''
        folder = File.join('variables', row.delete('folder').to_s)
        FileUtils.mkpath folder
        hash = {}
        id = row.delete('id')
        hash['id'] = id
        hash['display_name'] = row.delete('display_name')
        hash['description'] = row.delete('description').to_s
        hash['type'] = row.delete('type')
        domain = row.delete('domain').to_s
        hash['domain'] = domain if domain != ''
        units = row.delete('units').to_s
        hash['units'] = units if units != ''
        calculation = row.delete('calculation').to_s
        hash['calculation'] = calculation if calculation != ''
        labels = row.delete('labels').to_s.split(';')
        hash['labels'] = labels if labels.size > 0
        hash['other'] = row unless row.empty?

        file_name = File.join(folder, id.downcase + '.json')
        File.open(file_name, 'w') do |file|
          file.write(JSON.pretty_generate(hash))
        end
        puts "      create".colorize( :green ) + "  #{file_name}"
      end
    else
      puts "\nPlease specify a valid CSV file.".colorize( :red ) + additional_csv_info
    end
  end
end