mpw/lib/UI/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 'readline'
require 'i18n'
require 'colorize'

require "#{APP_ROOT}/lib/MPW"

class Cli

	# Constructor
	# @args: lang -> the operating system language
	#        config_file -> a specify config file
	def initialize(config)
		@config = config
	end

	# Sync the data with the server
	# @rtnr: true if the synchro is finish
	def sync
		if not defined?(@sync)
			case @config.sync_type
			when 'mpw'
				require "#{APP_ROOT}/lib/Sync/MPWSync"
				@sync = MPW::Sync::MPWSync.new
			when 'sftp', 'scp', 'ssh'
				require "#{APP_ROOT}/lib/Sync/SSH"
				@sync = MPW::Sync::SSH.new
			when 'ftp'
				require "#{APP_ROOT}/lib/Sync/FTP"
				@sync = MPW::Sync::FTP.new
			else
				return false
			end
		end

		if  not @config.sync_host.nil? and not @config.sync_port.nil?
			if not @sync.connect(@config.sync_host, @config.sync_user, @config.sync_pwd, @config.sync_path, @config.sync_port)
				puts "#{I18n.t('display.error')} #1: #{@sync.error_msg}".red
			end
		end

		if @sync.enable
			if not @mpw.sync(@sync.get(@config.key, @passwd), @config.last_update)
				puts "#{I18n.t('display.error')} #2: #{@mpw.error_msg}".red  if not @mpw.error_msg.nil?
				puts "#{I18n.t('display.error')} #3: #{@sync.error_msg}".red if not @sync.error_msg.nil?

			elsif not @sync.update(File.open(@config.file_gpg).read)
				puts "#{I18n.t('display.error')} #4: #{@sync.error_msg}".red

			elsif not @config.set_last_update
				puts "#{I18n.t('display.error')} #5: #{@config.error_msg}".red

			elsif not @mpw.encrypt
				puts "#{I18n.t('display.error')} #6: #{@mpw.error_msg}".red

			else
				return true
			end
		end
	rescue Exception => e
		puts "#{I18n.t('display.error')} #7: #{e}".red
		puts @sync.error_msg   if not @sync.error_msg.nil?
		puts @config.error_msg if not @config.error_msg.nil?
		puts @mpw.error_msg    if not @mpw.error_msg.nil?
	else
		return false
	end

	# Create a new config file
	# @args: lang -> the software language
	def setup(lang)
		puts I18n.t('form.setup.title')
		puts '--------------------'
		language    = ask(I18n.t('form.setup.lang', lang: lang)).to_s
		key         = ask(I18n.t('form.setup.gpg_key')).to_s
		share_keys  = ask(I18n.t('form.setup.share_gpg_keys')).to_s
		file_gpg    = ask(I18n.t('form.setup.gpg_file', home: @conf.dir_config)).to_s
		sync_type   = ask(I18n.t('form.setup.sync_type')).to_s

		if ['ssh', 'ftp', 'mpw'].include?(sync_type)
			sync_host   = ask(I18n.t('form.setup.sync_host')).to_s
			sync_port   = ask(I18n.t('form.setup.sync_port')).to_s
			sync_user   = ask(I18n.t('form.setup.sync_user')).to_s
			sync_pwd    = ask(I18n.t('form.setup.sync_pwd')).to_s
			sync_path   = ask(I18n.t('form.setup.sync_path')).to_s
		end
		
		if language.nil? or language.empty?
			language = lang
		end
		I18n.locale = language.to_sym

		sync_type = sync_type.nil? or sync_type.empty? ? nil : sync_type
		sync_host = sync_host.nil? or sync_host.empty? ? nil : sync_host
		sync_port = sync_port.nil? or sync_port.empty? ? nil : sync_port.to_i
		sync_user = sync_user.nil? or sync_user.empty? ? nil : sync_user
		sync_pwd  = sync_pwd.nil?  or sync_pwd.empty?  ? nil : sync_pwd
		sync_path = sync_path.nil? or sync_path.empty? ? nil : sync_path

		if @config.setup(key, share_keys, language, file_gpg, sync_type, sync_host, sync_port, sync_user, sync_pwd, sync_path)
			puts "#{I18n.t('form.setup.valid')}".green
		else
			puts "#{I18n.t('display.error')} #8: #{@config.error_msg}".red
			exit 2
		end

		if not @config.checkconfig
			puts "#{I18n.t('display.error')} #9: #{@config.error_msg}".red
			exit 2
		end
	end

	# Setup a new GPG key
	def setup_gpg_key
		puts I18n.t('form.setup_gpg_key.title')
		puts '--------------------'
		ask      = ask(I18n.t('form.setup_gpg_key.ask')).to_s
		
		if not ['Y', 'y', 'O', 'o'].include?(ask)
			puts I18n.t('form.setup_gpg_key.no_create')
			exit 2
		end

		name     = ask(I18n.t('form.setup_gpg_key.name')).to_s
		password = ask(I18n.t('form.setup_gpg_key.password')) {|q| q.echo = false}
		confirm  = ask(I18n.t('form.setup_gpg_key.confirm_password')) {|q| q.echo = false}

		if password != confirm 
			puts I18n.t('form.setup_gpg_key.error_password')
			exit 2
		end

		length   = ask(I18n.t('form.setup_gpg_key.length')).to_s
		expire   = ask(I18n.t('form.setup_gpg_key.expire')).to_s
		password = password.to_s

		length = length.nil? or length.empty? ? 2048 : length.to_i
		expire = expire.nil? or expire.empty? ? 0    : expire.to_i

		puts I18n.t('form.setup_gpg_key.wait')
		
		if @config.setup_gpg_key(password, name, length, expire)
			puts "#{I18n.t('form.setup_gpg_key.valid')}".green
		else
			puts "#{I18n.t('display.error')} #10: #{@config.error_msg}".red
			exit 2
		end
	end

	# Request the GPG password and decrypt the file
	def decrypt
		if not defined?(@mpw)
			@mpw = MPW::MPW.new(@config.file_gpg, @config.key, @config.share_keys)
		end

		@passwd = ask(I18n.t('display.gpg_password')) {|q| q.echo = false}
		if not @mpw.decrypt(@passwd)
			puts "#{I18n.t('display.error')} #11: #{@mpw.error_msg}".red
			exit 2
		end
	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)
		result = @mpw.search(search, group, protocol)

		case result.length
		when 0
			puts I18n.t('display.nothing')
		when 1
			display_item(result.first)
		else
			i = 1
			result.each do |r|
				print "#{i}: ".cyan
				print r['name']
				print " -> #{r['comment']}".magenta if not r['comment'].to_s.empty?
				print "\n"

				i += 1
			end
			choice = ask(I18n.t('form.select')).to_i

			if choice >= 1 and choice < i 
				display_item(result[choice-1])
			else
				puts "#{I18n.t('display.warning')}: #{I18n.t('warning.select')}".yellow
			end
		end
	end

	# Display an item in the default format
	# @args: item -> an array with the item information
	def display_item(item)
		puts '--------------------'.cyan
		print 'Id: '.cyan
		puts  item['id']
		print "#{I18n.t('display.name')}: ".cyan
		puts  item['name']
		print "#{I18n.t('display.group')}: ".cyan
		puts  item['group']
		print "#{I18n.t('display.server')}: ".cyan
		puts  item['host']
		print "#{I18n.t('display.protocol')}: ".cyan
		puts  item['protocol']
		print "#{I18n.t('display.login')}: ".cyan
		puts  item['login']
		print "#{I18n.t('display.password')}: ".cyan
		puts  item['password']
		print "#{I18n.t('display.port')}: ".cyan
		puts  item['port']
		print "#{I18n.t('display.comment')}: ".cyan
		puts  item['comment']
	end

	# Form to add a new item
	def add
		row = []
		puts I18n.t('form.add.title')
		puts '--------------------'
		name     = ask(I18n.t('form.add.name')).to_s
		group    = ask(I18n.t('form.add.group')).to_s
		server   = ask(I18n.t('form.add.server')).to_s
		protocol = ask(I18n.t('form.add.protocol')).to_s
		login    = ask(I18n.t('form.add.login')).to_s
		passwd   = ask(I18n.t('form.add.password')).to_s
		port     = ask(I18n.t('form.add.port')).to_s
		comment  = ask(I18n.t('form.add.comment')).to_s

		if @mpw.update(name, group, server, protocol, login, passwd, port, comment)
			if @mpw.encrypt
				sync
				puts "#{I18n.t('form.add.valid')}".green
			else
				puts "#{I18n.t('display.error')} #12: #{@mpw.error_msg}".red
			end
		else
			puts "#{I18n.t('display.error')} #13: #{@mpw.error_msg}".red
		end
	end

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

		if not row.empty?
			puts I18n.t('form.update.title')
			puts '--------------------'
			name     = ask(I18n.t('form.update.name'    , name:     row['name'])).to_s
			group    = ask(I18n.t('form.update.group'   , group:    row['group'])).to_s
			server   = ask(I18n.t('form.update.server'  , server:   row['host'])).to_s
			protocol = ask(I18n.t('form.update.protocol', protocol: row['protocol'])).to_s
			login    = ask(I18n.t('form.update.login'   , login:    row['login'])).to_s
			passwd   = ask(I18n.t('form.update.password')).to_s
			port     = ask(I18n.t('form.update.port'    , port:     row['port'])).to_s
			comment  = ask(I18n.t('form.update.comment' , comment:  row['comment'])).to_s
				
			if @mpw.update(name, group, server, protocol, login, passwd, port, comment, id)
				if @mpw.encrypt
					sync
					puts "#{I18n.t('form.update.valid')}".green
				else
					puts "#{I18n.t('display.error')} #14: #{@mpw.error_msg}".red
				end
			else
				puts "#{I18n.t('display.error')} #15: #{@mpw.error_msg}".red
			end
		else
			puts I18n.t('display.nothing')
		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 = @mpw.search_by_id(id)

			if result.length > 0
				display_item(result)

				confirm = ask("#{I18n.t('form.delete.ask', id: id)} (y/N) ").to_s
				if confirm =~ /^(y|yes|YES|Yes|Y)$/
					force = true
				end
			else
				puts I18n.t('display.nothing')
			end
		end

		if force
			if @mpw.remove(id)
				if @mpw.encrypt
					sync
					puts "#{I18n.t('form.delete.valid', id: id)}".green
				else
					puts "#{I18n.t('display.error')} #16: #{@mpw.error_msg}".red
				end
			else
				puts I18n.t('form.delete.not_valid')
			end
		end
	end

	# Export the items in a CSV file
	# @args: file -> the destination file
	def export(file, type=:yaml)
		if @mpw.export(file, type)
			puts "#{I18n.t('export.valid', file)}".green
		else
			puts "#{I18n.t('display.error')} #17: #{@mpw.error_msg}".red
		end

	end

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

		if not force
			result = @mpw.import_preview(file, type)
			if result.is_a?(Array) and not result.empty?
				result.each do |r|
					display_item(r)
				end

				confirm = ask("#{I18n.t('form.import.ask', file: file)} (y/N) ").to_s
				if confirm =~ /^(y|yes|YES|Yes|Y)$/
					force = true
				end
			else
				puts I18n.t('form.import.not_valid')
			end
		end

		if force
			if @mpw.import(file, type) and @mpw.encrypt
				sync
				puts "#{I18n.t('form.import.valid')}".green
			else
				puts "#{I18n.t('display.error')} #18: #{@mpw.error_msg}".red
			end
		end
	end

end