feat: filter by region (#107)

Signed-off-by: Chris Gianelloni <[email protected]>

Author: wolf31o2

cmd/vpn-indexer/main.go

@@ -87,7 +87,11 @@ func main() {
 	}
 
 	slog.Info(
-		fmt.Sprintf("vpn-indexer %s started", version.GetVersionString()),
+		fmt.Sprintf(
+			"vpn-indexer %s started for region %s",
+			version.GetVersionString(),
+			cfg.Vpn.Region,
+		),
 	)
 
 	// Start debug listener

internal/config/config.go

@@ -84,6 +84,7 @@ type S3Config struct {
 
 type VpnConfig struct {
 	Domain string `yaml:"domain"`
+	Region string `yaml:"region"`
 	Port   int    `yaml:"port"`
 }
 
@@ -134,6 +135,7 @@ var globalConfig = &Config{
 	},
 	Vpn: VpnConfig{
 		Domain: "test.domain",
+		Region: "test",
 		Port:   443,
 	},
 	Crl: CrlConfig{

internal/database/client.go

@@ -48,7 +48,8 @@ func (d *Database) AddClient(
 
 func (d *Database) ExpiredClients() ([]Client, error) {
 	var ret []Client
-	result := d.db.Where("expiration < datetime('now')").
+	result := d.db.
+		Where("expiration < datetime('now') AND region == ?", d.config.Vpn.Region).
 		Order("expiration").
 		Find(&ret)
 	if result.Error != nil {

internal/database/database.go

@@ -32,6 +32,7 @@ import (
 var ErrRecordNotFound = gorm.ErrRecordNotFound
 
 type Database struct {
+	config *config.Config
 	db     *gorm.DB
 	logger *slog.Logger
 }
@@ -72,6 +73,7 @@ func New(cfg *config.Config, logger *slog.Logger) (*Database, error) {
 		return nil, err
 	}
 	d := &Database{
+		config: cfg,
 		db:     db,
 		logger: logger,
 	}

internal/indexer/indexer.go

@@ -318,6 +318,11 @@ func (i *Indexer) handleEventClient(txOutput lcommon.Utxo) error {
 	if err != nil {
 		return err
 	}
+	// Check region match
+	if i.cfg.Vpn.Region != string(clientDatum.Region) {
+		// Region doesn't match, return without error
+		return nil
+	}
 	// Generate client
 	tmpClient := client.New(i.cfg, i.ca, assetName)
 	vpnHost := fmt.Sprintf(