download: allow to specify a password

gnojus · Oct 5, 2024 · 5ddef54 · 5ddef54
1 parent 6ce8c85
commit 5ddef54
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 10 deletions.
diff --git a/cli/args.go b/cli/args.go
@@ -8,6 +8,7 @@ type args struct {
 	Url    string
 	Output string
 	Path   string
+	Password string
 	Silent bool
 	Force  bool
 	Info   bool

diff --git a/cli/cli.go b/cli/cli.go
@@ -16,7 +16,7 @@ func Eval(opts map[string]interface{}) (err error) {
 	if err != nil {
 		return
 	}
-	resp, r, err := transfer.GetDlResponse(parsed.Url)
+	resp, r, err := transfer.GetDlResponse(parsed.Url, parsed.Password)
 	if err != nil {
 		return
 	}
@@ -52,4 +52,4 @@ func Eval(opts map[string]interface{}) (err error) {
 		io.Copy(writer, resp.Body)
 	}
 	return
-}
+}
diff --git a/transfer/download.go b/transfer/download.go
@@ -19,6 +19,7 @@ type headers map[string]string
 
 type requestData struct {
 	SecurityHash string `json:"security_hash"`
+	Password     string `json:"password"`
 	RecipientId  string `json:"recipient_id,omitempty"`
 	Intent       string `json:"intent"`
 }
@@ -35,7 +36,7 @@ type DlResponse struct {
 	DlFilename string `json:"dl_filename"`
 }
 
-func GetDlResponse(URL string) (resp *http.Response, r DlResponse, err error) {
+func GetDlResponse(URL string, password string) (resp *http.Response, r DlResponse, err error) {
 	client := &http.Client{}
 	req, err := createRequest("GET", URL, nil, nil)
 	if err != nil {
@@ -49,6 +50,7 @@ func GetDlResponse(URL string) (resp *http.Response, r DlResponse, err error) {
 	if err != nil {
 		return
 	}
+	data.reqData.Password = password
 	link, err := getDownloadLink(client, data)
 	if err != nil {
 		return

diff --git a/wedl.go b/wedl.go
@@ -35,13 +35,14 @@ Usage:
   wedl [options] <URL>
 
 Options:
-  -h --help              Show this screen.
-  -v --version           Print version and exit.
-  -o FILE --output=FILE  Output file. Use - for stdout.
-  -p PATH --path=PATH	 Downloaded files directory.
-  -s --silent            Silent. Do not output anything to stderr.
-  -f --force             Overwrite files if needed.
-  -i --info              Write download info to stdout and exit.
+  -h --help                Show this screen.
+  -v --version             Print version and exit.
+  -o FILE --output=FILE    Output file. Use - for stdout.
+  -p PATH --path=PATH	   Downloaded files directory.
+  -P PASS --password=PASS  Use a password.
+  -s --silent              Silent. Do not output anything to stderr.
+  -f --force               Overwrite files if needed.
+  -i --info                Write download info to stdout and exit.
   `
 	opts, _ := docopt.ParseArgs(usage, os.Args[1:], resolveVersion())
 	err := cli.Eval(opts)