synapsecns/sanguine

// Package base provides a base store for the screener-api.
package base

import (
    "context"
    "errors"
    "fmt"

    "github.com/synapsecns/sanguine/contrib/screener-api/db"
    "github.com/synapsecns/sanguine/core/metrics"
    "gorm.io/gorm"
    "gorm.io/gorm/clause"
)

// Store is a store that implements an underlying gorm db.
type Store struct {
    db      *gorm.DB
    metrics metrics.Handler
}

// NewStore creates a new store.
func NewStore(db *gorm.DB, metrics metrics.Handler) *Store {
    return &Store{db: db, metrics: metrics}
}

// GetAllModels gets all models to migrate.
// see: https://medium.com/@SaifAbid/slice-interfaces-8c78f8b6345d for an explanation of why we can't do this at initialization time
func GetAllModels() (allModels []interface{}) {
    allModels = append(allModels, &db.BlacklistedAddress{})

    return allModels
}

// GetBlacklistedAddress queries the db for the blacklisted address.
// Returns true if the address is blacklisted, false otherwise.
// Not used currently.
func (s *Store) GetBlacklistedAddress(ctx context.Context, address string) (*db.BlacklistedAddress, error) {
    var blacklistedAddress db.BlacklistedAddress

    if err := s.db.WithContext(ctx).Where("address = ?", address).
        First(&blacklistedAddress).Error; err != nil {
        if errors.Is(err, gorm.ErrRecordNotFound) {
            return nil, db.ErrNoAddressNotFound
        }
        return nil, fmt.Errorf("failed to get blacklisted address: %w", err)
    }

    return &blacklistedAddress, nil
}

// PutBlacklistedAddress puts the blacklisted address in the underlying db.
func (s *Store) PutBlacklistedAddress(ctx context.Context, body db.BlacklistedAddress) error {
    dbTx := s.db.WithContext(ctx).Model(&db.BlacklistedAddress{}).
        Clauses(clause.OnConflict{
            Columns: []clause.Column{
                {Name: idName},
            },
            DoUpdates: clause.AssignmentColumns([]string{
                idName, typeName, dataAddressName, dataRemarkName, dataNetworkName, dataTagName},
            ),
        }).Create(&body)
    if dbTx.Error != nil {
        return fmt.Errorf("failed to store blacklisted address: %w", dbTx.Error)
    }

    return nil
}

// UpdateBlacklistedAddress updates the blacklisted address in the underlying db.
func (s *Store) UpdateBlacklistedAddress(ctx context.Context, id string, body db.BlacklistedAddress) error {
    dbTx := s.db.WithContext(ctx).Model(&db.BlacklistedAddress{}).
        Where("id = ?", id).Updates(body)
    if dbTx.Error != nil {
        return fmt.Errorf("failed to update blacklisted address: %w", dbTx.Error)
    }

    return nil
}

// DeleteBlacklistedAddress deletes the blacklisted address from the underlying db.
func (s *Store) DeleteBlacklistedAddress(ctx context.Context, id string) error {
    if dbTx := s.db.WithContext(ctx).Where(
        "id = ?", id).Delete(&db.BlacklistedAddress{}); dbTx.Error != nil {
        return fmt.Errorf("failed to delete blacklisted address: %w", dbTx.Error)
    }
    return nil
}