feat: pay & repay

This commit is contained in:
snsd0805 2023-06-06 01:41:38 +08:00
parent 9d91ac234d
commit c0af84f886
Signed by: snsd0805
GPG Key ID: 569349933C77A854
3 changed files with 54 additions and 5 deletions

View File

@ -7,15 +7,26 @@ contract Bank {
using Counters for Counters.Counter; using Counters for Counters.Counter;
Counters.Counter private id_counter; Counters.Counter private id_counter;
struct Order {
bool isFinished;
uint amount;
}
SoulboundToken sbt; SoulboundToken sbt;
address owner; address owner;
mapping(address => uint) arrears; mapping(address => uint) arrears;
mapping(uint => Order) order_amount;
mapping(address => uint[]) client_orders;
mapping(address => uint) credits;
mapping(address => uint) sbt_number; mapping(address => uint) sbt_number;
constructor(address SBT_addr) { constructor(address SBT_addr) {
sbt = SoulboundToken(SBT_addr); sbt = SoulboundToken(SBT_addr);
owner = msg.sender;
} }
receive() external payable {}
modifier onlyBank { modifier onlyBank {
require(msg.sender == owner, "Only the owner can access this function."); require(msg.sender == owner, "Only the owner can access this function.");
_; _;
@ -26,6 +37,10 @@ contract Bank {
_; _;
} }
function setCredit(address client, uint amount) public onlyBank {
credits[client] = amount;
}
function register(uint number) public { function register(uint number) public {
uint target = sbt.getAccountNumber(msg.sender); uint target = sbt.getAccountNumber(msg.sender);
require(target != 0, "You don't have SBT."); require(target != 0, "You don't have SBT.");
@ -33,4 +48,38 @@ contract Bank {
sbt_number[msg.sender] = target; sbt_number[msg.sender] = target;
} }
function pay(address shop, uint amount) public onlyClient {
require(amount <= credits[msg.sender], "You don't have enough credit.");
id_counter.increment();
uint id = id_counter.current();
arrears[msg.sender] += amount;
order_amount[id] = Order(false, amount);
client_orders[msg.sender].push(id);
payable(shop).transfer(amount);
sbt.logBorrowing(msg.sender, id, amount);
}
function repay() public payable onlyClient returns (uint[] memory, uint) {
uint value = msg.value;
uint repay_amount = 0;
uint should_pay;
uint[] memory unfinished = new uint[](client_orders[msg.sender].length);
uint refund;
uint item_counter = 0;
for (uint i=0; i<client_orders[msg.sender].length; i++) {
should_pay = order_amount[client_orders[msg.sender][i]].amount;
if ((value - repay_amount) >= should_pay){
repay_amount += should_pay;
order_amount[client_orders[msg.sender][i]].isFinished = true;
} else {
unfinished[item_counter] = (client_orders[msg.sender][i]);
item_counter += 1;
}
}
refund = value - repay_amount;
if (refund != 0){
payable(msg.sender).transfer(refund);
}
return (unfinished, refund);
}
} }

View File

@ -20,7 +20,7 @@ contract SoulboundToken is ERC721, Ownable {
mapping(address => uint) private address_to_number; mapping(address => uint) private address_to_number;
event Borrow(address client, address bank, uint id, uint amount); event Borrow(address client, address bank, uint id, uint amount);
event Repay(address client, address bank, uint id, uint amount, bool finish); event Repay(address client, address bank, uint id, uint amount);
event Warning(address client, address bank); event Warning(address client, address bank);
constructor() ERC721("Credit System Soulbound Token", "CS_SBT") {} constructor() ERC721("Credit System Soulbound Token", "CS_SBT") {}
@ -90,8 +90,8 @@ contract SoulboundToken is ERC721, Ownable {
emit Borrow(client, msg.sender, id, amount); emit Borrow(client, msg.sender, id, amount);
} }
function logRepaying(address client, uint id, uint amount, bool finish) public onlyBank { function logRepaying(address client, uint id, uint amount) public onlyBank {
emit Repay(client, msg.sender, id, amount, finish); emit Repay(client, msg.sender, id, amount);
} }
function logWarning(address client) public onlyBank { function logWarning(address client) public onlyBank {

View File

@ -6,6 +6,6 @@ module.exports = function(_deployer) {
_deployer.deploy(SBT).then((SBT_instance) => { _deployer.deploy(SBT).then((SBT_instance) => {
return _deployer.deploy(Bank, SBT_instance.address).then((Bank_instance) => { return _deployer.deploy(Bank, SBT_instance.address).then((Bank_instance) => {
return SBT_instance.addReliableBank(Bank_instance.address); return SBT_instance.addReliableBank(Bank_instance.address);
}) });
}) });
}; };