# This file is part of Maker Keeper Framework.
#
# Copyright (C) 2017-2018 reverendus
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging
import time
from typing import Tuple

from eth_account.messages import defunct_hash_message
from eth_utils import encode_hex
from web3 import Web3

from pymaker import Address
from pymaker.keys import _registered_accounts
from pymaker.util import bytes_to_hexstring


def eth_sign(message: bytes, web3: Web3, key=None):
    assert(isinstance(message, bytes))
    assert(isinstance(web3, Web3))

    local_account = _registered_accounts.get((web3, Address(web3.eth.defaultAccount)))

    if key is None:
        pkey = local_account.privateKey
    else:
        pkey = key

    if local_account:
        start_time = time.time()
        start_clock = time.clock()
        try:
            message_hash = defunct_hash_message(primitive=message)
            signature = web3.eth.account.signHash(message_hash, private_key=pkey).signature.hex()
        finally:
            end_time = time.time()
            end_clock = time.clock()

        logging.debug(f"Local signing took {end_time - start_time:.3f}s time, {end_clock - start_clock:.3f}s clock")

        return signature

    else:
        signature = bytes_to_hexstring(web3.manager.request_blocking(
            "eth_sign", [web3.eth.defaultAccount, encode_hex(message)],
        ))

        # for `EthereumJS TestRPC/v2.2.1/ethereum-js`
        if signature.endswith("00"):
            signature = signature[:-2] + "1b"

        if signature.endswith("01"):
            signature = signature[:-2] + "1c"

        return signature


def to_vrs(signature: str) -> Tuple[int, bytes, bytes]:
    assert(isinstance(signature, str))
    assert(signature.startswith("0x"))

    signature_hex = signature[2:]
    r = bytes.fromhex(signature_hex[0:64])
    s = bytes.fromhex(signature_hex[64:128])
    v = ord(bytes.fromhex(signature_hex[128:130]))

    return v, r, s

@@ -1231,24 +1231,51 @@
 Account)))%0A%0A
+    if local_account:%0A%0A    
     if key i
@@ -1282,16 +1282,20 @@
 s None:%0A
+    
         
@@ -1326,16 +1326,20 @@
 vateKey%0A
+    
     else
@@ -1332,32 +1332,36 @@
 y%0A        else:%0A
+    
         pkey = k
@@ -1364,38 +1364,16 @@
  = key%0A%0A
-    if local_account:%0A
         

Fix unit tests, set pkey only if local_account