getter.gno
5.74 Kb ยท 177 lines
1package v1
2
3import (
4 "strconv"
5
6 u256 "gno.land/p/gnoswap/uint256"
7 ufmt "gno.land/p/nt/ufmt"
8
9 pl "gno.land/r/gnoswap/pool"
10 "gno.land/r/gnoswap/position"
11)
12
13// MustGetPosition returns a position for a given position ID.
14// panics if position doesn't exist
15func (p *positionV1) mustGetPosition(positionId uint64) *position.Position {
16 pos, exist := p.GetPosition(positionId)
17 if !exist {
18 panic(newErrorWithDetail(
19 errPositionDoesNotExist,
20 ufmt.Sprintf("position with position ID(%d) doesn't exist", positionId),
21 ))
22 }
23
24 return &pos
25}
26
27// GetPositionCount returns the total number of positions.
28func (p *positionV1) GetPositionCount() int {
29 return p.store.GetPositions().Size()
30}
31
32// GetPositionIDs returns a paginated list of position IDs.
33func (p *positionV1) GetPositionIDs(offset, count int) []uint64 {
34 positions := p.store.GetPositions()
35 positionIDs := make([]uint64, 0)
36
37 positions.IterateByOffset(offset, count, func(key string, _ any) bool {
38 id, err := strconv.ParseUint(key, 10, 64)
39 if err != nil {
40 return false
41 }
42
43 positionIDs = append(positionIDs, id)
44
45 return false
46 })
47
48 return positionIDs
49}
50
51// GetPosition returns a position for a given position ID.
52// Returns false if position doesn't exist
53func (p *positionV1) GetPosition(id uint64) (position.Position, bool) {
54 return p.store.GetPosition(id)
55}
56
57// ExistPosition checks if a position exists for a given position ID
58func (p *positionV1) ExistPosition(positionId uint64) bool {
59 return p.store.HasPosition(positionId)
60}
61
62func (p *positionV1) IsBurned(positionId uint64) bool {
63 return p.mustGetPosition(positionId).Burned()
64}
65
66func (p *positionV1) IsInRange(positionId uint64) bool {
67 position := p.mustGetPosition(positionId)
68 poolPath := position.PoolKey()
69 poolCurrentTick := pl.GetSlot0Tick(poolPath)
70
71 return position.TickLower() <= poolCurrentTick && poolCurrentTick < position.TickUpper()
72}
73
74func (p *positionV1) GetPositionOperator(positionId uint64) address {
75 return p.mustGetPosition(positionId).Operator()
76}
77
78func (p *positionV1) GetPositionPoolKey(positionId uint64) string {
79 return p.mustGetPosition(positionId).PoolKey()
80}
81
82func (p *positionV1) GetPositionTickLower(positionId uint64) int32 {
83 return p.mustGetPosition(positionId).TickLower()
84}
85
86func (p *positionV1) GetPositionTickUpper(positionId uint64) int32 {
87 return p.mustGetPosition(positionId).TickUpper()
88}
89
90func (p *positionV1) GetPositionLiquidity(positionId uint64) *u256.Uint {
91 return p.mustGetPosition(positionId).Liquidity()
92}
93
94func (p *positionV1) GetPositionFeeGrowthInside0LastX128(positionId uint64) *u256.Uint {
95 return p.mustGetPosition(positionId).FeeGrowthInside0LastX128()
96}
97
98func (p *positionV1) GetPositionFeeGrowthInside1LastX128(positionId uint64) *u256.Uint {
99 return p.mustGetPosition(positionId).FeeGrowthInside1LastX128()
100}
101
102func (p *positionV1) GetPositionTokensOwed0(positionId uint64) *u256.Uint {
103 return p.mustGetPosition(positionId).TokensOwed0()
104}
105
106func (p *positionV1) GetPositionTokensOwed1(positionId uint64) *u256.Uint {
107 return p.mustGetPosition(positionId).TokensOwed1()
108}
109
110func (p *positionV1) GetPositionOwner(positionId uint64) address {
111 owner, err := p.nftAccessor.OwnerOf(positionIdFrom(positionId))
112 if err != nil {
113 panic(newErrorWithDetail(
114 errDataNotFound, err.Error()))
115 }
116
117 return owner
118}
119
120func (p *positionV1) GetUnclaimedFee(positionId uint64) (*u256.Uint, *u256.Uint) {
121 // ref: https://blog.uniswap.org/uniswap-v3-math-primer-2#calculating-uncollected-fees
122 position := p.mustGetPosition(positionId)
123
124 liquidity := position.Liquidity()
125 tickLower := position.TickLower()
126 tickUpper := position.TickUpper()
127
128 poolKey := position.PoolKey()
129
130 currentTick := pl.GetSlot0Tick(poolKey)
131
132 feeGrowthGlobal0X128, feeGrowthGlobal1X128 := pl.GetFeeGrowthGlobalX128(poolKey)
133 tickUpperFeeGrowthOutside0X128, tickUpperFeeGrowthOutside1X128 := pl.GetTickFeeGrowthOutsideX128(poolKey, tickUpper)
134 tickLowerFeeGrowthOutside0X128, tickLowerFeeGrowthOutside1X128 := pl.GetTickFeeGrowthOutsideX128(poolKey, tickLower)
135
136 feeGrowthInside0LastX128 := position.FeeGrowthInside0LastX128()
137 feeGrowthInside1LastX128 := position.FeeGrowthInside1LastX128()
138
139 var tickLowerFeeGrowthBelow0, tickLowerFeeGrowthBelow1, tickUpperFeeGrowthAbove0, tickUpperFeeGrowthAbove1 *u256.Uint
140
141 if currentTick >= tickUpper {
142 tickUpperFeeGrowthAbove0 = subUint256(feeGrowthGlobal0X128, tickUpperFeeGrowthOutside0X128)
143 tickUpperFeeGrowthAbove1 = subUint256(feeGrowthGlobal1X128, tickUpperFeeGrowthOutside1X128)
144 } else {
145 tickUpperFeeGrowthAbove0 = tickUpperFeeGrowthOutside0X128
146 tickUpperFeeGrowthAbove1 = tickUpperFeeGrowthOutside1X128
147 }
148
149 if currentTick >= tickLower {
150 tickLowerFeeGrowthBelow0 = tickLowerFeeGrowthOutside0X128
151 tickLowerFeeGrowthBelow1 = tickLowerFeeGrowthOutside1X128
152 } else {
153 tickLowerFeeGrowthBelow0 = subUint256(feeGrowthGlobal0X128, tickLowerFeeGrowthOutside0X128)
154 tickLowerFeeGrowthBelow1 = subUint256(feeGrowthGlobal1X128, tickLowerFeeGrowthOutside1X128)
155 }
156
157 feeGrowthInside0X128 := subUint256(feeGrowthGlobal0X128, tickLowerFeeGrowthBelow0)
158 feeGrowthInside0X128 = subUint256(feeGrowthInside0X128, tickUpperFeeGrowthAbove0)
159
160 feeGrowthInside1X128 := subUint256(feeGrowthGlobal1X128, tickLowerFeeGrowthBelow1)
161 feeGrowthInside1X128 = subUint256(feeGrowthInside1X128, tickUpperFeeGrowthAbove1)
162
163 diffGrowthInside0X128 := subUint256(feeGrowthInside0X128, feeGrowthInside0LastX128)
164 unclaimedFee0X128, overflow := u256.Zero().MulOverflow(liquidity, diffGrowthInside0X128)
165 if overflow {
166 panic(errOverflow)
167 }
168 unclaimedFee0 := u256.Zero().Div(unclaimedFee0X128, q128)
169
170 diffGrowthInside1X128 := subUint256(feeGrowthInside1X128, feeGrowthInside1LastX128)
171 unclaimedFee1X128, overflow := u256.Zero().MulOverflow(liquidity, diffGrowthInside1X128)
172 if overflow {
173 panic(errOverflow)
174 }
175
176 return unclaimedFee0, u256.Zero().Div(unclaimedFee1X128, q128)
177}
