bit1001/tdl
1
1
Fork 0
mirror of https://github.com/bit1001/tdl.git synced 2024-09-20 06:35:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(cmd): flags getter

Browse source
This commit is contained in:
iyear 2022-09-14 10:38:08 +08:00
parent 0723f1de9f
commit 670dd3940d
3 changed files with 3 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cmd/chat/ls.go
View file

@ -10,16 +10,6 @@ var cmdList = &cobra.Command{
Short: "List your all chats with info",
Example: "",
RunE: func(cmd *cobra.Command, args []string) error {
proxy, err := cmd.Flags().GetString("proxy")
if err != nil {
return err
}
ns, err := cmd.Flags().GetString("ns")
if err != nil {
return err
}
return chat.List(cmd.Context(), ns, proxy)
return chat.List(cmd.Context(), cmd.Flag("ns").Value.String(), cmd.Flag("proxy").Value.String())
},
}

12
cmd/dl/url.go
View file

@ -14,17 +14,7 @@ var cmdURL = &cobra.Command{
Short: "Download in url mode",
Example: "tdl dl url -n iyear --proxy socks5://127.0.0.1:1080 -u https://t.me/tdl/1 -u https://t.me/tdl/2 -s 262144 -t 16 -l 3",
RunE: func(cmd *cobra.Command, args []string) error {
proxy, err := cmd.Flags().GetString("proxy")
if err != nil {
return err
}
ns, err := cmd.Flags().GetString("ns")
if err != nil {
return err
}
return dlurl.Run(cmd.Context(), ns, proxy, partSize, threads, limit, urls)
return dlurl.Run(cmd.Context(), cmd.Flag("ns").Value.String(), cmd.Flag("proxy").Value.String(), partSize, threads, limit, urls)
},
}

16
cmd/login/login.go
View file

@ -1,7 +1,6 @@
package login
import (
"fmt"
"github.com/iyear/tdl/app/login"
"github.com/spf13/cobra"
)
@ -11,19 +10,6 @@ var Cmd = &cobra.Command{
Short: "Login to Telegram",
Example: "tdl login -n my-tdl --proxy socks5://localhost:1080",
RunE: func(cmd *cobra.Command, args []string) error {
proxy, err := cmd.Flags().GetString("proxy")
if err != nil {
return err
}
ns, err := cmd.Flags().GetString("ns")
if err != nil {
return err
}
if err := login.Run(cmd.Context(), ns, proxy); err != nil {
return fmt.Errorf("login failed: %v", err)
}
return nil
return login.Run(cmd.Context(), cmd.Flag("ns").Value.String(), cmd.Flag("proxy").Value.String())
},
}