mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +08:00
334 lines
8.8 KiB
Go
334 lines
8.8 KiB
Go
// Copyright (c) 2020 tickstep.
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
package command
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/tickstep/aliyunpan-api/aliyunpan"
|
||
"github.com/tickstep/aliyunpan-api/aliyunpan/apierror"
|
||
"github.com/tickstep/aliyunpan/cmder"
|
||
"github.com/tickstep/aliyunpan/cmder/cmdtable"
|
||
"github.com/tickstep/aliyunpan/internal/config"
|
||
"github.com/tickstep/library-go/logger"
|
||
"github.com/urfave/cli"
|
||
"os"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
func CmdShare() cli.Command {
|
||
return cli.Command{
|
||
Name: "share",
|
||
Usage: "分享文件/目录",
|
||
UsageText: cmder.App().Name + " share",
|
||
Category: "阿里云盘",
|
||
Before: cmder.ReloadConfigFunc,
|
||
Action: func(c *cli.Context) error {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
return nil
|
||
},
|
||
Subcommands: []cli.Command{
|
||
{
|
||
Name: "set",
|
||
Aliases: []string{"s"},
|
||
Usage: "设置分享文件/目录",
|
||
UsageText: cmder.App().Name + " share set <文件/目录1> <文件/目录2> ...",
|
||
Description: `
|
||
示例:
|
||
|
||
创建文件 1.mp4 的分享链接
|
||
aliyunpan share set 1.mp4
|
||
|
||
创建文件 1.mp4 的分享链接，并指定分享密码为2333
|
||
aliyunpan share set -sharePwd 2333 1.mp4
|
||
|
||
创建文件 1.mp4 的分享链接，并指定有效期为1天
|
||
aliyunpan share set -time 1 1.mp4
|
||
`,
|
||
Action: func(c *cli.Context) error {
|
||
if c.NArg() < 1 {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
return nil
|
||
}
|
||
if config.Config.ActiveUser() == nil {
|
||
fmt.Println("未登录账号")
|
||
return nil
|
||
}
|
||
et := ""
|
||
timeFlag := "0"
|
||
if c.IsSet("time") {
|
||
timeFlag = c.String("time")
|
||
}
|
||
now := time.Now()
|
||
if timeFlag == "1" {
|
||
et = now.Add(time.Duration(1) * time.Hour * 24).Format("2006-01-02 15:04:05")
|
||
} else if timeFlag == "2" {
|
||
et = now.Add(time.Duration(7) * time.Hour * 24).Format("2006-01-02 15:04:05")
|
||
} else {
|
||
et = ""
|
||
}
|
||
|
||
sharePwd := ""
|
||
if c.IsSet("sharePwd") {
|
||
sharePwd = c.String("sharePwd")
|
||
}
|
||
|
||
modeFlag := "1"
|
||
if c.IsSet("mode") {
|
||
modeFlag = c.String("mode")
|
||
}
|
||
if modeFlag == "1" {
|
||
if sharePwd == "" {
|
||
sharePwd = RandomStr(4)
|
||
}
|
||
} else {
|
||
sharePwd = ""
|
||
}
|
||
RunShareSet(parseDriveId(c), c.Args(), et, sharePwd)
|
||
return nil
|
||
},
|
||
Flags: []cli.Flag{
|
||
cli.StringFlag{
|
||
Name: "driveId",
|
||
Usage: "网盘ID",
|
||
Value: "",
|
||
},
|
||
cli.StringFlag{
|
||
Name: "time",
|
||
Usage: "有效期，0-永久，1-1天，2-7天",
|
||
Value: "0",
|
||
},
|
||
cli.StringFlag{
|
||
Name: "mode",
|
||
Usage: "有效期，1-私密分享，2-公开分享",
|
||
Value: "1",
|
||
},
|
||
cli.StringFlag{
|
||
Name: "sharePwd",
|
||
Usage: "自定义私密分享密码，4个字符，没有指定则随机生成",
|
||
Value: "",
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "list",
|
||
Aliases: []string{"l"},
|
||
Usage: "列出已分享文件/目录",
|
||
UsageText: cmder.App().Name + " share list",
|
||
Action: func(c *cli.Context) error {
|
||
RunShareList()
|
||
return nil
|
||
},
|
||
Flags: []cli.Flag{
|
||
},
|
||
},
|
||
{
|
||
Name: "cancel",
|
||
Aliases: []string{"c"},
|
||
Usage: "取消分享文件/目录",
|
||
UsageText: cmder.App().Name + " share cancel <shareid_1> <shareid_2> ...",
|
||
Description: `目前只支持通过分享id (shareid) 来取消分享.`,
|
||
Action: func(c *cli.Context) error {
|
||
if c.NArg() < 1 {
|
||
cli.ShowCommandHelp(c, c.Command.Name)
|
||
return nil
|
||
}
|
||
RunShareCancel(c.Args())
|
||
return nil
|
||
},
|
||
},
|
||
// {
|
||
// Name: "mc",
|
||
// Aliases: []string{},
|
||
// Usage: "创建秒传链接",
|
||
// UsageText: cmder.App().Name + " share mc <文件/目录1> <文件/目录2> ...",
|
||
// Description: `
|
||
//创建文件秒传链接，秒传链接只能是文件，如果是文件夹则会创建文件夹包含的所有文件的秒传链接。秒传链接可以通过RapidUpload命令或者Import命令进行导入到自己的网盘。
|
||
//示例:
|
||
// 创建文件 1.mp4 的秒传链接
|
||
// aliyunpan share mc 1.mp4
|
||
//
|
||
// 创建文件 1.mp4 的秒传链接，但链接隐藏相对路径
|
||
// aliyunpan share mc -hp 1.mp4
|
||
//
|
||
// 创建文件夹 share_folder 下面所有文件的秒传链接
|
||
// aliyunpan share mc share_folder/
|
||
//`,
|
||
// Action: func(c *cli.Context) error {
|
||
// if c.NArg() < 1 {
|
||
// cli.ShowCommandHelp(c, c.Command.Name)
|
||
// return nil
|
||
// }
|
||
// if config.Config.ActiveUser() == nil {
|
||
// fmt.Println("未登录账号")
|
||
// return nil
|
||
// }
|
||
// hp := false
|
||
// if c.IsSet("hp") {
|
||
// hp = c.Bool("hp")
|
||
// }
|
||
// RunShareMc(parseDriveId(c), hp, c.Args())
|
||
// return nil
|
||
// },
|
||
// Flags: []cli.Flag{
|
||
// cli.StringFlag{
|
||
// Name: "driveId",
|
||
// Usage: "网盘ID",
|
||
// Value: "",
|
||
// },
|
||
// cli.BoolFlag{
|
||
// Name: "hp",
|
||
// Usage: "hide path, 隐藏相对目录",
|
||
// },
|
||
// },
|
||
// },
|
||
},
|
||
}
|
||
}
|
||
|
||
// RunShareSet 执行分享
|
||
func RunShareSet(driveId string, paths []string, expiredTime string, sharePwd string) {
|
||
panClient := GetActivePanClient()
|
||
fileList, _, err := GetFileInfoByPaths(paths[:len(paths)]...)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return
|
||
}
|
||
|
||
fidList := []string{}
|
||
for _,f := range fileList {
|
||
fidList = append(fidList, f.FileId)
|
||
}
|
||
|
||
if len(fidList) == 0 {
|
||
fmt.Printf("没有指定有效的文件\n")
|
||
return
|
||
}
|
||
|
||
r, err1 := panClient.ShareLinkCreate(aliyunpan.ShareCreateParam{
|
||
DriveId: driveId,
|
||
SharePwd: sharePwd,
|
||
Expiration: expiredTime,
|
||
FileIdList: fidList,
|
||
})
|
||
|
||
if err1 != nil || r == nil {
|
||
if err1.Code == apierror.ApiCodeFileShareNotAllowed {
|
||
fmt.Printf("创建分享链接失败: 该文件类型不允许分享\n")
|
||
} else {
|
||
fmt.Printf("创建分享链接失败: %s\n", err)
|
||
}
|
||
return
|
||
}
|
||
|
||
fmt.Printf("创建分享链接成功\n")
|
||
if len(sharePwd) > 0 {
|
||
fmt.Printf("链接：%s 提取码：%s\n", r.ShareUrl, r.SharePwd)
|
||
} else {
|
||
fmt.Printf("链接：%s\n", r.ShareUrl)
|
||
}
|
||
|
||
}
|
||
|
||
// RunShareList 执行列出分享列表
|
||
func RunShareList() {
|
||
activeUser := GetActiveUser()
|
||
records, err := activeUser.PanClient().ShareLinkList(activeUser.UserId)
|
||
if err != nil {
|
||
fmt.Printf("获取分享列表失败: %s\n", err)
|
||
return
|
||
}
|
||
|
||
tb := cmdtable.NewTable(os.Stdout)
|
||
tb.SetHeader([]string{"#", "ShARE_ID", "分享链接", "提取码", "文件名", "FILE_ID", "过期时间", "状态"})
|
||
now := time.Now()
|
||
for k, record := range records {
|
||
et := "永久有效"
|
||
if len(record.Expiration) > 0 {
|
||
et = record.Expiration
|
||
}
|
||
status := "有效"
|
||
if record.FirstFile == nil {
|
||
status = "已删除"
|
||
} else {
|
||
cz := time.FixedZone("CST", 8*3600)
|
||
if len(record.Expiration) > 0 {
|
||
expiredTime, _ := time.ParseInLocation("2006-01-02 15:04:05", record.Expiration, cz)
|
||
if expiredTime.Unix() < now.Unix() {
|
||
status = "已过期"
|
||
}
|
||
}
|
||
}
|
||
tb.Append([]string{strconv.Itoa(k), record.ShareId, record.ShareUrl, record.SharePwd,
|
||
record.ShareName,
|
||
record.FileIdList[0],
|
||
et,
|
||
status})
|
||
}
|
||
tb.Render()
|
||
}
|
||
|
||
// RunShareCancel 执行取消分享
|
||
func RunShareCancel(shareIdList []string) {
|
||
if len(shareIdList) == 0 {
|
||
fmt.Printf("取消分享操作失败, 没有任何 shareid\n")
|
||
return
|
||
}
|
||
|
||
activeUser := GetActiveUser()
|
||
r, err := activeUser.PanClient().ShareLinkCancel(shareIdList)
|
||
if err != nil {
|
||
fmt.Printf("取消分享操作失败: %s\n", err)
|
||
return
|
||
}
|
||
|
||
if r != nil && len(r) > 0 {
|
||
fmt.Printf("取消分享操作成功\n")
|
||
} else {
|
||
fmt.Printf("取消分享操作失败\n")
|
||
}
|
||
}
|
||
|
||
// 创建秒传链接
|
||
func RunShareMc(driveId string, hideRelativePath bool, panPaths []string) {
|
||
activeUser := config.Config.ActiveUser()
|
||
panClient := activeUser.PanClient()
|
||
|
||
totalCount := 0
|
||
for _,panPath := range panPaths {
|
||
panPath = activeUser.PathJoin(driveId, panPath)
|
||
panClient.FilesDirectoriesRecurseList(driveId, panPath, func(depth int, _ string, fd *aliyunpan.FileEntity, apiError *apierror.ApiError) bool {
|
||
if apiError != nil {
|
||
logger.Verbosef("%s\n", apiError)
|
||
return true
|
||
}
|
||
|
||
// 只需要文件即可
|
||
if !fd.IsFolder() {
|
||
item := newRapidUploadItemFromFileEntity(fd)
|
||
jstr := item.createRapidUploadLink(hideRelativePath)
|
||
if len(jstr) <= 0 {
|
||
logger.Verboseln("create rapid upload link err")
|
||
return false
|
||
}
|
||
// print
|
||
fmt.Println(jstr)
|
||
totalCount += 1
|
||
time.Sleep(time.Duration(100) * time.Millisecond)
|
||
}
|
||
return true
|
||
})
|
||
}
|
||
fmt.Printf("\n秒传链接总数量: %d\n", totalCount)
|
||
} |