init commit
This commit is contained in:
40
plugins/MediumFollowersCountPlugin.rb
Normal file
40
plugins/MediumFollowersCountPlugin.rb
Normal file
@@ -0,0 +1,40 @@
|
||||
require_relative 'Plugin'
|
||||
require 'net/http'
|
||||
require 'nokogiri'
|
||||
require 'uri'
|
||||
|
||||
class MediumFollowersCountPlugin < Plugin
|
||||
attr_reader :data, :username
|
||||
|
||||
def initialize(data)
|
||||
@data = data
|
||||
@username = data[0]["username"]
|
||||
end
|
||||
|
||||
def execute
|
||||
return load_medium_followers("https://medium.com/@#{@username}")
|
||||
end
|
||||
|
||||
|
||||
def load_medium_followers(url, limit = 10)
|
||||
return 0 if limit.zero?
|
||||
|
||||
uri = URI(url)
|
||||
response = Net::HTTP.get_response(uri)
|
||||
case response
|
||||
when Net::HTTPSuccess then
|
||||
document = Nokogiri::HTML(response.body)
|
||||
|
||||
follower_count_element = document.at('span.pw-follower-count > a')
|
||||
follower_count = follower_count_element&.text&.split(' ')&.first
|
||||
|
||||
return follower_count || 0
|
||||
when Net::HTTPRedirection then
|
||||
location = response['location']
|
||||
return load_medium_followers(location, limit - 1)
|
||||
else
|
||||
return 0
|
||||
end
|
||||
end
|
||||
|
||||
end
|
Reference in New Issue
Block a user