module Ddr
  module Models
    module Indexing

      include Ddr::Index::Fields

      def to_solr(solr_doc=Hash.new, opts={})
        solr_doc = super(solr_doc, opts)
        solr_doc.merge index_fields
      end

      def index_fields
        fields = {
          ACCESS_ROLE           => roles.to_json,
          ADMIN_SET             => admin_set,
          BOX_NUMBER_FACET      => desc_metadata_values('box_number'),
          CREATOR_FACET         => creator,
          DATE_FACET            => date,
          DATE_SORT             => date_sort,
          DEPOSITOR             => depositor,
          DISPLAY_FORMAT        => display_format,
          DOI                   => adminMetadata.doi,
          IDENTIFIER_ALL        => all_identifiers,
          INTERNAL_URI          => internal_uri,
          LICENSE               => license,
          LICENSE_DESCRIPTION   => rightsMetadata.license.description.first,
          LICENSE_TITLE         => rightsMetadata.license.title.first,
          LICENSE_URL           => rightsMetadata.license.url.first,
          LOCAL_ID              => local_id,
          PERMANENT_ID          => permanent_id,
          PERMANENT_URL         => permanent_url,
          POLICY_ROLE           => roles.in_policy_scope.agents,
          PUBLISHER_FACET       => publisher,
          RESEARCH_HELP_CONTACT => research_help_contact,
          RESOURCE_ROLE         => roles.in_resource_scope.agents,
          SERIES_FACET          => desc_metadata_values('series'),
          SPATIAL_FACET         => desc_metadata_values('spatial'),
          TITLE                 => title_display,
          TYPE_FACET            => type,
          WORKFLOW_STATE        => workflow_state,
          YEAR_FACET            => year_facet,
        }
        if respond_to? :fixity_checks
          last_fixity_check = fixity_checks.last
          fields.merge!(last_fixity_check.to_solr) if last_fixity_check
        end
        if respond_to? :virus_checks
          last_virus_check = virus_checks.last
          fields.merge!(last_virus_check.to_solr) if last_virus_check
        end
        if has_content?
          fields[CONTENT_CONTROL_GROUP] = content.controlGroup
          fields[CONTENT_SIZE] = content_size
          fields[CONTENT_SIZE_HUMAN] = content_human_size
          fields[MEDIA_TYPE] = content_type
          fields[MEDIA_MAJOR_TYPE] = content_major_type
          fields[MEDIA_SUB_TYPE] = content_sub_type
          fields.merge! techmd.index_fields
        end
        if has_multires_image?
          fields[MULTIRES_IMAGE_FILE_PATH] = multires_image_file_path
        end
        if has_struct_metadata?
          fields[STRUCT_MAPS] = structure.struct_maps.to_json
        end
        if has_extracted_text?
          fields[EXTRACTED_TEXT] = extractedText.content
        end
        if is_a? Component
          fields[COLLECTION_URI] = collection_uri
        end
        if is_a? Collection
          fields[DEFAULT_LICENSE_DESCRIPTION] = defaultRights.license.description.first
          fields[DEFAULT_LICENSE_TITLE] = defaultRights.license.title.first
          fields[DEFAULT_LICENSE_URL] = defaultRights.license.url.first
          fields[ADMIN_SET_FACET] = admin_set_facet
          fields[COLLECTION_FACET] = collection_facet
        end
        if is_a? Item
          fields[ADMIN_SET_FACET] = admin_set_facet
          fields[COLLECTION_FACET] = collection_facet
        end
        fields
      end

      def title_display
        return title.first if title.present?
        return identifier.first if identifier.present?
        return original_filename if respond_to?(:original_filename) && original_filename.present?
        "[#{pid}]"
      end

      def all_identifiers
        identifier + [local_id, permanent_id, pid].compact
      end

      def associated_collection
        admin_policy
      end

      def admin_set_facet
        if admin_set.present?
          admin_set
        elsif associated_collection.present?
          associated_collection.admin_set
        end
      end

      def collection_facet
        associated_collection.internal_uri if associated_collection.present?
      end

      def date_sort
        date.first
      end

      def year_facet
        YearFacet.call(self)
      end

    end
  end
end