✨ 自定义脚本

README.md

@@ -42,17 +42,16 @@
 #### TODO
 
 - [x] 排序
-- [ ] 文件预览
+- [x] 文件预览
   - [x] 图片
   - [x] 视频 
   - [x] 音频
-  - [ ] 文档
 - [x] `Readme`渲染
 - [x] 密码加密
 - [ ] 搜索与翻页
-- [ ] 文件直链
+- [x] 文件直链
 - [ ] 路径优化
-- [ ] 缓存
+- [x] 缓存
 
 ### 如何使用
 

alidrive/auth.go

@@ -12,7 +12,7 @@ func TokenLogin() (*TokenLoginResp, error) {
 	log.Infof("尝试使用token登录...")
 	url:="https://auth.aliyundrive.com/v2/oauth/token_login"
 	req:=TokenLoginReq{Token:conf.Conf.AliDrive.LoginToken}
-	log.Debugf("token_login_req:%v",req)
+	log.Debugf("token_login_req:%+v",req)
 	var tokenLogin TokenLoginResp
 	if body, err := DoPost(url, req,false); err != nil {
 		log.Errorf("tokenLogin-doPost出错:%s",err.Error())

alidrive/request.go

@@ -146,7 +146,7 @@ func DoPost(url string,request interface{},auth bool) (body []byte, err error) {
 		log.Errorf("创建requestBody出错:%s",err.Error())
 	}
 	req,err:=http.NewRequest("POST",url,requestBody)
-	log.Debugf("do_post_req:%v",req)
+	log.Debugf("do_post_req:%+v",req)
 	if err != nil {
 		log.Errorf("创建request出错:%s",err.Error())
 		return

bootstrap/alidrive.go

@@ -24,13 +24,13 @@ func InitAliDrive() bool {
 	}else {
 		conf.Authorization=conf.Bearer+conf.Conf.AliDrive.AccessToken
 	}
-	log.Infof("token:%s",conf.Authorization)
+	log.Debugf("token:%s",conf.Authorization)
 	user,err:=alidrive.GetUserInfo()
 	if err != nil {
 		log.Errorf("初始化用户失败:%s",err.Error())
 		return false
 	}
-	log.Infof("当前用户信息:%v",user)
+	log.Infof("当前用户信息:%+v",user)
 	alidrive.User=user
 	return true
 }

bootstrap/cmd.go

@@ -39,6 +39,7 @@ func printASC()  {
 func start() {
 	InitLog()
 	printASC()
+	CheckUpdate()
 	if !ReadConf(conf.Con) {
 		log.Errorf("读取配置文件时出现错误,启动失败.")
 		return

bootstrap/config.go

@@ -24,6 +24,6 @@ func ReadConf(config string) bool {
 		log.Errorf("加载配置文件时发生错误:%s",err.Error())
 		return false
 	}
-	log.Debugf("config:%v",conf.Conf)
+	log.Debugf("config:%+v",conf.Conf)
 	return true
 }

bootstrap/update.go

@@ -0,0 +1,41 @@
+package bootstrap
+
+import (
+	"encoding/json"
+	"github.com/Xhofe/alist/conf"
+	log "github.com/sirupsen/logrus"
+	"io/ioutil"
+	"net/http"
+)
+
+type GithubRelease struct {
+	TagName	string `json:"tag_name"`
+	HtmlUrl	string `json:"html_url"`
+	Body	string `json:"body"`
+}
+
+func CheckUpdate() {
+	url:="https://api.github.com/repos/Xhofe/alist/releases/latest"
+	resp,err:=http.Get(url)
+	if err!=nil {
+		log.Warnf("检查更新失败:%s",err.Error())
+		return
+	}
+	body,err:=ioutil.ReadAll(resp.Body)
+	if err!=nil {
+		log.Warnf("读取更新内容失败:%s",err.Error())
+		return
+	}
+	var release GithubRelease
+	err = json.Unmarshal(body,&release)
+	if err!=nil {
+		log.Warnf("解析更新失败:%s",err.Error())
+		return
+	}
+	if conf.VERSION == release.TagName {
+		log.Infof("当前已是最新版本:%s",release.TagName)
+	}else {
+		log.Infof("发现新版本:%s",release.TagName)
+		log.Infof("请至'%s'获取更新.",release.HtmlUrl)
+	}
+}

conf.yml.example

@@ -6,14 +6,16 @@ info:
   footer_text: Xhofe's Blog #网页底部文字
   footer_url: https://www.nn.ci #网页底部文字链接
   music_img: https://img.oez.cc/2020/12/19/0f8b57866bdb5.gif #预览音乐文件时的图片
+  check_update: true #前端是否显示更新
+  script: #自定义脚本,可以是脚本的链接，也可以直接是脚本内容,如document.querySelector('body').style="background-image:url('https://api.mtyqx.cn/api/random.php');background-attachment:fixed"
 server:
   port: "5244" #程序监听端口
   search: true #是否开启搜索接口,开启搜索之后密码和根目录都会失效,所以前端暂时不做搜索
   static: dist #前端文件目录
 cache:
   enable: true #是否开启缓存
-  expiration: 5 #缓存失效时间(单位:分钟)
-  cleanup_interval: 10 #清理失效缓存间隔
+  expiration: 60 #缓存失效时间(单位:分钟)
+  cleanup_interval: 120 #清理失效缓存间隔
   refresh_password: password #手动清理缓存密码
 ali_drive:
   api_url: https://api.aliyundrive.com/v2 #阿里云盘api,无需修改

conf/config.go

@@ -9,6 +9,8 @@ type Config struct {
 		FooterText		string		`yaml:"footer_text" json:"footer_text"`
 		FooterUrl		string		`yaml:"footer_url" json:"footer_url"`
 		MusicImg		string		`yaml:"music_img" json:"music_img"`
+		CheckUpdate		bool		`yaml:"check_update" json:"check_update"`
+		Script			string 		`yaml:"script" json:"script"`
 	}	`yaml:"info"`
 	Server struct{
 		Port 			string 		`yaml:"port"`//端口

conf/const.go

@@ -18,7 +18,7 @@ var(
 var Conf = new(Config)
 
 const (
-	VERSION="0.1.0"
+	VERSION="v0.1.3"
 
 	ImageThumbnailProcess="image/resize,w_50"
 	VideoThumbnailProcess="video/snapshot,t_0,f_jpg,w_50"

server/controllers/get.go

@@ -1,32 +1,31 @@
 package controllers
 
 import (
-	"fmt"
 	"github.com/Xhofe/alist/alidrive"
-	"github.com/Xhofe/alist/conf"
 	"github.com/gin-gonic/gin"
-	"github.com/patrickmn/go-cache"
 	log "github.com/sirupsen/logrus"
 	"strings"
 )
 
+// 因为下载地址有时效，所以去掉了文件请求和直链的缓存
+
 func Get(c *gin.Context) {
 	var get alidrive.GetReq
 	if err := c.ShouldBindJSON(&get); err != nil {
 		c.JSON(200,metaResponse(400,"Bad Request"))
 		return
 	}
-	log.Debugf("get:%v",get)
+	log.Debugf("get:%+v",get)
 	// cache
-	cacheKey:=fmt.Sprintf("%s-%s","g",get.FileId)
-	if conf.Conf.Cache.Enable {
-		file,exist:=conf.Cache.Get(cacheKey)
-		if exist {
-			log.Debugf("使用了缓存:%s",cacheKey)
-			c.JSON(200,dataResponse(file))
-			return
-		}
-	}
+	//cacheKey:=fmt.Sprintf("%s-%s","g",get.FileId)
+	//if conf.Conf.Cache.Enable {
+	//	file,exist:=conf.Cache.Get(cacheKey)
+	//	if exist {
+	//		log.Debugf("使用了缓存:%s",cacheKey)
+	//		c.JSON(200,dataResponse(file))
+	//		return
+	//	}
+	//}
 	file,err:=alidrive.GetFile(get.FileId)
 	if err !=nil {
 		c.JSON(200,metaResponse(500,err.Error()))
@@ -38,9 +37,9 @@ func Get(c *gin.Context) {
 		return
 	}
 	file.Paths=*paths
-	if conf.Conf.Cache.Enable {
-		conf.Cache.Set(cacheKey,file,cache.DefaultExpiration)
-	}
+	//if conf.Conf.Cache.Enable {
+	//	conf.Cache.Set(cacheKey,file,cache.DefaultExpiration)
+	//}
 	c.JSON(200,dataResponse(file))
 }
 
@@ -48,23 +47,23 @@ func Down(c *gin.Context) {
 	fileIdParam:=c.Param("file_id")
 	log.Debugf("down:%s",fileIdParam)
 	fileId:=strings.Split(fileIdParam,"/")[1]
-	cacheKey:=fmt.Sprintf("%s-%s","d",fileId)
-	if conf.Conf.Cache.Enable {
-		downloadUrl,exist:=conf.Cache.Get(cacheKey)
-		if exist {
-			log.Debugf("使用了缓存:%s",cacheKey)
-			c.Redirect(301,downloadUrl.(string))
-			return
-		}
-	}
+	//cacheKey:=fmt.Sprintf("%s-%s","d",fileId)
+	//if conf.Conf.Cache.Enable {
+	//	downloadUrl,exist:=conf.Cache.Get(cacheKey)
+	//	if exist {
+	//		log.Debugf("使用了缓存:%s",cacheKey)
+	//		c.Redirect(301,downloadUrl.(string))
+	//		return
+	//	}
+	//}
 	file,err:=alidrive.GetFile(fileId)
 	if err != nil {
 		c.JSON(200, metaResponse(500,err.Error()))
 		return
 	}
-	if conf.Conf.Cache.Enable {
-		conf.Cache.Set(cacheKey,file.DownloadUrl,cache.DefaultExpiration)
-	}
+	//if conf.Conf.Cache.Enable {
+	//	conf.Cache.Set(cacheKey,file.DownloadUrl,cache.DefaultExpiration)
+	//}
 	c.Redirect(301,file.DownloadUrl)
 	return
 }

server/controllers/list.go

@@ -9,13 +9,18 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
+type ListReq struct {
+	Password	string	`json:"password"`
+	alidrive.ListReq
+}
+
 func List(c *gin.Context) {
 	var list ListReq
 	if err := c.ShouldBindJSON(&list);err!=nil {
 		c.JSON(200, metaResponse(400,"Bad Request"))
 		return
 	}
-	log.Debugf("list:%v",list)
+	log.Debugf("list:%+v",list)
 	// cache
 	cacheKey:=fmt.Sprintf("%s-%s-%s","l",list.ParentFileId,list.Password)
 	if conf.Conf.Cache.Enable {

server/controllers/req_bean.go

@@ -1,8 +0,0 @@
-package controllers
-
-import "github.com/Xhofe/alist/alidrive"
-
-type ListReq struct {
-	Password	string	`json:"password"`
-	alidrive.ListReq
-}

server/controllers/search.go

@@ -19,7 +19,7 @@ func Search(c *gin.Context) {
 		c.JSON(200, metaResponse(400,"Bad Request"))
 		return
 	}
-	log.Debugf("search:%v",search)
+	log.Debugf("search:%+v",search)
 	// cache
 	cacheKey:=fmt.Sprintf("%s-%s","s",search.Query)
 	if conf.Conf.Cache.Enable {