Noosfero/noosfero

class CustomFormsPlugin::Graph
  # @query_results have the format
  # [
  #   {
  #     "summary"=>{}, "data"=>{"foo"=>1, "bar"=>1, "teste"=>1},
  #     "show_as"=>"radio", "field"=>"some form field"
  #   }
  # ]
  # Each 'data' key on @query_results represents the data that will be used by
  # chartkick lib, to render a graph based on the show_as value.

  AVAILABLE_FIELDS = %w(check_box radio select multiple_select text)

  def initialize(form)
    @form = form
    data = get_data()
    @query_results = format_data(data)
  end

  def chart_to_show_data(show_as)
    return "pizza" if ["radio", "select"].include?(show_as)
    return "column" if ["check_box", "multiple_select"].include?(show_as)
    return "text" if show_as == "text"
  end

  def query_results
    @query_results
  end

  private

    def get_select_fields
      # TODO: Use left_joins after Rails 5 upgrade
      @form.fields
           .joins(:alternatives)
           .joins('LEFT JOIN custom_forms_plugin_form_answers
                   ON custom_forms_plugin_alternatives.id = custom_forms_plugin_form_answers.alternative_id')
           .group("custom_forms_plugin_fields.id, custom_forms_plugin_alternatives.id")
           .order("custom_forms_plugin_alternatives.id")
           .select('custom_forms_plugin_fields.name AS field_name,
                     custom_forms_plugin_fields.show_as AS show_as,
                     custom_forms_plugin_alternatives.id AS alternative,
                     custom_forms_plugin_alternatives.label AS label,
                     COUNT(custom_forms_plugin_form_answers.alternative_id) AS answer_count')
    end

    def get_text_fields
      @form.fields
           .joins("INNER JOIN custom_forms_plugin_answers as ans ON ans.field_id = custom_forms_plugin_fields.id
                   INNER JOIN custom_forms_plugin_submissions as s ON s.id = ans.submission_id")
           .group("custom_forms_plugin_fields.id, custom_forms_plugin_fields.name")
           .select("custom_forms_plugin_fields.id as id,
                     custom_forms_plugin_fields.name as field_name,
                     STRING_AGG(ans.value, ',') as answers,
                     STRING_AGG(s.author_name, ',') as users,
                     custom_forms_plugin_fields.show_as as show_as,
                     STRING_AGG(ans.imported::text, ',') as imported")
           .where("custom_forms_plugin_fields.type = 'CustomFormsPlugin::TextField'")
    end

    def format_text_fields(unformatted_text_fields)
      formated_text_fields = []
      unformatted_text_fields.each do |f|
        text_answers = {
          "data" => {
            "answers" => f.answers.map(&:value),
            "users" => f.users.split(","),
            "imported" => f.imported.split(",")
          },
          "show_as" => "text",
          "field" => f.field_name
        }

        formated_text_fields << text_answers
      end

      formated_text_fields
    end

    # TODO: Maybe there's a better way to do this
    def format_select_fields(unformatted_select_fields)
      computed_fields = []
      computed_data = []

      unformatted_select_fields.map do |r|
        if computed_fields.include?(r.field_name)
          computed_data.each do |c|
            if c["field"] == r.field_name
              c["data"][r.label] = r.answer_count
              break
            end
          end
        else
          computed_fields << r.field_name
          data = {
            "data" => { r.label => r.answer_count },
            "field" => r.field_name,
            "show_as" => chart_to_show_data(r.show_as),
            "summary" => {}
          }
          computed_data << data
        end
      end

      computed_data
    end

    def get_data
      select_fields = get_select_fields
      text_fields = get_text_fields

      { text_fields: text_fields, select_fields: select_fields }
    end

    def format_data(data)
      select_fields = []
      text_fields = []

      unless data[:text_fields] == []
        text_fields = format_text_fields(data[:text_fields])
      end

      unless data[:select_fields] == []
        select_fields = format_select_fields(data[:select_fields])
      end

      select_fields + text_fields
    end
end