clients/dns: add acme dns clientclients/dns/v0.1.0

2 files changed, 58 insertions, 0 deletions

diff --git a/clients/dns/acme.go b/clients/dns/acme.go
new file mode 100644
index 0000000..a827807
--- /dev/null
+++ b/clients/dns/acme.go
@@ -0,0 +1,55 @@
+package dns
+
+import (
+        "bytes"
+        "context"
+        "encoding/json"
+        "fmt"
+        "io/ioutil"
+        "net/http"
+        "time"
+)
+
+type AcmeDNSServiceClient struct {
+        URL    string
+        ApiKey string
+}
+
+func (c *AcmeDNSServiceClient) makeRequest(cin context.Context, method, zone, chal string) error {
+        out, err := json.Marshal(map[string]string{"challenge": chal, "zone": zone})
+        if err != nil {
+                return err
+        }
+
+        // Autocert has an overall timeout of 5 minutes
+        ctx, cancel := context.WithTimeout(cin, 3*time.Minute)
+        defer cancel()
+
+        r, err := http.NewRequestWithContext(ctx, method, c.URL, bytes.NewBuffer(out))
+        if err != nil {
+                return err
+        }
+        r.Header.Set("Content-Type", "application/json")
+        r.SetBasicAuth("u", c.ApiKey)
+
+        res, err := http.DefaultClient.Do(r)
+        if err != nil {
+                return err
+        }
+
+        // The service will only return one of these codes on success
+        if res.StatusCode != http.StatusCreated && res.StatusCode != http.StatusNoContent {
+                rb, _ := ioutil.ReadAll(res.Body)
+                return fmt.Errorf("DNS service HTTP error: %s", rb)
+        }
+
+        return nil
+}
+
+func (c *AcmeDNSServiceClient) Fulfill(ctx context.Context, domain string, record string) error {
+        return c.makeRequest(ctx, http.MethodPost, domain, record)
+}
+
+func (c *AcmeDNSServiceClient) Cleanup(ctx context.Context, domain string, record string) {
+        c.makeRequest(ctx, http.MethodDelete, domain, record)
+}
diff --git a/clients/dns/go.mod b/clients/dns/go.mod
new file mode 100644
index 0000000..c4cef7d
--- /dev/null
+++ b/clients/dns/go.mod
@@ -0,0 +1,3 @@
+module code.crute.us/mcrute/golib/clients/dns
+
+go 1.18