112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
|
package query
|
||
|
|
|
||
|
|
import (
|
||
|
|
"dehasher/internal/sqlite"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DehashedClient struct {
|
||
|
|
key string
|
||
|
|
email string
|
||
|
|
results []sqlite.Result
|
||
|
|
client *http.Client
|
||
|
|
query string
|
||
|
|
params string
|
||
|
|
printBal bool
|
||
|
|
total int
|
||
|
|
balance int
|
||
|
|
}
|
||
|
|
|
||
|
|
var baseUrl = "https://api.dehashed.com/v2/search"
|
||
|
|
|
||
|
|
func NewDehashedClient(key, email string, printBal bool) *DehashedClient {
|
||
|
|
return &DehashedClient{key: key, email: email, results: make([]sqlite.Result, 0), client: &http.Client{}, printBal: printBal}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) getKey() string {
|
||
|
|
return dc.key
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) getEmail() string {
|
||
|
|
return dc.email
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) GetResults() sqlite.DehashedResults {
|
||
|
|
return sqlite.DehashedResults{Results: dc.results}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) buildQuery(params map[string]string) {
|
||
|
|
urlParams := url.Values{}
|
||
|
|
urlString := baseUrl
|
||
|
|
|
||
|
|
if len(params) > 0 {
|
||
|
|
urlString += "?query="
|
||
|
|
|
||
|
|
for k, v := range params {
|
||
|
|
if len(v) > 0 {
|
||
|
|
urlParams.Add(k, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
tmp, _ := url.QueryUnescape(urlParams.Encode())
|
||
|
|
tmp2 := strings.Replace(tmp, "=", ":", -1)
|
||
|
|
dc.params = tmp2
|
||
|
|
urlString += dc.params
|
||
|
|
dc.query = urlString
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) setResults(results int) {
|
||
|
|
dc.query = fmt.Sprintf("%s?query=%s&size=%d", baseUrl, dc.params, results)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) setPage(page int) {
|
||
|
|
dc.query = fmt.Sprintf("%s&nextPage=%d", dc.query, page)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) Do() int {
|
||
|
|
fmt.Printf("\n\t[*] Performing Request...")
|
||
|
|
req, err := http.NewRequest("GET", dc.query, nil)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("[!] Error constructing request: %v", err)
|
||
|
|
os.Exit(-1)
|
||
|
|
}
|
||
|
|
|
||
|
|
dc.setAuth(req)
|
||
|
|
req.Header.Add("Dehashed-Api-Key", dc.getKey())
|
||
|
|
req.Header.Add("Accept", "application/json")
|
||
|
|
resp, err := dc.client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("[!] Error performing request: %s\n%v", dc.query, err)
|
||
|
|
os.Exit(-1)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.StatusCode != 200 {
|
||
|
|
dhErr := GetDehashedError(resp.StatusCode)
|
||
|
|
fmt.Println()
|
||
|
|
log.Fatal(dhErr.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
entries, balance, total := sqlite.NewDehashedResults(resp.Body)
|
||
|
|
dc.results = append(dc.results, entries...)
|
||
|
|
dc.balance = balance
|
||
|
|
dc.total += total
|
||
|
|
if dc.printBal {
|
||
|
|
fmt.Printf("\n\t\t[*] Balance Remaining: %d", balance)
|
||
|
|
}
|
||
|
|
return total
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) setAuth(r *http.Request) {
|
||
|
|
r.SetBasicAuth(dc.email, dc.key)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dc *DehashedClient) GetDomains() int {
|
||
|
|
return dc.balance
|
||
|
|
}
|