sdk/referral_code_view_controller.go at main · urnetwork/sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package sdk

import (
	"context"
	"sync"

	"github.com/urnetwork/glog"

	"github.com/urnetwork/connect"
)

type ReferralCodeListener interface {
	ReferralCodeUpdated(string)
}

type ReferralCodeViewController struct {
	ctx    context.Context
	cancel context.CancelFunc

	stateLock sync.Mutex

	isFetching bool

	device Device

	referralCodeListeners *connect.CallbackList[ReferralCodeListener]
}

func newReferralCodeViewController(ctx context.Context, device Device) *ReferralCodeViewController {

	cancelCtx, cancel := context.WithCancel(ctx)

	vc := &ReferralCodeViewController{
		ctx:    cancelCtx,
		cancel: cancel,
		device: device,

		isFetching: false,

		referralCodeListeners: connect.NewCallbackList[ReferralCodeListener](),
	}
	return vc
}

func (self *ReferralCodeViewController) AddReferralCodeListener(listener ReferralCodeListener) Sub {
	callbackId := self.referralCodeListeners.Add(listener)
	return newSub(func() {
		self.referralCodeListeners.Remove(callbackId)
	})
}

func (self *ReferralCodeViewController) referralCodeChanged(code string) {
	for _, listener := range self.referralCodeListeners.Get() {
		connect.HandleError(func() {
			listener.ReferralCodeUpdated(code)
		})
	}
}

func (self *ReferralCodeViewController) Start() {
	go connect.HandleError(self.fetchNetworkReferralCode)
}

func (self *ReferralCodeViewController) Stop() {}

func (self *ReferralCodeViewController) Close() {
	glog.Info("[rcvc]close")

	self.cancel()
}

func (self *ReferralCodeViewController) setIsFetching(isFetching bool) {
	func() {
		self.stateLock.Lock()
		defer self.stateLock.Unlock()
		self.isFetching = isFetching
	}()
}

func (self *ReferralCodeViewController) fetchNetworkReferralCode() {

	if !self.isFetching {
		self.device.GetApi().GetNetworkReferralCode(
			GetNetworkReferralCodeCallback(
				connect.NewApiCallback[*GetNetworkReferralCodeResult](
					func(result *GetNetworkReferralCodeResult, err error) {
						if err != nil {
							self.setIsFetching(false)
							glog.Infof("[rcvc]error fetching referral code: %s", err)
							return
						}

						if result != nil && result.ReferralCode != "" {
							self.referralCodeChanged(result.ReferralCode)
						}

						self.setIsFetching(false)
					},
				),
			),
		)
	}
}