lib/qiita_trend/trend.rb in qiita_trend-0.1.1 vs lib/qiita_trend/trend.rb in qiita_trend-0.2.0

@@ -3,24 +3,17 @@
 require 'nokogiri'
 require 'open-uri'
 require 'json'
 
 module QiitaTrend
-  # Qiitaトレンドクラス
-  # Qiitaのトレンド機能を提供する
   class Trend
     attr_reader :data
 
-    QIITA_URI = 'https://qiita.com/'
-    USER_AGENT = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/71.0.3578.98 Safari/537.36'
+    def initialize(ua = 'Mac Safari')
+      page = QiitaTrend::Page.new(ua)
+      parsed_html = Nokogiri::HTML.parse(page.html)
 
-    def initialize
-      html = URI.parse(QIITA_URI)
-      html = html.read('User-Agent' => USER_AGENT)
-      char = html.charset
-      parsed_html = Nokogiri::HTML.parse(html, nil, char)
-
       trends_data = JSON.parse(parsed_html.xpath('//div[@data-hyperapp-app="Trend"]')[0]['data-hyperapp-props'])
       @data = trends_data['trend']['edges']
     end
 
     def items
@@ -29,10 +22,10 @@
         result['title'] = trend['node']['title']
         result['user_name'] = trend['node']['author']['urlName']
         result['user_image'] = trend['node']['author']['profileImageUrl']
         result['likes_count'] = trend['node']['likesCount']
         result['is_new_arrival'] = trend['isNewArrival']
-        result['article'] = "#{QIITA_URI}#{trend['node']['author']['urlName']}/items/#{trend['node']['uuid']}"
+        result['article'] = "#{Page::QIITA_URI}#{trend['node']['author']['urlName']}/items/#{trend['node']['uuid']}"
         value << result
       end
     end
 
     def new_items