nishiki/mpw
1
0
Fork 0
mirror of https://github.com/nishiki/manage-password.git synced 2024-11-25 06:43:05 +00:00
Code Issues Releases Activity
mpw/MPW/UI/Cli.rb

415 lines
12 KiB
Ruby
Raw Blame History

#!/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 'yaml'
require "#{APP_ROOT}/MPW/MPW"
require "#{APP_ROOT}/MPW/Sync/MPWSync"
require "#{APP_ROOT}/MPW/Sync/SSH"
require "#{APP_ROOT}/MPW/Sync/FTP"
class Cli
# Constructor
# @args: lang -> the operating system language
# config_file -> a specify config file
def initialize(lang, config)
@config = config
end
# Sync the data with the server
# @rtnr: true if the synchro is finish
def sync
if !defined?(@sync)
case @config.sync_type
when 'mpw'
@sync = MPW::Sync::MPWSync.new
when 'sftp', 'scp', 'ssh'
@sync = MPW::Sync::SSH.new
when 'ftp'
@sync = MPW::Sync::FTP.new
else
return false
end
end
if !@config.sync_host.nil? && !@config.sync_port.nil?
if !@sync.connect(@config.sync_host, @config.sync_user, @config.sync_pwd, @config.sync_path, @config.sync_port)
puts "#{I18n.t('display.error')}: #{@sync.error_msg}"
end
end
if @sync.enable
if !@mpw.sync(@sync.get(@passwd), @config.last_update)
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}" if !@mpw.error_msg.nil?
puts "#{I18n.t('display.error')}: #{@sync.error_msg}" if !@sync.error_msg.nil?
elsif !@sync.update(File.open(@config.file_gpg).read)
puts "#{I18n.t('display.error')}: #{@sync.error_msg}"
elsif !@config.set_last_update
puts "#{I18n.t('display.error')}: #{@config.error_msg}"
elsif !@mpw.encrypt
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
else
return true
end
end
rescue Exception => e
puts "#{I18n.t('display.error')}: #{e}"
puts @sync.error_msg if @sync.error_msg.nil?
puts @config.error_msg if @config.error_msg.nil?
puts @mpw.error_msg if @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 => Dir.home)).to_s
timeout_pwd = ask(I18n.t('form.setup.timeout')).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? || language.empty?
language = lang
end
I18n.locale = language.to_sym
sync_type = sync_type.nil? || sync_type.empty? ? nil : sync_type
sync_host = sync_host.nil? || sync_host.empty? ? nil : sync_host
sync_port = sync_port.nil? || sync_port.empty? ? nil : sync_port.to_i
sync_user = sync_user.nil? || sync_user.empty? ? nil : sync_user
sync_pwd = sync_pwd.nil? || sync_pwd.empty? ? nil : sync_pwd
sync_path = sync_path.nil? || sync_path.empty? ? nil : sync_path
if @config.setup(key, share_keys, language, file_gpg, timeout_pwd, sync_type, sync_host, sync_port, sync_user, sync_pwd, sync_path)
puts I18n.t('form.setup.valid')
else
puts "#{I18n.t('display.error')}: #{@config.error_msg}"
exit 2
end
if not @config.checkconfig
puts "#{I18n.t('display.error')}: #{@config.error_msg}"
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 !['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? || length.empty? ? 2048 : length.to_i
expire = expire.nil? || 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')
else
puts "#{I18n.t('display.error')}: #{@config.error_msg}"
exit 2
end
end
# Request the GPG password and decrypt the file
def decrypt
if !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 !@mpw.decrypt(@passwd)
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
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, format=nil)
result = @mpw.search(search, group, protocol)
if not result.empty?
result.each do |r|
if format.nil? || !format
displayFormat(r)
else
displayFormatAlt(r)
end
end
else
puts I18n.t('display.nothing')
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::MPW::ID]}"
puts "#{I18n.t('display.name')}: #{item[MPW::MPW::NAME]}"
puts "#{I18n.t('display.group')}: #{item[MPW::MPW::GROUP]}"
puts "#{I18n.t('display.server')}: #{item[MPW::MPW::SERVER]}"
puts "#{I18n.t('display.protocol')}: #{item[MPW::MPW::PROTOCOL]}"
puts "#{I18n.t('display.login')}: #{item[MPW::MPW::LOGIN]}"
puts "#{I18n.t('display.password')}: #{item[MPW::MPW::PASSWORD]}"
puts "#{I18n.t('display.port')}: #{item[MPW::MPW::PORT]}"
puts "#{I18n.t('display.comment')}: #{item[MPW::MPW::COMMENT]}"
end
# Display an item in the alternative format
# @args: item -> an array with the item information
def displayFormatAlt(item)
port = item[MPW::MPW::PORT].nil? ? '' : ":#{item[MPW::MPW::PORT]}"
if item[MPW::MPW::PASSWORD].nil? || item[MPW::MPW::PASSWORD].empty?
if item[MPW::MPW::LOGIN].include('@')
puts "# #{item[MPW::MPW::ID]} #{item[MPW::MPW::PROTOCOL]}://#{item[MPW::MPW::LOGIN]}@#{item[MPW::MPW::SERVER]}#{port}"
else
puts "# #{item[MPW::MPW::ID]} #{item[MPW::MPW::PROTOCOL]}://{#{item[MPW::MPW::LOGIN]}}@#{item[MPW::MPW::SERVER]}#{port}"
end
else
puts "# #{item[MPW::MPW::ID]} #{item[MPW::MPW::PROTOCOL]}://{#{item[MPW::MPW::LOGIN]}:#{item[MPW::MPW::PASSWORD]}}@#{item[MPW::MPW::SERVER]}#{port}"
end
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')
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
end
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
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[MPW::MPW::NAME])).to_s
group = ask(I18n.t('form.update.group' , :group => row[MPW::MPW::GROUP])).to_s
server = ask(I18n.t('form.update.server' , :server => row[MPW::MPW::SERVER])).to_s
protocol = ask(I18n.t('form.update.protocol', :protocol => row[MPW::MPW::PROTOCOL])).to_s
login = ask(I18n.t('form.update.login' , :login => row[MPW::MPW::LOGIN])).to_s
passwd = ask(I18n.t('form.update.password')).to_s
port = ask(I18n.t('form.update.port' , :port => row[MPW::MPW::PORT])).to_s
comment = ask(I18n.t('form.update.comment' , :comment => row[MPW::MPW::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')
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
end
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
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
displayFormat(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)
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
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)
if @mpw.export(file)
puts "The export in #{file} is succesfull!"
else
puts "#{I18n.t('display.error')}: #{@mpw.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 = @mpw.import_preview(file)
if not force
if result.is_a?(Array) && !result.empty?
result.each do |r|
displayFormat(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) && @mpw.encrypt
sync
puts I18n.t('form.import.valid')
else
puts "#{I18n.t('display.error')}: #{@mpw.error_msg}"
end
end
end
# Interactive mode
def interactive
group = nil
last_access = Time.now.to_i
while buf = Readline.readline('<mpw> ', true)
if @config.timeout_pwd < Time.now.to_i - last_access
passwd_confirm = ask(I18n.t('interactive.ask_password')) {|q| q.echo = false}
if @passwd.eql?(passwd_confirm)
last_access = Time.now.to_i
else
puts I18n.t('interactive.bad_password')
next
end
else
last_access = Time.now.to_i
end
command = buf.split(' ')
case command[0]
when 'display', 'show', 'd', 's'
if !command[1].nil? && !command[1].empty?
display(command[1], group, command[2])
end
when 'add', 'a'
add
when 'update', 'u'
if !command[1].nil? && !command[1].empty?
update(command[1])
end
when 'remove', 'delete', 'r', 'd'
if !command[1].nil? && !command[1].empty?
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 I18n.t('interactive.option.title')
puts '--------------------'
puts "display, show, d, s SEARCH #{I18n.t('interactive.option.show')}"
puts "group, g #{I18n.t('interactive.option.group')}"
puts "add, a #{I18n.t('interactive.option.add')}"
puts "update, u ID #{I18n.t('interactive.option.update')}"
puts "remove, delete, r, d ID #{I18n.t('interactive.option.remove')}"
puts "help, h, ? #{I18n.t('interactive.option.help')}"
puts "quit, exit, q #{I18n.t('interactive.option.quit')}"
when 'quit', 'exit', 'q'
puts I18n.t('interactive.goodbye')
break
else
if !command[0].nil? && !command[0].empty?
puts I18n.t('interactive.unknown_command')
end
end
end
end
end
Reference in a new issue View git blame Copy permalink