Get TXT record with RandomID

oursky · Dec 15, 2023 · c4d3181 · c4d3181
1 parent 11bf4bc
commit c4d3181
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 14 deletions.
diff --git a/internal/models/domain_verification.go b/internal/models/domain_verification.go
@@ -1,9 +1,7 @@
 package models
 
 import (
-	"crypto/rand"
 	"fmt"
-	"math/big"
 	"time"
 )
 
@@ -22,23 +20,15 @@ type DomainVerification struct {
 }
 
 func NewDomainVerification(now time.Time, domain string, appID string) *DomainVerification {
-	num1, err := rand.Int(rand.Reader, big.NewInt(999999))
-	if err != nil {
-		panic(err)
-	}
-	num2, err := rand.Int(rand.Reader, big.NewInt(999999))
-	if err != nil {
-		panic(err)
-	}
 	return &DomainVerification{
 		ID:            newID("domain_verification"),
 		CreatedAt:     now,
 		UpdatedAt:     now,
 		DeletedAt:     nil,
 		Domain:        domain,
 		AppID:         appID,
-		DomainPrefix:  fmt.Sprintf("%06d", num1),
-		Value:         fmt.Sprintf("%06d", num2),
+		DomainPrefix:  RandomID(4),
+		Value:         RandomID(4),
 		WillCheckAt:   &now,
 		LastCheckedAt: nil,
 	}

diff --git a/internal/models/domain_verification_test.go b/internal/models/domain_verification_test.go
@@ -11,6 +11,6 @@ import (
 func TestDomainVerification(t *testing.T) {
 	v := models.NewDomainVerification(time.Now(), "example.com", "appId")
 	domain, value := v.GetTxtRecord()
-	assert.Regexp(t, "\\d{6}\\._pageship\\.example\\.com", domain)
-	assert.Regexp(t, "\\d{6}", value)
+	assert.Regexp(t, "\\w{7}\\._pageship\\.example\\.com", domain)
+	assert.Regexp(t, "\\w{7}", value)
 }