88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package bootstrap
|
|
|
|
import (
|
|
"github.com/Xhofe/alist/conf"
|
|
"github.com/Xhofe/alist/utils"
|
|
log "github.com/sirupsen/logrus"
|
|
"gopkg.in/yaml.v2"
|
|
"io/ioutil"
|
|
"strings"
|
|
)
|
|
|
|
// read config file
|
|
func ReadConf(config string) bool {
|
|
log.Infof("读取配置文件...")
|
|
if !utils.Exists(config) {
|
|
log.Infof("找不到配置文件:%s", config)
|
|
if !Write(config) {
|
|
return false
|
|
}
|
|
}
|
|
confFile, err := ioutil.ReadFile(config)
|
|
if err != nil {
|
|
log.Errorf("读取配置文件时发生错误:%s", err.Error())
|
|
return false
|
|
}
|
|
err = yaml.Unmarshal(confFile, conf.Conf)
|
|
if err != nil {
|
|
log.Errorf("加载配置文件时发生错误:%s", err.Error())
|
|
return false
|
|
}
|
|
log.Debugf("config:%+v", conf.Conf)
|
|
conf.Conf.Info.Roots = utils.GetNames()
|
|
conf.Origins = strings.Split(conf.Conf.Server.SiteUrl, ",")
|
|
conf.AllowProxies = strings.Split(conf.Conf.Server.AllowProxy,",")
|
|
return true
|
|
}
|
|
func Write(path string) bool {
|
|
log.Infof("创建默认配置文件")
|
|
file, err := utils.CreatNestedFile(path)
|
|
if err != nil {
|
|
log.Errorf("无法创建配置文件, %s", err)
|
|
return false
|
|
}
|
|
defer func() {
|
|
_ = file.Close()
|
|
}()
|
|
str := `
|
|
info:
|
|
title: AList #标题
|
|
logo: "" #网站logo 如果填写,则会替换掉默认的
|
|
footer_text: Xhofe's Blog #网页底部文字
|
|
footer_url: https://www.nn.ci #网页底部文字链接
|
|
music_img: https://img.xhofe.top/2020/12/19/0f8b57866bdb5.gif #预览音乐文件时的图片
|
|
check_update: true #前端是否显示更新
|
|
script: #自定义脚本,可以是脚本的链接，也可以直接是脚本内容
|
|
autoplay: true #视频是否自动播放
|
|
preview:
|
|
text: [txt,htm,html,xml,java,properties,sql,js,md,json,conf,ini,vue,php,py,bat,gitignore,yml,go,sh,c,cpp,h,hpp] #要预览的文本文件的后缀，可以自行添加
|
|
server:
|
|
address: "0.0.0.0"
|
|
port: "5244"
|
|
search: true
|
|
download: true
|
|
static: dist
|
|
site_url: '*'
|
|
password: password #用于重建目录
|
|
allow_proxy: vtt
|
|
ali_drive:
|
|
api_url: https://api.aliyundrive.com/v2
|
|
max_files_count: 100
|
|
drives:
|
|
- refresh_token: xxx #refresh_token
|
|
root_folder: root #根目录的file_id
|
|
name: drive0 #盘名，多个盘不可重复，这里只是示例，不是一定要叫这个名字，可随意修改
|
|
password: pass #该盘密码，空（''）则不设密码，修改需要重建生效
|
|
hide: false #是否在主页隐藏该盘，不可全部隐藏，至少暴露一个
|
|
database:
|
|
type: sqlite3
|
|
dBFile: alist.db
|
|
`
|
|
_, err = file.WriteString(str)
|
|
if err != nil {
|
|
log.Errorf("无法写入配置文件, %s", err)
|
|
return false
|
|
}
|
|
return true
|
|
}
|