Merge pull request #38 from Lackoftactics/revert-34-refactor-to-support-view-models

Temporary Revert "Refactor code to support ViewModels"

Author: Przemyslaw Mroczek

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    facebook_data_analyzer (0.1.1)
+    facebook_data_analyzer (0.1.0)
       axlsx (= 3.0.0.pre)
       json
       micro-optparse

lib/.DS_Store

@@ -2,27 +2,20 @@
 
 # My script for 'I analyzed my facebook data and it's story of shyness,
 # loneliness and change'
-require 'facebook_data_analyzer/view_model_generators/view_model_generator'
-require 'facebook_data_analyzer/view_model_generators/friends_view_model_generator'
-require 'facebook_data_analyzer/view_model_generators/contacts_view_model_generator'
-require 'facebook_data_analyzer/view_model_generators/messages_view_model_generator'
-require 'facebook_data_analyzer/mixins/excel_exporter_mixin'
-require 'facebook_data_analyzer/mixins/export_views_mixin'
-require 'facebook_data_analyzer/analyzeable'
-require 'facebook_data_analyzer/contacts'
-require 'facebook_data_analyzer/friends'
-require 'facebook_data_analyzer/messages'
-require 'facebook_data_analyzer/models/contact'
-require 'facebook_data_analyzer/models/friend'
-require 'facebook_data_analyzer/models/message'
-require 'facebook_data_analyzer/models/table'
-
 require 'axlsx'
 require 'parallel'
 require 'json'
 require 'workbook'
 require 'set'
 
+require 'facebook_data_analyzer/analyzeables/analyzeable'
+require 'facebook_data_analyzer/analyzeables/contacts'
+require 'facebook_data_analyzer/analyzeables/friends'
+require 'facebook_data_analyzer/analyzeables/messages'
+require 'facebook_data_analyzer/contact'
+require 'facebook_data_analyzer/friend'
+require 'facebook_data_analyzer/message'
+
 module FacebookDataAnalyzer
   def self.run(options = {})
     file_output    = 'facebook_data_analyzer'

lib/facebook_data_analyzer.rb

@@ -10,7 +10,6 @@ def self.parse
 
     GROUP_BY = [].freeze
     COUNT_BY = [].freeze
-    EXPORTS  = [].freeze
 
     def initialize(threads_supported: nil, processes_supported: nil, parallel: false)
       # Grouped by is weird and needs a hash for each GROUP_BY, hash for each unique group, and hash for attributes

lib/facebook_data_analyzer/analyzeable.rb renamed to lib/facebook_data_analyzer/analyzeables/analyzeable.rb

@@ -2,11 +2,7 @@
 
 module FacebookDataAnalyzer
   class Contacts < Analyzeable
-    include FacebookDataAnalyzer::ExportViewsMixin
     attr_reader :contacts
-
-    EXPORTS = [:contact_list].freeze
-
     def initialize(catalog:)
       @catalog = catalog
       @directory = "#{catalog}/html/"
@@ -36,5 +32,22 @@ def analyze
         end
       end
     end
+
+    def export(package:)
+      contact_list_sheet(package: package)
+    end
+
+    private
+
+    def contact_list_sheet(package:)
+      package.workbook.add_worksheet(name: 'Contact list') do |sheet|
+        sheet.add_row ['Contact list']
+        sheet.add_row ["Facebook imported #{@contacts.length} of your contacts"]
+        sheet.add_row ['Name', 'Email']
+        @contacts.sort_by(&:name).each do |contact|
+          sheet.add_row [contact.name, contact.details]
+        end
+      end
+    end
   end
 end

lib/facebook_data_analyzer/contacts.rb renamed to lib/facebook_data_analyzer/analyzeables/contacts.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module FacebookDataAnalyzer
+  class Friends < Analyzeable
+    attr_reader :friends
+    # year, day_of_week, day, month: {#unit: count ...}
+    # weekend: {weekend: count,
+    #           weekday: count}
+    # month_year: {#month - #year: count ...}
+    # week_year: {#week - #year: count ...}
+    COUNT_BY = %i[year day_of_week day month weekend month_year week_year].freeze
+
+    def initialize(catalog:)
+      @catalog = catalog
+      @directory = "#{catalog}/html/"
+      @file_pattern = 'friends.htm'
+      @friends = []
+
+      super()
+    end
+
+    def analyze
+      Dir.chdir(@directory) do
+        content = File.open(@file_pattern).read
+        doc = Nokogiri::HTML(content)
+        friends_list = doc.css('div.contents > ul')[0].css('li')
+
+        friends_list.each do |friend_element|
+          friend = Friend.parse(friend_element: friend_element)
+
+          @friends << friend
+          count(analyzeable: friend)
+        end
+      end
+    end
+
+    def export(package:)
+      making_friends_sheet(package: package)
+    end
+
+    private
+
+    def making_friends_sheet(package:)
+      package.workbook.add_worksheet(name: 'Making friends') do |sheet|
+        sheet.add_row ['Making friends']
+        sheet.add_row ['']
+
+        sheet.add_row ['Making friends by year']
+        sheet.add_row ['Year', 'Number of friends added']
+        @counted_by[:year].sort_by { |year, _count| year }.each do |year, count|
+          sheet.add_row [year, count]
+        end
+
+        sheet.add_row ['Making friends by week day']
+        sheet.add_row ['Day of week', 'Number of friends added']
+        @counted_by[:day_of_week].sort_by { |_day, count| count }.reverse.each do |day, count|
+          sheet.add_row [day, count]
+        end
+
+        sheet.add_row ['Making friends by month']
+        sheet.add_row ['Month', 'Number of friends added']
+        @counted_by[:month].sort_by { |_month, count| count }.reverse.each do |month, count|
+          sheet.add_row [month, count]
+        end
+
+        sheet.add_row ['Making friends on weekend vs. working days']
+        sheet.add_row ['Working day or weekend', 'Number of friends added']
+        @counted_by[:weekend].each do |type_of_day, count|
+          sheet.add_row [type_of_day.to_s, count]
+        end
+
+        sheet.add_row ['Most busy weeks for making friends (week number and year)']
+        sheet.add_row ['Week and year', 'Number of friends added']
+        @counted_by[:week_year].sort_by { |_week_year, count| count }.reverse.each do |week_year, count|
+          sheet.add_row [week_year, count]
+        end
+
+        sheet.add_row ['Most busy month-year by friends added']
+        sheet.add_row ['Month year', 'Number of friends added']
+        @counted_by[:month_year].sort_by { |_month_year, count| count }.reverse.each do |month_year, count|
+          sheet.add_row [month_year, count]
+        end
+
+        sheet.add_row ['Most busy making friends days']
+        sheet.add_row ['Day', 'Number of friends added']
+        @counted_by[:day].sort_by { |_day, count| count }.reverse.each do |day, count|
+          sheet.add_row [day, count]
+        end
+      end
+    end
+  end
+end