db

type CompletePredictionParams struct {
	Status string    `json:"status"`
	Result []byte    `json:"result"`
	Error  *string   `json:"error"`
	ID     uuid.UUID `json:"id"`
}

type CreateLoginHistoryParams struct {
	UserID        uuid.UUID   `json:"user_id"`
	Success       bool        `json:"success"`
	FailureReason *string     `json:"failure_reason"`
	IpAddress     *netip.Addr `json:"ip_address"`
	UserAgent     *string     `json:"user_agent"`
	Location      *string     `json:"location"`
}

type CreateNewPredictionParams struct {
	UserID    uuid.UUID `json:"user_id"`
	TrashScan string    `json:"trash_scan"`
	Status    string    `json:"status"`
}

type CreateRefreshTokenParams struct {
	UserID    uuid.UUID `json:"user_id"`
	TokenHash string    `json:"token_hash"`
	ExpiresAt time.Time `json:"expires_at"`
}

type CreateUserParams struct {
	Name           string `json:"name"`
	Login          string `json:"login"`
	HashedPassword string `json:"hashed_password"`
	Role           string `json:"role"`
}

type DBTX interface {
	Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
	Query(context.Context, string, ...interface{}) (pgx.Rows, error)
	QueryRow(context.Context, string, ...interface{}) pgx.Row
}

type GetAdminUserByIDRow struct {
	ID            uuid.UUID          `json:"id"`
	Login         string             `json:"login"`
	Name          string             `json:"name"`
	Role          string             `json:"role"`
	Avatar        *string            `json:"avatar"`
	Deleted       bool               `json:"deleted"`
	CreatedAt     time.Time          `json:"created_at"`
	UpdatedAt     time.Time          `json:"updated_at"`
	Status        *string            `json:"status"`
	Rating        *int32             `json:"rating"`
	FilesScanned  *int32             `json:"files_scanned"`
	TotalWeight   *float64           `json:"total_weight"`
	LastScannedAt pgtype.Timestamptz `json:"last_scanned_at"`
	LastLoginAt   interface{}        `json:"last_login_at"`
}

type GetAdminUsersParams struct {
	Limit  int32 `json:"limit"`
	Offset int32 `json:"offset"`
}

type GetAdminUsersRow struct {
	ID            uuid.UUID          `json:"id"`
	Login         string             `json:"login"`
	Name          string             `json:"name"`
	Role          string             `json:"role"`
	Avatar        *string            `json:"avatar"`
	Deleted       bool               `json:"deleted"`
	CreatedAt     time.Time          `json:"created_at"`
	UpdatedAt     time.Time          `json:"updated_at"`
	Status        *string            `json:"status"`
	Rating        *int32             `json:"rating"`
	FilesScanned  *int32             `json:"files_scanned"`
	TotalWeight   *float64           `json:"total_weight"`
	LastScannedAt pgtype.Timestamptz `json:"last_scanned_at"`
	LastLoginAt   interface{}        `json:"last_login_at"`
}

type GetLoginHistoryByUserParams struct {
	UserID uuid.UUID `json:"user_id"`
	Limit  int32     `json:"limit"`
	Offset int32     `json:"offset"`
}

type GetPredictionsByUserIDParams struct {
	UserID uuid.UUID `json:"user_id"`
	Limit  int32     `json:"limit"`
	Offset int32     `json:"offset"`
}

type LoginHistory struct {
	ID            uuid.UUID   `json:"id"`
	UserID        uuid.UUID   `json:"user_id"`
	Success       bool        `json:"success"`
	FailureReason *string     `json:"failure_reason"`
	IpAddress     *netip.Addr `json:"ip_address"`
	UserAgent     *string     `json:"user_agent"`
	Location      *string     `json:"location"`
	CreatedAt     time.Time   `json:"created_at"`
}

type Prediction struct {
	ID        uuid.UUID `json:"id"`
	UserID    uuid.UUID `json:"user_id"`
	TrashScan string    `json:"trash_scan"`
	Status    string    `json:"status"`
	Result    []byte    `json:"result"`
	Error     *string   `json:"error"`
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

type Querier interface {
	CompletePrediction(ctx context.Context, arg CompletePredictionParams) error
	CountUsers(ctx context.Context) (int64, error)
	CreateLoginHistory(ctx context.Context, arg CreateLoginHistoryParams) (uuid.UUID, error)
	CreateNewPrediction(ctx context.Context, arg CreateNewPredictionParams) (Prediction, error)
	CreateRefreshToken(ctx context.Context, arg CreateRefreshTokenParams) (uuid.UUID, error)
	CreateUser(ctx context.Context, arg CreateUserParams) (uuid.UUID, error)
	DeleteUser(ctx context.Context, id uuid.UUID) error
	GetActiveTokensByUser(ctx context.Context, userID uuid.UUID) ([]RefreshToken, error)
	GetAdminUserByID(ctx context.Context, id uuid.UUID) (GetAdminUserByIDRow, error)
	GetAdminUsers(ctx context.Context, arg GetAdminUsersParams) ([]GetAdminUsersRow, error)
	GetLoginHistoryByUser(ctx context.Context, arg GetLoginHistoryByUserParams) ([]LoginHistory, error)
	GetPrediction(ctx context.Context, id uuid.UUID) (Prediction, error)
	GetPredictionsByUserID(ctx context.Context, arg GetPredictionsByUserIDParams) ([]Prediction, error)
	GetRefreshTokenByHash(ctx context.Context, tokenHash string) (RefreshToken, error)
	GetStatsByUserID(ctx context.Context, userID uuid.UUID) (Stat, error)
	GetUserByID(ctx context.Context, id uuid.UUID) (User, error)
	GetUserByLogin(ctx context.Context, login string) (User, error)
	RevokeAllUserTokens(ctx context.Context, userID uuid.UUID) error
	RevokeRefreshToken(ctx context.Context, tokenHash string) error
	UpdateStats(ctx context.Context, arg UpdateStatsParams) error
	UpdateUser(ctx context.Context, arg UpdateUserParams) error
	UpdateUserAvatar(ctx context.Context, arg UpdateUserAvatarParams) error
	UpdateUserPassword(ctx context.Context, arg UpdateUserPasswordParams) error
}

type Queries struct {
	// contains filtered or unexported fields
}

type RefreshToken struct {
	ID        uuid.UUID          `json:"id"`
	UserID    uuid.UUID          `json:"user_id"`
	TokenHash string             `json:"token_hash"`
	ExpiresAt time.Time          `json:"expires_at"`
	Revoked   bool               `json:"revoked"`
	RevokedAt pgtype.Timestamptz `json:"revoked_at"`
	CreatedAt time.Time          `json:"created_at"`
	UpdatedAt time.Time          `json:"updated_at"`
}

type Stat struct {
	ID            uuid.UUID          `json:"id"`
	UserID        uuid.UUID          `json:"user_id"`
	Status        string             `json:"status"`
	Rating        int32              `json:"rating"`
	FilesScanned  int32              `json:"files_scanned"`
	TotalWeight   float64            `json:"total_weight"`
	Achievements  []byte             `json:"achievements"`
	TrashByTypes  []byte             `json:"trash_by_types"`
	LastScannedAt pgtype.Timestamptz `json:"last_scanned_at"`
	CreatedAt     time.Time          `json:"created_at"`
	UpdatedAt     time.Time          `json:"updated_at"`
}

type UpdateStatsParams struct {
	Status        string             `json:"status"`
	Rating        int32              `json:"rating"`
	FilesScanned  int32              `json:"files_scanned"`
	TotalWeight   float64            `json:"total_weight"`
	Achievements  []byte             `json:"achievements"`
	TrashByTypes  []byte             `json:"trash_by_types"`
	LastScannedAt pgtype.Timestamptz `json:"last_scanned_at"`
	ID            uuid.UUID          `json:"id"`
}

type UpdateUserAvatarParams struct {
	Avatar *string   `json:"avatar"`
	ID     uuid.UUID `json:"id"`
}

type UpdateUserParams struct {
	Name string    `json:"name"`
	ID   uuid.UUID `json:"id"`
}

type UpdateUserPasswordParams struct {
	HashedPassword string    `json:"hashed_password"`
	ID             uuid.UUID `json:"id"`
}

type User struct {
	ID             uuid.UUID `json:"id"`
	Login          string    `json:"login"`
	Name           string    `json:"name"`
	HashedPassword string    `json:"hashed_password"`
	Role           string    `json:"role"`
	Avatar         *string   `json:"avatar"`
	Deleted        bool      `json:"deleted"`
	CreatedAt      time.Time `json:"created_at"`
	UpdatedAt      time.Time `json:"updated_at"`
}