feat(115_share): add QR code source selection

SheltonZhu · Jan 15, 2024 · 47f60b2 · 47f60b2
1 parent cd5fe1c
commit 47f60b2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/drivers/115_share/meta.go b/drivers/115_share/meta.go
@@ -6,12 +6,13 @@ import (
 )
 
 type Addition struct {
-	Cookie      string  `json:"cookie" type:"text" help:"one of QR code token and cookie required"`
-	QRCodeToken string  `json:"qrcode_token" type:"text" help:"one of QR code token and cookie required"`
-	PageSize    int64   `json:"page_size" type:"number" default:"20" help:"list api per page size of 115 driver"`
-	LimitRate   float64 `json:"limit_rate" type:"number" default:"2" help:"limit all api request rate (1r/[limit_rate]s)"`
-	ShareCode   string  `json:"share_code" type:"text" required:"true" help:"share code of 115 share link"`
-	ReceiveCode string  `json:"receive_code" type:"text" required:"true" help:"receive code of 115 share link"`
+	Cookie       string  `json:"cookie" type:"text" help:"one of QR code token and cookie required"`
+	QRCodeToken  string  `json:"qrcode_token" type:"text" help:"one of QR code token and cookie required"`
+	QRCodeSource string  `json:"qrcode_source" type:"select" options:"web,android,ios,linux,mac,windows,tv" default:"linux" help:"select the QR code device, default linux"`
+	PageSize     int64   `json:"page_size" type:"number" default:"20" help:"list api per page size of 115 driver"`
+	LimitRate    float64 `json:"limit_rate" type:"number" default:"2" help:"limit all api request rate (1r/[limit_rate]s)"`
+	ShareCode    string  `json:"share_code" type:"text" required:"true" help:"share code of 115 share link"`
+	ReceiveCode  string  `json:"receive_code" type:"text" required:"true" help:"receive code of 115 share link"`
 	driver.RootID
 }
 

diff --git a/drivers/115_share/utils.go b/drivers/115_share/utils.go
@@ -93,7 +93,7 @@ func (d *Pan115Share) login() error {
 		s := &driver115.QRCodeSession{
 			UID: d.QRCodeToken,
 		}
-		if cr, err = d.client.QRCodeLogin(s); err != nil {
+		if cr, err = d.client.QRCodeLoginWithApp(s, driver115.LoginApp(d.QRCodeSource)); err != nil {
 			return errors.Wrap(err, "failed to login by qrcode")
 		}
 		d.Cookie = fmt.Sprintf("UID=%s;CID=%s;SEID=%s", cr.UID, cr.CID, cr.SEID)