unstable: fix sync via ssh

lib/Sync/SyncMPW.rb

@@ -0,0 +1,132 @@
+#!/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(gpg_key, gpg_password)
+			return nil 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 nil
+			elsif not msg['error'].nil?
+				@error_msg = I18n.t(msg['error'])
+				return nil
+			elsif msg['data'].nil? or msg['data'].empty?
+				return {}
+			else
+				file_tmp = Tempfile.new('mpw-')
+				File.open(file_tmp, 'w') do |file|
+					file << msg['data']
+				end
+				
+				mpw = MPW.new(file_tmp, gpg_key)
+				raise mpw.error_msg if not mpw.decrypt(gpg_password)
+				
+				file_tmp.close(true)
+
+				puts 'test'
+				return mpw.list
+			end
+		rescue Exception => e
+			@error_msg = "#{I18n.t('error.sync.download')}\n#{e}"
+			file_tmp.close(true)
+			return nil
+		end
+	
+		# Update the remote data
+		# @args: data -> the data to send on server
+		# @rtrn: false if there is a problem
+		def update(data)
+			return true if not @enable
+	
+			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