106 lines
1.9 KiB
Go
106 lines
1.9 KiB
Go
package clients
|
|
|
|
import (
|
|
"dero-swap/coin"
|
|
"log"
|
|
|
|
"github.com/lesismal/nbio/nbhttp/websocket"
|
|
)
|
|
|
|
type Swap_External struct {
|
|
Nickname string `json:"nickname"`
|
|
Dero float64 `json:"dero"`
|
|
XMR float64 `json:"xmr,omitempty"`
|
|
}
|
|
|
|
func IsExternalSwapAvailable(user string, pair string, amount float64) (ok bool, client *websocket.Conn) {
|
|
|
|
Clients.Range(func(key any, value any) bool {
|
|
c := value.(ClientInfo)
|
|
for _, p := range c.PairInfo {
|
|
if p.Pair == pair && p.Balance >= amount && user == c.Nickname {
|
|
ok = true
|
|
client = key.(*websocket.Conn)
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
return
|
|
}
|
|
|
|
func PrepareExternalSwap(user string, pair string, amount float64) (bool, *websocket.Conn) {
|
|
|
|
// only XMR swaps
|
|
if pair != coin.XMRDERO && pair != coin.DEROXMR {
|
|
log.Println("Only 3rd party XMR swaps")
|
|
return false, nil
|
|
}
|
|
|
|
ok, conn := IsExternalSwapAvailable(user, pair, amount)
|
|
if !ok {
|
|
log.Println("No 3rd party swaps available")
|
|
return false, nil
|
|
}
|
|
|
|
return true, conn
|
|
}
|
|
|
|
func (c *SwapState) ChangeClientState(mode uint, conn *websocket.Conn) {
|
|
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
|
|
if mode == LOCK {
|
|
c.Client[conn] = true
|
|
} else {
|
|
c.Client[conn] = false
|
|
}
|
|
}
|
|
|
|
func (c *SwapState) CheckClientState(conn *websocket.Conn) bool {
|
|
|
|
c.RLock()
|
|
defer c.RUnlock()
|
|
|
|
return c.Client[conn]
|
|
}
|
|
|
|
func (c *SwapState) AddOrigin(conn *websocket.Conn, target *websocket.Conn) {
|
|
|
|
c.Lock()
|
|
defer c.Unlock()
|
|
|
|
c.Result[conn] = target
|
|
}
|
|
|
|
func (c *SwapState) GetOrigin(conn *websocket.Conn) *websocket.Conn {
|
|
|
|
c.RLock()
|
|
defer c.RUnlock()
|
|
|
|
return c.Result[conn]
|
|
}
|
|
|
|
func GetExternalBalances() (list []Swap_External) {
|
|
|
|
var entry Swap_External
|
|
|
|
Clients.Range(func(key any, value any) bool {
|
|
c := value.(ClientInfo)
|
|
entry.Nickname = c.Nickname
|
|
for _, p := range c.PairInfo {
|
|
switch p.Pair {
|
|
case coin.DEROXMR:
|
|
entry.XMR = p.Balance
|
|
case coin.XMRDERO:
|
|
entry.Dero = p.Balance
|
|
}
|
|
}
|
|
list = append(list, entry)
|
|
return true
|
|
})
|
|
|
|
return
|
|
}
|