mirror of
https://github.com/nishiki/manage-password.git
synced 2024-11-30 08:53:05 +00:00
304 lines
7.3 KiB
Ruby
304 lines
7.3 KiB
Ruby
#!/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 "ERROR: #{@m.error_msg}"
|
|
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
|