From a6bbff2199dbd37aae8544a95f28154932f3dcc7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BE=AE=E5=87=89?= <927625802@qq.com>
Date: Wed, 26 May 2021 19:14:58 +0800
Subject: [PATCH] =?UTF-8?q?:sparkler:=20=E5=88=9B=E5=BB=BA=E9=BB=98?=
 =?UTF-8?q?=E8=AE=A4=E9=85=8D=E7=BD=AE=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bootstrap/config.go | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/bootstrap/config.go b/bootstrap/config.go
index 790d3635..20bc7da9 100644
--- a/bootstrap/config.go
+++ b/bootstrap/config.go
@@ -1,13 +1,11 @@
 package bootstrap
 
 import (
-	"bufio"
 	"github.com/Xhofe/alist/conf"
 	"github.com/Xhofe/alist/utils"
 	log "github.com/sirupsen/logrus"
 	"gopkg.in/yaml.v2"
 	"io/ioutil"
-	"os"
 	"strings"
 )
 
@@ -16,8 +14,9 @@ func ReadConf(config string) bool {
 	log.Infof("读取配置文件...")
 	if !utils.Exists(config) {
 		log.Infof("找不到配置文件:%s", config)
-		Write()
-		return false
+		if !Write(config) {
+			return false
+		}
 	}
 	confFile, err := ioutil.ReadFile(config)
 	if err != nil {
@@ -34,15 +33,16 @@ func ReadConf(config string) bool {
 	conf.Origins = strings.Split(conf.Conf.Server.SiteUrl, ",")
 	return true
 }
-func Write()  {
+func Write(path string) bool {
 	log.Infof("创建默认配置文件")
-	filePath :="conf.yml"
-	file,err :=os.OpenFile(filePath,os.O_WRONLY | os.O_CREATE,0644)
+	file, err := utils.CreatNestedFile(path)
 	if err != nil {
-		log.Infof("文件创建失败...")
-		return
+		log.Errorf("无法创建配置文件, %s", err)
+		return false
 	}
-	defer file.Close()
+	defer func() {
+		_ = file.Close()
+	}()
 	str := `
 info:
   title: AList #标题
@@ -75,7 +75,10 @@ database:
   type: sqlite3
   dBFile: alist.db
 `
-	writer := bufio.NewWriter(file)
-	writer.WriteString(str)
-	writer.Flush()
+	_, err = file.WriteString(str)
+	if err != nil {
+		log.Errorf("无法写入配置文件, %s", err)
+		return false
+	}
+	return true
 }