dragons_keep 0.5.0

data/Gemfile

@@ -0,0 +1,29 @@
+source "http://rubygems.org"
+
+# ruby gems needed for Dragon's Keep
+gem 'ezcrypto'
+gem 'uuid'
+# using datamapper instead of activerecord
+# Active record Gems
+#gem 'activerecord', "3.0.0.beta3"
+#gem 'sqlite3-ruby'
+#
+# Data Mapper gems 0.10.2
+gem 'dm-core', "0.10.2"
+gem 'do_sqlite3', '0.10.2'
+
+# Data Mapper gems 1.0.0
+#gem 'data_mapper', "1.0.0"
+#gem 'dm-sqlite-adapter', "1.0.0"
+
+# Wxruby renames gem for 1.9.1 support
+if (RUBY_PLATFORM =~ /linux$/) == nil
+  #Decide which gem to load based on version of ruby installed
+  if (RUBY_VERSION =~/^1.9/) != nil
+    gem 'wxruby-ruby19'
+  else
+    gem 'wxruby'
+  end
+else
+  gem 'wxruby', "2.0.1"
+end

data/LICENSE

@@ -0,0 +1,3 @@
+== dragons_keep
+
+Put appropriate LICENSE for your project here.

data/README

@@ -0,0 +1,3 @@
+== dragons_keep
+
+You should document your project here.

data/Rakefile

@@ -0,0 +1,59 @@
+# 
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+ 
+
+require 'rubygems'
+require 'rake'
+require 'rake/clean'
+require 'rake/gempackagetask'
+require 'rake/rdoctask'
+require 'rake/testtask'
+require 'bundler'
+
+
+spec = Gem::Specification.new do |s|
+  s.name = 'dragons_keep'
+  s.version = '0.5.0'
+  s.has_rdoc = true
+  s.extra_rdoc_files = ['README', 'LICENSE']
+  s.summary = 'Secure Password Keeper Application'
+  s.description = s.summary
+  s.author = 'Allan Davis'
+  s.email = 'javaalley@gmail.com'
+  s.homepage = 'http://github.com/javaalley/dragons_keep'
+  s.executables = ['dragons_keep']
+  s.files = %w(LICENSE README Rakefile Gemfile) + Dir.glob("{bin,lib,spec,db}/**/*")
+  s.require_path = "lib"
+  s.bindir = "bin"
+  s.add_bundler_dependencies
+end
+
+Rake::GemPackageTask.new(spec) do |p|
+  p.gem_spec = spec
+  p.need_tar = true
+  p.need_zip = true
+end
+
+Rake::RDocTask.new do |rdoc|
+  files =['README', 'LICENSE', 'lib/**/*.rb']
+  rdoc.rdoc_files.add(files)
+  rdoc.main = "README" # page to start on
+  rdoc.title = "dragons_keep Docs"
+  rdoc.rdoc_dir = 'doc/rdoc' # rdoc output folder
+  rdoc.options << '--line-numbers'
+end
+
+Rake::TestTask.new do |t|
+  t.test_files = FileList['test/**/*.rb']
+end
+
+#desc "Migrate the database through migrations scripts"
+#task :migrate => :enviroment do
+#  ActiveRecord::Migrator.migrate('db/migrations', ENV["VERSION"]? ENV["VERSION"].to_i : nil)
+#end
+#
+#task :enviroment do
+#  ActiveRecord::Base.establish_connection(YAML::load(File.open('config/database.yml')))
+#  ActiveRecord::Base.logger = Logger.new(File.open('database.log', 'a'))
+#end

data/bin/dragons_keep

@@ -0,0 +1,17 @@
+#! /usr/bin/env ruby
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+require 'rubygems'
+begin
+  require 'bundler'
+  Bundler.setup
+rescue 
+end
+
+# Set the path to load from the lib and main directory
+path = File.expand_path(File.dirname(__FILE__))
+$: << path
+$: << File.join(path, "..", "lib")
+
+load "dragons_keep.rb"
+

data/bin/dragons_keep.bat

@@ -0,0 +1 @@
+@"ruby.exe" "%~dpn0" %*

data/db/migrations/201005101750_create_accounts.rb

@@ -0,0 +1,19 @@
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+
+class CreateAccounts < ActiveRecord::Migration
+  def self.up
+    create_table :accounts do |t|
+      t.string :name
+      t.string :password
+      t.string :salt
+      t.string :url
+      t.string :user_name
+      
+      t.timestamps
+    end
+  end
+  def self.down
+    drop_table :accounts
+  end
+end

data/lib/dragons_keep.rb

@@ -0,0 +1,33 @@
+#! /usr/etc ruby
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+require 'rubygems'
+#require 'bundler'
+#Bundler.setup
+
+# Set the path to load from the lib and main directory
+path = File.expand_path(File.dirname(__FILE__))
+$LOAD_PATH.insert 0, path
+#puts "LOAD_PATH"
+#puts $LOAD_PATH
+#$: << File.join(path, "..", "lib")
+
+require 'wx'
+require 'dragons_keep/keeps_main'
+require 'ezcrypto'
+require 'uuid'
+
+module DragonsKeep
+
+  class DragonsKeepApp < Wx::App
+    def on_init()
+      self.app_name = "Dragon's Keep"
+      @frame = KeepsMain.new "Dragon's Keep"
+      @frame.center_on_screen(Wx::BOTH)
+      @frame.show
+    end
+  end
+end
+
+dk = DragonsKeep::DragonsKeepApp.new
+dk.main_loop

data/lib/dragons_keep/account.rb

@@ -0,0 +1,99 @@
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+require 'dm-core'
+require 'ezcrypto'
+module DragonsKeep
+  # Account class to store accounts and passwords into a keeper system
+  class Account 
+    include DataMapper::Resource
+    #Data Mapper properties
+    property :id, Serial
+    property :name, String, :length => 100
+    property :password, String, :length => 100
+    property :salt, String, :length => 100
+    property :user_name, String, :length => 100
+    property :url, String, :length => 100
+
+    #register callbacks
+    before :save, :before_create
+    
+    
+    # Transient storage of unencrypted password and confirmation field
+    attr_accessor :unencrypted_password
+    attr_reader :password_confirmation
+
+    # create writer for password confirmation
+    def password_confirmation=(value)
+      @password_confirmation = value
+      @unencrypted = true
+    end
+
+    # loading data validate password filed and check status
+    def after_initialize   
+      @unencrypted = self.password.nil?
+    end
+
+    # Before create data set salt and encrypt password
+    def before_create
+      raise PasswordException, "Password is not Encrypted"  if self.new_password? && self.unencrypted?
+    end
+
+    # are we re defining the account password
+    def new_password?()
+      return !(self.unencrypted_password.blank? and self.password_confirmation.blank?)
+    end
+    
+    # Encrypt the password
+    # encrypt_pass = Given password when user created the account information
+    def encrypt_password(encrypt_pass)
+      if self.unencrypted? && self.unencrypted_password == self.password_confirmation
+        self.create_salt
+        self.password = Base64.encode64(EzCrypto::Key.encrypt_with_password(encrypt_pass, self.salt, self.unencrypted_password))
+        @unencrypted = false
+      end
+    end
+    
+    # Decrypt the password 
+    # encrypt_pass = Given password when user created the account information
+    def decrpyt_password(encrypt_pass)
+      if !(self.unencrypted?)
+        self.unencrypted_password = EzCrypto::Key.decrypt_with_password(encrypt_pass, self.salt, Base64.decode64( self.password))
+        @unencrypted = true
+        self.password_confirmation=""
+      end
+    end
+    
+    # Is the password encrypted?
+    def unencrypted?      
+      return (@unencrypted.nil?)? self.password.nil?: @unencrypted
+    end
+    
+    # Generate a random password
+    # length_of_password = Length of the password to generate
+    # special_char = Can this password need to contain Special Characters
+    def generate_password( length_of_pass, special_char )
+      chars = []
+      ("a".."z").each {|ele| chars << ele}
+      ("A".."Z").each {|ele| chars << ele}
+      ("0".."9").each {|ele| chars << ele}
+      if(special_char)
+        ["@", "!", "_",].each {|ele| chars << ele}
+      end
+      newpass = ""
+      1.upto(length_of_pass) { |i| newpass << chars[rand(chars.size-1)] }
+      #self.password
+      self.unencrypted_password = newpass
+      self.password_confirmation = newpass
+      @unencrypted = true
+    end
+    
+    # Create the salt
+    def create_salt
+      self.salt = UUID.generate(:compact)
+    end
+  end
+
+  # Error on Passwords Encryption or validation
+  class PasswordException < RuntimeError    
+  end
+end

data/lib/dragons_keep/account_controller.rb

@@ -0,0 +1,85 @@
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+require 'dragons_keep/account'
+require 'dm-core'
+require 'ezcrypto'
+require 'uuid'
+require 'digest/sha1'
+
+module DragonsKeep
+  class AccountController
+    attr_reader :encrypt_pass
+    attr_accessor :database
+
+    def encrypt_pass=(pass)
+      @encrypt_pass = Digest::SHA1.hexdigest(pass)
+    end
+
+    def establish_connection
+      migrate = false
+      if !(File.exist?(self.database))
+        migrate = true
+      end
+     
+      #DataMapper::Logger.new $stdout, :debug
+
+      DataMapper.setup :default, "sqlite3://#{self.database}"
+      if migrate
+        DataMapper.auto_migrate!
+      else
+        validate_connection
+      end
+      @connection = true
+    end
+    # Check the first record in the database and decrypt the password to see if the password is valid
+    def validate_connection
+      account = Account.first
+      if ! account.nil?
+        self.decrypt!(account)
+      end
+    end
+
+    def initialize(database=nil, password=nil)
+      @connection = false
+      if(!(database.nil?))
+        self.database = File.expand_path database
+      end
+      if (!(password.nil?))
+        self.encrypt_pass = password
+      end
+    end
+
+    def list()
+      if @connection
+        return Account.all
+      end
+      
+    end
+    def save!(account)
+      if @connection
+        if account.unencrypted?
+          account.encrypt_password(self.encrypt_pass)
+        end
+        return account.save()
+      end      
+    end
+
+    def get(id)
+      if @connection
+        return Account.get id
+      end
+    end
+
+    def decrypt!(account)
+      begin
+        account.decrpyt_password self.encrypt_pass
+      rescue OpenSSL::Cipher::CipherError
+        raise PasswordException, "Password is invalid"
+      end
+    end
+
+    def delete(account)
+      account.destroy
+    end
+  end
+end

data/lib/dragons_keep/account_dialog.rb

@@ -0,0 +1,103 @@
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+require 'wx'
+require 'dragons_keep/generate_password_dialog'
+
+module DragonsKeep
+  class AccountDialog < Wx::Dialog
+    ID_USER_NAME = 103
+    ID_ACCOUNT_NAME = 101
+    ID_URL = 102
+    ID_PASSWORD = 104
+    ID_GENERATE_PASSWORD = 105
+    ID_SAVE = 106
+    ID_COPY_CLIP = 107
+    ID_CANCEL = 108
+
+    def initialize parent, id, title
+      super parent, id, title
+      self.set_size(Wx::Size.new(400, 300))
+      main_sizer = Wx::BoxSizer.new(Wx::VERTICAL)
+      grid_sizer = Wx::FlexGridSizer.new(4,2,5,5)
+
+      # Create name entry      
+      label = Wx::StaticText.new(self, :label=>"Name:")
+      grid_sizer.add label, 0,  Wx::ALL |Wx::ALIGN_RIGHT
+      @name = Wx::TextCtrl.new self, ID_ACCOUNT_NAME
+      grid_sizer.add @name, 1, Wx::EXPAND
+      
+      # Create url entry      
+      label = Wx::StaticText.new(self, :label=>"URL:")
+      grid_sizer.add label, 0,Wx::ALL |Wx::ALIGN_RIGHT
+      @url = Wx::TextCtrl.new self, ID_URL
+      grid_sizer.add @url, 1, Wx::EXPAND
+      
+      # Create User_name entry      
+      label = Wx::StaticText.new(self, :label=>"User Name:")
+      grid_sizer.add label, 0, Wx::ALL |Wx::ALIGN_RIGHT
+      @user = Wx::TextCtrl.new self, ID_USER_NAME
+      grid_sizer.add @user, 1, Wx::EXPAND
+      
+      # Create password entry      
+      label = Wx::StaticText.new(self, :label=>"Password:")
+      grid_sizer.add label, 0, Wx::ALL |Wx::ALIGN_RIGHT
+      @pass = Wx::TextCtrl.new self, ID_PASSWORD, :style=> Wx::TE_PASSWORD
+      grid_sizer.add @pass, 1, Wx::EXPAND
+      main_sizer.add grid_sizer, 0, Wx::GROW|Wx::ALIGN_CENTER_VERTICAL|Wx::ALL, 5
+      grid_sizer.add_growable_col(1)
+      # add spacer to grid
+      grid_sizer.add(25, 25)
+      # Add generate password button
+      @gen_button = Wx::Button.new self, ID_GENERATE_PASSWORD, "Generate Password..."
+      grid_sizer.add @gen_button, 1, Wx::EXPAND
+      evt_button(ID_GENERATE_PASSWORD){|evt| self.gen_pass_click(evt)} 
+      main_sizer.add self.create_separated_button_sizer(Wx::OK|Wx::CANCEL), 1,  Wx::ALIGN_RIGHT
+      self.set_sizer main_sizer
+
+    end
+
+    def gen_pass_click(event)
+      # display generate pass dialog
+      gen_dialog = GeneratePasswordDialog.new self, -1, "Generate Password"
+      if gen_dialog.show_modal() == Wx::ID_OK
+        @account.generate_password gen_dialog.password_length, gen_dialog.use_special_chars?
+        @pass.value = @account.unencrypted_password
+      end      
+    end
+
+    def account
+      save_account
+      @account
+    end
+
+    def account=(account)
+      @account = account
+      load_account
+    end
+
+    private
+    def save_account
+      if not @account.blank?
+        if @account.unencrypted_password != @pass.value && @account.password_confirmation.blank?
+           password_dialog = Wx::PasswordEntryDialog.new(self, "Please confirm the password")
+           if password_dialog.show_modal == Wx::ID_OK
+             @account.password_confirmation = password_dialog.get_value
+             @account.unencrypted_password = @pass.value
+           end           
+        end
+        @account.name = @name.value
+        @account.user_name = @user.value
+        @account.url = @url.value
+      end
+    end
+    def load_account
+      if not @account.name.blank?
+        @name.value = @account.name
+        @url.value = @account.url
+        @user.value = @account.user_name
+        @pass.value = @account.unencrypted_password.blank? ? @account.password : @account.unencrypted_password
+      end
+    end
+
+  end
+end