argosnap 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ebd85f5ba30c97bd0f6152d0c904d27bdf13ac88
-  data.tar.gz: f315bbc294fb0011eae5ab82027e3aa6eab611de
+  metadata.gz: a7a15c8ff5820ce085fbe1a69147fe44e2516f31
+  data.tar.gz: b24ede47dff5a3e52e9db0bf3be9799f25728920
 SHA512:
-  metadata.gz: 8fe3912b47321cafd27786c6174e4cd24fad9982ae8a33ff2895533b5b5d70f2f185296ea6abfb5529ccec6af9a783de77cff8ba9b7d94d51ea6ceee57d89bc1
-  data.tar.gz: 2b933606d435e2455895779bcaa2001be2bd113e7d91347ecfdc1b7ee9506784bf1c0d4732bc5782d9077c4862caa0c68ebc5ff8c29ed7f2b035c6ced22b8698
+  metadata.gz: 9e3c652266467c06d5234fc673e16aa433ae80b175d80b1ce19e7522a9ffaef84820ac22a48ecbd67ad1e1a6e6ac34999a6918be850c3bb8a589d9fe3832ee02
+  data.tar.gz: 2946547f4570be8b426b71ddb2dd4c6233bf47aa08bdedccc4f8acc0f2561c2bb44539806d5ae46b4d959869e24a5b7fbb457630c5eb2af40a39a3769a78465f

data/.gitignore

@@ -15,3 +15,5 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+.DS_Store
+notes.txt

data/Gemfile

@@ -1,7 +1,19 @@
 source 'https://rubygems.org'
 gem 'mechanize', '2.7.3'
-gem 'terminal-notifier', '1.6.0'
-gem 'plist', '3.1.0'
+gem 'haml', '4.0.7'
+gem 'mail', '2.6.3'
+
+# osx related
+gem 'terminal-notifier'
+gem 'plist'
+
+group :development do
+  gem 'minitest'
+  gem 'minitest-reporters'
+  gem 'guard' # NOTE: this is necessary in newer versions
+  gem 'guard-minitest'
+  gem 'ruby_gntp'
+end
 
 # Specify your gem's dependencies in argosnap.gemspec
 gemspec

data/Guardfile

@@ -0,0 +1,25 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+## Uncomment and set this to only include directories you want to watch
+ directories %w(bin lib files test) \
+  .select{|d| Dir.exists?(d) ? d : UI.warning("Directory #{d} does not exist")}
+
+## Note: if you are using the `directories` clause above and you are not
+## watching the project directory ('.'), then you will want to move
+## the Guardfile to a watched dir and symlink it back, e.g.
+#
+#  $ mkdir config
+#  $ mv Guardfile config/
+#  $ ln -s config/Guardfile .
+#
+# and, you'll have to watch "config/Guardfile" instead of "Guardfile"
+
+notification :gntp, host: '127.0.0.1', password: ENV['GNTP_PASSWORD']
+
+guard :minitest do
+  # with Minitest::Unit
+  watch(%r{^test/(.*)\/?test_(.*)\.rb$})
+  watch(%r{^lib/(.*/)?([^/]+)\.rb$})     { |m| "test/#{m[1]}test_#{m[2]}.rb" }
+  watch(%r{^test/test_helper\.rb$})      { 'test' }
+end

data/README.md

@@ -2,13 +2,13 @@
 
 This gem allows you to displays your current amount of picodollars along with [OSX notifications](http://support.apple.com/kb/ht5362)  when your [tarsnap account](http://www.tarsnap.com/) falls bellow a predefined threshold of [picodollars](http://www.tarsnap.com/picoUSD-why.html).
 
-## Installation
+# Installation
 
 Install the gem via rubygems:
 
     $ gem install argosnap plist mechanize terminal-notifier
 
-## Configure
+# Configure
 
 Run the following command for the script to create a configuration file like: 
 
@@ -27,7 +27,7 @@ Just put your tarsnap login credentials. You can omit the other two variables if
     $ argosnap -p
 
 
-## Configure Launchd under MacOSX
+# Configure Launchd under MacOSX
 
 In the `config.yml` adjust the `threshold` and `seconds` options as you see fit. Threshold is the amount of *picodollars* bellow which you'd like to start seeing notifications and the `seconds` consists of the time window. Then type:
 
@@ -35,7 +35,7 @@ In the `config.yml` adjust the `threshold` and `seconds` options as you see fit.
 
 Now just start the installed `.plist` file following instruction you'll see on the screen and you're done.
 
-## Options
+# Options
 
 To run an osx notification use the command:
 
@@ -49,7 +49,7 @@ To get the amount of picodollars as an integer type:
 
 That's all :-)
 
-## Contributing
+# Contributing
 
 1. Fork it ( http://github.com/atmosx/argosnap/fork )
 2. Create your feature branch (`git checkout -b my-new-feature`)

data/argosnap.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new do |spec|
   spec.version       = Argosnap::VERSION
   spec.authors       = ["Panagiotis Atmatzidis"]
   spec.email         = ["atma@convalesco.org"]
-  spec.summary       = %q{Display osx notifications when tarsnap account runs out of picodollars}
-  spec.description   = %q{A ruby script displays OSX notifications with current ballance. If the balance of picodollars falls bellow a specific threshold}
+  spec.summary       = %q{Read your tarsnap picoUSD amount in cli and display notifications when it runs out!.}
+  spec.description   = %q{A ruby script that displays tarsnap current ballance in picoUSD. The script can send notifications when the acount balance falls below the predefined threshold. Argosnap supports email, pushover and OSX notifications.}
   spec.homepage      = "https://github.com/atmosx/argosnap"
   spec.license       = "MIT"
 

data/bin/argosnap

@@ -7,25 +7,37 @@ options = {}
 opt_parser = OptionParser.new do |opt|
   opt.banner = "argosnap #{Argosnap::VERSION} ( https://github.com/atmosx/argosnap/ )\nUsage: argosnap [OPTIONS]"
   opt.separator  ""
-  opt.separator  "     -v:        dislay version"
-  opt.separator  "     -i config: install configuration files"
-  opt.separator  "     -i cron:   installs Launchd script under OSX"
-  opt.separator  "     -p:        prints the current amount in picodollars"
-  opt.separator  "     -p osx:    prints desktop osx notification"
-  opt.separator  "     -p clean:  prints only the picollars (float rounded in 4 decimals), to use in cli"
+  opt.separator  "     -v:           dislay version"
+  opt.separator  "     -i config:    install configuration files"
+  opt.separator  "     -i cron:      display sample cron entry"
+  opt.separator  "     -i plist:     install plist file for OSX"
+  opt.separator  "     -p:           prints the current amount in picoUSD"
+  opt.separator  "     -p clean:     prints only the picollars (float rounded in 4 decimals), to use in cli"
+  opt.separator  "     -n mail:      send notification via email"
+  opt.separator  "     -n pushover:  send notification via pushover"
+  opt.separator  "     -n osx:       display osx notification"
+  opt.separator  "     -n all:       send notifications everywhere"
   opt.separator  ""
 
   opt.on("-v","--version","display version") do |version|
     options[:version] = version
   end
 
+  opt.on("-c","--cron","run notifications - to use with cron") do |version|
+    options[:cron] = cron
+  end
+
   opt.on("-i","--install [OPTION]", "install configuration files") do |install|
     options[:install] = install || 'config'
   end
 
-  opt.on("-p","--print [OPTION]","fetch current amount in picodollars") do |print|
+  opt.on("-p","--print [OPTION]","fetch current amount in picoUSD") do |print|
     options[:print] = print || 'default'
   end
+  
+  opt.on("-n","--notification [OPTION]","send notification via email") do |notification|
+    options[:notification] = notification || 'notification'
+  end
 
   opt.on("-h","--help","help") do |h|
     options[:help] = h
@@ -40,30 +52,43 @@ begin
     puts Argosnap::VERSION
   elsif options[:install]
     if options[:install] == 'config'
-      Argosnap::Install.new.config
+      Argosnap::Install.new.install
     elsif options[:install] == 'cron'
-      Argosnap::OSXNotifications.new.install_launchd_script
+      Argosnap::Install.new.cron_entry
+    elsif options[:install] == 'plist'
+      Argosnap::Helpers.new.install_plist
     end
+  elsif options[:notification]
+    if options[:notification] == 'mail'
+      Argosnap::Notifications.new.send_mail
+    elsif options[:notification] == 'pushover'
+      Argosnap::Notifications.new.send_pushover
+    elsif options[:notification] == 'osx'
+      Argosnap::Notifications.new.send_osx_notification
+    elsif options[:notification] == 'osx_check'
+      Argosnap::Notifications.new.osx_check
+    elsif options[:notification] == 'all'
+      Argosnap::Notifications.new.notify
+    else
+      puts "You need to need to speficy notification option: 'mail' or 'pushover'"
+    end
+  elsif options[:cron]
+    Argosnap::Notifications.notify
   elsif options[:help]
     # avoids double printing of 'opt_parser'
   elsif options[:print]
-    if options[:print] == 'osx'
-      Argosnap::OSXNotifications.new.display
-    elsif options[:print] == 'osx_check'
-      Argosnap::OSXNotifications.new.display_check
-    elsif options[:print] == 'default'
+    if options[:print] == 'default'
       b = Argosnap::Fetch.new.balance
-      puts "Current balance (picodollars): #{b}"
+      puts "Current picoUSD balance: #{b}"
     elsif options[:print] == 'clean'
-      b = Argosnap::Fetch.new.balance
-      puts b
+      puts Argosnap::Fetch.new.balance
     else
       puts "Option not recognized"
     end
   else
     puts opt_parser
   end
-rescue OptionParser::InvalidOption => e
+rescue OptionParser::InvalidOption 
   puts "No such option! Type 'argosnap -h' for help!"
   exit
 end

data/files/local.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+SH=/bin/sh
+
 # Load RVM into a shell session *as a function*
 if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
 	# First try to load from a user install
@@ -7,9 +9,15 @@ if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
 elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
 	# Then try to load from a root install
 	source "/usr/local/rvm/scripts/rvm"
+elif [[ -s "$HOME/.rbenv/bin" ]] ; then
+  # try to load rbenv
+  $SH -c 'export PATH="$HOME/.rbenv/bin:$PATH" ; eval "$(rbenv init -)"'
+elif [[ -s `which ruby` ]] ; then
+  # Nothing to do here(?)
+  echo "Using system's ruby" >> ~/.argosnap/argosnap.log
 else
-	printf "ERROR: An RVM installation was not found.\n"
+	printf "ERROR: No ruby installation was found!\n"
 fi
 
 # Executable now should be in path
-argosnap -p osx_check
+argosnap -n osx_check

data/files/mail.body.haml

@@ -0,0 +1,24 @@
+%p Hello
+%p
+  Your
+  %a{:href => "https://www.tarsnap.com/"} tarsnap account
+  is running out of funds.
+  
+  Your current amount of picoUSD is
+  = succeed "." do
+    %strong #{self.to_s}
+
+%p
+  Have a nice day!
+
+%p
+  argosnap
+
+\--
+
+%p
+  %strong Argosnap: 
+  Receive notifications when your tarsnap account is running out of picoUSD!
+%p
+  %strong URL:
+  %a{:href => "https://github.com/atmosx/argosnap"} https://github.com/atmosx/argosnap

data/lib/argosnap.rb

@@ -1,6 +1,8 @@
-# argosnap module
+# load argosnap classes and modules
 
 require_relative File.expand_path("../argosnap/version", __FILE__)
 require_relative File.expand_path("../argosnap/install", __FILE__)
+require_relative File.expand_path("../argosnap/config", __FILE__)
+require_relative File.expand_path("../argosnap/helpers", __FILE__)
 require_relative File.expand_path("../argosnap/balance", __FILE__)
-require_relative File.expand_path("../argosnap/osxnotify", __FILE__) if Gem::Platform.local.os == 'darwin'
+require_relative File.expand_path("../argosnap/notifications", __FILE__)

data/lib/argosnap/balance.rb

@@ -1,39 +1,48 @@
-require 'yaml'
 require 'mechanize'
-require 'logger'
 
 module Argosnap
-  # Given a username and a password, fetch balance from http://www.tarsnap.com
+  # Given a username and a password, fetch balance from https://www.tarsnap.com
   class Fetch
-    def initialize         
-      begin
-        user = ENV['USER']
-        raise ArgumentError.new("Please make sure you run 'argosnap install' in order to install configuration files, before running asnap!") unless File.exists?("#{Dir.home}/.argosnap/config.yml")
-        r = YAML::load_file("#{Dir.home}/.argosnap/config.yml")
-        logfile = "#{Dir.home}/.argosnap/argosnap.log"
-        @email, @password, @threshold, @logger, @agent = r[:email], r[:password], r[:threshold], Logger.new(logfile), Mechanize.new
-      rescue ArgumentError => e
-        puts e.message
-        # puts e.backtrace
-        exit
+
+    # Elementary configuration file check
+    def check_configuration
+      Install.new.ensure_compatibility
+      Install.new.ensure_installation
+      if config.data[:email].empty?
+        config.log_and_abort("Please add your tarsnap email to #{config}")
+      end
+      if config.data[:password].empty?
+        config.log_and_abort("Please add your tarsnap password to #{config}")
       end
     end
 
-    # Fetch balance from tarsnap using mechanize
+    # Fetch balance from tarsnap using
     def balance
-      begin
-        page = @agent.get('https://www.tarsnap.com/account.html')
-        form = page.form_with(:action => 'https://www.tarsnap.com/manage.cgi')
-        form.address = @email
-        form.password = @password
-        panel = @agent.submit(form)
-        picodollars = panel.parser.to_s.scan(/\$\d+\.\d+/)[0].scan(/\d+\.\d+/)[0].to_f.round(4) # I'm the feeling that this can be done in a prettier way
-        @logger.info "Execution ended successfully! Picodollars: #{picodollars}" 
-        picodollars
-      rescue Exception => e
-        @logger.error "A problem with tarsnap notification occured:"
-        @logger.error "#{e}"
+      check_configuration
+      agent         = Mechanize.new
+      page          = agent.get('https://www.tarsnap.com/account.html')
+      form          = page.form_with(:action => 'https://www.tarsnap.com/manage.cgi')
+      form.address  = config.data[:email]
+      form.password = config.data[:password]
+      panel         = agent.submit(form)
+      wrong_email_message = 'No user exists with the provided email address; please try again.'
+      wrong_password_message = 'Password is incorrect; please try again.'
+      if panel.body.include?(wrong_email_message)
+        config.log_and_abort('Password is incorrect; please try again.')
+      elsif panel.body.include?(wrong_password_message)
+        config.log_and_abort('Bad password. Please check your configuration file tarsnap password!')
       end
+      picodollars   = panel.parser.to_s.scan(/\$\d+\.\d+/)[0].scan(/\d+\.\d+/)[0].to_f.round(4) 
+      config.logger.info("Current amount of picoUSD: #{picodollars}")
+      picodollars
+    end
+
+    # private methods
+    private 
+
+    # load configuration files
+    def config 
+      Configuration.new
     end
   end
 end

data/lib/argosnap/config.rb

@@ -0,0 +1,40 @@
+require 'yaml'
+require 'logger'
+
+module Argosnap
+  class Configuration
+    # configuration files
+    def files
+      {
+        config: "#{Dir.home}/.argosnap/config.yml", 
+        home: "#{Dir.home}/.argosnap/", 
+        logfile: "#{Dir.home}/.argosnap/argosnap.log"
+      }
+    end
+
+    # logger object
+    def logger
+      Logger.new(files[:logfile], 10, 1024000)
+    end
+
+    # Config data in hash 
+    def data
+      YAML::load_file(files[:config])
+    end
+
+    # Check if gem is available
+    def gem_available?(name)
+         Gem::Specification.find_by_name(name)
+    rescue Gem::LoadError
+         false
+    rescue
+         Gem.available?(name)
+    end
+
+    # Log and Abort!
+    def log_and_abort(msg)
+      logger.error(msg)
+      Kernel.abort(msg)
+    end
+  end
+end

data/lib/argosnap/helpers.rb

@@ -0,0 +1,41 @@
+module Argosnap
+
+  # Helper Class
+  class Helpers
+
+    # Creates launchd script with user's variables
+    def install_plist
+      if configuration.gem_available?('plist')
+        require 'plist'
+        time_interval = configuration.data[:seconds]
+        # Gem::Platform.local.os # => osx
+        user = ENV['USER']
+        launch_agents = "/Users/#{user}/Library/LaunchAgents/"
+        # start_script loads ruby - optimized for 'rvm'
+        start_script = File.expand_path('../../../files/local.sh', __FILE__)
+        if Dir.exists?(launch_agents)
+          filename = "org.#{user}.argosnap.plist"
+          if File.exist?("#{launch_agents}#{filename}")
+            puts "Please delete file: #{launch_agents}#{filename} before proceeding!"
+          else
+            hash = {"Label"=>"#{filename.scan(/(.+)(?:\.[^.]+)/).flatten[0]}", "ProgramArguments"=>["#{start_script}"], "StartInterval"=> time_interval, "RunAtLoad"=>true} 
+            File.open("#{launch_agents}#{filename}", 'w') {|f| f.write(hash.to_plist)}
+            puts "Launchd script is installed. Type 'launchctl load -w #{launch_agents}#{filename}' to load the plist."
+          end
+        else
+          puts "No '#{launch_agents}' directory found! Aborting installation."
+        end
+      else
+        configuration.log_and_abort("Please install plist!")
+      end
+    end
+
+    # private
+    private
+    def configuration
+      Configuration.new
+    end
+
+  end
+
+end

data/lib/argosnap/install.rb

@@ -1,22 +1,92 @@
-require 'logger'
-require 'yaml'
-require 'plist'
+require 'fileutils'
 
 module Argosnap
+
   class Install
-    # Install 'config.yml'
-    def config
-      url = 'http://atmosx.github.com/argosnap'
-      raise ArgumentError.new("This gem is made for UNIX! Please check the website for details #{url} !") unless %w{ darwin linux freebsd openbsd netbsd }.include? Gem::Platform.local.os 
+
+    # Install configuration files in the system
+    def install
+      ensure_compatibility
       begin
-        Dir.mkdir(File.join(Dir.home, ".argosnap"), 0700) unless Dir.exists?("/Users/#{ENV['USER']}/.argosnap")# permissions are a-rwx,u+rwx
-        config = {email: 'tarsnap_email', password: 'tarsnap_password', threshold: 10, seconds: 7200}
-        File.open("#{Dir.home}/.argosnap/config.yml", "w") {|f| f.write(config.to_yaml)}
-        puts "1. Edit the configuration file: #{Dir.home}/.argosnap/config.yml"
-        puts "2. Launch argosnap by typing in the terminal: argosnap -p"
-      rescue Exception => e
+        # permissions are a-rwx, u+rwx
+        if File.exists? configuration.files[:config]
+          puts "Config file exists!"
+          puts "Please remove config: #{configuration.files[:config]} manually and re-install the configuration file." 
+        else
+          Dir.mkdir(File.join(Dir.home, ".argosnap"), 0700) unless File.exist? configuration.files[:home]
+          FileUtils.touch(configuration.files[:logfile]) unless File.exist? configuration.files[:logfile]
+          config = {
+            email: 'tarsnap_email', 
+            password: 'tarsnap_password', 
+            threshold: 10, 
+            seconds: 86400,
+            notifications_osx: false,
+            notifications_email: false,
+            smtp: {
+              email_delivery_method: 'smtp', 
+              smtpd_user: 'username',
+              smtpd_password: 'password',
+              smtpd_address: "gmail.google.com", 
+              smtpd_port: 587, 
+              smtpd_from: 'user@host.domain.com', 
+              smtpd_to: 'your@email.com', 
+              format: 'txt'},
+            notifications_pushover: false,
+            pushover: {
+              key: 'user_key',
+              token: 'app_token'}
+          }
+          File.open(configuration.files[:config], "w") {|f| f.write(config.to_yaml)}
+
+          puts ""
+          puts "Configuration file created! To setup argosnap: "
+          puts "1. Edit the configuration file: #{configuration.files[:config]}"
+          puts "2. Run argosnap by typing in the terminal 'argosnap -p' to fetch current balance"
+          puts "3. To configure notifications see: <link>"
+          puts ""
+
+        end
+      rescue Errno::ENOENT => e
+        puts "Can't configuration locate file!"
         puts e.message
+        puts e.backtrace
+      end
+    end
+
+    def cron_entry
+      puts "\nA standard cron entry looks like this:\n"
+      puts "\n5 8 * * 6 <user> <command>\n"
+      puts "\nFor detailed instructions see here: <wiki-link>"
+      puts ""
+    end
+
+    # Check OS support and configuration file
+    def ensure_compatibility
+      # Check the operating system. Windows is not supported at this time.
+      unless %w{ darwin linux freebsd openbsd netbsd }.include? Gem::Platform.local.os 
+        Kernel.abort("\nThis gem is made for UNIX! Please check the website for details #{url} !\n\n")
+      end
+    end
+
+    def ensure_installation
+      if File.exist?(configuration.files[:home])
+        if (File.exist?(configuration.files[:logfile]) && File.exist?(configuration.files[:config]))
+          true
+        else
+          Kernel.abort("\nNo configuration files found! Please run 'argosnap -i config'\n\n")
+        end
+      else
+        Kernel.abort("\nNo configuration directory found! Please run 'argosnap -i config'\n\n")
       end
     end
+
+    # private methods
+    private
+
+    # Load onfiguration files
+    def configuration
+      Configuration.new
+    end
+
   end
 end

data/lib/argosnap/notifications.rb

@@ -0,0 +1,78 @@
+module Argosnap
+  class Notifications
+
+    # When tarsnap balance is bellow threshold notify me
+    # Method is chosen based on threshold
+    def notify
+      if config.data[:threshold] >= Fetch.new.balance
+        send_mail
+        send_pushover
+        send_osx_notification
+      end
+    end
+
+    # Use with plist file
+    def osx_check
+      if config.data[:threshold] >= Fetch.new.balance
+        send_osx_notification
+      end
+    end
+
+    # send notification via email
+    def send_mail
+      if config.data[:notifications_email]
+        if config.gem_available?('mail')
+          require 'mail'
+          require 'socket'
+          require_relative File.expand_path("../notifications/mailer", __FILE__)
+          Mailer.new(config.data, config.logger).send_mail
+        else
+          config.log_and_abort("Please install 'mail' gem by executing: '$ gem install mail'.")
+        end
+      end
+    end
+
+    # send notification via pushover.net
+    def send_pushover
+      if config.data[:notifications_pushover]
+        require "net/https"
+        require_relative File.expand_path("../notifications/pushover", __FILE__)
+        key     = config.data[:pushover][:key]
+        token   = config.data[:pushover][:token]
+        amount = Fetch.new.balance
+        message = "Your current tarsnap amount is #{amount} picoUSD!"
+        Pushover.send(token, key, message, config.logger, amount)
+      end
+    end
+
+    # send notification to osx notifications (desktop)
+    def send_osx_notification
+      if config.data[:notifications_osx]
+        if Gem::Platform.local.os == 'darwin'
+          # load 'terminal-notifier'
+          if config.gem_available?('terminal-notifier')
+            require 'terminal-notifier'
+            require_relative File.expand_path("../notifications/osxnotifications", __FILE__)
+            message = "Your picoUSD amount is: #{Fetch.new.balance}"
+            title  = 'argosnap'
+            subtitle = 'running out of picoUSD!'
+            OSXnotify.send(message, title, subtitle)
+          else
+             config.log_and_abort("You need to install 'terminal-notifier' gem!")
+          end
+        else
+          config.log_and_abort("OSX notifications are available only under OSX > 10.8.x")
+        end
+      end
+    end
+
+    # private methods
+    private
+
+    # load configuration files
+    def config 
+      Configuration.new
+    end
+
+  end
+end

data/lib/argosnap/notifications/mailer.rb

@@ -0,0 +1,101 @@
+module Argosnap
+  class Mailer
+
+    # load configuration files
+    def config 
+      Configuration.new
+    end
+
+    attr_reader :logger, :smtpd_user, :smtpd_password, :smtpd_address, :smtpd_port, :smtpd_from, :smtpd_to, :method, :format, :balance
+    def initialize(config, logger)
+      @logger              = logger
+      @smtpd_user          = config[:smtp][:smtpd_user]
+      @smtpd_password      = config[:smtp][:smtpd_password]
+      @smtpd_address       = config[:smtp][:smtpd_address]
+      @smtpd_port          = config[:smtp][:smtpd_port]
+      @smtpd_from          = config[:smtp][:smtpd_from]
+      @smtpd_to            = config[:smtp][:smtpd_to]
+      @method              = config[:smtp][:email_delivery_method]
+      @format              = config[:smtp][:format]
+      @balance             = Fetch.new.balance.to_s
+    end
+
+    def ensure_mail_configuration
+      if smtpd_address.empty?
+        config.log_and_abort("There is no 'smtpd_address' in #{config}. Please check your configuration file.")
+      elsif smtpd_to.empty?
+        config.log_and_abort("There is no 'smtpd_to' in #{config}. Please check your configuration file.")
+      end
+    end
+
+    def text_message(amount)
+      message = <<EOF
+
+Hello
+
+Your tarsnap account is running out of funds. Your current amount of picoUSD is #{amount}.
+
+This automated email message was sent by argosnap. Please do not reply to this email.
+
+Have a nice day!
+
+argosnap
+--
+
+Argosnap: Receive notifications when your tarsnap account is running out of picoUSD!
+URL:      https://github.com/atmosx/argosnap
+
+EOF
+
+      message
+    end
+
+
+    # send email. HTML format is the default.
+    def send_mail
+      # ensure compatibility first
+      Install.new.ensure_compatibility
+      ensure_mail_configuration
+
+      # create the 'mail' object
+      mail           = Mail.new
+
+      # configure mail options
+      mail[:from]    =  smtpd_from
+      mail[:to]      =  smtpd_to
+      mail[:subject] = "ARGOSNAP: your tarsnap balance."
+
+      # configure mail format: Use HTML if the user wants it, otherwise default to 'txt'
+      if format == 'html'
+        if config.gem_available?('haml')
+          require 'haml'
+          mail['content-type'] = 'text/html; charset=UTF-8'
+          # convert HAML to HTML
+          Haml::Engine.new(File.read(File.expand_path('../../../../files/mail.body.haml', __FILE__))).def_method(balance, :render)
+          mail[:body] = balance.render
+        else
+          log.error("Please install haml gem to receive HTML emails: '$ gem install haml'")
+          mail[:body] = text_message(balance)
+        end
+      else
+        mail[:body] =  text_message(balance)
+      end
+
+      # SMTPd configuration. Defaults to local 'sendmail'
+      if method == 'smtp'
+        if smtpd_port == 465
+          # use smtps with some defaults
+          mail.delivery_method :smtp, address: smtpd_address, port: smtpd_port, user_name: smtpd_user, password: smtpd_password, ssl: true, openssl_verify_mode: 0
+        else
+          mail.delivery_method :smtp, address: smtpd_address, port: smtpd_port
+        end
+      end
+
+      # dispatch email
+      mail.deliver
+      logger.info("Argosnap sent an email to #{smtpd_to}!")
+      logger.info("Current amount in picoUSD: #{balance}")
+    end
+
+  end
+end

data/lib/argosnap/notifications/osxnotifications.rb

@@ -0,0 +1,5 @@
+module OSXnotify
+  def self.send(message, title, subtitle)
+    TerminalNotifier.notify(message, :title => title, :subtitle => subtitle) 
+  end
+end

data/lib/argosnap/notifications/pushover.rb

@@ -0,0 +1,32 @@
+# pushover.net module
+module Pushover
+  def self.send(token, user, message, log, amount)
+    begin
+      # be nice to Pushover API
+      sleep(5)
+      url = URI.parse("https://api.pushover.net/1/messages.json")
+      req = Net::HTTP::Post.new(url.path)
+      req.set_form_data({
+        token: token,
+        user: user,
+        message: message
+      })
+      res = Net::HTTP.new(url.host, url.port)
+      res.use_ssl = true
+      res.verify_mode = OpenSSL::SSL::VERIFY_PEER
+      res.start do |http|
+        status = http.request(req).code
+        if status == '200'
+          log.info('Notification sent via Pushover!')
+          log.info("Current amount of picoUSD: #{amount}")
+        elsif (400...500).to_a.include?(status)
+          log.error("Please revise your Pushover credentials. API request failed.")
+        else
+          log.error("Request returned status code: #{status}, Pushover notification probably failed.")
+        end
+      end
+    end
+  rescue SocketError
+    log.error("No internet connection available, Pushover notification failed.")
+  end
+end

data/lib/argosnap/version.rb

@@ -1,3 +1,3 @@
 module Argosnap
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: argosnap
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Panagiotis Atmatzidis
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-12 00:00:00.000000000 Z
+date: 2015-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -38,8 +38,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: A ruby script displays OSX notifications with current ballance. If the
-  balance of picodollars falls bellow a specific threshold
+description: A ruby script that displays tarsnap current ballance in picoUSD. The
+  script can send notifications when the acount balance falls below the predefined
+  threshold. Argosnap supports email, pushover and OSX notifications.
 email:
 - atma@convalesco.org
 executables:
@@ -49,16 +50,23 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - Gemfile
+- Guardfile
 - LICENSE.txt
 - README.md
 - Rakefile
 - argosnap.gemspec
 - bin/argosnap
 - files/local.sh
+- files/mail.body.haml
 - lib/argosnap.rb
 - lib/argosnap/balance.rb
+- lib/argosnap/config.rb
+- lib/argosnap/helpers.rb
 - lib/argosnap/install.rb
-- lib/argosnap/osxnotify.rb
+- lib/argosnap/notifications.rb
+- lib/argosnap/notifications/mailer.rb
+- lib/argosnap/notifications/osxnotifications.rb
+- lib/argosnap/notifications/pushover.rb
 - lib/argosnap/version.rb
 - test/lib/argosnap/options_test.rb
 - test/lib/argosnap/version_test.rb
@@ -83,10 +91,11 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.2
+rubygems_version: 2.4.8
 signing_key: 
 specification_version: 4
-summary: Display osx notifications when tarsnap account runs out of picodollars
+summary: Read your tarsnap picoUSD amount in cli and display notifications when it
+  runs out!.
 test_files:
 - test/lib/argosnap/options_test.rb
 - test/lib/argosnap/version_test.rb

data/lib/argosnap/osxnotify.rb

@@ -1,58 +0,0 @@
-require 'terminal-notifier'
-require_relative File.expand_path('../balance', __FILE__)
-require 'yaml'
-
-module Argosnap
-  class OSXNotifications
-    def initialize
-      begin
-        user = ENV['USER']
-        raise ArgumentError.new("This command is made for Darwin! Please check the website for details #{url} !") unless Gem::Platform.local.os == 'darwin'
-        r = YAML::load_file("#{Dir.home}/.argosnap/config.yml")
-        logfile = "#{Dir.home}/.argosnap/argosnap.log"
-        @threshold, @logger, @picodollars, @time_interval = r[:threshold], Logger.new(logfile), Argosnap::Fetch.new.balance, r[:seconds]
-      rescue ArgumentError => e
-        puts e.message
-        # puts e.backtrace
-        exit
-      end
-    end
-
-    # Creates launchd script with user's variables
-    def install_launchd_script 
-      raise ArgumentError.new("Please make sure you run 'argosnap install' in order to install configuration files, before running asnap!") unless File.exists?("#{Dir.home}/.argosnap/config.yml")
-      user = ENV['USER']
-      c = "#{Dir.home}/.argosnap/config.yml"
-      begin
-        launch_agents = "/Users/#{user}/Library/LaunchAgents/"
-        # start_script loads the RVM environment.
-        start_script = File.expand_path('../../../files/local.sh', __FILE__)
-        if Dir.exists?(launch_agents)
-          filename = "org.#{user}.argosnap.plist"
-          if File.exist?("#{launch_agents}#{filename}")
-            puts "Please delete file: #{launch_agents}#{filename} before proceeding!"
-          else
-            hash = {"Label"=>"#{filename.scan(/(.+)(?:\.[^.]+)/).flatten[0]}", "ProgramArguments"=>["#{start_script}"], "StartInterval"=> @time_interval, "RunAtLoad"=>true} 
-            File.open("#{launch_agents}#{filename}", 'w') {|f| f.write(hash.to_plist)}
-            puts "Launchd script is installed. Type 'launchctl load -w #{launch_agents}#{filename}' to load the plist."
-          end
-        else
-          puts "No '#{launch_agents}' directory found! Aborting installation."
-        end
-      rescue Exception => e
-        puts e.message
-        # puts e.backtrace
-      end
-    end
-
-    # Display notification if threshold is surpassed
-    def display_check
-      TerminalNotifier.notify("Current balance: #{@picodollars}", :title => 'TarSnap', :subtitle => 'balance running out') if @picodollars < @threshold.to_i
-    end
-
-    # Display current ballance in OSX style notifications
-    def display
-      TerminalNotifier.notify("Current balance: #{@picodollars}", :title => 'TarSnap', :subtitle => 'balance running out') 
-    end
-  end
-end