2015-08-19 05:15:46 +08:00
|
|
|
class EmbeddableHost < ActiveRecord::Base
|
2016-01-12 00:06:09 +08:00
|
|
|
validate :host_must_be_valid
|
2015-08-19 05:15:46 +08:00
|
|
|
belongs_to :category
|
|
|
|
|
|
|
|
before_validation do
|
|
|
|
self.host.sub!(/^https?:\/\//, '')
|
|
|
|
self.host.sub!(/\/.*$/, '')
|
|
|
|
end
|
|
|
|
|
2016-08-24 02:55:52 +08:00
|
|
|
def self.record_for_url(uri)
|
|
|
|
|
|
|
|
if uri.is_a?(String)
|
2017-02-28 01:17:52 +08:00
|
|
|
uri = URI(URI.encode(uri)) rescue nil
|
2016-08-24 02:55:52 +08:00
|
|
|
end
|
2015-08-19 05:15:46 +08:00
|
|
|
return false unless uri.present?
|
|
|
|
|
|
|
|
host = uri.host
|
|
|
|
return false unless host.present?
|
|
|
|
|
2017-02-28 01:17:52 +08:00
|
|
|
if uri.port.present? && uri.port != 80 && uri.port != 443
|
|
|
|
host << ":#{uri.port}"
|
|
|
|
end
|
|
|
|
|
2016-08-27 00:47:21 +08:00
|
|
|
path = uri.path
|
|
|
|
path << "?" << uri.query if uri.query.present?
|
|
|
|
|
2017-02-18 01:39:33 +08:00
|
|
|
where("lower(host) = ?", host).each do |eh|
|
|
|
|
return eh if eh.path_whitelist.blank? || !Regexp.new(eh.path_whitelist).match(path).nil?
|
|
|
|
end
|
|
|
|
|
|
|
|
nil
|
|
|
|
end
|
2016-08-27 00:47:21 +08:00
|
|
|
|
2017-02-18 01:39:33 +08:00
|
|
|
def self.url_allowed?(url)
|
2017-02-28 01:17:52 +08:00
|
|
|
uri = URI(URI.encode(url)) rescue nil
|
2017-02-18 01:39:33 +08:00
|
|
|
uri.present? && record_for_url(uri).present?
|
2015-08-19 05:15:46 +08:00
|
|
|
end
|
|
|
|
|
2016-01-12 00:06:09 +08:00
|
|
|
private
|
|
|
|
|
|
|
|
def host_must_be_valid
|
2017-04-15 19:06:51 +08:00
|
|
|
if host !~ /\A[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,10}(:[0-9]{1,5})?(\/.*)?\Z/i &&
|
2017-03-16 05:16:34 +08:00
|
|
|
host !~ /\A(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})(:[0-9]{1,5})?(\/.*)?\Z/ &&
|
2016-07-23 03:33:21 +08:00
|
|
|
host !~ /\A([a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.)?localhost(\:[0-9]{1,5})?(\/.*)?\Z/i
|
2016-01-12 00:06:09 +08:00
|
|
|
errors.add(:host, I18n.t('errors.messages.invalid'))
|
|
|
|
end
|
|
|
|
end
|
2015-08-19 05:15:46 +08:00
|
|
|
end
|
2015-09-18 08:41:10 +08:00
|
|
|
|
|
|
|
# == Schema Information
|
|
|
|
#
|
|
|
|
# Table name: embeddable_hosts
|
|
|
|
#
|
2016-10-31 17:32:11 +08:00
|
|
|
# id :integer not null, primary key
|
|
|
|
# host :string not null
|
|
|
|
# category_id :integer not null
|
|
|
|
# created_at :datetime
|
|
|
|
# updated_at :datetime
|
|
|
|
# path_whitelist :string
|
2017-05-13 02:46:57 +08:00
|
|
|
# class_name :string
|
2015-09-18 08:41:10 +08:00
|
|
|
#
|