create record if it does not exist

Author: qdm12

internal/settings/providers/aliyun/create.go

@@ -0,0 +1,65 @@
+package aliyun
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net"
+	"net/http"
+	"net/url"
+
+	"github.com/qdm12/ddns-updater/internal/settings/constants"
+	"github.com/qdm12/ddns-updater/internal/settings/errors"
+	"github.com/qdm12/ddns-updater/internal/settings/utils"
+)
+
+func (p *Provider) createRecord(ctx context.Context,
+	client *http.Client, ip net.IP) (recordID string, err error) {
+	recordType := constants.A
+	if ip.To4() == nil {
+		recordType = constants.AAAA
+	}
+
+	u := &url.URL{
+		Scheme: "https",
+		Host:   "alidns.aliyuncs.com",
+	}
+	values := newURLValues(p.accessKeyID)
+	values.Set("Action", "AddDomainRecord")
+	values.Set("DomainName", p.domain)
+	values.Set("RR", p.host)
+	values.Set("Type", recordType)
+	values.Set("Value", ip.String())
+
+	sign(http.MethodGet, values, p.accessSecret)
+
+	u.RawQuery = values.Encode()
+
+	request, err := http.NewRequestWithContext(ctx, http.MethodGet, u.String(), nil)
+	if err != nil {
+		return "", fmt.Errorf("%w: %w", errors.ErrBadRequest, err)
+	}
+
+	response, err := client.Do(request)
+	if err != nil {
+		return "", fmt.Errorf("doing HTTP request: %w", err)
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("%w: %d: %s",
+			errors.ErrBadHTTPStatus, response.StatusCode,
+			utils.BodyToSingleLine(response.Body))
+	}
+
+	var data struct {
+		RecordID string `json:"RecordId"`
+	}
+	decoder := json.NewDecoder(response.Body)
+	err = decoder.Decode(&data)
+	if err != nil {
+		return "", fmt.Errorf("%w: %w", errors.ErrUnmarshalResponse, err)
+	}
+
+	return data.RecordID, nil
+}

internal/settings/providers/aliyun/getrecord.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"io"
 	"net/http"
 	"net/url"
 
@@ -41,6 +42,23 @@ func (p *Provider) getRecordID(ctx context.Context, client *http.Client,
 	switch response.StatusCode {
 	case http.StatusOK:
 	case http.StatusNotFound:
+		return "", fmt.Errorf("%w", errors.ErrRecordNotFound)
+	case http.StatusBadRequest:
+		bodyBytes, err := io.ReadAll(response.Body)
+		if err != nil {
+			return "", fmt.Errorf("reading response body: %w", err)
+		}
+
+		var data struct {
+			Code string `json:"Code"`
+		}
+		err = json.Unmarshal(bodyBytes, &data)
+		if err != nil || data.Code != "InvalidDomainName.NoExist" {
+			return "", fmt.Errorf("%w: %d: %s",
+				errors.ErrBadHTTPStatus, response.StatusCode,
+				utils.BodyToSingleLine(response.Body))
+		}
+
 		return "", fmt.Errorf("%w", errors.ErrRecordNotFound)
 	default:
 		return "", fmt.Errorf("%w: %d: %s",

internal/settings/providers/aliyun/provider.go

@@ -3,6 +3,7 @@ package aliyun
 import (
 	"context"
 	"encoding/json"
+	stderrors "errors"
 	"fmt"
 	"net"
 	"net/http"
@@ -103,7 +104,12 @@ func (p *Provider) Update(ctx context.Context, client *http.Client, ip net.IP) (
 	}
 
 	recordID, err := p.getRecordID(ctx, client, recordType)
-	if err != nil {
+	if stderrors.Is(err, errors.ErrRecordNotFound) {
+		recordID, err = p.createRecord(ctx, client, ip)
+		if err != nil {
+			return nil, fmt.Errorf("%w: %w", errors.ErrCreateRecord, err)
+		}
+	} else if err != nil {
 		return nil, fmt.Errorf("%w: %w", errors.ErrGetRecordID, err)
 	}