require 'tempfile' module Ddr::Models class File < ActiveFedora::File extend Deprecation CONTENT = "content".freeze DESC_METADATA = "descMetadata".freeze EXTRACTED_TEXT = "extractedText".freeze FITS = "fits".freeze STRUCT_METADATA = "structMetadata".freeze THUMBNAIL = "thumbnail".freeze DEFAULT_FILE_EXTENSION = "bin" STRFTIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%LZ" def dsid Deprecation.warn(File, "`dsid` is no longer a file method. Use `::File.basename(id)`.") if id ::File.basename(id) end end def dsCreateDate Deprecation.warn(File, "`dsCreateDate` is no longer a file method. Use `create_date` instead.") create_date end def validate_checksum!(checksum_value, checksum_type=nil) raise Error, "Checksum cannot be validated on new file." if new_record? raise Error, "Checksum cannot be validated on unpersisted content." if content_changed? raise ChecksumInvalid, "The repository internal checksum validation failed." unless check_fixity algorithm = checksum_type || checksum.algorithm calculated_checksum = if algorithm == checksum.algorithm checksum.value else content_digest(algorithm) end if checksum_value == calculated_checksum "The checksum #{algorithm}:#{checksum_value} is valid for file #{dsid}." else raise ChecksumInvalid, "The checksum #{algorithm}:#{checksum_value} is not valid for file #{dsid}." end end def create_date_string dsCreateDate.strftime(STRFTIME_FORMAT) if dsCreateDate end def content_digest(algorithm) Ddr::Utils.digest(content, algorithm) end # Return default file extension for file based on MIME type def default_file_extension mimetypes = MIME::Types[mime_type] return mimetypes.first.extensions.first unless mimetypes.empty? case mime_type when 'application/n-triples' 'txt' else DEFAULT_FILE_EXTENSION end end def default_file_prefix (id && id.gsub(/\//, "_")) || "NEW" end # Return default file name def default_file_name [ default_file_prefix, default_file_extension ].join(".") end def tempfile(prefix: nil, suffix: nil) if empty? raise Error, "Refusing to create tempfile for empty file!" end prefix ||= default_file_prefix + "--" suffix ||= "." + default_file_extension Tempfile.open [prefix, suffix], encoding: Encoding::ASCII_8BIT do |f| f.write(content) f.close yield f end end end end