pulibrary / orangelight / 00dbc10b-d747-4ad7-b211-7b26d753abbb

require 'csv'

class BookmarksController < CatalogController

  include Blacklight::Bookmarks

  configure_blacklight do |_config|

    blacklight_config.show.document_actions[:print] =

      {

        partial: 'document_action',

        name: :print,

        modal: false

      }

  end

  def print

    fetch_bookmarked_documents

    @url_gen_params = {}

    render('orangelight/record_mailer/email_record', formats: [:text])

  end

  def citation

    bookmarks = token_or_current_or_guest_user.bookmarks

    bookmark_ids = bookmarks.collect { |bookmark| bookmark.document_id.to_s }

    @documents = search_service.fetch(bookmark_ids, { rows: bookmark_ids.count, fl: "author_citation_display, title_citation_display, pub_citation_display, number_of_pages_citation_display, pub_date_start_sort, edition_display, format" })

  end

  def csv

    fetch_bookmarked_documents

    send_data csv_output, type: 'text/csv', filename: "bookmarks-#{Time.zone.today}.csv"

  end

  private

    def fetch_bookmarked_documents

      bookmark_ids = token_or_current_or_guest_user.bookmarks.collect { |bookmark| bookmark.document_id.to_s }

      @documents = search_service.fetch(bookmark_ids, rows: bookmark_ids.length, fl: '*')

    end

    def csv_bom

      %w[EF BB BF].map { |a| a.hex.chr }.join

    end

    def csv_fields

      {

        id: 'ID',

        title_citation_display: ['Title', 'Title (Original Script)'],

        author_display: ['Author', 'Author (Original Script)'],

        format: 'Format',

        language_facet: 'Language',

        pub_citation_display: 'Published/Created',

        pub_date_display: 'Date',

        description_display: 'Description',

        series_display: 'Series',

        location: 'Library',

        location_display: 'Location',

        call_number_display: 'Call Number',

        notes_display: 'Notes',

        edition_display: 'Edition'

      }

    end

    def csv_output

      CSV.generate(csv_bom, headers: true) do |csv|

        csv << csv_fields.values.flatten

        @documents.each do |doc|

          csv << csv_fields.keys.map { |field| csv_values(doc, field) }.flatten

        end

      end

    end

    def csv_values(doc, field)

      if csv_fields[field] == 'ID'

        "'#{doc[field]}'"

      elsif csv_fields[field].is_a?(Array)

        two_values(doc[field])

      else

        Array(doc[field]).join('; ')

      end

    end

    def two_values(arr)

      values = arr || ['', '']

      values << '' if values.length == 1

      values.map { |v| v.chomp(' /') }

    end

end