51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package curve
|
|
|
|
import (
|
|
"git.lehouerou.net/laurent/evm"
|
|
"git.lehouerou.net/laurent/evm/curve/contracts"
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/pkg/errors"
|
|
"github.com/shopspring/decimal"
|
|
)
|
|
|
|
type Pool struct {
|
|
client evm.Client
|
|
contract *contracts.Pool
|
|
|
|
lptoken evm.Token
|
|
tokens []evm.Token
|
|
}
|
|
|
|
func NewPool(client evm.Client, address common.Address) (*Pool, error) {
|
|
contract, err := contracts.NewPool(address, client)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "init contract")
|
|
}
|
|
|
|
lptokenaddress, err := contract.LpToken(&bind.CallOpts{})
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "getting lp token address")
|
|
}
|
|
lptoken, err := client.TokenService().TokenByAddress(lptokenaddress)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "getting lp token")
|
|
}
|
|
return &Pool{
|
|
client: client,
|
|
contract: contract,
|
|
lptoken: lptoken,
|
|
}, nil
|
|
}
|
|
|
|
func (p *Pool) VirtualPrice() (decimal.Decimal, error) {
|
|
price, err := p.contract.GetVirtualPrice(&bind.CallOpts{})
|
|
if err != nil {
|
|
return decimal.Zero, errors.Wrap(err, "calling contract")
|
|
}
|
|
|
|
return decimal.NewFromBigInt(price, -18), nil
|
|
}
|
|
|
|
//func (p *Pool) CalculateTokenAmount(amounts []decimal.Decimal)
|