70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
|
package search
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/PuerkitoBio/goquery"
|
||
|
|
)
|
||
|
|
|
||
|
|
func getLibGenDownloadURL(md5 string, _ Source) ([]string, error) {
|
||
|
|
// Get Page
|
||
|
|
body, err := getPage("http://libgen.li/ads.php?md5=" + md5)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer body.Close()
|
||
|
|
|
||
|
|
// Parse
|
||
|
|
doc, err := goquery.NewDocumentFromReader(body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Return Download URL
|
||
|
|
downloadPath, exists := doc.Find("body > table > tbody > tr > td > a").Attr("href")
|
||
|
|
if !exists {
|
||
|
|
return nil, fmt.Errorf("Download URL not found")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Possible Funky URL
|
||
|
|
downloadPath = strings.ReplaceAll(downloadPath, "\\", "/")
|
||
|
|
return []string{fmt.Sprintf("http://libgen.li/%s", downloadPath)}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func getLibraryDownloadURL(md5 string, source Source) ([]string, error) {
|
||
|
|
// Derive Info URL
|
||
|
|
var infoURL string
|
||
|
|
switch source {
|
||
|
|
case SOURCE_LIBGEN_FICTION, SOURCE_ANNAS_ARCHIVE:
|
||
|
|
infoURL = "http://library.lol/fiction/" + md5
|
||
|
|
case SOURCE_LIBGEN_NON_FICTION:
|
||
|
|
infoURL = "http://library.lol/main/" + md5
|
||
|
|
default:
|
||
|
|
return nil, errors.New("invalid source")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get Page
|
||
|
|
body, err := getPage(infoURL)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer body.Close()
|
||
|
|
|
||
|
|
// Parse
|
||
|
|
doc, err := goquery.NewDocumentFromReader(body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Return Download URL
|
||
|
|
// downloadURL, _ := doc.Find("#download [href*=cloudflare]").Attr("href")
|
||
|
|
downloadURL, exists := doc.Find("#download h2 a").Attr("href")
|
||
|
|
if !exists {
|
||
|
|
return nil, errors.New("Download URL not found")
|
||
|
|
}
|
||
|
|
|
||
|
|
return []string{downloadURL}, nil
|
||
|
|
}
|