mpw/lib/mpw/cli.rb

#!/usr/bin/ruby
# MPW is a software to crypt and manage your passwords
# Copyright (C) 2016  Adrien Waksberg <mpw@yae.im>
# 
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

require 'readline'
require 'locale'
require 'i18n'
require 'colorize'
require 'highline/import'
require 'clipboard'
require 'tmpdir'
require 'mpw/item'
require 'mpw/mpw'

module MPW
class Cli

	# Constructor
	# @args: config -> the config
	#        sync -> boolean for sync or not
	def initialize(config, sync=true)
		@config    = config
		@sync      = sync
	end

	# Change a parameter int the config after init
	# @args: options -> param to change
	def set_config(options)
		gpg_key    = options[:gpg_key]    || @config.key
		lang       = options[:lang]       || @config.lang
		wallet_dir = options[:wallet_dir] || @config.wallet_dir
		gpg_exe    = options[:gpg_exe]    || @config.gpg_exe

		@config.setup(gpg_key, lang, wallet_dir, gpg_exe)
	rescue Exception => e
		puts "#{I18n.t('display.error')} #15: #{e}".red
		exit 2
	end

	# Create a new config file
	# @args: options -> set param
	def setup(options)
		lang = options[:lang] || Locale::Tag.parse(ENV['LANG']).to_simple.to_s[0..1]

		I18n.locale = lang.to_sym

		@config.setup(options[:gpg_key], lang, options[:wallet_dir], options[:gpg_exe])

		load_config

		puts "#{I18n.t('form.setup_config.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #8: #{e}".red
		exit 2
	end
	
	# Setup a new GPG key
	# @args: gpg_key -> the key name
	def setup_gpg_key(gpg_key)
		return if @config.check_gpg_key?

		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 
			raise I18n.t('form.setup_gpg_key.error_password')
		end

		@password = password.to_s 

		puts I18n.t('form.setup_gpg_key.wait')
		
		@config.setup_gpg_key(@password, gpg_key)

		puts "#{I18n.t('form.setup_gpg_key.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #8: #{e}".red
		exit 2
	end

	# Setup wallet config for sync
	# @args: options -> value to change
	def setup_wallet_config(options={})
		if not options[:password].nil?
			options[:password] = ask(I18n.t('form.setup_wallet.password')) {|q| q.echo = false}
		end

		#wallet_file = wallet.nil? ? "#{@config.wallet_dir}/default.mpw" : "#{@config.wallet_dir}/#{wallet}.mpw"

		@mpw = MPW.new(@config.key, @wallet_file, @password, @config.gpg_exe)
		@mpw.read_data
		@mpw.set_config(options)
		@mpw.write_data

		puts "#{I18n.t('form.setup_wallet.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #10: #{e}".red
		exit 2
	end
	
	# Load config
	def load_config
		@config.load_config

	rescue Exception => e
		puts "#{I18n.t('display.error')} #10: #{e}".red
		exit 2
	end

	# Request the GPG password and decrypt the file
	def decrypt
		if not defined?(@mpw)
			@password = ask(I18n.t('display.gpg_password')) {|q| q.echo = false}
			@mpw      = MPW.new(@config.key, @wallet_file, @password, @config.gpg_exe)
		end

		@mpw.read_data
		@mpw.sync if @sync
	rescue Exception => e
		puts "#{I18n.t('display.error')} #11: #{e}".red
		exit 2
	end

	# Format items on a table
	def table(items=[])
		group        = '.'
		i            = 1
		length_total = 10
		data         = { id:       { length: 3,  color: 'cyan' },
		                 host:     { length: 9,  color: 'yellow' },
		                 user:     { length: 7,  color: 'green' },
		                 protocol: { length: 9,  color: 'white' },
		                 port:     { length: 5,  color: 'white' },
		                 otp:      { length: 4,  color: 'white' },
		                 comment:  { length: 14, color: 'magenta' },
		               }

		items.each do |item|
			data.each do |k, v|
				next if k == :id or k == :otp

				v[:length] = item.send(k.to_s).length + 3 if item.send(k.to_s).to_s.length >= v[:length]
			end
		end
		data[:id][:length]  = items.length.to_s.length + 2 if items.length.to_s.length > data[:id][:length]
		
		data.each_value { |v| length_total += v[:length] }
		items.sort!     { |a, b| a.group.to_s.downcase <=> b.group.to_s.downcase }

		items.each do |item|
			if group != item.group
				group = item.group

				if group.to_s.empty?
					puts "\n#{I18n.t('display.no_group')}".red
				else
					puts "\n#{group}".red
				end

				print ' '
				length_total.times { print '=' }
				print "\n "
				data.each do |k, v|
					case k
					when :id
						print ' ID'
					when :otp
						print '| OTP'
					else
						print "| #{k.to_s.capitalize}"
					end

					(v[:length] - k.to_s.length).times { print ' ' }
				end
				print "\n "
				length_total.times { print '=' }
				print "\n"
			end

			print "  #{i}".send(data[:id][:color])
			(data[:id][:length] - i.to_s.length).times { print ' ' }
			data.each do |k, v|
				next if k == :id

				if k == :otp
					print '| '
					if item.otp;  print ' X  ' else 4.times { print ' ' } end

					next
				end

				print '| '
				print "#{item.send(k.to_s)}".send(v[:color])
				(v[:length] - item.send(k.to_s).to_s.length).times { print ' ' }
			end
			print "\n"

			i += 1
		end

		print "\n"
	end

	# Display the query's result
	# @args: options -> the option to search
	def list(options={})
		result = @mpw.list(options)

		if result.length == 0
			puts I18n.t('display.nothing')

		else
			table(result)
		end
	end

	# Get an item when multiple choice
	# @args: items -> array of items
	# @rtrn: item
	def get_item(items)
		return items[0] if items.length == 1

		items.sort! { |a,b| a.group.to_s.downcase <=> b.group.to_s.downcase }
		choice = ask(I18n.t('form.select')).to_i

		if choice >= 1 and choice <= items.length
			return items[choice-1]
		else
			return nil
		end
	end

	# Copy in clipboard the login and password
	# @args: item -> the item
	#        clipboard -> enable clipboard
	def clipboard(item, clipboard=true)
		pid = nil

		# Security: force quit after 90s
		Thread.new do
			sleep 90
			exit
		end
		
		while true
			choice = ask(I18n.t('form.clipboard.choice')).to_s
			
			case choice
			when 'q', 'quit'
				break

			when 'l', 'login'
				if clipboard
					Clipboard.copy(item.user)
					puts I18n.t('form.clipboard.login').green
				else
					puts item.user
				end

			when 'p', 'password'
				if clipboard
					Clipboard.copy(@mpw.get_password(item.id))
					puts I18n.t('form.clipboard.password').yellow

					Thread.new do
						sleep 30

						Clipboard.clear
					end
				else
					puts @mpw.get_password(item.id)
				end

			when 'o', 'otp'
				if clipboard
					Clipboard.copy(@mpw.get_otp_code(item.id))
				else
					puts @mpw.get_otp_code(item.id)
				end
				puts I18n.t('form.clipboard.otp', time: @mpw.get_otp_remaining_time).yellow

			else
				puts "----- #{I18n.t('form.clipboard.help.name')} -----".cyan
				puts I18n.t('form.clipboard.help.login')
				puts I18n.t('form.clipboard.help.password')
				puts I18n.t('form.clipboard.help.otp_code')
				puts I18n.t('form.clipboard.help.quit')
				next
			end
		end

		Clipboard.clear
	rescue SystemExit, Interrupt
		Clipboard.clear
	end

	# List all wallets
	def list_wallet
		wallets = Dir.glob("#{@config.wallet_dir}/*.mpw")

		wallets.each do |wallet|
			puts File.basename(wallet, '.mpw')
		end
	end

	# Display the wallet
	# @args: wallet -> the wallet name
	def get_wallet(wallet=nil)
		if wallet.to_s.empty?
			wallets = Dir.glob("#{@config.wallet_dir}/*.mpw")

			if wallets.length == 1
				@wallet_file = wallets[0]
			else
				@wallet_file = "#{@config.wallet_dir}/default.mpw"
			end
		else
			@wallet_file = "#{@config.wallet_dir}/#{wallet}.mpw"
		end
	end

	# Add a new public key
	# args: key -> the key name to add
	#       file -> gpg public file to import
	def add_key(key, file=nil)
		@mpw.add_key(key, file)
		@mpw.write_data
		@mpw.sync(true) if @sync

		puts "#{I18n.t('form.add_key.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #13: #{e}".red
	end

	# Add new public key
	# args: key -> the key name to delete
	def delete_key(key)
		@mpw.delete_key(key)
		@mpw.write_data
		@mpw.sync(true) if @sync

		puts "#{I18n.t('form.delete_key.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #15: #{e}".red
	end

	# Text editor interface
	# @args: template -> template name
	#        item -> the item to edit
	#        password -> disable field password
	def text_editor(template_name, item=nil, password=false)
		editor        = ENV['EDITOR'] || 'nano'
		options       = {}
		opts          = {}
		template_file = "#{File.expand_path('../../../templates', __FILE__)}/#{template_name}.erb"
		template      = ERB.new(IO.read(template_file))

		Dir.mktmpdir do |dir|
			tmp_file = "#{dir}/#{template_name}.yml"

			File.open(tmp_file, 'w') do |f|
				f << template.result(binding)
			end

			system("#{editor} #{tmp_file}")

			opts = YAML::load_file(tmp_file)
		end

		opts.delete_if { |k,v| v.to_s.empty? }

		opts.each do |k,v|
			options[k.to_sym] = v
		end

		return options
	end

	# Form to add a new item
	# @args: password -> generate a random password
	def add(password=false)
		options = text_editor('add_form', nil, password)	
		item    = Item.new(options)

		if password
			options[:password] = MPW::password(length: 24)
		end
		
		@mpw.add(item)
		@mpw.set_password(item.id, options[:password]) if options.has_key?(:password)
		@mpw.set_otp_key(item.id, options[:otp_key])   if options.has_key?(:otp_key)
		@mpw.write_data
		@mpw.sync(true) if @sync

		puts "#{I18n.t('form.add_item.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #13: #{e}".red
	end

	# Update an item
	# @args: options -> the option to search
	def update(options={})
		items = @mpw.list(options)
		
		if items.length == 0
			puts "#{I18n.t('display.warning')}: #{I18n.t('warning.select')}".yellow
		else
			table(items) if items.length > 1

			item    = get_item(items)
			options = text_editor('update_form', item)

			item.update(options)
			@mpw.set_password(item.id, options[:password]) if options.has_key?(:password)
			@mpw.set_otp_key(item.id, options[:otp_key])   if options.has_key?(:otp_key)
			@mpw.write_data
			@mpw.sync(true) if @sync

			puts "#{I18n.t('form.update_item.valid')}".green
		end
	rescue Exception => e
		puts "#{I18n.t('display.error')} #14: #{e}".red
	end

	# Remove an item
	# @args: options -> the option to search
	def delete(options={})
		items = @mpw.list(options)
		
		if items.length == 0
			puts "#{I18n.t('display.warning')}: #{I18n.t('warning.select')}".yellow
		else
			table(items)

			item    = get_item(items)
			confirm = ask("#{I18n.t('form.delete_item.ask')} (y/N) ").to_s
	
			if not confirm =~ /^(y|yes|YES|Yes|Y)$/
				return false
			end
	
			item.delete
			@mpw.write_data
			@mpw.sync(true) if @sync
	
			puts "#{I18n.t('form.delete_item.valid')}".green
		end
	rescue Exception => e
		puts "#{I18n.t('display.error')} #16: #{e}".red
	end

	# Copy a password, otp, login
	# @args: clipboard -> enable clipboard
	#        options -> the option to search
	def copy(clipboard=true, options={})
		items = @mpw.list(options)
		
		if items.length == 0
			puts "#{I18n.t('display.warning')}: #{I18n.t('warning.select')}".yellow
		else
			table(items)

			item = get_item(items)
			clipboard(item, clipboard)
		end
	rescue Exception => e
		puts "#{I18n.t('display.error')} #14: #{e}".red
	end

	# Export the items in a CSV file
	# @args: file -> the destination file
	#        options -> option to search
	def export(file, options)
		file  = 'export-mpw.yml' if file.to_s.empty?
		items = @mpw.list(options)
		data  = {}
		i     = 1

		items.each do |item|
			data.merge!(i => { 'host'      => item.host,
			                   'user'      => item.user,
			                   'group'     => item.group,
			                   'password'  => @mpw.get_password(item.id),
			                   'protocol'  => item.protocol,
			                   'port'      => item.port,
			                   'otp_key'   => @mpw.get_otp_key(item.id),
			                   'comment'   => item.comment,
			                   'last_edit' => item.last_edit,
			                   'created'   => item.created,
			                 }
			            )

			i += 1
		end

		File.open(file, 'w') {|f| f << data.to_yaml}

		puts "#{I18n.t('export.valid', file)}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #17: #{e}".red
	end

	# Import items from a YAML file
	# @args: file -> the import file
	def import(file)
		raise I18n.t('form.import.file_empty')     if file.to_s.empty?
		raise I18n.t('form.import.file_not_exist') if not File.exist?(file)

		YAML::load_file(file).each_value do |row|

			item = Item.new(group:    row['group'],
			                host:     row['host'],
			                protocol: row['protocol'],
			                user:     row['user'],
			                port:     row['port'],
			                comment:  row['comment'],
			               )

			next if item.empty?

			@mpw.add(item)
			@mpw.set_password(item.id, row['password']) if not row['password'].to_s.empty?
			@mpw.set_otp_key(item.id, row['otp_key'])   if not row['otp_key'].to_s.empty?
		end

		@mpw.write_data

		puts "#{I18n.t('form.import.valid')}".green
	rescue Exception => e
		puts "#{I18n.t('display.error')} #18: #{e}".red
	end
end
end