duke-libraries/ddr-models

module Ddr
  module Managers
    class WorkflowManager < Manager

      PUBLISHED = "published"
      UNPUBLISHED = "unpublished"

      def published?
        object.workflow_state == PUBLISHED
      end

      def unpublished?
        object.workflow_state == UNPUBLISHED
      end

      def publish!(include_descendants: true)
        unless published?
          publish
          object.save!
        end
        if include_descendants && object.respond_to?(:children)
          solr_results = object.children(response_format: :solr)
          ActiveFedora::SolrService.lazy_reify_solr_results(solr_results).each do |child|
            child.publish!(include_descendants: include_descendants)
          end
        end
      end

      def unpublish!
        if published?
          unpublish
          object.save!
        end
        if object.respond_to?(:children)
          object.children.each { |child| child.unpublish! }
        end
      end

      private

      def publish
        object.workflow_state = PUBLISHED
      end

      def unpublish
        object.workflow_state = UNPUBLISHED
      end

    end
  end
end