ministryofjustice/Claim-for-Crown-Court-Defence

module API
  module Entities
    module CCR
      class Offence < API::Entities::BaseEntity
        expose :unique_code_check, as: :unique_code
        expose :offence_class,
               if: ->(instance, _opts) { instance.scheme_nine? },
               using: API::Entities::CCR::OffenceClass

        private

        def unique_code_check
          object.unique_code.split('~').first
        end
      end
    end
  end
end