feat: add dns site check, services list, and additional twitter/x api endpointsv1.0.6

7 files changed, 195 insertions, 0 deletions

diff --git a/internal/client/client.go b/internal/client/client.go
index eaf87cb..4aa6f06 100644
--- a/internal/client/client.go
+++ b/internal/client/client.go
@@ -126,3 +126,7 @@ func (c *Client) Patch(path string, body interface{}) ([]byte, error) {
 func (c *Client) Put(path string, body interface{}) ([]byte, error) {
 	return c.doRequest(http.MethodPut, path, nil, body)
 }
+
+func (c *Client) ListServices() ([]byte, error) {
+	return c.Get("/", nil)
+}
diff --git a/internal/client/dns.go b/internal/client/dns.go
index 9f9d4e8..96ae8e1 100644
--- a/internal/client/dns.go
+++ b/internal/client/dns.go
@@ -59,3 +59,18 @@ func (c *Client) CheckDNSTLDStream(params *models.DNSTLDParams, callback func(re
 
 	return nil
 }
+
+func (c *Client) CheckDNSSite(siteURL string) (*models.DNSResponse, error) {
+	path := fmt.Sprintf("/dns/site/%s", url.PathEscape(siteURL))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.DNSResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse DNS site response: %w", err)
+	}
+
+	return &response, nil
+}
diff --git a/internal/client/x.go b/internal/client/x.go
index b6ddba8..f385b9b 100644
--- a/internal/client/x.go
+++ b/internal/client/x.go
@@ -207,3 +207,93 @@ func (c *Client) SearchTweetsStream(query string, limit int, callback func(resul
 
 	return nil
 }
+
+func (c *Client) CheckDeletedTweets(handle string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/deleted-tweets/%s", url.PathEscape(handle))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse deleted tweets response: %w", err)
+	}
+
+	return &response, nil
+}
+
+func (c *Client) CheckDomainAge(domain string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/domain/%s", url.PathEscape(domain))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse domain age response: %w", err)
+	}
+
+	return &response, nil
+}
+
+func (c *Client) GetHoldersMap(address string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/holders/map/%s", url.PathEscape(address))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse holders map response: %w", err)
+	}
+
+	return &response, nil
+}
+
+func (c *Client) GetTwitterID(handle string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/id/%s", url.PathEscape(handle))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse Twitter ID response: %w", err)
+	}
+
+	return &response, nil
+}
+
+func (c *Client) ReverseImageSearch(address string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/image/%s", url.PathEscape(address))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse image search response: %w", err)
+	}
+
+	return &response, nil
+}
+
+func (c *Client) GetTwitterUserID(handle string) (*models.XResponse, error) {
+	path := fmt.Sprintf("/x/user-id/%s", url.PathEscape(handle))
+	data, err := c.Get(path, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	var response models.XResponse
+	if err := json.Unmarshal(data, &response); err != nil {
+		return nil, fmt.Errorf("failed to parse Twitter user ID response: %w", err)
+	}
+
+	return &response, nil
+}
diff --git a/internal/formatter/admin.go b/internal/formatter/admin.go
index 914eaaa..12ffcd0 100644
--- a/internal/formatter/admin.go
+++ b/internal/formatter/admin.go
@@ -185,3 +185,4 @@ func FormatAccountToggled(message string, asJSON bool) (string, error) {
 
 	return fmt.Sprintf("%s %s\n", Blue("ℹ"), message), nil
 }
+
diff --git a/internal/formatter/dns.go b/internal/formatter/dns.go
index 6abd5b0..8e66039 100644
--- a/internal/formatter/dns.go
+++ b/internal/formatter/dns.go
@@ -1,6 +1,7 @@
 package formatter
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 	"strings"
@@ -55,3 +56,37 @@ func FormatDNSResults(result *models.DomainResult, asJSON bool) (string, error)
 
 	return sb.String(), nil
 }
+
+func FormatDNSSite(resp *models.DNSResponse, asJSON bool) (string, error) {
+	if asJSON {
+		data, err := json.MarshalIndent(resp, "", "  ")
+		if err != nil {
+			return "", fmt.Errorf("failed to marshal JSON: %w", err)
+		}
+		return string(data), nil
+	}
+
+	var buf bytes.Buffer
+
+	if resp.Error != "" {
+		fmt.Fprintf(&buf, "%s\n", Red(fmt.Sprintf("Error: %s", resp.Error)))
+		return buf.String(), nil
+	}
+
+	if resp.Message != "" {
+		fmt.Fprintf(&buf, "%s\n", Bold(Cyan(resp.Message)))
+	}
+
+	if resp.Response != "" {
+		fmt.Fprintf(&buf, "%s\n", resp.Response)
+	}
+
+	if resp.Data != nil && len(resp.Data) > 0 {
+		dataJSON, err := json.MarshalIndent(resp.Data, "", "  ")
+		if err == nil {
+			fmt.Fprintf(&buf, "\n%s\n", string(dataJSON))
+		}
+	}
+
+	return buf.String(), nil
+}
diff --git a/internal/formatter/x.go b/internal/formatter/x.go
index f701797..a79d173 100644
--- a/internal/formatter/x.go
+++ b/internal/formatter/x.go
@@ -377,3 +377,46 @@ func wrapText(text string, width int) []string {
 
 	return lines
 }
+
+func FormatXGeneric(resp *models.XResponse, asJSON bool) (string, error) {
+	if asJSON {
+		data, err := json.MarshalIndent(resp, "", "  ")
+		if err != nil {
+			return "", fmt.Errorf("failed to marshal JSON: %w", err)
+		}
+		return string(data), nil
+	}
+
+	var buf bytes.Buffer
+
+	if resp.Error != "" {
+		fmt.Fprintf(&buf, "%s\n", Red(fmt.Sprintf("Error: %s", resp.Error)))
+		return buf.String(), nil
+	}
+
+	if resp.Message != "" {
+		fmt.Fprintf(&buf, "%s\n\n", Bold(Cyan(resp.Message)))
+	}
+
+	if resp.Response != "" {
+		fmt.Fprintf(&buf, "%s\n\n", resp.Response)
+	}
+
+	if resp.Data != nil {
+		dataJSON, err := json.MarshalIndent(resp.Data, "", "  ")
+		if err == nil {
+			fmt.Fprintf(&buf, "%s\n", string(dataJSON))
+		}
+	}
+
+	if resp.Credits.Remaining > 0 || resp.Credits.Unlimited {
+		fmt.Fprintf(&buf, "\n%s ", Blue("Credits Remaining:"))
+		if resp.Credits.Unlimited {
+			fmt.Fprintf(&buf, "%s\n", Green("Unlimited"))
+		} else {
+			fmt.Fprintf(&buf, "%s\n", FormatCredits(int64(resp.Credits.Remaining)))
+		}
+	}
+
+	return buf.String(), nil
+}
diff --git a/internal/models/dns.go b/internal/models/dns.go
index 51d9e50..776e3d6 100644
--- a/internal/models/dns.go
+++ b/internal/models/dns.go
@@ -11,3 +11,10 @@ type DomainResult struct {
 	Title  string   `json:"title,omitempty"`
 	Tech   []string `json:"tech,omitempty"`
 }
+
+type DNSResponse struct {
+	Message  string                 `json:"message,omitempty"`
+	Data     map[string]interface{} `json:"data,omitempty"`
+	Error    string                 `json:"error,omitempty"`
+	Response string                 `json:"response,omitempty"`
+}