mpw/lib/MPWConfig.rb

#!/usr/bin/ruby
# author: nishiki
# mail: nishiki@yaegashi.fr
# info: a simple script who manage your passwords

require 'rubygems'
require 'yaml'
require 'i18n'

class MPWConfig
	
	attr_accessor :error_msg

	attr_accessor :key
	attr_accessor :lang
	attr_accessor :file_gpg
	attr_accessor :timeout_pwd
	attr_accessor :sync_host
	attr_accessor :sync_port
	attr_accessor :sync_pwd
	attr_accessor :sync_sufix

	# Constructor
	# @args: file_config -> the specify config file
	def initialize(file_config=nil)
		@error_msg  = nil
		@file_config = "#{Dir.home()}/.mpw.cfg"

		if !file_config.nil? && !file_config.empty?
			@file_config = file_config
		end
	end

	# Create a new config file
	# @args: key -> the gpg key to encrypt
	#        lang -> the software language
	#        file_gpg -> the file who is encrypted
	#        timeout_pwd -> time to save the password 
	# @rtrn: true if le config file is create
	def setup(key, lang, file_gpg, timeout_pwd)

		if not key =~ /[a-zA-Z0-9.-_]+\@[a-zA-Z0-9]+\.[a-zA-Z]+/
			@error_msg = I18n.t('error.config.key_bad_format')
			return false
		end
		
		if file_gpg.empty?
			file_gpg = "#{Dir.home()}/.mpw.gpg"
		end

		timeout_pwd.empty? ? (timeout_pwd = 60) : (timeout_pwd = timeout_pwd.to_i)

		config = {'config' => {'key'         => key,
		                       'lang'        => lang,
		                       'file_gpg'    => file_gpg,
		                       'timeout_pwd' => timeout_pwd,
		                       'sync_host'   => host,
		                       'sync_port'   => port,
		                       'sync_pwd'    => password,
		                       'sync_suffix' => suffix}}

		begin
			File.open(@file_config, 'w') do |file|
				file << config.to_yaml
			end
		rescue Exception => e 
			@error_msg = "#{I18n.t('error.config.write')}\n#{e}"
			return false
		end

		return true
	end

	# Check the config file
	# @rtrn: true if the config file is correct
	def checkconfig()
		begin
			config = YAML::load_file(@file_config)
			@key         = config['config']['key']
			@lang        = config['config']['lang']
			@file_gpg    = config['config']['file_gpg']
			@timeout_pwd = config['config']['timeout_pwd'].to_i
			@sync_host   = config['config']['sync_host']
			@sync_port   = config['config']['sync_port']
			@sync_pwd    = config['config']['sync_pwd']
			@sync_sufix  = config['config']['sync_suffix']

			if @key.empty? || @file_gpg.empty? 
				@error_msg = I18n.t('error.config.check')
				return false
			end

			I18n.locale = @lang.to_sym

		rescue Exception => e 
			@error_msg = "#{I18n.t('error.config.check')}\n#{e}"
			return false
		end

		return true
	end
	
end