From 16a5da155cd2ba51a1e2251643e2d1b558e09082 Mon Sep 17 00:00:00 2001
From: Mmx233 <admin@mmxblog.com>
Date: Fri, 7 Apr 2023 10:52:50 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=B8=A4=E5=A4=84=E5=85=BC=E5=AE=B9?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/controllers/daemon.go            | 5 ++---
 internal/pkg/dns/cloudflare/cloudflare.go | 3 ++-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/internal/controllers/daemon.go b/internal/controllers/daemon.go
index 747c853..7a0f3a8 100644
--- a/internal/controllers/daemon.go
+++ b/internal/controllers/daemon.go
@@ -3,7 +3,6 @@ package controllers
 import (
 	"fmt"
 	"github.com/Mmx233/BitSrunLoginGo/internal/global"
-	"github.com/Mmx233/tool"
 	"github.com/howeyc/fsnotify"
 	"os"
 	"time"
@@ -22,12 +21,12 @@ var Daemon = daemon{
 
 // MarkDaemon 写入后台标记文件
 func (a *daemon) MarkDaemon() error {
-	return tool.File.WriteAll(a.Path, []byte(a.Mark))
+	return os.WriteFile(a.Path, []byte(a.Mark), 0600)
 }
 
 // CheckDaemon 检查后台标记文件
 func (a *daemon) CheckDaemon() bool {
-	if data, err := tool.File.ReadAll(a.Path); err != nil {
+	if data, err := os.ReadFile(a.Path); err != nil {
 		return false
 	} else {
 		return string(data) == a.Mark
diff --git a/internal/pkg/dns/cloudflare/cloudflare.go b/internal/pkg/dns/cloudflare/cloudflare.go
index c134cdd..4d0d408 100644
--- a/internal/pkg/dns/cloudflare/cloudflare.go
+++ b/internal/pkg/dns/cloudflare/cloudflare.go
@@ -60,9 +60,10 @@ func (a DnsProvider) SetDomainRecord(domain, ip string) error {
 		if record.Content == ip {
 			return nil
 		}
-		return a.Api.UpdateDNSRecord(context.Background(), a.ZoneResource, cloudflare.UpdateDNSRecordParams{
+		_, e = a.Api.UpdateDNSRecord(context.Background(), a.ZoneResource, cloudflare.UpdateDNSRecordParams{
 			ID:      record.ID,
 			Content: ip,
 		})
+		return e
 	}
 }