Mukhtar Akere
72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package account
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/puzpuzpuz/xsync/v4"
|
|
"github.com/sirrobot01/decypharr/internal/request"
|
|
"github.com/sirrobot01/decypharr/pkg/debrid/types"
|
|
)
|
|
|
|
type Account struct {
|
|
Debrid string `json:"debrid"` // The debrid service name, e.g. "realdebrid"
|
|
links *xsync.Map[string, types.DownloadLink] // key is the sliced file link
|
|
Index int `json:"index"` // The index of the account in the config
|
|
Disabled atomic.Bool `json:"disabled"`
|
|
Token string `json:"token"`
|
|
TrafficUsed atomic.Int64 `json:"traffic_used"` // Traffic used in bytes
|
|
Username string `json:"username"` // Username for the account
|
|
httpClient *request.Client
|
|
}
|
|
|
|
func (a *Account) Equals(other *Account) bool {
|
|
if other == nil {
|
|
return false
|
|
}
|
|
return a.Token == other.Token && a.Debrid == other.Debrid
|
|
}
|
|
|
|
func (a *Account) Client() *request.Client {
|
|
return a.httpClient
|
|
}
|
|
|
|
// slice download link
|
|
func (a *Account) sliceFileLink(fileLink string) string {
|
|
if a.Debrid != "realdebrid" {
|
|
return fileLink
|
|
}
|
|
if len(fileLink) < 39 {
|
|
return fileLink
|
|
}
|
|
return fileLink[0:39]
|
|
}
|
|
|
|
func (a *Account) GetDownloadLink(fileLink string) (types.DownloadLink, error) {
|
|
slicedLink := a.sliceFileLink(fileLink)
|
|
dl, ok := a.links.Load(slicedLink)
|
|
if !ok {
|
|
return types.DownloadLink{}, types.ErrDownloadLinkNotFound
|
|
}
|
|
return dl, nil
|
|
}
|
|
|
|
func (a *Account) StoreDownloadLink(dl types.DownloadLink) {
|
|
slicedLink := a.sliceFileLink(dl.Link)
|
|
a.links.Store(slicedLink, dl)
|
|
}
|
|
func (a *Account) DeleteDownloadLink(fileLink string) {
|
|
slicedLink := a.sliceFileLink(fileLink)
|
|
a.links.Delete(slicedLink)
|
|
}
|
|
func (a *Account) ClearDownloadLinks() {
|
|
a.links.Clear()
|
|
}
|
|
func (a *Account) DownloadLinksCount() int {
|
|
return a.links.Size()
|
|
}
|
|
func (a *Account) StoreDownloadLinks(dls map[string]*types.DownloadLink) {
|
|
for _, dl := range dls {
|
|
a.StoreDownloadLink(*dl)
|
|
}
|
|
}
|