# encoding: utf-8

require 'ostruct'
require 'csv'

require 'flextures/flextures_base_config'
require 'flextures/flextures_extension_modules'
require 'flextures/flextures'
require 'flextures/flextures_factory'

module Flextures
  # Dumperと違ってデータの吐き出し処理をまとめたクラス
  module Loader 
    PARENT = Flextures

    # 型に応じて勝手にdefault値を設定する
    COMPLETER = {
      binary:->{ 0 },
      boolean:->{ false },
      date:->{ DateTime.now },
      datetime:->{ DateTime.now },
      decimal:->{ 0 },
      float:->{ 0.0 },
      integer:->{ 0 },
      string:->{ "" },
      text:->{ "" },
      time:->{ DateTime.now },
      timestamp:->{ DateTime.now },
    }

    # 型の変換を行う
    TRANSLATER = {
      binary:->(d){
        return d if d.nil?
        Base64.encode64(d)
      },
      boolean:->(d){
        return d if d.nil?
        (0==d || ""==d || !d) ? false : true
      },
      date:->(d){
        return d   if d.nil?
        return nil if d==""
        Date.parse(d.to_s)
      },
      datetime:->(d){
        return d   if d.nil?
        return nil if d==""
        DateTime.parse(d.to_s)
      },
      decimal:->(d){
        return d   if d.nil?
        d.to_i
      },
      float:->(d){
        return d   if d.nil?
        d.to_f
      },
      integer:->(d){
        return d   if d.nil?
        d.to_i
      },
      string:->(d){
        return d if d.nil?
        return d if d.is_a?(Hash)
        return d if d.is_a?(Array)
        d.to_s
      },
      text:->(d){
        return d if d.nil?
        return d if d.is_a?(Hash)
        return d if d.is_a?(Array)
        d.to_s
      },
      time:->(d){
        return d   if d.nil?
        return nil if d==""
        DateTime.parse(d.to_s)
      },
      timestamp:->(d){
        return d   if d.nil?
        return nil if d==""
        DateTime.parse(d.to_s)
      },
    }

    # どのファイルが存在するかチェック
    # @param [Hash] format ロードしたいファイルの情報
    # @return 存在するファイルの種類(csv,yml)、どちも存在しないならnil
    def self.file_exist format, type = [:csv,:yml]
      table_name = format[:table].to_s
      file_name = format[:file] || format[:table]
      dir_name = format[:dir] || LOAD_DIR

      ext=->{
        if    type.member?(:csv) and File.exist? "#{dir_name}#{file_name}.csv"
          :csv
        elsif type.member?(:yml) and File.exist? "#{dir_name}#{file_name}.yml"
          :yml
        else
          nil
        end
      }.call

      [table_name, "#{dir_name}#{file_name}",ext]
    end

    # fixturesをまとめてロード、主にテストtest/unit, rspec で使用する
    #
    # 全テーブルが対象
    # fixtures :all
    # テーブル名で一覧する
    # fixtures :users, :items
    # ハッシュで指定
    # fixtures :users => :users2
    def self.flextures *fixtures
      # :allですべてのfixtureを反映
      fixtures = Flextures::deletable_tables if fixtures.size== 1 and :all == fixtures.first
      fixtures_hash = fixtures.pop if fixtures.last and fixtures.last.is_a? Hash # ハッシュ取り出し
      fixtures.each{ |table_name| Loader::load table: table_name, loader: :fun }
      fixtures_hash.each{ |k,v| Loader::load table: k, file: v,   loader: :fun } if fixtures_hash
      fixtures
    end

    # csv 優先で存在している fixtures をロード
    def self.load format
      table_name, file_name, method = file_exist format
      if method
        send(method, format)
      else
        # ファイルが存在しない時
        print "Warning: #{file_name}  is not exist!\n" 
      end
    end

    # CSVのデータをロードする
    def self.csv format
      table_name, file_name, ext = file_exist format, [:csv]

      print "try loading #{file_name}.csv\n" unless [:fun].include? format[:loader]
      return nil unless File.exist? "#{file_name}.csv"

      klass = PARENT::create_model table_name
      attributes = klass.columns.map &:name
      filter = create_filter klass, Factory[table_name], file_name, :csv
      # rails3_acts_as_paranoid がdelete_allで物理削除しないことの対策
      klass.send( klass.respond_to?(:delete_all!) ? :delete_all! : :delete_all )
      CSV.open( "#{file_name}.csv" ) do |csv|
        keys = csv.shift # keyの設定
        warning "CSV", attributes, keys
        csv.each do |values|
          h = values.extend(Extensions::Array).to_hash(keys)
          o = filter.call h
          o.save
        end
      end
      "#{file_name}.csv"
    end

    # YAML形式でデータをロードする
    def self.yml format
      table_name, file_name, ext = file_exist format, [:yml]

      print "try loading #{file_name}.yml\n" unless [:fun].include? format[:loader]
      return nil unless File.exist? "#{file_name}.yml"

      attributes, filter = ->{
        klass = PARENT::create_model table_name
        # rails3_acts_as_paranoid がdelete_allで物理削除しないことの対策
        klass.send( klass.respond_to?(:delete_all!) ? :delete_all! : :delete_all )
        attributes = klass.columns.map &:name
        filter = create_filter klass, Factory[table_name], file_name, :yml
        [attributes, filter]
      }.call

      yaml = YAML.load(File.open("#{file_name}.yml"))
      return false unless yaml # ファイルの中身が空の場合
      yaml.each do |k,h|
        warning "YAML", attributes, h.keys
        o = filter.call h
        o.save
      end
      "#{file_name}.yml"
    end

    # 欠けたカラムを検知してメッセージを出しておく
    def self.warning format, attributes, keys
      (attributes-keys).each { |name| print "Warning: #{format} colum is missing! [#{name}]\n" }
      (keys-attributes).each { |name| print "Warning: #{format} colum is left over! [#{name}]\n" }
    end

    # フィクスチャから取り出した値を、加工して欲しいデータにするフィルタを作成して返す
    def self.create_filter klass, factory, filename, ext
      columns = klass.columns
      # テーブルからカラム情報を取り出し
      column_hash = {}
      columns.each { |col| column_hash[col.name] = col }
      # 自動補完が必要なはずのカラム
      lack_columns = columns.select { |c| !c.null and !c.default }.map{ |o| o.name.to_sym }
      not_nullable_columns = columns.select { |c| !c.null }.map &:name
      # ハッシュを受け取って、必要な値に加工してからハッシュで返すラムダを返す
      return->(h){
        # テーブルに存在しないキーが定義されているときは削除
        h.select! { |k,v| column_hash[k] }
        o = klass.new 
        # 値がnilでないなら型をDBで適切なものに変更
        h.each{ |k,v| nil==v || o[k] = (TRANSLATER[column_hash[k].type] ? TRANSLATER[column_hash[k].type].call(v) : nil) }
        # FactoryFilterを動作させる
        factory.call(*[o, :load, filename, ext][0,factory.arity]) if factory
        # 値がnilの列にデフォルト値を補間
        not_nullable_columns.each{ |k| o[k]==nil && o[k] = (COMPLETER[column_hash[k].type] ? COMPLETER[column_hash[k].type].call : nil) }
        # 列ごと抜けているデータを保管
        lack_columns.each { |k| nil==o[k] && o[k] = COMPLETER[column_hash[k].type].call }
        o
      }
    end
  end
end