Class: Dor::ContentMetadataDS

Inherits:
ActiveFedora::OmDatastream
  • Object
show all
Includes:
Upgradable
Defined in:
lib/dor/datastreams/content_metadata_ds.rb

Instance Method Summary collapse

Methods included from Upgradable

add_upgrade_callback, included, run_upgrade_callbacks, #upgrade!

Instance Method Details

#add_file(file, resource_name) ⇒ Nokogiri::XML::Node 



129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 129

def add_file(file, resource_name)
  resource_nodes = ng_xml.search('//resource[@id=\'' + resource_name + '\']')
  raise 'resource doesnt exist.' if resource_nodes.length == 0
  node = resource_nodes.first
  file_node = Nokogiri::XML::Node.new('file', ng_xml)
  file_node['id'] = file[:name]
  file_node['shelve'  ] = file[:shelve  ] ? file[:shelve  ] : ''
  file_node['publish' ] = file[:publish ] ? file[:publish ] : ''
  file_node['preserve'] = file[:preserve] ? file[:preserve] : ''
  node.add_child(file_node)

  [:md5, :sha1].each do |algo|
    next unless file[algo]
    checksum_node = Nokogiri::XML::Node.new('checksum', ng_xml)
    checksum_node['type'] = algo.to_s
    checksum_node.content = file[algo]
    file_node.add_child(checksum_node)
  end
  file_node['size'    ] = file[:size     ] if file[:size     ]
  file_node['mimetype'] = file[:mime_type] if file[:mime_type]
  self.content = ng_xml.to_s
  file_node
end

#add_resource(files, resource_name, position, type = 'file') ⇒ Nokogiri::XML::Node 



183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 183

def add_resource(files, resource_name, position, type = 'file')
  raise "resource #{resource_name} already exists" if ng_xml.search('//resource[@id=\'' + resource_name + '\']').length > 0
  max = ng_xml.search('//resource').map { |node| node['sequence'].to_i }.max
  # renumber all of the resources that will come after the newly added one
  while max > position
    node = ng_xml.search('//resource[@sequence=\'' + position + '\']')
    node.first[sequence] = max + 1 if node.length > 0
    max -= 1
  end
  node = Nokogiri::XML::Node.new('resource', ng_xml)
  node['sequence'] = position.to_s
  node['id']       = resource_name
  node['type']     = type
  files.each do |file|
    file_node = Nokogiri::XML::Node.new('file', ng_xml)
    %w(shelve publish preserve).each {|x| file_node[x] = file[x.to_sym] ? file[x.to_sym] : '' }
    file_node['id'] = file[:name]
    node.add_child(file_node)

    [:md5, :sha1].each { |algo|
      next if file[algo].nil?
      checksum_node = Nokogiri::XML::Node.new('checksum', ng_xml)
      checksum_node['type'] = algo.to_s
      checksum_node.content = file[algo]
      file_node.add_child(checksum_node)
    }
    file_node['size'] = file[:size] if file[:size]
  end
  ng_xml.search('//contentMetadata').first.add_child(node)
  self.content = ng_xml.to_s
  node
end

#add_virtual_resource(child_druid, child_resource) ⇒ Nokogiri::XML::Element

Copies the child’s resource into the parent (self) as a virtual resource. Assumes the resource isn’t a duplicate of an existing virtual or real resource.



158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 158

def add_virtual_resource(child_druid, child_resource)
  # create a virtual resource element with attributes linked to the child and omit label
  sequence_max = ng_xml.search('//resource').map { |node| node[:sequence].to_i }.max
  resource = Nokogiri::XML::Element.new('resource', ng_xml)
  resource[:sequence] = sequence_max + 1
  resource[:id] = "#{pid.gsub(/^druid:/, '')}_#{resource[:sequence]}"
  resource[:type] = child_resource[:type]

  # iterate over all the published files and link to them
  child_resource.search('file[@publish=\'yes\']').each do |file|
    resource << generate_external_file_node(child_druid, child_resource[:id], file[:id], file[:mimetype])
  end
  resource << generate_also_available_as_node(child_druid)

  # attach the virtual resource as a sibling and return
  ng_xml.root << resource
  self.content = ng_xml.to_s
  resource
end

#generate_also_available_as_node(objectId) ⇒ Nokogiri::XML::Element

Generates the XML tree for virtual resource relationship reference. For example:

<relationship type="alsoAvailableAs" objectId="druid:mn123pq4567" />



79
80
81
82
83
84
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 79

def generate_also_available_as_node(objectId)
  relationship = ng_xml.create_element 'relationship'
  relationship[:type] = 'alsoAvailableAs'
  relationship[:objectId] = objectId
  relationship
end

#generate_external_file_node(objectId, resourceId, fileId, mimetype) ⇒ Nokogiri::XML::Element

Generates the XML tree for externalFile references. For example:

<externalFile objectId="druid:mn123pq4567" resourceId="Image01" fileId="image_01.jp2000" mimetype="image/jp2" />



66
67
68
69
70
71
72
73
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 66

def generate_external_file_node(objectId, resourceId, fileId, mimetype)
  externalFile = ng_xml.create_element 'externalFile'
  externalFile[:objectId]   = objectId
  externalFile[:resourceId] = resourceId
  externalFile[:fileId]     = fileId
  externalFile[:mimetype]   = mimetype
  externalFile
end

#move_resource(resource_name, new_position) ⇒ Nokogiri::XML::Element

You just had to have ordered lists in XML, didn’t you? Re-enumerate the sequence numbers affected



310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 310

def move_resource(resource_name, new_position)
  node = singular_node('//resource[@id=\'' + resource_name + '\']')
  position = node['sequence'].to_i
  new_position = new_position.to_i # tolerate strings as a Legacy behavior
  return node if position == new_position
  # otherwise, is the resource being moved earlier in the sequence or later?
  up = new_position > position
  others = new_position..(up ? position - 1 : position + 1) # a range
  others.each do |i|
    item = ng_xml.at_xpath('/resource[@sequence=\'' + i.to_s + '\']')
    item['sequence'] = (up ? i - 1 : i + 1).to_s # if you're going up, everything else comes down and vice versa
  end
  node['sequence'] = new_position.to_s # set the node we already had last, so we don't hit it twice!
  self.content = ng_xml.to_s
  node
end

#public_xml ⇒ Nokogiri::XML::Document 



38
39
40
41
42
43
44
45
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 38

def public_xml
  result = ng_xml.clone
  result.xpath('/contentMetadata/resource[not(file[(@deliver="yes" or @publish="yes")])]'   ).each(&:remove)
  result.xpath('/contentMetadata/resource/file[not(@deliver="yes" or @publish="yes")]'      ).each(&:remove)
  result.xpath('/contentMetadata/resource/file').xpath('@preserve|@shelve|@publish|@deliver').each(&:remove)
  result.xpath('/contentMetadata/resource/file/checksum'                                    ).each(&:remove)
  result
end

#remove_file(file_name) ⇒ Object 



231
232
233
234
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 231

def remove_file(file_name)
  ng_xml.search('//file[@id=\'' + file_name + '\']').each(&:remove)
  self.content = ng_xml.to_s
end

#remove_resource(resource_name) ⇒ Object 



217
218
219
220
221
222
223
224
225
226
227
228
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 217

def remove_resource(resource_name)
  node = singular_node('//resource[@id=\'' + resource_name + '\']')
  position = node['sequence'].to_i + 1
  node.remove
  loop do
    res = ng_xml.search('//resource[@sequence=\'' + position.to_s + '\']')
    break if res.length == 0
    res['sequence'] = position.to_s
    position += 1
  end
  self.content = ng_xml.to_s
end

#rename_file(old_name, new_name) ⇒ Nokogiri::XML::Element 



273
274
275
276
277
278
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 273

def rename_file(old_name, new_name)
  file_node = ng_xml.search('//file[@id=\'' + old_name + '\']').first
  file_node['id'] = new_name
  self.content = ng_xml.to_s
  file_node
end

#set_content_type(old_type, old_resource_type, new_type, new_resource_type) ⇒ Object

Set the content type (e.g. “book”) and the resource type (e.g. “book”) for all resources



332
333
334
335
336
337
338
339
340
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 332

def set_content_type(old_type, old_resource_type, new_type, new_resource_type)
  ng_xml.search('/contentMetadata[@type=\'' + old_type + '\']').each do |node|
    node['type'] = new_type
    ng_xml.search('//resource[@type=\'' + old_resource_type + '\']').each do |resource|
      resource['type'] = new_resource_type
    end
  end
  self.content = ng_xml.to_s
end

#singular_node(xpath) ⇒ Nokogiri::XML::Element

Only use this when you want the behavior of raising an exception if anything besides exactly one matching node is found. Otherwise just use .xpath, .at_xpath or .search.



51
52
53
54
55
56
57
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 51

def singular_node(xpath)
  node = ng_xml.search(xpath)
  len  = node.length
  raise "#{xpath} not found" if len < 1
  raise "#{xpath} duplicated: #{len} found" if len != 1
  node.first
end

#to_solr(solr_doc = {}, *args) ⇒ Object

Terminology-based solrization is going to be painfully slow for large contentMetadata streams. Just select the relevant elements instead. TODO: Call super()?



89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 89

def to_solr(solr_doc = {}, *args)
  doc = ng_xml
  return solr_doc unless doc.root['type']

  preserved_size = 0
  counts = Hash.new(0)                # default count is zero
  resource_type_counts = Hash.new(0)  # default count is zero
  first_shelved_image = nil

  doc.xpath('contentMetadata/resource').sort { |a, b| a['sequence'].to_i <=> b['sequence'].to_i }.each do |resource|
    counts['resource'] += 1
    resource_type_counts[resource['type']] += 1 if resource['type']
    resource.xpath('file').each do |file|
      counts['content_file'] += 1
      preserved_size += file['size'].to_i if file['preserve'] == 'yes'
      next unless file['shelve'] == 'yes'
      counts['shelved_file'] += 1
      first_shelved_image ||= file['id'] if file['id'].match(/jp2$/)
    end
  end
  solr_doc['content_type_ssim'              ] = doc.root['type']
  solr_doc['content_file_count_itsi'        ] = counts['content_file']
  solr_doc['shelved_content_file_count_itsi'] = counts['shelved_file']
  solr_doc['resource_count_itsi'            ] = counts['resource']
  solr_doc['preserved_size_dbtsi'           ] = preserved_size # double (trie) to support very large sizes
  solr_doc['resource_types_ssim'            ] = resource_type_counts.keys if resource_type_counts.size > 0
  resource_type_counts.each do |key, count|
    solr_doc["#{key}_resource_count_itsi"] = count
  end
  # first_shelved_image is neither indexed nor multiple
  solr_doc['first_shelved_image_ss'] = first_shelved_image unless first_shelved_image.nil?
  solr_doc
end

#update_attributes(file_name, publish, shelve, preserve) ⇒ Object 



240
241
242
243
244
245
246
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 240

def update_attributes(file_name, publish, shelve, preserve)
  file_node = ng_xml.search('//file[@id=\'' + file_name + '\']').first
  file_node['publish' ] = publish
  file_node['shelve'  ] = shelve
  file_node['preserve'] = preserve
  self.content = ng_xml.to_s
end

#update_file(file, old_file_id) ⇒ Object 



250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 250

def update_file(file, old_file_id)
  file_node = ng_xml.search('//file[@id=\'' + old_file_id + '\']').first
  file_node['id'] = file[:name]
  [:md5, :sha1].each { |algo|
    next if file[algo].nil?
    checksum_node = ng_xml.search('//file[@id=\'' + old_file_id + '\']/checksum[@type=\'' + algo.to_s + '\']').first
    if checksum_node.nil?
      checksum_node = Nokogiri::XML::Node.new('checksum', ng_xml)
      file_node.add_child(checksum_node)
    end
    checksum_node['type'] = algo.to_s
    checksum_node.content = file[algo]
  }

  [:size, :shelve, :preserve, :publish].each { |x|
    file_node[x.to_s] = file[x] if file[x]
  }
  self.content = ng_xml.to_s
end

#update_resource_label(resource_name, new_label) ⇒ Nokogiri::XML::Element

Updates old label OR creates a new one if necessary



284
285
286
287
288
289
290
291
292
293
294
295
296
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 284

def update_resource_label(resource_name, new_label)
  node = singular_node('//resource[@id=\'' + resource_name + '\']')
  labels = node.xpath('./label')
  if labels.length == 0
    label_node = Nokogiri::XML::Node.new('label', ng_xml) # create a label
    label_node.content = new_label
    node.add_child(label_node)
  else
    labels.first.content = new_label
  end
  self.content = ng_xml.to_s
  node
end

#update_resource_type(resource_name, new_type) ⇒ Object 



300
301
302
303
 
# File 'lib/dor/datastreams/content_metadata_ds.rb', line 300

def update_resource_type(resource_name, new_type)
  singular_node('//resource[@id=\'' + resource_name + '\']')['type'] = new_type
  self.content = ng_xml.to_s
end