Add rpc_refund method

Author: cpacia

api/restapi.py

@@ -1324,6 +1324,7 @@ def refund(self, request):
             request.finish()
             return server.NOT_DONE_YET
 
+
 class RestAPI(Site):
 
     def __init__(self, mserver, kserver, openbazaar_protocol, username, password,

market/network.py

@@ -997,7 +997,7 @@ def get_result(result):
         d = self.protocol.callGetRatings(node_to_ask, listing_hash)
         return d.addCallback(get_result)
 
-    def refund(self, order_id, reason):
+    def refund(self, order_id):
         """
         Refund the given order_id. If this is a direct payment he transaction will be
         immediately broadcast to the Bitcoin network otherwise the refund message sent
@@ -1023,7 +1023,6 @@ def refund(self, order_id, reason):
 
         refund_json = {"refund": {}}
         refund_json["refund"]["order_id"] = order_id
-        refund_json["refund"]["reason"] = reason
         if "moderator" in contract["buyer_order"]["order"]:
             sigs = tx.create_signature(vendor_priv, redeem_script)
             refund_json["refund"]["value"] = tx.get_out_value()
@@ -1057,6 +1056,7 @@ def parse_response(response):
         self.log.info("sending refund message to %s" % buyer_guid)
         return self.kserver.resolve(unhexlify(buyer_guid)).addCallback(get_node)
 
+
     @staticmethod
     def cache(file_to_save, filename):
         """

market/protocol.py

@@ -1,17 +1,24 @@
 __author__ = 'chris'
 
+import bitcointools
 import json
+import os
+import pickle
 import nacl.signing
 import nacl.utils
 import nacl.encoding
 import nacl.hash
+from binascii import unhexlify
 from collections import OrderedDict
+from config import DATA_FOLDER
 from interfaces import MessageProcessor, BroadcastListener, MessageListener, NotificationListener
 from keys.bip32utils import derive_childkey
+from keys.keychain import KeyChain
 from log import Logger
 from market.contracts import Contract
 from market.moderation import process_dispute, close_dispute
 from market.profile import Profile
+from market.transactions import BitcoinTransaction
 from nacl.public import PublicKey, Box
 from net.rpcudp import RPCProtocol
 from protos.message import GET_CONTRACT, GET_IMAGE, GET_PROFILE, GET_LISTINGS, GET_USER_METADATA,\
@@ -346,6 +353,73 @@ def rpc_get_ratings(self, sender, listing_hash=None):
             self.log.warning("could not load ratings for contract %s" % a)
             return None
 
+    def rpc_refund(self, sender, pubkey, encrypted):
+        try:
+            box = Box(self.signing_key.to_curve25519_private_key(), PublicKey(pubkey))
+            refund = box.decrypt(encrypted)
+            refund_json = json.loads(refund, object_pairs_hook=OrderedDict)
+            order_id = refund_json["order_id"]
+
+            file_path = DATA_FOLDER + "purchases/in progress/" + order_id + ".json"
+            with open(file_path, 'r') as filename:
+                order = json.load(filename, object_pairs_hook=OrderedDict)
+            order["refund"] = refund_json["refund"]
+
+            if "txid" not in refund_json:
+                outpoints = pickle.loads(self.db.sales.get_outpoint(order_id))
+                refund_address = order["buyer_order"]["order"]["refund_address"]
+                redeem_script = order["buyer_order"]["order"]["payment"]["redeem_script"]
+                tx = BitcoinTransaction.make_unsigned(outpoints, refund_address,
+                                                      testnet=self.multiplexer.testnet,
+                                                      out_value=long(refund_json["refund"]["value"]))
+                chaincode = order["buyer_order"]["order"]["payment"]["chaincode"]
+                masterkey_b = bitcointools.bip32_extract_key(KeyChain(self.db).bitcoin_master_privkey)
+                buyer_priv = derive_childkey(masterkey_b, chaincode, bitcointools.MAINNET_PRIVATE)
+                buyer_sigs = tx.create_signature(buyer_priv, redeem_script)
+                vendor_sigs = refund_json["refund"]["signature(s)"]
+
+                signatures = []
+                for i in range(len(outpoints)):
+                    for vendor_sig in vendor_sigs:
+                        if vendor_sig["index"] == i:
+                            v_signature = vendor_sig["signature"]
+                    for buyer_sig in buyer_sigs:
+                        if buyer_sig["index"] == i:
+                            b_signature = buyer_sig["signature"]
+                    signature_obj = {"index": i, "signatures": [b_signature, v_signature]}
+                    signatures.append(signature_obj)
+
+                tx.multisign(signatures, redeem_script)
+                tx.broadcast(self.multiplexer.blockchain)
+                self.log.info("Broadcasting refund tx %s to network" % tx.get_hash())
+
+            self.db.sales.update_status(order_id, 7)
+            file_path = DATA_FOLDER + "purchases/trade receipts/" + order_id + ".json"
+            with open(file_path, 'w') as outfile:
+                outfile.write(json.dumps(order, indent=4))
+            file_path = DATA_FOLDER + "purchases/in progress/" + order_id + ".json"
+            if os.path.exists(file_path):
+                os.remove(file_path)
+
+            title = order["vendor_offer"]["listing"]["item"]["title"]
+            if "image_hashes" in order["vendor_offer"]["listing"]["item"]:
+                image_hash = unhexlify(order["vendor_offer"]["listing"]["item"]["image_hashes"][0])
+            else:
+                image_hash = ""
+            buyer_guid = self.contract["buyer_order"]["order"]["id"]["guid"]
+            if "blockchain_id" in self.contract["buyer_order"]["order"]["id"]:
+                handle = self.contract["buyer_order"]["order"]["id"]["blockchain_id"]
+            else:
+                handle = ""
+            self.get_notification_listener().notify(buyer_guid, handle, "refund", order_id, title, image_hash)
+
+            self.router.addContact(sender)
+            self.log.info("order %s refunded by vendor" % refund_json["refund"]["order_id"])
+            return ["True"]
+        except Exception:
+            self.log.error("unable to parse refund message from %s" % sender)
+            return ["False"]
+
     def callGetContract(self, nodeToAsk, contract_hash):
         d = self.get_contract(nodeToAsk, contract_hash)
         return d.addCallback(self.handleCallResponse, nodeToAsk)
@@ -444,6 +518,7 @@ def get_notification_listener(self):
                 return listener
             except DoesNotImplement:
                 pass
+
     def get_message_listener(self):
         for listener in self.listeners:
             try: