mirror of
https://github.com/nishiki/manage-password.git
synced 2024-12-18 07:23:30 +00:00
124 lines
2.7 KiB
Ruby
124 lines
2.7 KiB
Ruby
#!/usr/bin/ruby
|
|
# author: nishiki
|
|
|
|
require 'rubygems'
|
|
require 'i18n'
|
|
require 'socket'
|
|
require 'json'
|
|
require 'timeout'
|
|
|
|
module MPW
|
|
class SyncMPW
|
|
|
|
attr_accessor :error_msg
|
|
attr_accessor :enable
|
|
|
|
# Constructor
|
|
# @args: host -> the server host
|
|
# port -> ther connection port
|
|
# gpg_key -> the gpg key
|
|
# password -> the remote password
|
|
# suffix -> the suffix file
|
|
def initialize(host, user, password, suffix, port=nil)
|
|
@error_msg = nil
|
|
@enable = false
|
|
|
|
@host = host
|
|
@port = !port.instance_of?(Integer) ? 2201 : port
|
|
@gpg_key = user
|
|
@password = password
|
|
@suffix = suffix
|
|
end
|
|
|
|
# Connect to server
|
|
# @rtrn: false if the connection fail
|
|
def connect
|
|
Timeout.timeout(10) do
|
|
begin
|
|
TCPSocket.open(@host, @port) do
|
|
@enable = true
|
|
end
|
|
rescue Errno::ENETUNREACH
|
|
retry
|
|
end
|
|
end
|
|
rescue Timeout::Error
|
|
@error_msg = "#{I18n.t('error.timeout')}\n#{e}"
|
|
@enable = false
|
|
rescue Exception => e
|
|
@error_msg = "#{I18n.t('error.sync.connection')}\n#{e}"
|
|
@enable = false
|
|
else
|
|
return @enable
|
|
end
|
|
|
|
# Get data on server
|
|
# @args: gpg_password -> the gpg password
|
|
# @rtrn: nil if nothing data or error
|
|
def get(file_tmp)
|
|
return false if not @enable
|
|
|
|
msg = nil
|
|
TCPSocket.open(@host, @port) do |socket|
|
|
send_msg = {action: 'get',
|
|
gpg_key: @gpg_key,
|
|
password: @password,
|
|
suffix: @suffix
|
|
}
|
|
|
|
socket.puts send_msg.to_json
|
|
msg = JSON.parse(socket.gets)
|
|
end
|
|
|
|
if not defined?(msg['error'])
|
|
@error_msg = I18n.t('error.sync.communication')
|
|
return false
|
|
elsif not msg['error'].nil?
|
|
@error_msg = I18n.t(msg['error'])
|
|
return false
|
|
end
|
|
|
|
File.open(file_tmp, 'w') do |file|
|
|
file << msg['data']
|
|
end
|
|
|
|
return true
|
|
rescue Exception => e
|
|
@error_msg = "#{I18n.t('error.sync.download')}\n#{e}"
|
|
return false
|
|
end
|
|
|
|
# Update the remote data
|
|
# @args: data -> the data to send on server
|
|
# @rtrn: false if there is a problem
|
|
def update(file_gpg)
|
|
return true if not @enable
|
|
|
|
data = File.open(file_gpg, 'r').read
|
|
|
|
msg = nil
|
|
TCPSocket.open(@host, @port) do |socket|
|
|
send_msg = {action: 'update',
|
|
gpg_key: @gpg_key,
|
|
password: @password,
|
|
suffix: @suffix,
|
|
data: data
|
|
}
|
|
|
|
socket.puts send_msg.to_json
|
|
msg = JSON.parse(socket.gets)
|
|
end
|
|
|
|
if not defined?(msg['error'])
|
|
@error_msg = I18n.t('error.sync.communication')
|
|
return false
|
|
elsif msg['error'].nil?
|
|
return true
|
|
else
|
|
@error_msg = I18n.t(msg['error'])
|
|
return false
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|