mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2024-11-15 12:45:14 +08:00
89 lines
2 KiB
Go
89 lines
2 KiB
Go
|
package internetbs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Api layer for Internet.bs
|
||
|
|
|
||
|
|
type api struct {
|
||
|
|
key string
|
||
|
|
password string
|
||
|
|
}
|
||
|
|
|
||
|
|
type requestParams map[string]string
|
||
|
|
|
||
|
|
type errorResponse struct {
|
||
|
|
TransactID string `json:"transactid"`
|
||
|
|
Status string `json:"status"`
|
||
|
|
Message string `json:"message,omitempty"`
|
||
|
|
Code uint `json:"code,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type domainRecord struct {
|
||
|
|
Nameserver []string `json:"nameserver"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *api) getNameservers(domain string) ([]string, error) {
|
||
|
|
var bodyString, err = c.get("/Domain/Info", requestParams{"Domain": domain})
|
||
|
|
if err != nil {
|
||
|
|
return []string{}, errors.Errorf("Error fetching nameservers list from Internet.bs: %s", err)
|
||
|
|
}
|
||
|
|
var dr domainRecord
|
||
|
|
json.Unmarshal(bodyString, &dr)
|
||
|
|
ns := []string{}
|
||
|
|
for _, nameserver := range dr.Nameserver {
|
||
|
|
ns = append(ns, nameserver)
|
||
|
|
}
|
||
|
|
return ns, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *api) updateNameservers(ns []string, domain string) error {
|
||
|
|
rec := requestParams{}
|
||
|
|
rec["Domain"] = domain
|
||
|
|
rec["Ns_list"] = strings.Join(ns, ",")
|
||
|
|
if _, err := c.get("/Domain/Update", rec); err != nil {
|
||
|
|
return errors.Errorf("Internet.ns: Error update NS : %s", err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *api) get(endpoint string, params requestParams) ([]byte, error) {
|
||
|
|
client := &http.Client{}
|
||
|
|
req, _ := http.NewRequest("GET", "https://api.internet.bs/"+endpoint, nil)
|
||
|
|
q := req.URL.Query()
|
||
|
|
|
||
|
|
// Add auth params
|
||
|
|
q.Add("ApiKey", c.key)
|
||
|
|
q.Add("Password", c.password)
|
||
|
|
q.Add("ResponseFormat", "JSON")
|
||
|
|
|
||
|
|
for pName, pValue := range params {
|
||
|
|
q.Add(pName, pValue)
|
||
|
|
}
|
||
|
|
|
||
|
|
req.URL.RawQuery = q.Encode()
|
||
|
|
|
||
|
|
resp, err := client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return []byte{}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
bodyString, _ := ioutil.ReadAll(resp.Body)
|
||
|
|
|
||
|
|
// Got error from API ?
|
||
|
|
var errResp errorResponse
|
||
|
|
err = json.Unmarshal(bodyString, &errResp)
|
||
|
|
if errResp.Status == "FAILURE" {
|
||
|
|
return bodyString, errors.Errorf("Internet.bs API error: %s code: %d transactid: %s URL:%s%s ",
|
||
|
|
errResp.Message, errResp.Code, errResp.TransactID,
|
||
|
|
req.Host, req.URL.RequestURI())
|
||
|
|
}
|
||
|
|
|
||
|
|
return bodyString, nil
|
||
|
|
}
|