clamby 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bcf4bae22a96d13a305f1ed6ea672a9e32ae76be
-  data.tar.gz: 955cb04265ba74a26c0d329db079db68755e4af6
+  metadata.gz: de1411f20e529bdebaaa6f32b34863c06abc38f0
+  data.tar.gz: d6e542a17e3dcec0939d5fd94017b9892613b581
 SHA512:
-  metadata.gz: 1bec2c40aaecbb5e439d5ab9e92c25d8c70c99447e3b2ca37b9bd5b69db999dc2c6941b378a288daa0c29187642eefc1d7637d11ddd1fef443edc94e3564bd85
-  data.tar.gz: c3d2ca5c777919b6f5f4956818d1bf6efa1954c4d76509aa8bbdc9168ef1ed467772dca5c27746ec189543e53d2b9793c624df7b5cc29d29af0dbc4538f3eac6
+  metadata.gz: 64d0b0e23f7a8dd5fd42c839e6f1854d73e55fbb72d4a7d22a31ed5db1a63ce3f95d760e7dcc506a49238a51acebae7ba54de65a4b9cce18378922bf45aeaf84
+  data.tar.gz: 58a13ecdbd9068dd7052b4ba1f806592234c2c751b6e8e93702d4a7b851980fc5f5e515f30c563b3508d1185e68d0e310eaf57f8925f0993de4bcce787a9dce6

data/README.md

@@ -50,8 +50,15 @@ It's good to note that Clamby will not by default delete files which had a virus
 
 Configuration is rather limited right now. You can exclude the check if `clamscan` exists which will save a bunch of time for scanning your files. However, for development purposes, your machine may not have `clamscan` installed and you may wonder why it's not working properly. This is just to give you a reminder to install `clamscan` on your development machine and production machine. You can add the following to a config file, `clamby_setup.rb` to your initializers directory.
 
+There has been added additional functionality where you can override exceptions. If you set the exceptions below to false, then there will not be a hard exception generated. Instead, it will post to your log that an error had occured. By default each one of these configuration options are set to true. You may want to set these to false in your production environment.
+
 ```ruby
-    Clamby.configure({check: false})
+    Clamby.configure({
+      :check => false,
+      :error_clamscan_missing => false,
+      :error_file_missing => false,
+      :error_file_virus => false
+    })
 ```
 
 #Dependencies

data/lib/clamby.rb

@@ -1,8 +1,13 @@
 require "clamby/version"
-
+require "clamby/exception"
 module Clamby
 
-  @config = {:check => true}
+  @config = {
+    :check => true,
+    :error_clamscan_missing => true,
+    :error_file_missing => true,
+    :error_file_virus => true
+  }
 
   @valid_config_keys = @config.keys
 
@@ -17,8 +22,12 @@ module Clamby
         if scanner
           return true
         elsif not scanner
-          puts "VIRUS DETECTED on #{Time.now}: #{path}"
-          return false
+          if @config[:error_file_virus]
+            raise Exceptions::VirusDetected.new("VIRUS DETECTED on #{Time.now}: #{path}")
+          else
+            puts "VIRUS DETECTED on #{Time.now}: #{path}"
+            return false
+          end
         end
       end
     end
@@ -28,23 +37,31 @@ module Clamby
     if @config[:check]
       scanner = system('clamscan')
       if not scanner
-        puts "CLAMSCAN NOT FOUND"
-        return false
+        if @config[:error_clamscan_missing]
+          raise Exceptions::ClamscanMissing.new("Clamscan application not found. Check your installation and path.")
+        else
+          puts "CLAMSCAN NOT FOUND"
+          return false
+        end
       else
         return true
       end
-  else
-    return true
-  end
+    else
+      return true
+    end
   end
 
   def self.file_exists?(path)
     if File.file?(path)
       return true
-  else
-    puts "FILE NOT FOUND on #{Time.now}: #{path}"
-    return false
-  end
+    else
+      if @config[:error_file_missing]
+        raise Exceptions::FileNotFound.new("File not found: #{path}")
+      else
+        puts "FILE NOT FOUND on #{Time.now}: #{path}"
+        return false
+      end
+    end
   end
 
   def self.update

data/lib/clamby/exception.rb

@@ -0,0 +1,6 @@
+module Exceptions
+  class Error < StandardError; end
+  class VirusDetected < Error; end
+  class ClamscanMissing < Error; end
+  class FileNotFound < Error; end
+end

data/lib/clamby/version.rb

@@ -1,3 +1,3 @@
 module Clamby
-  VERSION = "1.0.2"
+  VERSION = "1.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: clamby
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - kobaltz
@@ -55,6 +55,7 @@ files:
 - Rakefile
 - clamby.gemspec
 - lib/clamby.rb
+- lib/clamby/exception.rb
 - lib/clamby/version.rb
 homepage: ''
 licenses: