package com.oneul.service;

import org.web3j.crypto.Credentials;
import org.web3j.protocol.Web3j;
import org.web3j.protocol.core.methods.response.TransactionReceipt;
import org.web3j.protocol.http.HttpService;
import org.web3j.tx.ClientTransactionManager;
import org.web3j.tx.TransactionManager;
import org.web3j.tx.gas.ContractGasProvider;
import org.web3j.tx.gas.StaticGasProvider;

import com.oneul.contract.DonationSystem;

import java.security.SecureRandom;

import java.math.BigInteger;

public class DonationSystemClient {

    private static final String PRIVATE_KEY = "geth에서 8545 포트에 배포했던 유저 개인키 값";
    private static final String INFURA_URL = "http://localhost:8545";
    private static final BigInteger GAS_PRICE = BigInteger.valueOf(20000000000L);
    private static final BigInteger GAS_LIMIT = BigInteger.valueOf(6721975L);

    private Web3j web3;
    private DonationSystem contract;
    private Credentials credentials;
    private TransactionManager transactionManager;
    private ContractGasProvider gasProvider;

    public DonationSystemClient() {
        this.web3 = Web3j.build(new HttpService(INFURA_URL));
        this.credentials = Credentials.create(PRIVATE_KEY);
        this.transactionManager = new ClientTransactionManager(web3, credentials.getAddress());
        this.gasProvider = new StaticGasProvider(GAS_PRICE, GAS_LIMIT);

        try {
            this.contract = DonationSystem.deploy(
                    web3,
                    transactionManager,
                    gasProvider).send();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public String registerUser() {
        try {
            while (true) {
                String userAddress = generateRandomAddress();
                TransactionReceipt registerReceipt = contract.register(userAddress).send();
                // 성공한 경우
                if (registerReceipt.isStatusOK()) {
                    return userAddress;
                }
                // 실패한 경우
                else {
                    System.out.println("Failed to register user, retrying...");
                    // 잠시 대기 후 재시도
                    Thread.sleep(1000); // 1초 대기
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            return "Failed to register user";
        }
    }

    public static String generateRandomAddress() {
        SecureRandom random = new SecureRandom();
        byte[] addressBytes = new byte[20]; // 20 바이트 크기의 주소를 생성
        random.nextBytes(addressBytes);
        return new BigInteger(1, addressBytes).toString(16); // 16진수로 변환하여 반환
    }

    public String fundAccount(String userAddress, BigInteger amount) {
        try {
            TransactionReceipt fundReceipt = contract.fundAccount(userAddress, amount, amount).send(); // 1 ETH
            System.out.println(fundReceipt);
            return "Account funded successfully with amount: " + amount.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return "Failed to fund account";
        }
    }

    public String withdrawFromAccount(String userAddress, BigInteger amount) {
        try {
            TransactionReceipt withdrawReceipt = contract.withdrawFromAccount(userAddress, amount).send(); // 1 ETH
            return "Amount withdrawn successfully: " + amount.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return "Failed to withdraw amount";
        }
    }

    public String getBalance(String userAddress) {
        try {
            System.out.println(userAddress);
            BigInteger balance = contract.getBalance(userAddress).send();
            return "Balance: " + balance.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return "Failed to get balance";
        }
    }

    public String transfer(String from, String to, BigInteger amount) {
        try {
            TransactionReceipt transferReceipt = contract.transfer(from, to, amount).send();
            return "Transfer completed successfully";
        } catch (Exception e) {

            return "Failed to transfer";
        }
    }
}

package com.oneul.controller;

import com.oneul.service.DonationSystemClient;
import org.springframework.web.bind.annotation.*;

import java.math.BigInteger;

@RestController
@RequestMapping("/donation")
public class DonationController {

    private final DonationSystemClient donationSystemClient;

    public DonationController() {
        this.donationSystemClient = new DonationSystemClient();
    }

    @PostMapping("/register")
    public String registerUser() {
        return donationSystemClient.registerUser();
    }

    @PostMapping("/fund")
    public String fundAccount(@RequestParam String userAddress, @RequestParam BigInteger amount) {
        return donationSystemClient.fundAccount(userAddress, amount);
    }

    @PostMapping("/withdraw")
    public String withdrawFromAccount(@RequestParam String userAddress, @RequestParam BigInteger amount) {
        return donationSystemClient.withdrawFromAccount(userAddress, amount);
    }

    @PostMapping("/balance")
    public String getBalance(@RequestParam String userAddress) {
        return donationSystemClient.getBalance(userAddress);
    }

    @PostMapping("/transfer")
    public String transfer(@RequestParam String from, @RequestParam String to, @RequestParam BigInteger amount) {
        return donationSystemClient.transfer(from, to, amount);
    }
}