fidius-cvedb 0.0.6 → 0.0.7

data/bin/fidius-cvedb

@@ -49,7 +49,31 @@ end
 def symlink_rake_tasks
   Dir.glob(File.join GEM_BASE, 'tasks', '*.rake') do |rake|
     dest = File.join 'lib', 'tasks', File.basename(rake)
-    FileUtils.ln_s(rake, dest)
+    begin
+      FileUtils.ln_s(rake, dest)
+    rescue Errno::EEXIST
+      puts "#{dest} already exists, do you want to override it? (y/n)"
+      if override_it?
+        FileUtils.rm(dest)
+        FileUtils.ln_s(rake, dest)
+        puts "Answer was 'yes', overriding the file."
+      else
+        puts "Answer was 'no', skipping this file."
+      end
+    end
+  end
+end
+
+def override_it?
+  while answer = gets
+    case answer
+    when "y\n"
+      return true
+    when "n\n"
+      return false
+    else
+      puts "I don't know what you mean, enter 'y' for yes or 'n' for no."
+    end
   end
 end
 

data/lib/fidius-cvedb/version.rb

@@ -1,5 +1,5 @@
 module FIDIUS
   module CveDb
-    VERSION = "0.0.6"
+    VERSION = "0.0.7"
   end
 end

data/lib/models/fidius/cve_db/cve_connection.rb

@@ -1,7 +1,13 @@
 class FIDIUS::CveDb::CveConnection < ActiveRecord::Base
+  self.abstract_class = true
+  
   # If the application is using the cve_db as foreign database this should be
   # uncommented.
   database_yml = YAML.load_file('config/database.yml')
+  unless database_yml['cve_db']
+    puts "WARNING: There is no \"cve_db\" entry in your database.yml, this "+
+         "will likely lead to an error."
+  end
   establish_connection database_yml['cve_db']
-  self.abstract_class = true
+  
 end

data/lib/tasks/nvd_migrate.rake

@@ -2,7 +2,7 @@ require 'yaml'
 
 namespace :nvd do 
   desc 'Execute NVD migrations'
-  task :migrate do
+  task :migrate => :environment do
     db_connect
     ActiveRecord::Migrator.migrate(File.join FIDIUS::CveDb::GEM_BASE, 'db', 'migrate')
   end

data/test/cve_parser_test.rb

@@ -5,6 +5,7 @@ $LOAD_PATH.unshift LIB_DIR
 require 'parser'
 require 'test/unit'
 
+# Tests the parsing funcionality of the NVD parser.
 class TestCveParser < Test::Unit::TestCase
   
   include FIDIUS::NVDParser

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 6
-  version: 0.0.6
+  - 7
+  version: 0.0.7
 platform: ruby
 authors: 
 - Andreas Bender
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-25 00:00:00 +01:00
+date: 2011-03-03 00:00:00 +01:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency