86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
|
package qbittorrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"net/http/cookiejar"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Authentication interface {
|
||
|
|
// Login cookie-based authentication, after calling NewClient, do not need to call Login again,
|
||
|
|
// it is the default behavior
|
||
|
|
Login(ctx context.Context) error
|
||
|
|
// Logout deactivate cookies
|
||
|
|
Logout(ctx context.Context) error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *client) Login(ctx context.Context) error {
|
||
|
|
if c.config.Username == "" || c.config.Password == "" {
|
||
|
|
return errors.New("username or password is empty")
|
||
|
|
}
|
||
|
|
|
||
|
|
formData := url.Values{}
|
||
|
|
formData.Set("username", c.config.Username)
|
||
|
|
formData.Set("password", c.config.Password)
|
||
|
|
encodedFormData := formData.Encode()
|
||
|
|
|
||
|
|
apiUrl := fmt.Sprintf("%s/api/v2/auth/login", c.config.Address)
|
||
|
|
|
||
|
|
result, err := c.doRequest(ctx, &requestData{
|
||
|
|
method: http.MethodPost,
|
||
|
|
url: apiUrl,
|
||
|
|
body: strings.NewReader(encodedFormData),
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if result.code != 200 {
|
||
|
|
return errors.New("login failed: " + string(result.body))
|
||
|
|
}
|
||
|
|
|
||
|
|
if string(result.body) == "Fails." {
|
||
|
|
return ErrAuthFailed
|
||
|
|
}
|
||
|
|
|
||
|
|
if string(result.body) != "Ok." {
|
||
|
|
return errors.New("login failed: " + string(result.body))
|
||
|
|
}
|
||
|
|
|
||
|
|
if c.cookieJar == nil {
|
||
|
|
c.cookieJar, err = cookiejar.New(nil)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
u, err := url.Parse(c.config.Address)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
c.cookieJar.SetCookies(u, result.cookies)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *client) Logout(ctx context.Context) error {
|
||
|
|
apiUrl := fmt.Sprintf("%s/api/v2/auth/logout", c.config.Address)
|
||
|
|
result, err := c.doRequest(ctx, &requestData{
|
||
|
|
method: http.MethodPost,
|
||
|
|
url: apiUrl,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if result.code != 200 {
|
||
|
|
return errors.New("logout failed: " + string(result.body))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|