8ecbc7e0aa
Added Db support to save last viewed Manga and chapter, and auto load it next time you start the server. Added Providers to add more Manga Platforms than just Bato in the future
65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"regexp"
|
|
"strconv"
|
|
)
|
|
|
|
func getTitleAndChapter(url string) (title string, chapter string, err error) {
|
|
reg, err := regexp.Compile(`/title/\d*-(.*?)/\d*-(.*)`)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
matches := reg.FindAllStringSubmatch(url, -1)
|
|
if len(matches) <= 0 {
|
|
return "", "", errors.New("no title or chapter found")
|
|
}
|
|
|
|
return matches[0][1], matches[0][2], nil
|
|
}
|
|
|
|
func getMangaIdAndChapterId(url string) (titleId int, chapterId int, err error) {
|
|
reg, err := regexp.Compile(`/title/(\d*)-.*?/(\d*)-.*`)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
|
|
matches := reg.FindAllStringSubmatch(url, -1)
|
|
if len(matches) <= 0 {
|
|
return 0, 0, errors.New("no title or chapter found")
|
|
}
|
|
t, err := strconv.Atoi(matches[0][1])
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
c, err := strconv.Atoi(matches[0][2])
|
|
|
|
return t, c, err
|
|
}
|
|
|
|
func addFileToRam(url string) (*bytes.Buffer, error) {
|
|
// Get the data
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer func(Body io.ReadCloser) {
|
|
err := Body.Close()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}(resp.Body)
|
|
|
|
buf := new(bytes.Buffer)
|
|
|
|
// Write the body to file
|
|
_, err = io.Copy(buf, resp.Body)
|
|
return buf, err
|
|
}
|