From eeb3f677cae9c13440e868a83c0a621c5b22b0b5 Mon Sep 17 00:00:00 2001 From: s Date: Fri, 7 Nov 2025 12:32:24 -0500 Subject: feat: add show-only flag to filter dns tld results by existence status --- internal/client/dns.go | 12 +++++++++++- internal/models/dns.go | 3 ++- 2 files changed, 13 insertions(+), 2 deletions(-) (limited to 'internal') diff --git a/internal/client/dns.go b/internal/client/dns.go index 08449e0..9f9d4e8 100644 --- a/internal/client/dns.go +++ b/internal/client/dns.go @@ -2,6 +2,7 @@ package client import ( "bufio" + "context" "encoding/json" "fmt" "net/http" @@ -14,6 +15,12 @@ func (c *Client) CheckDNSTLDStream(params *models.DNSTLDParams, callback func(re term := url.PathEscape(params.Term) fullURL := c.config.BaseURL + "/dns/tld/" + term + if params.ShowOnly != "" { + queryParams := url.Values{} + queryParams.Set("showOnly", params.ShowOnly) + fullURL = fullURL + "?" + queryParams.Encode() + } + req, err := http.NewRequest(http.MethodGet, fullURL, nil) if err != nil { return fmt.Errorf("failed to create request: %w", err) @@ -21,7 +28,10 @@ func (c *Client) CheckDNSTLDStream(params *models.DNSTLDParams, callback func(re req.Header.Set("User-Agent", c.config.UserAgent) - resp, err := c.httpClient.Do(req) + streamClient := &http.Client{} + req = req.WithContext(context.Background()) + + resp, err := streamClient.Do(req) if err != nil { return fmt.Errorf("request failed: %w", err) } diff --git a/internal/models/dns.go b/internal/models/dns.go index b08d8ba..51d9e50 100644 --- a/internal/models/dns.go +++ b/internal/models/dns.go @@ -1,7 +1,8 @@ package models type DNSTLDParams struct { - Term string + Term string + ShowOnly string } type DomainResult struct { -- cgit v1.2.3