require 'data_mapper'
require 'date'

require 'playwhe/storage/models'

module PlayWhe

  module Storage

    def self.connect(path_to_db, log_level = :debug)
      DataMapper::Logger.new($stdout, log_level)
      DataMapper.setup(:default, "sqlite://#{File.expand_path(path_to_db)}")
    end

    def self.create(path = nil, log_level = :debug)
      setup(path, log_level)

      DataMapper.auto_migrate!
    end

    def self.update(path = nil, log_level = :debug)
      setup(path, log_level)

      (PlayWhe::BIRTHDAY.year..Date.today.year).each do |year|
        task = Task.first_or_create({ year: year },
          { month: year == PlayWhe::BIRTHDAY.year ? PlayWhe::BIRTHDAY.month : 1,
            status: 'pending' })

        unless task.completed?
          DataMapper.logger << "retrieving results for year #{year}..."

          before = Proc.new do |month|
            if month < task.month
              DataMapper.logger << "  - skipping #{Date::MONTHNAMES[month]}"
              true # skip
            else
              DataMapper.logger << "  + at #{Date::MONTHNAMES[month]}..."
              false # do not skip
            end
          end

          after = Proc.new do |month, flag, results|
            case flag
            when :ok
              Result.transaction do |t|
                begin
                  results.each do |r|
                    result = Result.first_or_new(draw: r[:draw], date: r[:date], period: r[:period], mark: r[:mark])

                    unless result.save
                      DataMapper.logger << "    - invalid data: #{result.draw} #{result.date} #{result.period} #{result.mark}"
                      Error.create(message: r.to_s)
                    end
                  end

                  today = Date.today
                  if Date.new(year, month) < Date.new(today.year, today.month)
                    if month == 12
                      task.status = 'completed'
                    else
                      task.month += 1
                    end

                    if task.save
                      DataMapper.logger << "  + completed #{Date::MONTHNAMES[month]}!"
                      DataMapper.logger << "completed year #{year}!" if task.completed?

                      true # continue
                    else
                      DataMapper.logger << '  - task not completed!'

                      false # do not continue
                    end
                  else
                    DataMapper.logger << 'All tasks completed!'

                    false # do not continue
                  end
                rescue
                  t.rollback

                  DataMapper.logger << 'Encountered an error during processing'

                  false # do not continue
                end
              end
            when :error
              DataMapper.logger << 'Encountered an error during processing'

              false # do not continue
            end
          end

          PlayWhe.results_for_year(year, before: before, after: after)
        end
      end
    end

  private

    def self.setup(path, log_level)
      path ||= File.join(Dir.home, '.playwhe')
      path = File.expand_path(path)
      Dir.mkdir(path) unless File.directory?(path)

      self.connect("#{path}/playwhe.db", log_level)
    end
  end
end