Class: EPUB::Searcher::Publication

Inherits:
Object
  • Object
show all
Defined in:
lib/epub/searcher/publication.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(package) ⇒ Publication

Returns a new instance of Publication.



20
21
22
 
# File 'lib/epub/searcher/publication.rb', line 20

def initialize(package)
  @package = package
end

Class Method Details

.search_by_cfi(package, cfi) ⇒ Object 



15
16
17
 
# File 'lib/epub/searcher/publication.rb', line 15

def search_by_cfi(package, cfi)
  new(package).search_by_cfi(cfi)
end

.search_element(package, css: nil, xpath: nil, namespaces: {}) ⇒ Object 



11
12
13
 
# File 'lib/epub/searcher/publication.rb', line 11

def search_element(package, css: nil, xpath: nil, namespaces: {})
  new(package).search_element(css: css, xpath: xpath, namespaces: namespaces)
end

.search_text(package, word, **options) ⇒ Object 



7
8
9
 
# File 'lib/epub/searcher/publication.rb', line 7

def search_text(package, word, **options)
  new(package).search_text(word, options)
end

Instance Method Details

#search_by_cfi(cfi) ⇒ Array

TODO:

Use XHTML module

TODO:

Handle CFI with offset

TODO:

Handle range CFI

Note:

Currenty can handle only location CFI without offset

Returns Path in EPUB Rendition.

Parameters:

  • cfi (EPUB::CFI)

Returns:

  • (Array) —

    Path in EPUB Rendition

Raises:

  • (NotImplementedError) 


84
85
86
87
88
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
 
# File 'lib/epub/searcher/publication.rb', line 84

def search_by_cfi(cfi)
  path_in_package = cfi.paths.first
  spine = @package.spine
  model = [@package.metadata, @package.manifest, spine, @package.guide, @package.bindings].compact[path_in_package.steps.first.value / 2 - 1]
  raise NotImplementedError, "Currently, #{__method__} supports spine only(#{cfi})" unless model == spine
  raise ArgumentError, "Cannot identify <itemref>'s child" if path_in_package.steps.length > 2

  step_to_itemref = path_in_package.steps[1]
  itemref = spine.itemrefs[step_to_itemref.value / 2 - 1]

  doc = itemref.item.content_document.nokogiri
  path_in_doc = cfi.paths[1]
  current_node = doc.root
  path_in_doc.steps.each do |step|
    if step.element?
      current_node = current_node.element_children[step.value / 2 - 1]
    else
      element_index = (step.value - 1) / 2 - 1
      if element_index == -1
        current_node = current_node.children.first
      else
        prev = current_node.element_children[element_index]
        break unless prev
        current_node = prev.next_sibling
        break unless current_node
      end
    end
  end

  raise NotImplementedError, "Currently, #{__method__} doesn't support deeper DOM tree such as including <iframe>" if cfi.paths[2]

  [itemref, current_node]
end

#search_element(css: nil, xpath: nil, namespaces: {}) ⇒ Array<Hash>

@todo: Refactoring

Returns:

  • (Array<Hash>) —

    An array of rearch results. Each result is composed of: :element: [Nokogiri::XML::ELement] Found element :itemref: [EPUB::Publication::Package::Spine::Itemref] Itemref that element’s document belongs to :location: [EPUB::CFI::Location] CFI that indicates the element :package: [EPUB::Publication::Package] Package that the element belongs to

Raises:

  • (ArgumentError) 


45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/epub/searcher/publication.rb', line 45

def search_element(css: nil, xpath: nil, namespaces: {})
  raise ArgumentError, 'Both css and xpath are nil' if css.nil? && xpath.nil?

  namespaces = EPUB::NAMESPACES.merge(namespaces)
  results = []

  spine_step = EPUB::CFI::Step.new((EPUB::Publication::Package::CONTENT_MODELS.index(:spine) + 1) * 2)
  @package.spine.each_itemref.with_index do |itemref, index|
    assertion = itemref.id ? EPUB::CFI::IDAssertion.new(itemref.id) : nil
    itemref_step = EPUB::CFI::Step.new((index + 1) * 2, assertion)
    path_to_itemref = EPUB::CFI::Path.new([spine_step, itemref_step])
    content_document = itemref.item.content_document
    next unless content_document
    doc = content_document.nokogiri
    elems = if xpath
              doc.xpath(xpath, namespaces)
            else
              doc.css(css)
            end
    elems.each do |elem|
      path = find_path(elem)
      results << {
        location: EPUB::CFI::Location.new([path_to_itemref, path]),
        package: @package,
        itemref: itemref,
        element: elem
      }
    end
  end

  results
end

#search_text(word, algorithm: :seamless) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/epub/searcher/publication.rb', line 24

def search_text(word, algorithm: :seamless)
  results = []

  spine = @package.spine
  spine_step = Result::Step.new(:element, 2, {:name => 'spine', :id => spine.id})
  spine.each_itemref.with_index do |itemref, index|
    itemref_step = Result::Step.new(:itemref, index, {:id => itemref.id})
    XHTML::ALGORITHMS[algorithm].search_text(Nokogiri.XML(itemref.item.read), word).each do |sub_result|
      results << Result.new([spine_step, itemref_step] + sub_result.parent_steps, sub_result.start_steps, sub_result.end_steps)
    end
  end

  results
end