forked from TrueCloudLab/lego
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package dreamhost
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/go-acme/lego/v4/log"
|
|
)
|
|
|
|
const (
|
|
defaultBaseURL = "https://api.dreamhost.com"
|
|
|
|
cmdAddRecord = "dns-add_record"
|
|
cmdRemoveRecord = "dns-remove_record"
|
|
)
|
|
|
|
type apiResponse struct {
|
|
Data string `json:"data"`
|
|
Result string `json:"result"`
|
|
}
|
|
|
|
func (d *DNSProvider) buildQuery(action, domain, txt string) (*url.URL, error) {
|
|
u, err := url.Parse(d.config.BaseURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
query := u.Query()
|
|
query.Set("key", d.config.APIKey)
|
|
query.Set("cmd", action)
|
|
query.Set("format", "json")
|
|
query.Set("record", domain)
|
|
query.Set("type", "TXT")
|
|
query.Set("value", txt)
|
|
query.Set("comment", url.QueryEscape("Managed By lego"))
|
|
u.RawQuery = query.Encode()
|
|
|
|
return u, nil
|
|
}
|
|
|
|
// updateTxtRecord will either add or remove a TXT record.
|
|
// action is either cmdAddRecord or cmdRemoveRecord.
|
|
func (d *DNSProvider) updateTxtRecord(u fmt.Stringer) error {
|
|
resp, err := d.config.HTTPClient.Get(u.String())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("request failed with HTTP status code %d", resp.StatusCode)
|
|
}
|
|
|
|
raw, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to read body: %w", err)
|
|
}
|
|
|
|
var response apiResponse
|
|
err = json.Unmarshal(raw, &response)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to decode API server response: %w: %s", err, string(raw))
|
|
}
|
|
|
|
if response.Result == "error" {
|
|
return fmt.Errorf("add TXT record failed: %s", response.Data)
|
|
}
|
|
|
|
log.Infof("dreamhost: %s", response.Data)
|
|
return nil
|
|
}
|