mpw/lib/Cli.rb

#!/usr/bin/ruby
# author: nishiki
# mail: nishiki@yaegashi.fr
# info: a simple script who m your passwords

require 'rubygems'
require 'highline/import'
require 'pathname'

require "#{APP_ROOT}/lib/MPW.rb"

class Cli

	# Constructor
	def initialize()
		@m = MPW.new()
		
		if not @m.checkconfig()
			self.setup()
		end

		if not self.decrypt()
			puts "ERROR: #{@m.error_msg}"
			exit 2
		end
	end

	# Create a new config file
	def setup()
		puts "# Setup a new config file"
		puts "# --------------------"
		key         = ask("Enter the GPG key: ")
		file_gpg    = ask("Enter the path to encrypt file [default=#{Dir.home()}/.mpw.gpg]: ")
		timeout_pwd = ask("Enter the timeout (in seconde) to GPG password [default=60]: ")
		
		if @m.setup(key, file_gpg, timeout_pwd)
			puts "The config file has been created!"
		else
			puts "ERROR: #{@m.error_msg}"
		end
	end

	# Request the GPG password and decrypt the file
	def decrypt()
		@passwd = ask("Password GPG: ") {|q| q.echo = false}
		return @m.decrypt(@passwd)
	end

	# Display the query's result
	# @args: search -> the string to search
	#        protocol -> search from a particular protocol
	def display(search, protocol=nil, group=nil, format=nil)
		result = @m.search(search, group, protocol)

		if not result.empty?
			result.each do |r|
				if format.nil? || !format
					self.displayFormat(r)
				else
					self.displayFormatAlt(r)
				end
			end
		else
			puts "Nothing result!"	
		end
	end

	# Display an item in the default format
	# @args: item -> an array with the item information
	def displayFormat(item)
		puts "# --------------------"
		puts "# Id: #{item[MPW::ID]}"
		puts "# Name: #{item[MPW::NAME]}"
		puts "# Group: #{item[MPW::GROUP]}"
		puts "# Server: #{item[MPW::SERVER]}"
		puts "# Protocol: #{item[MPW::PROTOCOL]}"
		puts "# Login: #{item[MPW::LOGIN]}"
		puts "# Password: #{item[MPW::PASSWORD]}"
		puts "# Port: #{item[MPW::PORT]}"
		puts "# Comment: #{item[MPW::COMMENT]}"
	end

	# Display an item in the alternative format
	# @args: item -> an array with the item information
	def displayFormatAlt(item)
		item[MPW::PORT].nil? ? (port = '') : (port = ":#{item[MPW::PORT]}")

		if item[MPW::PASSWORD].nil? || item[MPW::PASSWORD].empty?
			if item[MPW::LOGIN].include('@')
				puts "# #{item[MPW::ID]} #{item[MPW::PROTOCOL]}://#{item[MPW::LOGIN]}@#{item[MPW::SERVER]}#{port}"
			else
				puts "# #{item[MPW::ID]} #{item[MPW::PROTOCOL]}://{#{item[MPW::LOGIN]}}@#{item[MPW::SERVER]}#{port}"
			end
		else
			puts "# #{item[MPW::ID]} #{item[MPW::PROTOCOL]}://{#{item[MPW::LOGIN]}:#{item[MPW::PASSWORD]}}@#{item[MPW::SERVER]}#{port}"
		end
	end

	# Form to add a new item
	def add()
		row = Array.new()
		puts "# Add a new item"
		puts "# --------------------"
		name     = ask("Enter the name: ")
		group    = ask("Enter the group [default=NoGroup]: ")
		server   = ask("Enter the hostname or ip: ")
		protocol = ask("Enter the protocol of the connection (ssh, http, other): ")
		login    = ask("Enter the login connection: ")
		passwd   = ask("Enter the the password: ")
		port     = ask("Enter the connection port (optinal): ")
		comment  = ask("Enter a comment (optinal): ")

		if @m.add(name, group, server, protocol, login, passwd, port, comment)
			if @m.encrypt()
				puts "Item has been added!"
			else
				puts "ERROR: #{@m.error_msg}"
			end
		else
			puts "ERROR: #{@m.error_msg}"
		end
	end

	# Update an item
	# @args: id -> the item's id
	def update(id)
		row = @m.searchById(id)

		if not row.empty?
			puts "# Update an item"
			puts "# --------------------"
			name     = ask("Enter the name [#{row[MPW::NAME]}]: ")
			group    = ask("Enter the group [#{row[MPW::GROUP]}]: ")
			server   = ask("Enter the hostname or ip [#{row[MPW::SERVER]}]: ")
			protocol = ask("Enter the protocol of the connection [#{row[MPW::PROTOCOL]}]: ")
			login    = ask("Enter the login connection [#{row[MPW::LOGIN]}]: ")
			passwd   = ask("Enter the the password: ")
			port     = ask("Enter the connection port [#{row[MPW::PORT]}]: ")
			comment  = ask("Enter a comment [#{row[MPW::COMMENT]}]: ")
				
			if @m.update(id, name, group, server, protocol, login, passwd, port, comment)
				if @m.encrypt()
					puts "Item has been updated!"
				else
					puts "ERROR: #{@m.error_msg}"
				end
			else
				puts "Nothing item has been updated!"
			end
		else
			puts "Nothing result!"
		end
	end

	# Remove an item
	# @args: id -> the item's id
	#        force -> no resquest a validation
	def remove(id, force=false)
		if not force
			result = @m.searchById(id)		

			if result.length > 0
				self.displayFormat(result)

				confirm = ask("Are you sure to remove the item: #{id} ? (y/N) ")
				if confirm =~ /^(y|yes|YES|Yes|Y)$/
					force = true
				end
			else
				puts "Nothing result!"
			end
		end

		if force
			if @m.remove(id)
				if @m.encrypt()
					puts "The item #{id} has been removed!"
				else
					puts "ERROR: #{@m.error_msg}"
				end
			else
				puts "Nothing item has been removed!"
			end
		end
	end

	# Export the items in a CSV file
	# @args: file -> the destination file
	def export(file)
		if @m.export(file)
			puts "The export in #{file} is succesfull!"
		else
			puts "ERROR: #{@m.error_msg}"
		end

	end

	# Import items from a CSV file
	# @args: file -> the import file
	#        force -> no resquest a validation
	def import(file, force=false)
		result = @m.importPreview(file)

		if not force
			if result.is_a?(Array) && !result.empty?
				result.each do |r|
					self.displayFormat(r)
				end

				confirm = ask("Are you sure to import this file: #{file} ? (y/N) ")
				if confirm =~ /^(y|yes|YES|Yes|Y)$/
					force = true
				end
			else
				puts "No data to import!"	
			end
		end

		if force
			if @m.import(file) && @m.encrypt()
				puts "The import is succesfull!"
			else
				puts "ERROR: #{@m.error_msg}"
			end
		end
	end

	# Interactive mode
	def interactive()
		group       = nil
		last_access = Time.now.to_i

		while true
			if @m.timeout_pwd < Time.now.to_i - last_access
				passwd_confirm = ask("Password GPG: ") {|q| q.echo = false}

				if @passwd.eql?(passwd_confirm)
					last_access = Time.now.to_i
				else
					puts 'Bad password!'
					next
				end
			else
				last_access = Time.now.to_i
			end

			command = ask("<mpw> ").split(' ')

			case command[0]
			when 'display', 'show', 'd', 's'
				if !command[1].nil? && !command[1].empty?
					self.display(command[1], group, command[2])
				end
			when 'add', 'a'
				add()
			when 'update', 'u'
				if !command[1].nil? && !command[1].empty?
					self.update(command[1])
				end
			when 'remove', 'delete', 'r', 'd'
				if !command[1].nil? && !command[1].empty?
					self.remove(command[1])
				end
			when 'group', 'g'
				if !command[1].nil? && !command[1].empty?
					group = command[1]
				else
					group = nil
				end
			when 'help', 'h', '?'
				puts '# Help'
				puts '# --------------------'
				puts '# Display an item:'
				puts '#	display SEARCH'
				puts '#	show SEARCH'
				puts '#	s SEARCH'
				puts '#	d SEARCH'
				puts '# Add an new item:'
				puts '#	add'
				puts '#	a'
				puts '# Update an item:'
				puts '#	update ID'
				puts '#	u ID'
				puts '# Remove an item:'
				puts '#	remove ID'
				puts '#	delete ID'
				puts '#	r ID'
				puts '#	d ID'
				puts '# Quit the program:'
				puts '#	quit'
				puts '#	exit'
				puts '#	q'
			when 'quit', 'exit', 'q'
				break
			else
				if !command[0].nil? && !command[0].empty?
					puts 'Unknow command!'
				end
			end

		end

	end
end