c21ff7e499
- Implemented a new task type for handling direct links downloads. - Added command handler for downloading multiple links via /dl command. - Introduced progress tracking for direct link downloads. - Enhanced filename parsing to support various encoding scenarios. - Updated enums to include direct links as a task type. - Refactored existing task structures to accommodate new functionality. - Improved error handling and logging throughout the download process.
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package parsers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/krau/SaveAny-Bot/parsers/js"
|
|
"github.com/krau/SaveAny-Bot/parsers/native/kemono"
|
|
"github.com/krau/SaveAny-Bot/parsers/native/twitter"
|
|
"github.com/krau/SaveAny-Bot/parsers/parsers"
|
|
"github.com/krau/SaveAny-Bot/pkg/parser"
|
|
)
|
|
|
|
func init() {
|
|
parsers.Add(new(twitter.TwitterParser), new(kemono.KemonoParser))
|
|
}
|
|
|
|
var (
|
|
ErrNoParserFound = fmt.Errorf("no parser found for the given URL")
|
|
)
|
|
|
|
func ParseWithContext(ctx context.Context, url string) (*parser.Item, error) {
|
|
ch := make(chan *parser.Item, 1)
|
|
errCh := make(chan error, 1)
|
|
|
|
go func() {
|
|
for _, pser := range parsers.Get() {
|
|
if !pser.CanHandle(url) {
|
|
continue
|
|
}
|
|
item, err := pser.Parse(ctx, url)
|
|
if err != nil {
|
|
errCh <- err
|
|
return
|
|
}
|
|
ch <- item
|
|
return
|
|
}
|
|
errCh <- ErrNoParserFound
|
|
}()
|
|
|
|
select {
|
|
case item := <-ch:
|
|
return item, nil
|
|
case err := <-errCh:
|
|
return nil, err
|
|
case <-ctx.Done():
|
|
return nil, ctx.Err()
|
|
}
|
|
}
|
|
|
|
// CanHandle checks if any registered parser can handle the given URL and returns the parser if found.
|
|
func CanHandle(url string) (bool, parser.Parser) {
|
|
for _, pser := range parsers.Get() {
|
|
if pser.CanHandle(url) {
|
|
return true, pser
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func LoadPlugins(ctx context.Context, dir string) error {
|
|
return js.LoadPlugins(ctx, dir)
|
|
}
|
|
|
|
func AddPlugin(ctx context.Context, code string, name string) error {
|
|
return js.AddPlugin(ctx, code, name)
|
|
}
|