change the data format csv to yaml

lib/MPW.rb

@@ -31,9 +31,8 @@ module MPW
 	
 			if File.exist?(@file_gpg)
 				crypto = GPGME::Crypto.new(armor: true)
-				data_decrypt = crypto.decrypt(IO.read(@file_gpg), password: passwd).read
+				data_decrypt = crypto.decrypt(IO.read(@file_gpg), password: passwd).read.force_encoding('utf-8')
-	
+				@data = YAML.load(data_decrypt)
-				@data = CSV.parse(data_decrypt, {headers: true, header_converters: :symbol})
 			end
 	
 			return true
@@ -45,17 +44,9 @@ module MPW
 		# Encrypt a file
 		# @rtrn: true if the file has been encrypted
 		def encrypt
-			FileUtils.cp(@file_gpg, "#{@file_gpg}.bk")
+			FileUtils.cp(@file_gpg, "#{@file_gpg}.bk") if File.exist?(@file_gpg)
 	
-			crypto = GPGME::Crypto.new(armor: true)
+			data_to_encrypt = @data.to_yaml
-			file_gpg = File.open(@file_gpg, 'w+')
-	
-			data_to_encrypt = CSV.generate(write_headers: true,
-			                               headers: ['id', 'name', 'group', 'protocol', 'host', 'login', 'password', 'port', 'comment', 'date']) do |csv|
-				@data.each do |r|
-					csv << [r[:id], r[:name], r[:group], r[:protocol], r[:host], r[:login], r[:password], r[:port], r[:comment], r[:date]]
-				end
-			end
 	
 			recipients = []
 			recipients.push(@key)
@@ -63,14 +54,16 @@ module MPW
 				@share_keys.split.each { |k| recipients.push(k) }
 			end
 
+			crypto = GPGME::Crypto.new(armor: true)
+			file_gpg = File.open(@file_gpg, 'w+')
 			crypto.encrypt(data_to_encrypt, recipients: recipients, output: file_gpg)
 			file_gpg.close
 	
-			FileUtils.rm("#{@file_gpg}.bk")
+			FileUtils.rm("#{@file_gpg}.bk") if File.exist?("#{@file_gpg}.bk")
 			return true
 		rescue Exception => e 
 			@error_msg = "#{I18n.t('error.gpg_file.encrypt')}\n#{e}"
-			FileUtils.mv("#{@file_gpg}.bk", @file_gpg)
+			FileUtils.mv("#{@file_gpg}.bk", @file_gpg) if File.exist?("#{@file_gpg}.bk")
 			return false
 		end
 		
@@ -84,12 +77,11 @@ module MPW
 			if not search.nil?
 				search = search.downcase
 			end
-			search = search.force_encoding('ASCII-8BIT')
 	
-			@data.each do |row|
+			@data.each do |id, row|
-				name    = row[:name].nil?    ? nil : row[:name].downcase
+				name    = row['name'].nil?    ? nil : row['name'].downcase
-				server  = row[:host].nil?  ? nil : row[:host].downcase
+				server  = row['host'].nil?    ? nil : row['host'].downcase
-				comment = row[:comment].nil? ? nil : row[:comment].downcase
+				comment = row['comment'].nil? ? nil : row['comment'].downcase
 	
 				if name =~ /^.*#{search}.*$/ or server =~ /^.*#{search}.*$/ or comment =~ /^.*#{search}.*$/ 
 					if (protocol.nil? or protocol.eql?(row[:protocol])) and (group.nil? or group.eql?(row[:group]))
@@ -102,13 +94,11 @@ module MPW
 		end
 	
 		# Search in some csv data
-		# @args: id -> the id item
+		# @args: id_search -> the id item
 		# @rtrn: a row with the resultat of the search
-		def search_by_id(id)
+		def search_by_id(id_search)
-			@data.each do |row|
+			@data.each do |id, row|
-				if row[:id] == id
+				return row if id == id_search
-					return row
-				end
 			end
 	
 			return []
@@ -130,13 +120,8 @@ module MPW
 			update = false
 	
 			i  = 0
-			@data.each do |r|
+			if @data.has_key?(id)
-				if r[:id] == id
+				row = @data[id]
-					row    = r
-					update = true
-					break
-				end
-				i += 1
 			end
 	
 			if port.to_i <= 0
@@ -144,35 +129,26 @@ module MPW
 			end
 	
 			row_update             = {}
-			row_update[:date] = Time.now.to_i
+			row_update['id']       = id.to_s.empty?       ? MPW.password(16) : id
+			row_update['name']     = name.to_s.empty?     ? row['name']      : name
+			row_update['group']    = group.to_s.empty?    ? row['group']     : group
+			row_update['host']     = server.to_s.empty?   ? row['host']      : server
+			row_update['protocol'] = protocol.to_s.empty? ? row['protocol']  : protocol
+			row_update['login']    = login.to_s.empty?    ? row['login']     : login
+			row_update['password'] = passwd.to_s.empty?   ? row['password']  : passwd
+			row_update['port']     = port.to_s.empty?     ? row['port']      : port
+			row_update['comment']  = comment.to_s.empty?  ? row['comment']   : comment
+			row_update['date']     = Time.now.to_i
 	
-			row_update[:id]       = id.nil?       or id.empty?       ? MPW.password(16)  : id
+			if row_update['name'].to_s.empty?
-			row_update[:name]     = name.nil?     or name.empty?     ? row[:name]        : name
-			row_update[:group]    = group.nil?    or group.empty?    ? row[:group]       : group
-			row_update[:host]     = server.nil?   or server.empty?   ? row[:host]        : server
-			row_update[:protocol] = protocol.nil? or protocol.empty? ? row[:protocol]    : protocol
-			row_update[:login]    = login.nil?    or login.empty?    ? row[:login]       : login
-			row_update[:password] = passwd.nil?   or passwd.empty?   ? row[:password]    : passwd
-			row_update[:port]     = port.nil?     or port.empty?     ? row[:port]        : port
-			row_update[:comment]  = comment.nil?  or comment.empty?  ? row[:comment]     : comment
-			
-			row_update[:name]     = row_update[:name].nil?     ? nil : row_update[:name].force_encoding('ASCII-8BIT')
-			row_update[:group]    = row_update[:group].nil?    ? nil : row_update[:group].force_encoding('ASCII-8BIT')
-			row_update[:host]     = row_update[:host].nil?     ? nil : row_update[:host].force_encoding('ASCII-8BIT')
-			row_update[:protocol] = row_update[:protocol].nil? ? nil : row_update[:protocol].force_encoding('ASCII-8BIT')
-			row_update[:login]    = row_update[:login].nil?    ? nil : row_update[:login].force_encoding('ASCII-8BIT')
-			row_update[:password] = row_update[:password].nil? ? nil : row_update[:password].force_encoding('ASCII-8BIT')
-			row_update[:comment]  = row_update[:comment].nil?  ? nil : row_update[:comment].force_encoding('ASCII-8BIT')
-	
-			if row_update[:name].nil? or row_update[:name].empty?
 				@error_msg = I18n.t('error.update.name_empty')
 				return false
 			end
 	
 			if update
-				@data[i] = row_update
+				@data[id] = row_update
 			else
-				@data.push(row_update)
+				@data[row_update['id']] = row_update
 			end
 	
 			return true
@@ -182,13 +158,11 @@ module MPW
 		# @args: id -> the item's identifiant
 		# @rtrn: true if the item has been deleted
 		def remove(id)
-			i = 0
+			@data.each do |k, row|
-			@data.each do |row|
+				if k == id
-				if row[:id] == id
+					@data.delete(id)
-					@data.delete_at(i)
 					return true
 				end
-				i += 1
 			end
 	
 			@error_msg = I18n.t('error.delete.id_no_exist', id: id)
@@ -197,36 +171,33 @@ module MPW
 	
 		# Export to csv
 		# @args: file -> file where you export the data
-		#        type -> data type
+		#        type -> udata type
 		# @rtrn: true if export work
 		def export(file, type=:csv)
 			case type
 			when :csv
 					CSV.open(file, 'w', write_headers: true,
 										headers: ['name', 'group', 'protocol', 'host', 'login', 'password', 'port', 'comment']) do |csv|
-						@data.each do |r|
+						@data.each do |id, r|
-							csv << [r[:name], r[:group], r[:protocol], r[:host], r[:login], r[:password], r[:port], r[:comment]]
+							csv << [r['name'], r['group'], r['protocol'], r['host'], r['login'], r['password'], r['port'], r['comment']]
 						end
 					end
 			when :yaml
 				data = {}
 
-				i = 0
+				@data.each do |id, r|
-				@data.each do |r|
+					data.merge!({id => {'id'       => r['id'],
-					data.merge!({i => {'id'       => r[:id],
+					                    'name'     => r['name'],
-					                   'name'     => r[:name],
+					                    'group'    => r['group'],
-					                   'group'    => r[:group],
+					                    'protocol' => r['protocol'],
-					                   'protocol' => r[:protocol],
+					                    'host'     => r['host'],
-					                   'host'     => r[:host],
+					                    'login'    => r['login'],
-					                   'login'    => r[:login],
+					                    'password' => r['password'],
-					                   'password' => r[:password],
+					                    'port'     => r['port'],
-					                   'port'     => r[:port],
+					                    'comment'  => r['comment'],
-					                   'comment'  => r[:comment]
 					                   }
 					            }
 					          )
-
-					i += 1
 				end
 
 				File.open(file, 'w') {|f| f << data.to_yaml}
@@ -243,13 +214,26 @@ module MPW
 	
 		# Import to csv
 		# @args: file -> path to file import
+		#        type -> udata type
 		# @rtrn: true if the import work
-		def import(file)
+		def import(file, type=:csv)
+			case type
+			when :csv
 				CSV.foreach(file, {headers: true, header_converters: :symbol}) do |row|
 					if not update(row[:name], row[:group], row[:host], row[:protocol], row[:login], row[:password], row[:port], row[:comment])
 						return false
 					end
 				end
+			when :yaml
+				YAML::load_file(file).each do |k, row| 
+					if not update(row['name'], row['group'], row['host'], row['protocol'], row['login'], row['password'], row['port'], row['comment'])
+						return false
+					end
+				end
+			else
+				@error_msg = "#{I18n.t('error.export.unknown_type', type: type)}"
+				return false
+			end
 	
 			return true
 		rescue Exception => e 
@@ -260,11 +244,21 @@ module MPW
 		# Return a preview import 
 		# @args: file -> path to file import
 		# @rtrn: an array with the items to import, if there is an error return false
-		def import_preview(file)
+		def import_preview(file, type=:csv)
 			result = []
-			CSV.foreach(file, {headers: true, header_converters: :symbol}) do |row|
+			case type
+			when :csv
+				CSV.foreach(file, {headers: true}) do |row|
 					result << row
 				end
+			when :yaml
+				YAML::load_file(file).each do |k, row| 
+					result << row
+				end
+			else
+				@error_msg = "#{I18n.t('error.export.unknown_type', type: type)}"
+				return false
+			end
 
 			return result
 		rescue Exception => e 

lib/UI/Cli.rb

@@ -77,7 +77,7 @@ class Cli
 		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_home)).to_s
+		file_gpg    = ask(I18n.t('form.setup.gpg_file', home: @conf.dir_config)).to_s
 		timeout_pwd = ask(I18n.t('form.setup.timeout')).to_s
 		sync_type   = ask(I18n.t('form.setup.sync_type')).to_s
 
@@ -187,15 +187,15 @@ class Cli
 	# @args: item -> an array with the item information
 	def displayFormat(item)
 		puts '--------------------'
-		puts "Id: #{item[:id]}"
+		puts "Id: #{item['id']}"
-		puts "#{I18n.t('display.name')}: #{item[:name]}"
+		puts "#{I18n.t('display.name')}: #{item['name']}"
-		puts "#{I18n.t('display.group')}: #{item[:group]}"
+		puts "#{I18n.t('display.group')}: #{item['group']}"
-		puts "#{I18n.t('display.server')}: #{item[:host]}"
+		puts "#{I18n.t('display.server')}: #{item['host']}"
-		puts "#{I18n.t('display.protocol')}: #{item[:protocol]}"
+		puts "#{I18n.t('display.protocol')}: #{item['protocol']}"
-		puts "#{I18n.t('display.login')}: #{item[:login]}"
+		puts "#{I18n.t('display.login')}: #{item['login']}"
-		puts "#{I18n.t('display.password')}: #{item[:password]}"
+		puts "#{I18n.t('display.password')}: #{item['password']}"
-		puts "#{I18n.t('display.port')}: #{item[:port]}"
+		puts "#{I18n.t('display.port')}: #{item['port']}"
-		puts "#{I18n.t('display.comment')}: #{item[:comment]}"
+		puts "#{I18n.t('display.comment')}: #{item['comment']}"
 	end
 
 	# Display an item in the alternative format
@@ -320,9 +320,9 @@ class Cli
 	# @args: file -> the import file
 	#        force -> no resquest a validation
 	def import(file, force=false)
-		result = @mpw.import_preview(file)
 
 		if not force
+			result = @mpw.import_preview(file)
 			if result.is_a?(Array) && !result.empty?
 				result.each do |r|
 					displayFormat(r)