from web3 import Web3
import time

# web3 = Web3(Web3.HTTPProvider("https://polygon-mainnet.infura.io/v3/017957497a0d4e9c80750c18a431ac1e"))
RPC = 'https://rpc-amoy.polygon.technology/'
CONTRACT_ADDRESS = '0x46411D9d30a56F5BDD0c30595b2f91701884EBE9'
CHAIN_ID = 80002
ABI = '[ { "inputs": [], "stateMutability": "nonpayable", "type": "constructor" }, { "anonymous": false, "inputs": [ { "indexed": false, "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "indexed": false, "internalType": "string", "name": "project", "type": "string" }, { "indexed": false, "internalType": "string", "name": "data", "type": "string" } ], "name": "conversion", "type": "event" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "deleteScriptPad", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "address", "name": "pubKey", "type": "address" }, { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" }, { "internalType": "string", "name": "_data", "type": "string" } ], "name": "UploadScriptPadData", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "userId", "type": "uint256" } ], "name": "getProjectId", "outputs": [ { "internalType": "string[]", "name": "", "type": "string[]" } ], "stateMutability": "view", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "getScriptPad", "outputs": [ { "components": [ { "internalType": "address", "name": "owner", "type": "address" }, { "internalType": "string", "name": "mydata", "type": "string" }, { "internalType": "bool", "name": "Write", "type": "bool" } ], "internalType": "struct ScriptPadDataBase.privateData", "name": "", "type": "tuple" } ], "stateMutability": "view", "type": "function" } ]'
web3 = Web3(Web3.HTTPProvider(RPC))
# mainnet
# CAddress = '0x3a5d033CdF38aC4a9fe116CE88EBA2E93260044c'
# abi = '[ { "inputs": [], "stateMutability": "nonpayable", "type": "constructor" }, { "anonymous": false, "inputs": [ { "indexed": false, "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "indexed": false, "internalType": "string", "name": "project", "type": "string" }, { "indexed": false, "internalType": "string", "name": "data", "type": "string" } ], "name": "conversion", "type": "event" }, { "inputs": [ { "internalType": "address", "name": "pubKey", "type": "address" }, { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" }, { "internalType": "string", "name": "_data", "type": "string" } ], "name": "UploadScriptAuditData", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "deleteScriptAudit", "outputs": [], "stateMutability": "nonpayable", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "userId", "type": "uint256" } ], "name": "getProjectId", "outputs": [ { "internalType": "string[]", "name": "", "type": "string[]" } ], "stateMutability": "view", "type": "function" }, { "inputs": [ { "internalType": "uint256", "name": "user_id", "type": "uint256" }, { "internalType": "string", "name": "project", "type": "string" } ], "name": "getScriptAudit", "outputs": [ { "components": [ { "internalType": "address", "name": "owner", "type": "address" }, { "internalType": "string", "name": "mydata", "type": "string" }, { "internalType": "bool", "name": "Write", "type": "bool" } ], "internalType": "struct ScriptAuditDataBase.privateData", "name": "", "type": "tuple" } ], "stateMutability": "view", "type": "function" } ]'
CAddress = CONTRACT_ADDRESS
abi = ABI
contractInst = web3.eth.contract(address=CAddress, abi=abi)

def UploadScriptPadData(privatekey,pubkey,user_id,project,data): 
    try:
        print("private Key",privatekey)
        acc1 = web3.eth.account.from_key(privatekey).address
        nonce = web3.eth.get_transaction_count(acc1)
        uploadData = contractInst.functions.UploadScriptPadData(pubkey,user_id,project,data).build_transaction({
        'gasPrice': web3.eth.gas_price, 'chainId': CHAIN_ID, 'from': acc1, 'nonce': nonce})
        signed_transaction = web3.eth.account.sign_transaction(uploadData, private_key=privatekey)
        transaction_hash = web3.eth.send_raw_transaction(signed_transaction.rawTransaction)
        print("transaction hash",transaction_hash)
        transaction_receipt = web3.eth.wait_for_transaction_receipt(transaction_hash)
        tx_id = transaction_hash.hex()
        return tx_id
    except Exception as e:
        # print({"error":str(e)})
        return e

def getScriptPad(private_key,user_id,project):
    ACCOUNT = web3.eth.account.from_key(private_key).address
    try:
        getData = contractInst.functions.getScriptPad(user_id,project).call({'from':ACCOUNT})
        print(getData)
        return True ,getData
    except Exception as e:
        print("Error calling function:", e)
        return False,e
    
def getUserprojectIds(user_id):
    try:
        getProjectId = contractInst.functions.getProjectId(user_id).call()
        return getProjectId
    except Exception as e:
        return e  

def deleteScriptPad(privatekey,user_id,project): 
    try:
        acc1 = web3.eth.account.from_key(privatekey).address
        nonce = web3.eth.getTransactionCount(acc1)
        deleteData = contractInst.functions.deleteScriptPad(user_id,project).buildTransaction({
        'gasPrice': web3.eth.gas_price, 'chainId': CHAIN_ID, 'from': acc1, 'nonce': nonce})
        signed_transaction = web3.eth.account.sign_transaction(deleteData, private_key=privatekey)
        transaction_hash = web3.eth.send_raw_transaction(signed_transaction.rawTransaction)
        transaction_receipt = web3.eth.wait_for_transaction_receipt(transaction_hash)
        tx_id = transaction_hash.hex()
        return tx_id
    except Exception as e:
        print({"PPT_conversion_delete_error":str(e)})
    
# script = UploadScriptPadData("6f06e1108b833b1918067042e13e60eda262705b80385a02d0330ce0db31d3ad",'0xd7F252B08B19e35344ac44DE7CCdd26D10Cc6e17',1,'1752aa50-a751-4149-84ad-680dcb932972',"fhdsjkdfd")
# print(script)
# getScriptPad("6f06e1108b833b1918067042e13e60eda262705b80385a02d0330ce0db31d3ad",1,'1752aa50-a751-4149-84ad-680dcb932972')