From 9a7e0ef0271f255f7e3d5970a887a544f503e565 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=96=9F=E9=85=8C=20=E9=B5=AC=E5=85=84?= <tgckpg@gmail.com>
Date: Mon, 14 Mar 2022 06:39:42 +0900
Subject: [PATCH] This should work

---
 main.go                             | 38 +++++++++++++++++++++++++----
 testdata/freedns-solver/config.json |  1 -
 2 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 305aee3..dd383ba 100644
--- a/main.go
+++ b/main.go
@@ -5,12 +5,14 @@ import (
 	"encoding/json"
 	"fmt"
 	"os"
+	"strings"
 
 	extapi "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/client-go/kubernetes"
 	"k8s.io/client-go/rest"
 
+	"github.com/cert-manager/webhook-freedns/freedns"
 	"github.com/jetstack/cert-manager/pkg/acme/webhook/apis/acme/v1alpha1"
 	"github.com/jetstack/cert-manager/pkg/acme/webhook/cmd"
 )
@@ -43,7 +45,8 @@ type customDNSProviderSolver struct {
 	// 3. uncomment the relevant code in the Initialize method below
 	// 4. ensure your webhook's service account has the required RBAC role
 	//    assigned to it for interacting with the Kubernetes APIs you need.
-	client *kubernetes.Clientset
+	client  *kubernetes.Clientset
+	freedns *freedns.FreeDNS
 }
 
 // customDNSProviderConfig is a structure that is used to decode into when
@@ -68,7 +71,6 @@ type customDNSProviderConfig struct {
 
 	//Email           string `json:"email"`
 	SecretRef string `json:"secretName"`
-	Domain    string `json:"domain"`
 	//APIKeySecretRef v1.SecretKeySelector `json:"apiKeySecretRef"`
 }
 
@@ -102,9 +104,22 @@ func (c *customDNSProviderSolver) Present(ch *v1alpha1.ChallengeRequest) error {
 	username := string(secretObj.Data["username"])
 	password := string(secretObj.Data["password"])
 
-	fmt.Printf("Domain: %s, Auth: %s %s\n", cfg.Domain, username, password)
+	freedns := freedns.FreeDNS{}
+	err = freedns.Login(username, password)
+	if err != nil {
+		return err
+	}
 
-	// TODO: add code that sets a record in the DNS provider's console
+	err = freedns.SelectDomain(ch.DNSName)
+	if err != nil {
+		return err
+	}
+
+	_zone := strings.TrimRight(ch.ResolvedZone, ch.DNSName)
+	_key := "\"" + ch.Key + "\""
+	freedns.AddRecord("TXT", _zone, _key, true, "")
+
+	c.freedns = &freedns
 	return nil
 }
 
@@ -115,7 +130,20 @@ func (c *customDNSProviderSolver) Present(ch *v1alpha1.ChallengeRequest) error {
 // This is in order to facilitate multiple DNS validations for the same domain
 // concurrently.
 func (c *customDNSProviderSolver) CleanUp(ch *v1alpha1.ChallengeRequest) error {
-	// TODO: add code that deletes a record from the DNS provider's console
+	_zone := strings.TrimRight(ch.ResolvedZone, ch.DNSName)
+	_key := "\"" + ch.Key + "\""
+	_id, err := c.freedns.FindRecord(_zone, "TXT", _key)
+
+	if err != nil {
+		return err
+	}
+
+	if _id != "" {
+		err = c.freedns.DeleteRecord(_id)
+		if err != nil {
+			return err
+		}
+	}
 	return nil
 }
 
diff --git a/testdata/freedns-solver/config.json b/testdata/freedns-solver/config.json
index f1723e9..b82e05a 100644
--- a/testdata/freedns-solver/config.json
+++ b/testdata/freedns-solver/config.json
@@ -1,4 +1,3 @@
 {
 	"secretName": "freedns-auth"
-	, "domain": "example.com"
 }