storage

type Bet struct {
	ID     int    `db:"id" sq:"id,omitempty"`
	UserID int    `sq:"user_id,omitempty"`
	ChatID int    `sq:"chat_id,omitempty"`
	Amount int64  `sq:"amount,omitempty"`
	Sign   string `sq:"sign,omitempty"`
	Won    bool   `sq:"won,omitempty"`
	Done   bool   `sq:"done,omitempty"`
}

type Bets struct {
	*sqlx.DB
}

type BetsStorage interface {
	ByID(id int) (bet Bet, _ error)
	Create(bet *azartio.Bet, chatID Chat) error
	ByUserID(chat Chat) (bets []Bet, _ error)
	Count(chat Chat) (int, error)
	NotDoneByUserID(user Chat, chat Chat) (bets []Bet, _ error)
	MakeDoneByChat(result *azartio.RollResult, user Chat, chat Chat) error
	NotDoneByChat(chat Chat) (bets []Bet, _ error)
}

type Chat interface {
	Recipient() string
}

type DB struct {
	*sqlx.DB
	Users    UsersStorage
	Groups   GroupsStorage
	Bets     BetsStorage
	Payments PaymentsStorage
}

type Group struct {
	ID        int64  `db:"id" sq:"id,omitempty"`
	State     string `sq:"state,omitempty"`
	MessageID int64  `db:"message_id" sq:"message_id,omitempty"`
}

type Groups struct {
	*sqlx.DB
}

type GroupsStorage interface {
	Create(chat Chat) error
	UpdateMessage(chat Chat, MessageID int) error
	UpdateState(chat Chat, state string) error
	ByID(chat Chat) (group Group, _ error)
	Exists(chat Chat) (exists bool, _ error)
}

type Payment struct {
	CreatedAt time.Time  `sq:"created_at,omitempty"`
	ID        int        `sq:"id,omitempty"`
	UserID    int        `sq:"user_id,omitempty"`
	Target    string     `sq:"target,omitempty"`
	Amount    string     `sq:"amount,omitempty"`
	Profit    string     `sq:"profit,omitempty"`
	PayAt     *time.Time `sq:"pay_at,omitempty"`
}

type Payments struct {
	*sqlx.DB
}

type PaymentsStorage interface {
	Create(payment Payment) (int, error)
	ByID(id int) (Payment, error)
	Pending(userID int, target, amount string) (Payment, error)
	Update(id int, profit string, payAt time.Time) error
}

type User struct {
	CreatedAt  time.Time           `sq:"created_at,omitempty"`
	UpdatedAt  time.Time           `sq:"updated_at,omitempty"`
	ID         int                 `sq:"id,omitempty"`
	Ref        string              `sq:"ref,omitempty"`
	Balance    int64               `sq:"balance,omitempty"`
	LastBonus  time.Time           `sq:"last_bonus,omitempty"`
	Subscribed bool                `sq:"subscribed,omitempty"`
	PerksArray pgtype.VarcharArray `db:"perks" sq:"perks,omitempty"`
}

type Users struct {
	*sqlx.DB
}

type UsersStorage interface {
	Create(chat Chat, ref string) error
	Exists(chat Chat) (bool, error)
	Charge(amount int64, chat Chat) error
	ByID(chat Chat) (usr User, _ error)
	Balance(chat Chat) (a int64, _ error)
	Friends(chat Chat) (int, error)
	IsLastBonusUsed(chat Chat) (bool, error)
	UseBonus(chat Chat) error
	Subscribed(chat Chat) bool
	SetSubscribed(chat Chat, s bool) error
	Leaderboard() (users []User, _ error)
	AddPerk(chat Chat, perk string) error
}