Sha256: 357ee9406a480dce13d4a639979d7392672b6b7a635a52a9261180d8694c800d
Contents?: true
Size: 1.96 KB
Versions: 16
Compression:
Stored size: 1.96 KB
Contents
# frozen_string_literal: true
require 'set'
module ::Nanoc::Extra
# @api private
class LinkCollector
URI_ATTRS = {
'a' => :href,
'audio' => :src,
'form' => :action,
'iframe' => :src,
'img' => :src,
'link' => :href,
'script' => :src,
'video' => :src,
}.freeze
def initialize(filenames, mode = nil)
Nanoc::Extra::JRubyNokogiriWarner.check_and_warn
@filenames = filenames
@filter =
case mode
when nil
->(_h) { true }
when :external
->(h) { external_href?(h) }
when :internal
->(h) { !external_href?(h) }
else
raise ArgumentError, 'Expected mode argument to be :internal, :external or nil'
end
end
def filenames_per_href
grouped_filenames { |filename| hrefs_in_file(filename) }
end
def filenames_per_resource_uri
grouped_filenames { |filename| resource_uris_in_file(filename) }
end
def external_href?(href)
href =~ %r{^(\/\/|[a-z\-]+:)}
end
def hrefs_in_file(filename)
uris_in_file filename, %w[a img]
end
def resource_uris_in_file(filename)
uris_in_file filename, %w[audio form img iframe link script video]
end
private
def grouped_filenames
require 'nokogiri'
grouped_filenames = {}
@filenames.each do |filename|
yield(filename).each do |resouce_uri|
grouped_filenames[resouce_uri] ||= Set.new
grouped_filenames[resouce_uri] << filename
end
end
grouped_filenames
end
def uris_in_file(filename, tag_names)
uris = Set.new
doc = Nokogiri::HTML(::File.read(filename))
tag_names.each do |tag_name|
attr = URI_ATTRS[tag_name]
doc.css(tag_name).each do |e|
uris << e[attr] unless e[attr].nil?
end
end
# Strip fragment
uris.map! { |href| href.gsub(/#.*$/, '') }
uris.select(&@filter)
end
end
end
Version data entries
16 entries across 16 versions & 1 rubygems