RecordsKeeperRubyLib 0.3.0 → 0.4.0

data/lib/RecordsKeeperRubyLib/transaction.rb

@@ -1,183 +1,213 @@
-#   Library to work with RecordsKeeper transactions.
-
-#   You can send transaction, create raw transaction, sign raw transaction, send raw transaction, send signed transaction,
-#     retrieve transaction information and calculate transaction's fees by using transaction class. You just have to pass
-#       parameters to invoke the pre-defined functions.
-
-require 'rubygems'
-require 'httparty'
-require 'json'
-require 'binary_parser'
-require 'yaml'
-require 'hex_string'
-
-module RecordsKeeperRubyLib
-
-	class Transaction
-
-		# Entry point for accessing Block class resources.
-		# Import values from config file.
-		cfg = YAML::load(File.open('config.yaml','r'))
-		@network = cfg['network']
-		@url = cfg['network']['url']
-		@user = cfg['network']['rkuser']
-		@password = cfg['network']['passwd']
-		@chain = cfg['network']['chain']
-
-		def self.variable
-			net = @network
-			return net
-		end
-
-	  # Function to send transaction on RecordsKeeper Blockchain
-	  def self.sendTransaction sender_address, receiveraddress, data, amount
-	    data_hex = data.to_hex_string
-	    datahex = data_hex.delete(' ')
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"createrawsendfrom","params":[sender_address, { receiveraddress => amount }, [datahex], "send"],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-	    txid = out[0]['result']
-	    return txid;
-	  end
-
-	  # Function to create transaction hex on RecordsKeeper Blockchain
-		def self.createRawTransaction sender_address, receiveraddress, data, amount
-	    data_hex = data.to_hex_string
-	    datahex = data_hex.delete(' ')
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"createrawsendfrom","params":[sender_address, {receiveraddress => amount}, [datahex], ''],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			txhex = out[0]['result']
-			return txhex;
-	  end
-
-	  # Function to sign transaction on RecordsKeeper Blockchain
-	  def self.signRawTransaction txHex, private_key
-			priv_key = []
-			priv_key.push(private_key)
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"signrawtransaction","params":[txHex, [], priv_key],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-		if out[0]['result']['complete']
-			signedHex = out[0]['result']['hex']
-		else
-			signedHex = "Transaction has not been signed."
-		end
-			return signedHex;
-	  end
-
-	  # Function to send raw transaction on RecordsKeeper Blockchain
-		def self.sendRawTransaction signed_txHex
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"sendrawtransaction","params":[signed_txHex],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			txn = out[0]['result']
-			if txn.nil?
-				txid = out[0]['error']['message']
-			else
-				txid = txn
-			end
-			return txid;
-	  end
-
-	  # Function to send signed transaction on RecordsKeeper Blockchain
-	  def self.sendSignedTransaction sender_address, receiveraddress, data, amount, private_key
-	    data_hex = data.to_hex_string
-	    datahex = data_hex.delete(' ')
-	    def self.createRawTransaction sender_address, receiveraddress, amount, datahex
-	      auth = {:username => @user, :password => @password}
-	      options = {
-	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	        :basic_auth => auth,
-	        :body => [ {"method":"createrawsendfrom","params":[sender_address, {receiveraddress => amount}, [datahex], ""],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	      }
-	      response = HTTParty.get(@url, options)
-	      out = response.parsed_response
-	      return out[0]['result']
-	    end
-	    txHex = createRawTransaction sender_address, receiveraddress, Float(amount), datahex
-	    def self.signRawTransaction txHex, private_key
-	      auth = {:username => @user, :password => @password}
-	      options = {
-	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	        :basic_auth => auth,
-	        :body => [ {"method":"signrawtransaction","params":[txHex, [], [private_key]],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	      }
-	      response = HTTParty.get(@url, options)
-	      out = response.parsed_response
-	      return out[0]['result']['hex']
-	    end
-	    signed_tx_hex = signRawTransaction txHex, private_key
-	    def self.sendRawTransaction signed_tx_hex
-	      auth = {:username => @user, :password => @password}
-	      options = {
-	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	        :basic_auth => auth,
-	        :body => [ {"method":"sendrawtransaction","params":[signed_tx_hex],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	      }
-	      response = HTTParty.get(@url, options)
-	      out = response.parsed_response
-	      return out[0]['result']
-	    end
-	    tx_id = sendRawTransaction signed_tx_hex
-	    return tx_id;
-	  end
-
-	  # Function to retrieve transaction on RecordsKeeper Blockchain
-	  def self.retrieveTransaction tx_id
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"getrawtransaction","params":[tx_id, 1],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-	    sent_hex_data = out[0]['result']['data'][0]
-	    sent_data = sent_hex_data.to_byte_string
-	    sent_amount = out[0]['result']['vout'][0]['value']
-			retrieve = {:sent_data => sent_data,:sent_amount => sent_amount }
-			retrievedinfo = JSON.generate retrieve
-			return retrievedinfo
-	  end
-
-	  # Function to calculate transaction's fee on RecordsKeeper Blockchain
-	  def self.getFee address, tx_id
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"getaddresstransaction","params":[address, tx_id, true],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-	    sent_amount = out[0]['result']['vout'][0]['amount']
-	    balance_amount = out[0]['result']['balance']['amount']
-	    fees = balance_amount.abs - sent_amount
-	    return fees;				#returns fees
-	  end
-	  
-
-	end
-
-end
+#   Library to work with RecordsKeeper transactions.
+
+#   You can send transaction, create raw transaction, sign raw transaction, send raw transaction, send signed transaction,
+#     retrieve transaction information and calculate transaction's fees by using transaction class. You just have to pass
+#       parameters to invoke the pre-defined functions.
+
+require 'rubygems'
+require 'httparty'
+require 'json'
+require 'binary_parser'
+require 'yaml'
+require 'hex_string'
+
+module RecordsKeeperRubyLib
+
+	class Transaction
+
+		# # Entry point for accessing Transaction class functions
+		if File.exist?('config.yaml')
+			# Import values from configuration file.
+			cfg = YAML::load(File.open('config.yaml','r'))
+			
+			@url = cfg['url']
+			@user = cfg['rkuser']
+			@password = cfg['passwd']
+			@chain = cfg['chain']
+		
+		else
+			#Import using ENV variables
+			
+			@url = ENV['url']
+    		@user = ENV['rkuser']
+    		@password = ENV['passwd']
+    		@chain = ENV['chain']	
+		end
+
+
+	  # Function to send transaction on RecordsKeeper Blockchain
+	  def self.sendTransaction sender_address, receiveraddress, data, amount
+	    data_hex = data.to_hex_string
+	    datahex = data_hex.delete(' ')
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"createrawsendfrom","params":[sender_address, { receiveraddress => amount }, [datahex], "send"],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+	    check = out[0]['result']
+
+		if check.nil?
+			txid = out[0]['error']['message']
+		else 
+	    	txid = out[0]['result']
+	    end
+	    return txid;
+	  end
+
+	  # Function to create transaction hex on RecordsKeeper Blockchain
+	  def self.createRawTransaction sender_address, receiveraddress, data, amount
+	    data_hex = data.to_hex_string
+	    datahex = data_hex.delete(' ')
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"createrawsendfrom","params":[sender_address, {receiveraddress => amount}, [datahex], ''],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+	    check = out[0]['result']
+
+		if check.nil?
+			txhex = out[0]['error']['message']
+		else 
+			txhex = out[0]['result']
+		end
+		return txhex;
+	  end
+
+	  # Function to sign transaction on RecordsKeeper Blockchain
+	  def self.signRawTransaction txHex, private_key
+		priv_key = []
+		priv_key.push(private_key)
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"signrawtransaction","params":[txHex, [], priv_key],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		if out[0]['result']['complete']
+			signedHex = out[0]['result']['hex']
+		else
+			signedHex = "Transaction has not been signed properly."
+		end
+			return signedHex;
+	  end
+
+	  # Function to send raw transaction on RecordsKeeper Blockchain
+	  def self.sendRawTransaction signed_txHex
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"sendrawtransaction","params":[signed_txHex],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		txn = out[0]['result']
+		if txn.nil?
+			txid = out[0]['error']['message']
+		else
+			txid = txn
+		end
+		return txid;
+	  end
+
+	  # Function to send signed transaction on RecordsKeeper Blockchain
+	  def self.sendSignedTransaction sender_address, receiveraddress, data, amount, private_key
+	    data_hex = data.to_hex_string
+	    datahex = data_hex.delete(' ')
+	    def self.createRawTransaction sender_address, receiveraddress, amount, datahex
+	      auth = {:username => @user, :password => @password}
+	      options = {
+	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	        :basic_auth => auth,
+	        :body => [ {"method":"createrawsendfrom","params":[sender_address, {receiveraddress => amount}, [datahex], ""],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	      }
+	      response = HTTParty.get(@url, options)
+	      out = response.parsed_response
+	      return out[0]['result']
+	    end
+	    txHex = createRawTransaction sender_address, receiveraddress, Float(amount), datahex
+	    def self.signRawTransaction txHex, private_key
+	      auth = {:username => @user, :password => @password}
+	      options = {
+	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	        :basic_auth => auth,
+	        :body => [ {"method":"signrawtransaction","params":[txHex, [], [private_key]],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	      }
+	      response = HTTParty.get(@url, options)
+	      out = response.parsed_response
+	      return out[0]['result']['hex']
+	    end
+	    signed_tx_hex = signRawTransaction txHex, private_key
+	    def self.sendRawTransaction signed_tx_hex
+	      auth = {:username => @user, :password => @password}
+	      options = {
+	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	        :basic_auth => auth,
+	        :body => [ {"method":"sendrawtransaction","params":[signed_tx_hex],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	      }
+	      response = HTTParty.get(@url, options)
+	      out = response.parsed_response
+	      return out[0]['result']
+	    end
+	    tx_id = sendRawTransaction signed_tx_hex
+	    return tx_id;
+	  end
+
+	  # Function to retrieve transaction on RecordsKeeper Blockchain
+	  def self.retrieveTransaction tx_id
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"getrawtransaction","params":[tx_id, 1],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+	    check = out[0]['result']
+
+		if check.nil?
+			retrievedinfo = out[0]['error']['message']
+		else 
+		    sent_hex_data = out[0]['result']['data'][0]
+		    sent_data = sent_hex_data.to_byte_string
+		    sent_amount = out[0]['result']['vout'][0]['value']
+			retrieve = {:sent_data => sent_data,:sent_amount => sent_amount }
+			retrievedinfo = JSON.generate retrieve
+		end
+		return retrievedinfo
+	  end
+
+	 # Function to calculate transaction's fee on RecordsKeeper Blockchain
+	 def self.getFee address, tx_id
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"getaddresstransaction","params":[address, tx_id, true],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+	    check = out[0]['result']
+
+		if check.nil?
+			fees = out[0]['error']['message']
+		else 
+		    sent_amount = out[0]['result']['vout'][0]['amount']
+		    balance_amount = out[0]['result']['balance']['amount']
+		    fees = balance_amount.abs - sent_amount
+		end
+	    return fees;				#returns fees
+	 end
+	  
+
+	end
+
+end

data/lib/RecordsKeeperRubyLib/version.rb

@@ -1,3 +1,3 @@
 module RecordsKeeperRubyLib
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

data/lib/RecordsKeeperRubyLib/wallet.rb

@@ -1,247 +1,263 @@
-#   Library to work with RecordsKeeper wallet.
-
-#   You can create wallet, create multisignature wallet, retrieve wallet's information, retrieve private key of a particular
-#     wallet address, sign message verify message, dump wallet file, backup wallet file, import wallet file, encrypt wallet by
-#       using wallet class. You just have to pass parameters to invoke the pre-defined functions.
-
-require 'rubygems'
-require 'httparty'
-require 'json'
-require 'binary_parser'
-require 'yaml'
-require 'hex_string'
-
-module RecordsKeeperRubyLib
-	class Wallet
-		# Entry point for accessing Block class resources.
-		# Import values from config file.
-		cfg = YAML::load(File.open('config.yaml','r'))
-		@network = cfg['network']
-		@url = cfg['network']['url']
-		@user = cfg['network']['rkuser']
-		@password = cfg['network']['passwd']
-		@chain = cfg['network']['chain']
-
-		def self.variable
-			net = @network
-			return net
-		end
-
-	  # Function to create wallet on RecordsKeeper Blockchain
-		def self.createWallet
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"createkeypairs","params":[],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			public_address = out[0]['result'][0]['address']			# returns public address of the wallet
-			private_key = out[0]['result'][0]['privkey']				# returns private key of the wallet
-			public_key = out[0]['result'][0]['pubkey']					# returns public key of the wallet
-
-			def self.importAddress public_address
-	      auth = {:username => @user, :password => @password}
-	      options = {
-	        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	        :basic_auth => auth,
-	        :body => [ {"method":"importaddress","params":[public_address, " ", false],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	      }
-	      response = HTTParty.get(@url, options)
-	      out = response.parsed_response
-				result = out[0]['result']
-				return result;
-	    end
-			import_address = importAddress public_address
-			retrieve = {:public_address => public_address,:private_key => private_key,:public_key => public_key}
-			retrievedinfo = JSON.generate retrieve
-			return retrievedinfo
-	  end
-
-		# Function to retrieve private key of a wallet on RecordsKeeper Blockchain
-		def self.getPrivateKey public_address
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"dumpprivkey","params":[public_address],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				private_key = out[0]['error']['message']
-			else
-				private_key = out[0]['result']
-	    end
-			return private_key;							#returns private key
-	  end
-
-		# Function to retrieve wallet's information on RecordsKeeper Blockchain
-		def self.retrieveWalletinfo
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"getwalletinfo","params":[],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			balance = out[0]['result']['balance']
-			tx_count = out[0]['result']['txcount']
-			unspent_tx = out[0]['result']['utxocount']
-			retrieve = {:balance => balance,:tx_count => tx_count,:unspent_tx => unspent_tx}
-			retrievedinfo = JSON.generate retrieve
-			return retrievedinfo
-	  end
-
-		# Function to create wallet's backup on RecordsKeeper Blockchain
-		def self.backupWallet filename
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"backupwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Backup successful!"
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to import wallet's backup on RecordsKeeper Blockchain
-		def self.importWallet filename
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"importwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Wallet is successfully imported"
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to dump wallet on RecordsKeeper Blockchain
-		def self.dumpWallet filename
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"dumpwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Wallet is successfully dumped"
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to lock wallet on RecordsKeeper Blockchain
-		def self.lockWallet password
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"encryptwallet","params":[password],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Wallet is successfully encrypted."
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to unlock wallet on RecordsKeeper Blockchain
-		def self.unlockWallet password, unlocktime
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"walletpassphrase","params":[password, unlocktime],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Wallet is successfully unlocked."
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to change password for wallet on RecordsKeeper Blockchain
-		def self.changeWalletPassword old_password, new_password
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"walletpassphrasechange","params":[old_password, new_password],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			result = out[0]['result']
-			if result.nil?
-				res = "Password successfully changed!"
-			else
-				res = out[0]['error']['message']
-	    end
-			return res;								#returns result
-	  end
-
-		# Function to sign message on RecordsKeeper Blockchain
-		def self.signMessage private_key, message
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"signmessage","params":[private_key, message],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			signedMessage = out[0]['result']
-			return signedMessage;									#returns private key
-	  end
-
-		# Function to verify message on RecordsKeeper Blockchain
-		def self.verifyMessage address, signedMessage, message
-	    auth = {:username => @user, :password => @password}
-	    options = {
-	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
-	      :basic_auth => auth,
-	      :body => [ {"method":"verifymessage","params":[address, signedMessage, message],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
-	    }
-	    response = HTTParty.get(@url, options)
-	    out = response.parsed_response
-			verifiedMessage = out[0]['result']
-			if verifiedMessage
-				validity = "Yes, message is verified"
-			else
-				validity = "No, signedMessage is not correct"
-	    end
-			return validity;										#returns validity
-	  end
-	end
-
-end
+#   Library to work with RecordsKeeper wallet.
+
+#   You can create wallet, create multisignature wallet, retrieve wallet's information, retrieve private key of a particular
+#     wallet address, sign message verify message, dump wallet file, backup wallet file, import wallet file, encrypt wallet by
+#       using wallet class. You just have to pass parameters to invoke the pre-defined functions.
+
+require 'rubygems'
+require 'httparty'
+require 'json'
+require 'binary_parser'
+require 'yaml'
+require 'hex_string'
+
+module RecordsKeeperRubyLib
+	class Wallet
+		
+		# # Entry point for accessing Wallet class functions
+		if File.exist?('config.yaml')
+			# Import values from configuration file.
+			cfg = YAML::load(File.open('config.yaml','r'))
+			
+			@url = cfg['url']
+			@user = cfg['rkuser']
+			@password = cfg['passwd']
+			@chain = cfg['chain']
+		
+		else
+			#Import using ENV variables
+			
+			@url = ENV['url']
+    		@user = ENV['rkuser']
+    		@password = ENV['passwd']
+    		@chain = ENV['chain']	
+		end
+
+	  # Function to create wallet on RecordsKeeper Blockchain
+		def self.createWallet
+		    auth = {:username => @user, :password => @password}
+		    options = {
+		      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+		      :basic_auth => auth,
+		      :body => [ {"method":"createkeypairs","params":[],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+		    }
+		    response = HTTParty.get(@url, options)
+		    out = response.parsed_response
+			public_address = out[0]['result'][0]['address']			# returns public address of the wallet
+			private_key = out[0]['result'][0]['privkey']				# returns private key of the wallet
+			public_key = out[0]['result'][0]['pubkey']					# returns public key of the wallet
+
+			def self.importAddress public_address
+		      auth = {:username => @user, :password => @password}
+		      options = {
+		        :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+		        :basic_auth => auth,
+		        :body => [ {"method":"importaddress","params":[public_address, " ", false],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+		      }
+		      response = HTTParty.get(@url, options)
+		      out = response.parsed_response
+			  result = out[0]['result']
+			  return result;
+		    end
+			import_address = importAddress public_address
+			retrieve = {:public_address => public_address,:private_key => private_key,:public_key => public_key}
+			retrievedinfo = JSON.generate retrieve
+			return retrievedinfo
+	    end
+
+		# Function to retrieve private key of a wallet address on RecordsKeeper Blockchain
+	   def self.getPrivateKey public_address
+		    auth = {:username => @user, :password => @password}
+		    options = {
+		      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+		      :basic_auth => auth,
+		      :body => [ {"method":"dumpprivkey","params":[public_address],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+		    }
+		    response = HTTParty.get(@url, options)
+		    out = response.parsed_response
+			result = out[0]['result']
+			if result.nil?
+				private_key = out[0]['error']['message']
+			else
+				private_key = out[0]['result']
+	    	end
+			return private_key;							#returns private key
+	   end
+
+		# Function to retrieve wallet's information on RecordsKeeper Blockchain
+      def self.retrieveWalletinfo
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"getwalletinfo","params":[],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		balance = out[0]['result']['balance']
+		tx_count = out[0]['result']['txcount']
+		unspent_tx = out[0]['result']['utxocount']
+		retrieve = {:balance => balance,:tx_count => tx_count,:unspent_tx => unspent_tx}
+		retrievedinfo = JSON.generate retrieve
+		return retrievedinfo
+	  end
+
+		# Function to create wallet's backup on RecordsKeeper Blockchain
+		def self.backupWallet filename
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"backupwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Backup successful!"
+		else
+			res = out[0]['error']['message']
+	    end
+			return res;								#returns result
+	  end
+
+		# Function to import wallet's backup on RecordsKeeper Blockchain
+	  def self.importWallet filename
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"importwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Wallet is successfully imported"
+		else
+			res = out[0]['error']['message']
+	    end
+			return res;								#returns result
+	  end
+
+		# Function to dump wallet on RecordsKeeper Blockchain
+	  def self.dumpWallet filename
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"dumpwallet","params":[filename],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Wallet is successfully dumped"
+		else
+			res = out[0]['error']['message']
+	    end
+			return res;								#returns result
+	  end
+
+		# Function to lock wallet on RecordsKeeper Blockchain
+	  def self.lockWallet password
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"encryptwallet","params":[password],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Wallet is successfully encrypted."
+		else
+			res = out[0]['error']['message']
+	    end
+			return res;								#returns result
+	  end
+
+		# Function to unlock wallet on RecordsKeeper Blockchain
+		def self.unlockWallet password, unlocktime
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"walletpassphrase","params":[password, unlocktime],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Wallet is successfully unlocked."
+		else
+			res = out[0]['error']['message']
+	    end
+			return res;								#returns result
+	  end
+
+		# Function to change password for wallet on RecordsKeeper Blockchain
+	  def self.changeWalletPassword old_password, new_password
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"walletpassphrasechange","params":[old_password, new_password],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		result = out[0]['result']
+		if result.nil?
+			res = "Password successfully changed!"
+		else
+			res = out[0]['error']['message']
+	    end
+		return res;								#returns result
+	  end
+
+		# Function to sign message on RecordsKeeper Blockchain
+	  def self.signMessage private_key, message
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"signmessage","params":[private_key, message],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+	    check = out[0]['result']
+
+		if check.nil?
+			signedMessage = out[0]['error']['message']
+		else 
+			signedMessage = out[0]['result']
+		end
+		return signedMessage;									#returns private key
+	  end
+
+		# Function to verify message on RecordsKeeper Blockchain
+	  def self.verifyMessage address, signedMessage, message
+	    auth = {:username => @user, :password => @password}
+	    options = {
+	      :headers => headers= {"Content-Type"=> "application/json","Cache-Control" => "no-cache"},
+	      :basic_auth => auth,
+	      :body => [ {"method":"verifymessage","params":[address, signedMessage, message],"jsonrpc":2.0,"id":"curltext","chain_name":@chain}].to_json
+	    }
+	    response = HTTParty.get(@url, options)
+	    out = response.parsed_response
+		verifiedMessage = out[0]['result']
+		error = out[0]['error']
+		if verifiedMessage
+			validity = "Yes, message is verified"
+		elsif error.nil?
+			validity = "No, signedMessage is not correct"
+		else 
+			validity = error['message']
+        end
+		return validity;										#returns validity
+	  end
+	end
+
+end
+