feat(thoth): make ASNChecker instances

Signed-off-by: Xe Iaso <me@xeiaso.net>
This commit is contained in:
Xe Iaso
2025-05-21 15:51:11 -04:00
parent 86ad85909c
commit 4aea22fac5
3 changed files with 22 additions and 3 deletions

View File

@@ -4,8 +4,11 @@ import (
"context"
"crypto/tls"
"fmt"
"strings"
"time"
"github.com/TecharoHQ/anubis/internal"
"github.com/TecharoHQ/anubis/lib/policy"
iptoasnv1 "github.com/TecharoHQ/thoth-proto/gen/techaro/thoth/iptoasn/v1"
grpcprom "github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus"
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/timeout"
@@ -70,3 +73,19 @@ func New(ctx context.Context, thothURL, apiToken string) (*Client, error) {
func (c *Client) Close() error {
return c.conn.Close()
}
func (c *Client) ASNCheckerFor(asns []uint32) policy.Checker {
asnMap := map[uint32]struct{}{}
var sb strings.Builder
fmt.Fprintln(&sb, "ASNChecker")
for _, asn := range asns {
asnMap[asn] = struct{}{}
fmt.Fprintln(&sb, "AS", asn)
}
return &ASNChecker{
iptoasn: c.iptoasn,
asns: asnMap,
hash: internal.SHA256sum(sb.String()),
}
}