nairobi/app/models/poll/stats.rb

class Poll
  class Stats

    def initialize(poll)
      @poll = poll
    end

    def generate
      stats = %w[total_participants total_participants_web total_web_valid total_web_white total_web_null
                 total_participants_booth total_booth_valid total_booth_white total_booth_null
                 total_valid_votes total_white_votes total_null_votes valid_percentage_web valid_percentage_booth
                 total_valid_percentage white_percentage_web white_percentage_booth total_white_percentage
                 null_percentage_web null_percentage_booth total_null_percentage total_participants_web_percentage
                 total_participants_booth_percentage]
      stats.map { |stat_name| [stat_name.to_sym, send(stat_name)] }.to_h
    end

    private

      def total_participants
        stats_cache("total_participants") { total_participants_web + total_participants_booth }
      end

      def total_participants_web
        stats_cache("total_participants_web") { total_web_valid + total_web_white + total_web_null }
      end

      def total_participants_web_percentage
        stats_cache("total_participants_web_percentage") do
          total_participants.zero? ? 0 : total_participants_web * 100 / total_participants
        end
      end

      def total_participants_booth
        stats_cache("total_participants_booth") { voters.where(origin: "booth").count }
      end

      def total_participants_booth_percentage
        stats_cache("total_participants_booth_percentage") do
          total_participants.zero? ? 0 : total_participants_booth * 100 / total_participants.to_f
        end
      end

      def total_web_valid
        stats_cache("total_web_valid") { voters.where(origin: "web").count }
      end

      def valid_percentage_web
        stats_cache("valid_percentage_web") do
          total_valid_votes.zero? ? 0 : total_web_valid * 100 / total_valid_votes.to_f
        end
      end

      def total_web_white
        stats_cache("total_web_white") { 0 }
      end

      def white_percentage_web
        stats_cache("white_percentage_web") { 0 }
      end

      def total_web_null
        stats_cache("total_web_null") { 0 }
      end

      def null_percentage_web
        stats_cache("null_percentage_web") { 0 }
      end

      def total_booth_valid
        stats_cache("total_booth_valid") { recounts.sum(:total_amount) }
      end

      def valid_percentage_booth
        stats_cache("valid_percentage_booth") do
          total_valid_votes.zero? ? 0 : total_booth_valid * 100 / total_valid_votes.to_f
        end
      end

      def total_booth_white
        stats_cache("total_booth_white") { recounts.sum(:white_amount) }
      end

      def white_percentage_booth
        stats_cache("white_percentage_booth") do
          total_white_votes.zero? ? 0 : total_booth_white * 100 / total_white_votes.to_f
        end
      end

      def total_booth_null
        stats_cache("total_booth_null") { recounts.sum(:null_amount) }
      end

      def null_percentage_booth
        stats_cache("null_percentage_booth") do
          total_null_votes.zero? ? 0 : total_booth_null * 100 / total_null_votes.to_f
        end
      end

      def total_valid_votes
        stats_cache("total_valid_votes") { total_web_valid + total_booth_valid }
      end

      def total_valid_percentage
        stats_cache("total_valid_percentage") do
          total_participants.zero? ? 0 : total_valid_votes * 100 / total_participants.to_f
        end
      end

      def total_white_votes
        stats_cache("total_white_votes") { total_web_white + total_booth_white }
      end

      def total_white_percentage
        stats_cache("total_white_percentage") do
          total_participants.zero? ? 0 : total_white_votes * 100 / total_participants.to_f
        end
      end

      def total_null_votes
        stats_cache("total_null_votes") { total_web_null + total_booth_null }
      end

      def total_null_percentage
        stats_cache("total_null_percentage") do
          total_participants.zero? ? 0 : total_null_votes * 100 / total_participants.to_f
        end
      end

      def voters
        stats_cache("voters") { @poll.voters }
      end

      def recounts
        stats_cache("recounts") { @poll.recounts }
      end

      def stats_cache(key, &block)
        Rails.cache.fetch("polls_stats/#{@poll.id}/#{key}", &block)
      end

  end
end