@wip Implementing analytics

lib/yt.rb

@@ -1,2 +1,3 @@
 require 'yt/config'
-require 'yt/models/account'
+require 'yt/models/account'
+require 'yt/models/content_owner'

lib/yt/actions/list.rb

@@ -4,7 +4,7 @@
 module Yt
   module Actions
     module List
-      delegate :count, :first, :any?, :each, :map, :find, to: :list
+      delegate :count, :first, :any?, :each, :map, :flat_map, :find, to: :list
       alias size count
 
       def first!
@@ -51,7 +51,7 @@ def fetch_page(params = {})
         request = Yt::Request.new params
         response = request.run
         token = response.body['nextPageToken']
-        items = response.body.fetch 'items', []
+        items = response.body.fetch items_key, []
         {items: items, token: token}
       end
 
@@ -65,6 +65,10 @@ def list_params
           params[:exptected_response] = Net::HTTPOK
         end
       end
+
+      def items_key
+        'items'
+      end
     end
   end
 end

lib/yt/associations.rb

@@ -14,6 +14,7 @@ module Associations
     autoload :Authentications
     autoload :Channels
     autoload :DetailsSets
+    autoload :Earnings
     autoload :Ids
     autoload :PlaylistItems
     autoload :Playlists

lib/yt/associations/earnings.rb

@@ -0,0 +1,36 @@
+require 'yt/collections/earnings'
+
+module Yt
+  module Associations
+    # Provides the `has_many :earnings` method to YouTube resources, which
+    # allows to invoke earning-related methods, such as .earnings_on.
+    # YouTube resources with earning are: channels.
+    module Earnings
+      def earnings_on(date)
+        earnings_within(date, date).values.first
+      end
+
+      def earnings_since(from)
+        # NOTE: Today's earnings are not available, and hardly yesterday's
+        earnings_within from, 1.day.ago
+      end
+
+      def earnings_within(from, to)
+        date_range = Range.new *[from, to].map(&:to_date)
+        Hash[*date_range.flat_map do |date|
+          [date, (@earnings ||= {})[date] ||= range_earnings(date_range)[date]]
+        end]
+      end
+
+    private
+
+      def range_earnings(date_range)
+        (@range_earnings ||= {})[date_range] ||= earnings.within date_range
+      end
+
+      def earnings
+        Collections::Earnings.of self
+      end
+    end
+  end
+end

lib/yt/collections/earnings.rb

@@ -0,0 +1,42 @@
+require 'yt/collections/base'
+
+module Yt
+  module Collections
+    class Earnings < Base
+
+      def within(days_range)
+        @days_range = sorted days_range
+        Hash[*flat_map{|daily_earning| daily_earning}]
+      end
+
+    private
+
+      def new_item(data)
+        # NOTE: could use column headers to be more precise
+        [Date.iso8601(data.first), data.last]
+      end
+
+      def list_params
+        super.tap do |params|
+          params[:path] = '/youtube/analytics/v1/reports'
+          params[:params] = {}.tap do |params|
+            params['ids'] = "contentOwner==#{@auth.id}"
+            params['filters'] = "channel==#{@parent.id}"
+            params['start-date'] = @days_range.begin
+            params['end-date'] = @days_range.end
+            params['metrics'] = :earnings
+            params['dimensions'] = :day
+          end
+        end
+      end
+
+      def items_key
+        'rows'
+      end
+
+      def sorted(range)
+        range.min ? range : (range.end..range.begin)
+      end
+    end
+  end
+end

lib/yt/models/account.rb

@@ -1,25 +1,11 @@
-require 'yt/models/base'
+require 'yt/models/authenticable'
 
 module Yt
   module Models
     # Provides methods to access a YouTube account.
-    class Account < Base
-      has_one :authentication, delegate: [:access_token, :refresh_token, :expires_at]
+    class Account < Authenticable
       has_one :channel, delegate: [:videos, :playlists, :create_playlist, :delete_playlists, :update_playlists]
       has_one :user_info, delegate: [:id, :email, :has_verified_email?, :gender, :name, :given_name, :family_name, :profile_url, :avatar_url, :locale, :hd]
-
-      def initialize(options = {})
-        @access_token = options[:access_token]
-        @refresh_token = options[:refresh_token]
-        @expires_at = options[:expires_at]
-        @authorization_code = options[:authorization_code]
-        @redirect_uri = options[:redirect_uri]
-        @scopes = options[:scopes]
-      end
-
-      def auth
-        self
-      end
     end
   end
 end

lib/yt/models/authenticable.rb

@@ -0,0 +1,22 @@
+require 'yt/models/base'
+
+module Yt
+  module Models
+    class Authenticable < Base
+      has_one :authentication, delegate: [:access_token, :refresh_token, :expires_at]
+
+      def initialize(options = {})
+        @access_token = options[:access_token]
+        @refresh_token = options[:refresh_token]
+        @expires_at = options[:expires_at]
+        @authorization_code = options[:authorization_code]
+        @redirect_uri = options[:redirect_uri]
+        @scopes = options[:scopes]
+      end
+
+      def auth
+        self
+      end
+    end
+  end
+end

lib/yt/models/channel.rb

@@ -6,6 +6,7 @@ class Channel < Resource
       has_many :subscriptions
       has_many :videos
       has_many :playlists
+      has_many :earnings # requires auth with an account with 'yt-analytics-monetary.readonly'
     end
   end
 end

lib/yt/models/content_owner.rb

@@ -0,0 +1,15 @@
+require 'yt/models/authenticable'
+
+module Yt
+  module Models
+    # Provides methods to access a YouTube content_owner.
+    class ContentOwner < Authenticable
+      attr_reader :id
+
+      def initialize(options = {})
+        @id = options[:id]
+        super options
+      end
+    end
+  end
+end