storage

type ReferralStatus string

type ReferralTracking struct {
	Sender         string         `db:"sender"`
	Receiver       string         `db:"receiver"`
	Status         ReferralStatus `db:"status"`
	RegisteredAt   time.Time      `db:"registered_at"`
	InstalledAt    sql.NullTime   `db:"installed_at"`
	ConfirmedAt    sql.NullTime   `db:"confirmed_at"`
	SenderReward   sql.NullInt32  `db:"sender_reward"`
	ReceiverReward sql.NullInt32  `db:"receiver_reward"`
}

type ReferralTrackingStats struct {
	Registered int `db:"registered"`
	Installed  int `db:"installed"`
	Confirmed  int `db:"confirmed"`
	Reward     int `db:"reward"`
}

type RegisterStats struct {
	Date  time.Time `json:"date"`
	Value int       `json:"value"`
}

type Request struct {
	Owner                    string         `db:"owner"`
	Email                    string         `db:"email"`
	Address                  string         `db:"address"`
	Code                     string         `db:"code"`
	CreatedAt                time.Time      `db:"created_at"`
	ConfirmedAt              sql.NullTime   `db:"confirmed_at"`
	OwnReferralCode          string         `db:"own_referral_code"`
	RegistrationReferralCode sql.NullString `db:"registration_referral_code"`
}

type Storage interface {
	// InTx runs code in transaction
	InTx(ctx context.Context, f func(s Storage) error) error
	// GetConfirmedRegistrationsTotal return a total number of all confirmed accounts (requests)
	GetConfirmedRegistrationsTotal(ctx context.Context) (int, error)
	// GetConfirmedRegistrationsStats return confirmed accounts stats for the last 30 days
	GetConfirmedRegistrationsStats(ctx context.Context) ([]*RegisterStats, error)
	// GetRequestByOwner returns request by owner.
	GetRequestByOwner(ctx context.Context, owner string) (*Request, error)
	// GetRequestByOwnReferralCode returns request by referral code.
	GetRequestByOwnReferralCode(ctx context.Context, ownReferralCode string) (*Request, error)
	// GetRequestByAddress returns request by address.
	GetRequestByAddress(ctx context.Context, address string) (*Request, error)
	// SetConfirmed sets request confirmed.
	SetConfirmed(ctx context.Context, owner string) error
	// UpsertRequest inserts request into storage.
	UpsertRequest(ctx context.Context, owner, email, address, code string, referralCode sql.NullString) error
	// CreateReferralTracking creates a new referral tracking
	CreateReferralTracking(ctx context.Context, receiver string, referralCode string) error
	// TransitionReferralTrackingToInstalled transitions referral tracking of the given referral code receiver as installed
	TransitionReferralTrackingToInstalled(ctx context.Context, receiver string) error
	// TransitionReferralTrackingToConfirmed transitions referral tracking as confirmed
	TransitionReferralTrackingToConfirmed(ctx context.Context, receiver string, senderReward, receiverReward int) error
	// GetReferralTrackingByReceiver returns referral tracking by the given receiver address
	GetReferralTrackingByReceiver(ctx context.Context, receiver string) (*ReferralTracking, error)
	// GetReferralTrackingStats returns referral tracking stats: total + 30 last days
	GetReferralTrackingStats(ctx context.Context, sender string) ([]*ReferralTrackingStats, error)
	// GetUnconfirmedReferralTracking returns referral tracking installed more than given days  ago
	GetUnconfirmedReferralTracking(ctx context.Context, days int) ([]*ReferralTracking, error)
}