shh 0.0.1 → 0.0.2

data/bin/shh

@@ -2,4 +2,4 @@
 
 $:.unshift File.dirname(__FILE__)+'/../lib'
 require 'shh'
-Shh::Cli.execute(ARGV)
+Shh::Cli.new.execute(*ARGV)

data/lib/shh/cli.rb

@@ -1,11 +1,83 @@
 require 'rubygems'
+require 'pathname2'
 require 'highline/import'
+require 'shh/crypt'
+require 'yaml'
 
 module Shh
   class Cli
-    def self.execute args
-      passphrase = ask('enter your passphrase') { |q| q.echo = false }
-      
+    def execute *args
+      passphrase = ask('Enter your passphrase') { |q| q.echo = false }
+
+      @folder = Pathname.new('secret')
+      @folder.mkdir_p
+      @crypt = Crypt.new(passphrase)
+
+      loop do
+        choose do |menu|
+          menu.layout = :menu_only
+          menu.shell  = true
+          menu.choice('list entries') { list }
+          menu.choice('create entry') { |command, details| create details }
+          menu.choice('edit entry') { |command, details| edit details }
+          menu.choice('view entry') { |command, details| view details }
+          menu.choice('quit') { exit }
+        end
+      end
+    end
+
+    def list
+      @folder.entries.each do |child|
+        say(child) unless ['.','..'].include?(child.to_s)
+      end
+    end
+
+    def create name=''
+      persist_entry check_name(name), prompt_loop({})
+    end
+
+    def edit name=''
+      name = check_name(name)
+      persist_entry name, prompt_loop(load_entry(name))
+    end
+
+    def view name=''
+      prompt_loop load_entry(check_name(name)), true
+    end
+
+private
+
+    def check_name name
+      name = ask('Enter the entry name') unless name.size > 0
+      name
+    end
+
+    def load_entry name
+      entry = (@folder + name).open('r') {|io| @crypt.decrypt(io) }
+      YAML::load(entry)
+    end
+
+    def persist_entry name, entry
+      (@folder + name).open('w') {|io| @crypt.encrypt(entry.to_yaml, io) }
+    end
+
+    def prompt_loop hash, read_only=false
+      loop do
+        choose do |menu|
+          menu.layout = :menu_only
+          menu.shell  = true
+          menu.choice('create key') { |command, name| hash[name] = new_value(name) } unless read_only
+          menu.choice('list keys') { say(hash.keys.sort.join(',')) }
+          menu.choice('show key') { |command, name| say(hash[name]) if hash[name] }
+          menu.choice('delete key') { |command, name| hash[name] = nil } unless read_only
+          menu.choice('quit') { return hash }
+        end
+      end
+    end
+
+    def new_value name
+      echo = (name =~ /pass/) ? false : true
+      ask("Enter new value for #{name}") {|q| q.echo = echo }
     end
   end
 end

data/lib/shh/crypt.rb

@@ -0,0 +1,25 @@
+require 'crypt/blowfish'
+
+module Shh
+  class Crypt
+    def initialize passphrase
+      @blowfish = ::Crypt::Blowfish.new(passphrase)
+    end
+
+    def encrypt text, out_io
+      in_io = StringIO.new(text)
+      while l = in_io.read(8) do
+        while l.size < 8 do l += "\0" end
+        out_io.print @blowfish.encrypt_block(l)
+      end
+    end
+
+    def decrypt in_io
+      out_io = StringIO.new
+      while l = in_io.read(8) do
+        out_io.print @blowfish.decrypt_block(l)
+      end
+      out_io.string.gsub("\0",'')
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: shh
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors: 
 - Mark Ryall
@@ -32,6 +32,26 @@ dependencies:
       - !ruby/object:Gem::Version 
         version: 2.1.0
     version: 
+- !ruby/object:Gem::Dependency 
+  name: crypt
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.1.4
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: pathname2
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: 1.6.3
+    version: 
 description: |
   A command line utility that manages accounts and passwords as individual encypted files
 
@@ -44,6 +64,7 @@ extra_rdoc_files: []
 
 files: 
 - lib/shh/cli.rb
+- lib/shh/crypt.rb
 - lib/shh.rb
 - bin/shh
 - README.rdoc