hunter2 0.0.1

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in hunter2.gemspec
+gemspec

data/MANIFEST

@@ -0,0 +1,16 @@
+Gemfile
+MANIFEST
+Rakefile
+bin/hunter2
+database.db
+lib/database.db
+lib/hunter2/command.rb
+lib/hunter2/model/password.rb
+lib/hunter2/spec/bacon/color_output.rb
+lib/hunter2/spec/helper.rb
+lib/hunter2/version.rb
+lib/hunter2.rb
+migrations/1313894524_passwords.rb
+task/build.rake
+task/db.rake
+task/test.rake

data/Rakefile

@@ -0,0 +1,7 @@
+require File.expand_path('../lib/hunter2', __FILE__)
+
+task_dir = File.expand_path('../task', __FILE__)
+
+Dir.glob("#{task_dir}/*.rake").each do |f|
+  import(f)
+end

data/bin/hunter2

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require File.expand_path('../../lib/hunter2', __FILE__)
+
+# Configure the database
+Hunter2.database = Sequel.connect(
+  :adapter   => 'sqlite',
+  :database  => '.hunter2.db',
+  :test      => false,
+  :encoding  => 'utf8'
+)
+
+Hunter2.init
+Shebang.run

data/lib/hunter2.rb

@@ -0,0 +1,39 @@
+require 'rubygems'
+require 'fast-aes'
+require 'shebang'
+require 'sequel'
+require 'base64'
+
+['version', 'command'].each do |file|
+  require File.expand_path("../hunter2/#{file}", __FILE__)
+end
+
+# Load AES
+key = '42#3b%c$dxyT,7a5=+5fUI3fa7352&^:'
+AES = FastAES.new(key)
+
+# Load all the required Sequel extensions/plugins
+Sequel.extension(:migration)
+Sequel::Model.plugin :validation_helpers, :timestamps
+
+module Hunter2
+  class << self
+    attr_accessor :database
+    attr_accessor :aes
+
+    def init
+      # Ensure we're connected to a database
+      if @database.nil?
+        $stderr.puts('No database connection has been set')
+        exit
+      end
+
+      # Load all the models after the database connection has been established.
+      Dir.glob(
+        File.expand_path('../hunter2/model/*.rb', __FILE__)
+      ).each do |f|
+        require(f)
+      end
+    end
+  end
+end

data/lib/hunter2/command.rb

@@ -0,0 +1,79 @@
+module Hunter2
+  class Command < ::Shebang::Command
+    command :default
+    banner  'Runs an example command.'
+    usage   '$ hunter2 [OPTIONS]'
+
+    o :h, :help    , 'Shows this help message'  , :method => :help
+    o :v, :version , 'Shows the current version', :method => :version
+
+    o :k, :key     , 'Key'     , :type => String
+    o :p, :password, 'Password', :type => String
+
+    def index
+
+    end
+
+    def add
+      if option(:p).nil?
+        puts "You need to enter a password."
+        exit
+      end
+
+      # Encrypt password using FastAES
+      encrypted_pass = AES.encrypt(option(:p))
+
+      # Save password and key to database
+      password = Hunter2::Model::Password.create_or_update(
+        :key      => option(:k),
+        :password => encrypted_pass
+      )
+
+      puts "Password #{option(:k)} successfully added. Use show -k "+
+        "#{option(:k)} to show your password."
+    end
+
+    def update
+      if option(:p).nil?
+        puts "You need to enter a password."
+        exit
+      end
+
+      # Encrypt password using FastAES
+      encrypted_pass = AES.encrypt(option(:p))
+
+      # Update password
+      password = Hunter2::Model::Password.filter(:key => option(:k)).limit(1)
+      password.update(:password => encrypted_pass)
+
+      puts "Password #{option(:k)} successfully updated."
+    end
+
+    def delete
+      password = Hunter2::Model::Password.filter(:key => option(:k)).limit(1)
+      password.delete
+
+      puts "Password #{option(:k)} successfully deleted."
+    end
+
+    def show
+      # Get encrypted password for this key
+      password = Hunter2::Model::Password.select(:password) \
+        .filter(:key => option(:k)) \
+        .limit(1) \
+        .single_value
+
+      # Decrypt password
+      password = AES.decrypt(password)
+
+      puts "Password for #{option(:key)}: #{password}"
+    end
+
+    protected
+
+    def version
+      puts Hunter2::Version
+      exit
+    end
+  end
+end

data/lib/hunter2/model/password.rb

@@ -0,0 +1,21 @@
+module Hunter2
+  module Model
+    class Password < Sequel::Model
+      plugin :timestamps, :created => :created_at
+
+      def self.create_or_update(params)
+        row = Password[params]
+
+        if row.nil?
+          return create(params)
+        else
+          return row.update(params)
+        end
+      end
+
+      def validate
+        validates_unique(:key)
+      end
+    end
+  end
+end

data/lib/hunter2/spec/bacon/color_output.rb

@@ -0,0 +1,39 @@
+#:nodoc:
+module Bacon
+  #:nodoc:
+  module ColorOutput
+    #:nodoc:
+    def handle_specification(name)
+      puts spaces + name
+      yield
+      puts if Counter[:context_depth] == 1
+    end
+
+    #:nodoc:
+    def handle_requirement(description)
+      error = yield
+
+      if !error.empty?
+        puts "#{spaces} \e[31m- #{description} [FAILED]\e[0m"
+      else
+        puts "#{spaces} \e[32m- #{description}\e[0m"
+      end
+    end
+
+    #:nodoc:
+    def handle_summary
+      print ErrorLog  if Backtraces
+      puts "%d specifications (%d requirements), %d failures, %d errors" %
+        Counter.values_at(:specifications, :requirements, :failed, :errors)
+    end
+
+    #:nodoc:
+    def spaces
+      if Counter[:context_depth] === 0
+        Counter[:context_depth] = 1
+      end
+
+      return ' ' * (Counter[:context_depth] - 1)
+    end
+  end # ColorOutput
+end # Bacon

data/lib/hunter2/spec/helper.rb

@@ -0,0 +1,65 @@
+require 'bacon'
+require 'stringio'
+require File.expand_path('../bacon/color_output', __FILE__)
+
+Bacon.extend(Bacon::ColorOutput)
+Bacon.summary_on_exit
+
+##
+# Runs the block in a new thread and redirects $stdout and $stderr. The output
+# normally stored in these variables is stored in an instance of StringIO which
+# is returned as a hash.
+#
+# @example
+#  out = catch_output do
+#    puts 'hello'
+#  end
+#
+#  puts out # => {:stdout => "hello\n", :stderr => ""}
+#
+# @author Yorick Peterse
+# @return [Hash]
+#
+def catch_output
+  data = {
+    :stdout => nil,
+    :stderr => nil
+  }
+
+  Thread.new do
+    $stdout, $stderr = StringIO.new, StringIO.new
+
+    yield
+
+    $stdout.rewind
+    $stderr.rewind
+
+    data[:stdout], data[:stderr] = $stdout.read, $stderr.read
+
+    $stdout, $stderr = STDOUT, STDERR
+  end.join
+
+  return data
+end
+
+##
+# Allows developers to create stubbed objects similar to Mocha's stub() method.
+#
+# @example
+#  obj = stub(:language => 'Ruby')
+#  puts obj.language # => "Ruby"
+#
+# @author Yorick Peterse
+# @param  [Hash] attributes A hash containing all the attributes to set and
+# their values.
+# @return [Class]
+#
+def stub(attributes)
+  obj = Struct.new(*attributes.keys).new
+
+  attributes.each do |k, v|
+    obj.send("#{k}=", v)
+  end
+
+  return obj
+end

data/lib/hunter2/version.rb

@@ -0,0 +1,3 @@
+module Hunter2
+  Version = '0.0.1'
+end

data/migrations/1313894524_passwords.rb

@@ -0,0 +1,16 @@
+Sequel.migration do
+  up do
+    create_table :passwords do
+      primary_key :id
+
+      String :key
+      String :password
+
+      Time :created_at
+    end
+  end
+
+  down do
+    drop_table(:passwords)
+  end
+end

data/task/build.rake

@@ -0,0 +1,55 @@
+require 'find'
+
+namespace :build do
+
+  desc 'Builds the documentation using YARD'
+  task :doc do
+    gem_path = File.expand_path('../../', __FILE__)
+    command  = "yard doc #{gem_path}/lib -m markdown -M rdiscount -o #{gem_path}/doc "
+    command += "-r #{gem_path}/README.md --private --protected "
+    command += "--files #{gem_path}/LICENSE"
+
+    sh(command)
+  end
+
+  desc 'Builds a new Gem and installs it'
+  task :gem do
+    gem_path = File.expand_path('../../', __FILE__)
+
+    # Build and install the gem
+    sh("gem build #{gem_path}/hunter2.gemspec")
+    sh("mv #{gem_path}/hunter2-#{Hunter2::Version}.gem #{gem_path}/pkg")
+    sh("gem install #{gem_path}/pkg/hunter2-#{Hunter2::Version}.gem")
+  end
+
+  desc 'Builds the MANIFEST file'
+  task :manifest do
+    gem_path     = File.expand_path('../../', __FILE__)
+    ignore_exts  = ['.gem', '.gemspec', '.swp']
+    ignore_files = ['.DS_Store', '.gitignore']
+    ignore_dirs  = ['.git', '.yardoc', 'spec', 'pkg', 'doc']
+    files        = ''
+    
+    Find.find(gem_path) do |f|
+      f[gem_path] = ''
+      f.gsub!(/^\//, '')
+
+      # Ignore directories
+      if !File.directory?(f) and !ignore_exts.include?(File.extname(f)) and !ignore_files.include?(File.basename(f))
+        files += "#{f}\n"
+      else
+        Find.prune if ignore_dirs.include?(f)
+      end
+    end
+    
+    # Time to write the MANIFEST file
+    begin
+      handle = File.open 'MANIFEST', 'w'
+      handle.write files.strip
+      puts "The MANIFEST file has been updated."
+    rescue
+      abort "The MANIFEST file could not be written."
+    end
+  end
+
+end

data/task/db.rake

@@ -0,0 +1,55 @@
+namespace :db do
+  desc 'Migrates the database to the most recent or the specified version'
+  task :migrate, :version do |task, args|
+    if !args[:version]
+      version = nil
+    else
+      version = args[:version].to_i
+    end
+
+    Sequel::Migrator.run(
+      Hunter2.database,
+      File.expand_path('../../migrations', __FILE__),
+      :target => version
+    )
+  end
+
+  desc 'Generates a new migration'
+  task :migration, :name do |task, args|
+    if !args[:name]
+      abort 'You need to specify a name for the migration'
+    end
+
+    # Generate the name of the migration
+    path = File.join(
+      File.expand_path('../../migrations', __FILE__),
+      "#{Time.new.to_i}_#{args[:name]}.rb"
+    )
+
+    if File.exist?(path)
+      abort "The migration #{path} already exists"
+    end
+
+    template = <<-TEMPLATE
+Sequel.migration do
+  up do
+
+  end
+
+  down do
+
+  end
+end
+TEMPLATE
+
+    # Write the migration
+    File.open(path, 'w') do |handle|
+      begin
+        handle.write(template.strip)
+        puts "Saved the migration in #{path}"
+      rescue => e
+        abort "Failed to write the migration: #{e.message}"
+      end
+    end
+  end
+end # namespace :db

data/task/test.rake

@@ -0,0 +1,6 @@
+desc 'Runs all the tests'
+task :test do
+  Dir.glob(File.expand_path('../../spec/hunter2/*.rb', __FILE__)).each do |f|
+    require(f)
+  end
+end

metadata

@@ -0,0 +1,147 @@
+--- !ruby/object:Gem::Specification 
+name: hunter2
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Abdelrahman Mahmoud
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-08-03 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: sequel
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 3.26.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: fast-aes
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.1.1
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: bcrypt-ruby
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 2.1.4
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.8.7
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 2.6.0
+  type: :development
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 0.7.1
+  type: :development
+  version_requirements: *id006
+- !ruby/object:Gem::Dependency 
+  name: rdiscount
+  prerelease: false
+  requirement: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.6.8
+  type: :development
+  version_requirements: *id007
+description: A CLI-based password manager in Ruby.
+email: abdel@aplusm.me
+executables: 
+- hunter2
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Gemfile
+- MANIFEST
+- Rakefile
+- bin/hunter2
+- database.db
+- lib/database.db
+- lib/hunter2/command.rb
+- lib/hunter2/model/password.rb
+- lib/hunter2/spec/bacon/color_output.rb
+- lib/hunter2/spec/helper.rb
+- lib/hunter2/version.rb
+- lib/hunter2.rb
+- migrations/1313894524_passwords.rb
+- task/build.rake
+- task/db.rake
+- task/test.rake
+has_rdoc: true
+homepage: https://github.com/abdelm/hunter2
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Kick-ass CLI-based password manager.
+test_files: []
+