diff --git a/drivers/189.go b/drivers/189.go index 8c1a2dca..8cbf3e8b 100644 --- a/drivers/189.go +++ b/drivers/189.go @@ -230,7 +230,7 @@ func (c Cloud189) Link(path string, account *model.Account) (string, error) { return resp.FileDownloadUrl, nil } -func (c Cloud189) Proxy(ctx *gin.Context) { +func (c Cloud189) Proxy(ctx *gin.Context, account *model.Account) { ctx.Request.Header.Del("Origin") } diff --git a/drivers/alidrive.go b/drivers/alidrive.go index bcbc3b0d..c9a7db79 100644 --- a/drivers/alidrive.go +++ b/drivers/alidrive.go @@ -103,7 +103,7 @@ func (a AliDrive) Items() []Item { } } -func (a AliDrive) Proxy(c *gin.Context) { +func (a AliDrive) Proxy(c *gin.Context, account *model.Account) { c.Request.Header.Del("Origin") c.Request.Header.Set("Referer", "https://www.aliyundrive.com/") } diff --git a/drivers/driver.go b/drivers/driver.go index 06db854b..99990a22 100644 --- a/drivers/driver.go +++ b/drivers/driver.go @@ -12,7 +12,7 @@ type Driver interface { Save(account *model.Account, old *model.Account) error Path(path string, account *model.Account) (*model.File, []*model.File, error) Link(path string, account *model.Account) (string, error) - Proxy(c *gin.Context) + Proxy(c *gin.Context, account *model.Account) Preview(path string, account *model.Account) (interface{}, error) // TODO //Search(path string, keyword string, account *model.Account) ([]*model.File, error) diff --git a/drivers/native.go b/drivers/native.go index b3b5254a..5f0a272a 100644 --- a/drivers/native.go +++ b/drivers/native.go @@ -31,7 +31,7 @@ func (n Native) Items() []Item { } } -func (n Native) Proxy(c *gin.Context) { +func (n Native) Proxy(c *gin.Context, account *model.Account) { // unnecessary } diff --git a/drivers/onedrive.go b/drivers/onedrive.go index 6292ec1e..54f3c620 100644 --- a/drivers/onedrive.go +++ b/drivers/onedrive.go @@ -301,7 +301,7 @@ func (o Onedrive) Save(account *model.Account, old *model.Account) error { return nil } -func (o Onedrive) Proxy(c *gin.Context) { +func (o Onedrive) Proxy(c *gin.Context, account *model.Account) { c.Request.Header.Del("Origin") }