ministryofjustice/Claim-for-Crown-Court-Defence

module API
  module Entities
    class Offence < BaseEntity
      expose :export_format, merge: true, if: export_format? do
        expose :description, as: :category
        expose :class_description, as: :class
      end

      expose :full_format, merge: true, unless: export_format? do
        expose :id
        expose :description
        expose :unique_code
        expose :offence_class_id, if: ->(instance, _opts) { instance.scheme_nine? }
        expose :offence_class,
               if: ->(instance, _opts) { instance.scheme_nine? },
               using: API::Entities::OffenceClass
        expose :act_of_law, if: ->(instance, _opts) { instance.post_agfs_reform? }
        expose :offence_band,
               if: ->(instance, _opts) { instance.post_agfs_reform? },
               using: API::Entities::OffenceBand
      end

      private

      def class_description
        object.offence_class.description
      end

      def act_of_law
        "#{object.contrary} #{object.year_chapter}"
      end
    end
  end
end