From a2a46bdeeba06dc40d28f0cbc4251c9ce08be764 Mon Sep 17 00:00:00 2001
From: tickstep <tickstep@outlook.com>
Date: Sat, 15 Oct 2022 22:25:22 +0800
Subject: [PATCH] locate can save download links to file

---
 internal/command/locate.go | 35 ++++++++++++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 3 deletions(-)

diff --git a/internal/command/locate.go b/internal/command/locate.go
index f8915dd..39f5f8d 100644
--- a/internal/command/locate.go
+++ b/internal/command/locate.go
@@ -20,8 +20,10 @@ import (
 	"github.com/tickstep/aliyunpan/cmder/cmdtable"
 	"github.com/tickstep/aliyunpan/internal/config"
 	"github.com/urfave/cli"
+	"io/ioutil"
 	"os"
 	"strconv"
+	"strings"
 )
 
 func CmdLocateUrl() cli.Command {
@@ -31,11 +33,16 @@ func CmdLocateUrl() cli.Command {
 		UsageText: cmder.App().Name + " locate <文件1> <文件2> <文件3> ...",
 		Description: `
 	获取文件下载直链，只支持文件，不支持文件夹。下载链接有效时间为4个小时。
+	导出的下载链接可以使用任何支持http下载的工具进行下载，如果是视频文件还可以使用支持在线流播放的视频软件进行在线播放。
+	
 	注意：由于阿里云盘网页端有防盗链设置所以不能直接使用网页Token登录，你必须使用手机扫码二维码登录(命令：login -QrCode)，否则获取的直链无法正常下载，会提示 403 Forbidden 下载被禁止。
 
 	示例:
 	获取 /我的资源/1.mp4 下载直链
 	aliyunpan locate /我的资源/1.mp4
+
+	获取 /我的资源/1.mp4 下载直链并保存到本地文件 /Volumes/Downloads/file_url.txt 中
+	aliyunpan locate -saveto "/Volumes/Downloads/file_url.txt" /我的资源/1.mp4
 `,
 		Category: "阿里云盘",
 		Before:   cmder.ReloadConfigFunc,
@@ -44,7 +51,11 @@ func CmdLocateUrl() cli.Command {
 				cli.ShowCommandHelp(c, c.Command.Name)
 				return nil
 			}
-			RunLocateUrl(parseDriveId(c), c.Args())
+			saveFilePath := ""
+			if c.IsSet("saveto") {
+				saveFilePath = c.String("saveto")
+			}
+			RunLocateUrl(parseDriveId(c), c.Args(), saveFilePath)
 			return nil
 		},
 		Flags: []cli.Flag{
@@ -53,12 +64,16 @@ func CmdLocateUrl() cli.Command {
 				Usage: "网盘ID",
 				Value: "",
 			},
+			cli.StringFlag{
+				Name:  "saveto",
+				Usage: "导出链接成文件并保存到指定的位置",
+			},
 		},
 	}
 }
 
 // RunLocateUrl 执行下载网盘内文件
-func RunLocateUrl(driveId string, paths []string) {
+func RunLocateUrl(driveId string, paths []string, saveFilePath string) {
 	useInternalUrl := config.Config.TransferUrlType == 2
 	activeUser := GetActiveUser()
 	activeUser.PanClient().EnableCache()
@@ -71,6 +86,7 @@ func RunLocateUrl(driveId string, paths []string) {
 		return
 	}
 
+	sb := &strings.Builder{}
 	failedList := []string{}
 	for k := range paths {
 		p := paths[k]
@@ -91,7 +107,20 @@ func RunLocateUrl(driveId string, paths []string) {
 			if useInternalUrl {
 				url = durl.InternalUrl
 			}
-			fmt.Printf("\n文件：%s\n%s\n", p, url)
+			if saveFilePath != "" {
+				fmt.Fprintf(sb, "\n文件：%s\n%s\n", p, url)
+			} else {
+				fmt.Printf("\n文件：%s\n%s\n", p, url)
+			}
+		}
+	}
+
+	if saveFilePath != "" {
+		// save file
+		if e := ioutil.WriteFile(saveFilePath, []byte(sb.String()), 0777); e == nil {
+			fmt.Printf("保存文件成功：%s\n", saveFilePath)
+		} else {
+			fmt.Printf("保存文件失败：%s\n", saveFilePath)
 		}
 	}