hbtrack 0.0.5 → 0.0.6

data/lib/hbtrack/habit_tracker.rb

@@ -1,26 +1,19 @@
 # frozen_string_literal: true
 
+require 'hbtrack/util'
+
 module Hbtrack
   # This class contains the methods to
   # handle the operation of mutliple habits
   class HabitTracker
-    attr_reader :habits, :hp
-
-    def self.help # Refactoring needed
-      puts 'usage: hbtrack list [-p] [ habit_name ]'
-      puts '       hbtrack add habit_name'
-      puts '       hbtrack done [-y] habit_name'
-      puts '       hbtrack undone [-y] habit_name'
-      puts '       hbtrack remove habit_name'
-    end
+    attr_accessor :habits
+    attr_reader :hp, :output_file_name
 
     def initialize(file = FILE_NAME,
                    output = FILE_NAME)
       @habits = []
       @file_name = file
       @output_file_name = output
-      @sf = CompleteSF.new
-      @hp = HabitPrinter.new(@sf)
       initialize_habits_from_file
     end
 
@@ -30,12 +23,6 @@ module Hbtrack
       input.each { |string| @habits << Habit.initialize_from_string(string) }
     end
 
-    def parse_arguments(args)
-      head = args.shift
-      tail = args
-      send(head, tail)
-    end
-
     # This methods find a habit based on the name given.
     # Blocks are executed when habit is not found.
     def find(habit_name)
@@ -46,102 +33,40 @@ module Hbtrack
       habit
     end
 
-    def longest_name
-      @habits.max_by(&:name_length).name
-    end
-
-    def method_missing(*args)
-      HabitTracker.help
-    end
-
-    def total_habits_stat 
-      @habits.reduce({done: 0, undone: 0}) do |stat, habit|
-        stat[:done] += habit.latest_stat[:done]
-        stat[:undone] += habit.latest_stat[:undone]
-        stat
-      end
-    end
-
-    def overall_stat_description
-      Util.title("Total") +
-      @sf.format(total_habits_stat)
-    end
-
-    private
-    def list(args)
-      habit_name, options = parse_options(args)
-      set_sf_based_on(options)
+    def find_or_create(habit_name)
       habit = find(habit_name) do
-        if habit_name.nil?
-          list_all_habits
-        else
-         raise_habit_not_found(habit_name)
-        end
-        return
-      end
-      puts Util.title habit.name 
-      puts @hp.print_all_progress(habit)
-    end
-
-    def set_sf_based_on(options)
-      return if options.empty?
-      if options[0] == '-p' || options[0] == '--percentage'
-        @sf = CompletionRateSF.new
-        @hp = HabitPrinter.new(@sf)
-      end
-    end
-
-    def list_all_habits 
-      puts Util.title Util.current_month
-      @habits.each_with_index do |h, index|
-        space = longest_name.length - h.name_length
-        puts "#{index + 1}. " +
-        "#{@hp.print_latest_progress(h, space)}"
+        habit = create(habit_name)
+        return habit
       end
-      puts "\n" + overall_stat_description
     end
 
-    def add(args)
-      habit_name, _options = parse_options(args)
-      find(habit_name) do
-        habit = create(habit_name)
-        save_to_file(habit, 'Add') unless habit.nil?
-        return
-      end
-      puts CLI.blue("#{habit_name} already existed!")
+    def longest_name
+      @habits.max_by(&:name_length).name
     end
 
-    def done(args)
-      habit_name, options = parse_options(args)
-      day = get_day_based_on(options)
-      habit = find_or_create(habit_name)
-      save_to_file(habit, 'Done') do
-        habit.done(true, day)
+    def total_habits_stat
+      @habits.each_with_object(done: 0, undone: 0) do |habit, stat|
+        stat[:done] += habit.latest_stat[:done]
+        stat[:undone] += habit.latest_stat[:undone]
       end
     end
 
-    def undone(args)
-      habit_name, options = parse_options(args)
-      day = get_day_based_on(options)
-      habit = find(habit_name) { raise_if_habit_error(habit_name) }
-      save_to_file(habit, 'Undone', 'blue') do
-        habit.done(false, day)
+    def done_count_for(date:)
+      habits.reduce(0) do |a, habit|
+        val = habit.done_for(date: date) == '1' ? 1 : 0
+        a + val
       end
     end
 
-    def remove(args)
-      habit_name, _options = parse_options(args)
-      habit = find(habit_name) { raise_if_habit_error(habit_name) }
-      save_to_file(habit, 'Remove', 'blue') do
-        @habits.delete(habit)
+    def undone_count_for(date:)
+      habits.reduce(0) do |a, habit|
+        val = habit.done_for(date: date) == '0' ? 1 : 0
+        a + val
       end
     end
 
-    def find_or_create(habit_name)
-      habit = find(habit_name) do
-        habit = create(habit_name)
-        return habit
-      end
+    def overall_stat_description(formatter)
+      Util.title('Total') + formatter.format(total_habits_stat)
     end
 
     def create(habit_name)
@@ -151,67 +76,14 @@ module Hbtrack
         return habit
       end
 
-      if habit_name && habit_name.length > 11 
-        raise_habit_name_too_long
-      else
-        raise_invalid_arguments
-      end
-    end
-
-    def parse_options(args)
-      options = args.select { |x| x =~ /\A-/ }
-      args -= options
-      [args[0], options]
-    end
-
-    def get_day_based_on(options)
-      yesterday = options[0] == '-y' || options[0] == '--yesterday'
-      return Date.today unless yesterday
-      Date.today - 1
-    end
-
-    def raise_error_msg(msg)
-      puts CLI.red msg 
-      return
-    end
-
-    def raise_if_habit_error(habit_name)       
-      return raise_invalid_arguments if habit_name.nil? 
-      raise_habit_not_found(habit_name)
-    end        
-
-    def raise_habit_not_found(habit_name)
-      raise_error_msg "Invalid habit: #{habit_name} not found."
-    end
-
-    def raise_invalid_arguments 
-      raise_error_msg "Invalid argument: habit_name is expected."
-    end
+      return ErrorHandler.raise_habit_name_too_long if habit_name && habit_name.length > 11
 
-    def raise_habit_name_too_long
-      raise_error_msg "habit_name too long."
+      ErrorHandler.raise_invalid_arguments
     end
 
     def invalid_habit_name?(habit_name)
-      habit_name.nil? || habit_name =~ /\s+/ || 
-      habit_name.length > 11
-    end
-
-    def save
-      File.open(@output_file_name, 'w') do |f|
-        @habits.each do |habit|
-          f.puts habit
-        end
-      end
-    end
-
-    def save_to_file(habit, action, color = 'green')
-      unless habit.nil?
-        yield if block_given?
-        save
-        output = "#{action} #{habit.name}!"
-        puts CLI.public_send(color, output)
-      end
+      habit_name.nil? || habit_name =~ /\s+/ ||
+        habit_name.length > 11
     end
   end
 end

data/lib/hbtrack/report/calendar.rb

@@ -0,0 +1,84 @@
+module Hbtrack
+  class Calendar
+
+    WEEKDAYS = [
+      "Mon", 
+      "Tue", 
+      "Wed", 
+      "Thu",
+      "Fri",
+      "Sat",
+      "Sun"
+    ]
+
+    # Generate weeks of the current month, 
+    # which contains 7 days.
+    def self.weeks(date)
+      range = date.beginning_of_month..date.end_of_month
+      result = range.chunk_while { |date| date.cwday < 7 }.to_a
+
+      if result.first.length < 7 
+        (7 - result.first.length).times do 
+          result.first.unshift(NullDay.new)
+        end
+      end
+
+      if result.last.length < 7 
+        (7 - result.last.length).times do 
+          result.last.push(NullDay.new)
+        end
+      end
+      
+      result
+    end
+
+    # Generate <td> elements for days
+    def self.td_for(date, progress)
+      progress = progress.split("")
+
+      className = if date.day.nil?
+        "null_day"      
+      elsif progress[date.day - 1] == "1"
+        "done"
+      elsif progress[date.day - 1] == "0"
+        "undone"
+      end
+      
+      string = <<~EOF
+      <td>
+        <p class="#{className}">
+          #{add_padding_to(date.day)}
+        </p>
+      </td>
+      EOF
+    end
+
+    def self.add_padding_to(day)
+      return nil if day.nil?
+      return "0#{day}" if day < 10 
+      day.to_s
+    end
+
+  end
+
+  # Null Object Pattern
+  class NullDay 
+    attr_reader :day
+    def initialize
+      @day = nil
+    end
+  end
+
+
+end
+
+# Extend Date Class
+class Date
+  def beginning_of_month
+    Date.new(self.year, self.month, 1)
+  end
+
+  def end_of_month
+    Date.new(self.year, self.month, -1)
+  end
+end

data/lib/hbtrack/report/generator.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require 'erb'
+require 'hbtrack/report/calendar'
+
+module Hbtrack
+  class Generator 
+    def initialize(hbt)
+      @hbt = hbt
+      initialize_stat
+      @week_days = Hbtrack::Calendar::WEEKDAYS
+      @weeks = Hbtrack::Calendar.weeks(Date.today)
+    end
+
+    def initialize_stat
+      @stat = @hbt.total_habits_stat
+      @stat[:total] = @stat[:done] + @stat[:undone]
+      @stat[:percentage] = (@stat[:done] / @stat[:total].to_f).round(2) * 100
+    end
+
+    def td_for(date)
+      Hbtrack::Calendar.td_for(date, @hbt.habits.last.latest_progress)
+    end
+
+    def generate
+      file_path = File.expand_path("../report.html.erb", __FILE__)
+      renderer = ERB.new(File.read(file_path))
+      File.write('report.html', renderer.result(binding))
+    end    
+  end
+end

data/lib/hbtrack/report/report.html.erb

@@ -0,0 +1,89 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title></title>
+    <link rel="stylesheet" type="text/css" href="https://cdnjs.cloudflare.com/ajax/libs/bulma/0.5.2/css/bulma.min.css">
+
+    <style type="text/css"> 
+      .text-center { margin-top: 10%; text-align: center; }
+
+      .calender { margin: 0 auto; width: 400px; height: 374.8571428571px; padding: 16px 18px; border: 1px solid #B3B3B3; overflow: hidden; }
+      .calender table { border-collapse: collapse; }
+      .calender .weekdays th { text-align: center; vertical-align: middle; text-transform: uppercase; height: 57.1428571429px; width: 57.1428571429px; }
+      .calender .days { height: 57.1428571429px; width: 57.1428571429px; }
+      .calender .days td { text-align: center; cursor: pointer; }
+      .calender .days td p { border-radius: 50%; width: 44px; height: 44px; margin: 0; display: inline-block; text-align: center; padding: 10px; }
+      .calender .days td p.null_day { display: none; }
+      .calender .days td p.undone { color: white; background-color: #FF1654; }
+      .calender .days td p.done { color: white; background-color: #70C1B3; }
+      .calender .days td:hover p { background-color: #E6E6E6; }
+      .calender .days td:hover p.undone { background-color: #CC1243; }
+      .calender .days td:hover p.done { background-color: #549187; }
+    </style>
+  </head>
+
+  <body>
+    <!-- Title -->
+    <section class="section">
+      <div class="container">
+        <div class="columns is-centered">
+          <h1 class="title">Hbtrack</h1>
+        </div>
+      </div>
+    </section>
+
+    <!-- Calendar -->
+    <section class="section>">
+    <div  class = "calender">
+      <table>
+        <!-- Generate Week Days -->
+        <tr class = "weekdays">
+          <% @week_days.each do |day| %>
+            <th><%= day %></th>
+          <% end %>
+        </tr>
+
+        <!-- Generate Days -->
+        <% @weeks.each do |week| %>
+          <tr class = "days">
+          <% week.each do |day| %>
+            <%= td_for(day) %>
+          <% end %>
+          </tr>
+        <% end %>
+      </table>
+    </div>
+    </section>
+
+    <!-- Stat -->
+    <section class="section">
+    <nav class="level">
+      <div class="level-item has-text-centered">
+        <div>
+          <p class="heading">Total</p>
+          <p class="title"><%= @stat[:total] %></p>
+        </div>
+      </div>
+      <div class="level-item has-text-centered">
+        <div>
+          <p class="heading">Done</p>
+          <p class="title"><%= @stat[:done] %></p>
+        </div>
+      </div>
+      <div class="level-item has-text-centered">
+        <div>
+          <p class="heading">Undone</p>
+          <p class="title"><%= @stat[:undone] %></p>
+        </div>
+      </div>
+      <div class="level-item has-text-centered">
+        <div>
+          <p class="heading">Completion Rate</p>
+          <p class="title"><%= @stat[:percentage] %>%</p>
+        </div>
+      </div>
+    </nav>
+    </section>
+    <!-- End of Stat -->
+  </body>
+</html>

data/lib/hbtrack/stat_formatter.rb

@@ -1,31 +1,21 @@
 # frozen_string_literal: true
 
-module Hbtrack
-  # This is an abstract class for classes that
-  # are used to format the progress of a Habit
-  # into string
-  class StatFormatter
-    def initialize
-    end
+require 'hbtrack/util'
 
-    def format
-      raise "Not Implemented"
-    end
-  end
-
-  class DoneUndoneSF < StatFormatter
+module Hbtrack
+  class DoneUndoneSF
     # Format in terms of the count of done and undone.
     # @param hash [Hash]
     # @option hash [String] :done total of done
     # @option hash [String] :undone total of undone
     # @return [String] formatted result
     def format(hash)
-      CLI.green("Done: #{hash[:done]}") + "\n" +
-      CLI.red("Undone: #{hash[:undone]}")
+      Util.green("Done: #{hash[:done]}") + "\n" +
+        Util.red("Undone: #{hash[:undone]}")
     end
   end
 
-  class CompleteSF < StatFormatter
+  class CompleteSF
     # Format in terms of the total and the count of
     # done and undone.
     # @param hash [Hash]
@@ -38,7 +28,7 @@ module Hbtrack
     end
   end
 
-  class CompletionRateSF < StatFormatter
+  class CompletionRateSF
     # Format in terms of the completion rate of the habit.
     # @param hash [Hash]
     # @option hash [String] :done total of done
@@ -46,7 +36,7 @@ module Hbtrack
     # @return [String] formatted result
     def format(hash)
       percentage = to_percentage(hash)[:done]
-      "Completion rate: #{'%.2f' % percentage}%"
+      sprintf('Completion rate: %.2f%', percentage)
     end
 
     # Convert the value in the hash into percentage
@@ -58,7 +48,7 @@ module Hbtrack
       total = hash[:done] + hash[:undone]
       done_p = 0
       undone_p = 0
-      unless total == 0
+      unless total.zero?
         done_p = hash[:done] / total.to_f * 100
         undone_p = hash[:undone] / total.to_f * 100
       end
@@ -66,4 +56,3 @@ module Hbtrack
     end
   end
 end
-

data/lib/hbtrack/store.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Hbtrack
+  class Store
+    def initialize(habits, file_name)
+      @habits = habits
+      @file_name = file_name
+    end
+
+    def save
+      File.open(@file_name, 'w') do |f|
+        @habits.each do |habit|
+          f.puts habit
+        end
+      end
+    end
+  end
+end

data/lib/hbtrack/util.rb

@@ -1,57 +1,68 @@
 # frozen_string_literal: true
 
 module Hbtrack
-  # This class contains the methods that 
+  # This class contains the methods that
   # are used to format the progress of a Habit
   # into string
   class Util
+    FONT_COLOR = {
+      green: "\e[32m",
+      red: "\e[31m",
+      blue: "\e[34m"
+    }.freeze
 
     class << self
-
+      # Define Util.green, Util.red, Util.blue
+      # which colorize string output in terminal
+      FONT_COLOR.each do |key, value|
+        define_method(key) do |string|
+          value + string + "\e[0m"
+        end
+      end
       # Format the string with title style.
       #
       # @param string [String] the string to be styled as title
       # @return [Nil]
-      # 
+      #
       # == Example
       #
       #   puts Util.title("Title")
       #   # Title
       #   # -----
-      #   #=> nil 
-      def title(string) 
-        string + "\n" + 
-        "-" * string.length + "\n"
+      #   #=> nil
+      def title(string)
+        string + "\n" +
+          '-' * string.length + "\n"
       end
 
       # Convert key into date in string form.
-      # 
-      # @param key [Symbol] The key of the progress in the 
+      #
+      # @param key [Symbol] The key of the progress in the
       #   form of :'year, month'. Example: :"2017,7"
-      # @param no_of_space [Numeric] number of space to be 
+      # @param no_of_space [Numeric] number of space to be
       #   added in front
       # @return [String] a string in date form.
-      # 
+      #
       # == Example
       #
       #   Util.convert_key_to_date(:"2017,7", 0)
       #   #=> "July 2016 : "
       def convert_key_to_date(key, no_of_space)
-        year = key.to_s.split(",")[0]
-        ' ' * no_of_space + get_month_from(key) + 
-        " #{year}" + " : "
+        year = key.to_s.split(',')[0]
+        ' ' * no_of_space + get_month_from(key) +
+          " #{year}" + ' : '
       end
 
       # Format the current month and year into string
-      # 
+      #
       # @return [String] Month and Year in String.
-      # 
+      #
       # == Example
       #
       #  Util.current_month
       #  #=> "August 2017"
       def current_month
-        Date.today.strftime("%B %Y")
+        Date.today.strftime('%B %Y')
       end
 
       # Get the month in string form from given key
@@ -60,7 +71,7 @@ module Hbtrack
       # @return [String] month
       #
       # == Example
-      #   
+      #
       #   Util.get_month_from(:"2017,7")
       #   #=> "July"
       def get_month_from(key)
@@ -70,4 +81,3 @@ module Hbtrack
     end
   end
 end
-

data/lib/hbtrack/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Hbtrack
-  VERSION = '0.0.5'
+  VERSION = '0.0.6'
 end