153 lines
2.9 KiB
Go
153 lines
2.9 KiB
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/PaulSonOfLars/gotgbot/v2"
|
|
"github.com/aki237/nscjar"
|
|
)
|
|
|
|
var cookiesCache = make(map[string][]*http.Cookie)
|
|
|
|
func GetLocationURL(
|
|
url string,
|
|
userAgent string,
|
|
) (string, error) {
|
|
req, err := http.NewRequest(http.MethodGet, url, nil)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to create request: %w", err)
|
|
}
|
|
if userAgent == "" {
|
|
userAgent = ChromeUA
|
|
}
|
|
req.Header.Set("User-Agent", userAgent)
|
|
session := GetDefaultHTTPClient()
|
|
resp, err := session.Do(req)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to send request: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
return resp.Request.URL.String(), nil
|
|
}
|
|
|
|
func IsUserAdmin(
|
|
bot *gotgbot.Bot,
|
|
chatID int64,
|
|
userID int64,
|
|
) bool {
|
|
chatMember, err := bot.GetChatMember(chatID, userID, nil)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if chatMember == nil {
|
|
return false
|
|
}
|
|
status := chatMember.GetStatus()
|
|
switch status {
|
|
case "creator":
|
|
return true
|
|
case "administrator":
|
|
if chatMember.MergeChatMember().CanChangeInfo {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return false
|
|
}
|
|
|
|
func EscapeCaption(str string) string {
|
|
// we wont use html.EscapeString
|
|
// cuz it will escape all the characters
|
|
// and we only need to escape < and >
|
|
chars := map[string]string{
|
|
"<": "<",
|
|
">": ">",
|
|
}
|
|
for k, v := range chars {
|
|
str = strings.ReplaceAll(str, k, v)
|
|
}
|
|
return str
|
|
}
|
|
|
|
func GetLastError(err error) error {
|
|
var lastErr error = err
|
|
for {
|
|
unwrapped := errors.Unwrap(lastErr)
|
|
if unwrapped == nil {
|
|
break
|
|
}
|
|
lastErr = unwrapped
|
|
}
|
|
return lastErr
|
|
}
|
|
|
|
func ParseCookieFile(fileName string) ([]*http.Cookie, error) {
|
|
cachedCookies, ok := cookiesCache[fileName]
|
|
if ok {
|
|
return cachedCookies, nil
|
|
}
|
|
cookiePath := filepath.Join("cookies", fileName)
|
|
cookieFile, err := os.Open(cookiePath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open cookie file: %w", err)
|
|
}
|
|
defer cookieFile.Close()
|
|
|
|
var parser nscjar.Parser
|
|
cookies, err := parser.Unmarshal(cookieFile)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse cookie file: %w", err)
|
|
}
|
|
cookiesCache[fileName] = cookies
|
|
return cookies, nil
|
|
}
|
|
|
|
func FixURL(url string) string {
|
|
return strings.ReplaceAll(url, "&", "&")
|
|
}
|
|
|
|
func CheckFFmpeg() bool {
|
|
_, err := exec.LookPath("ffmpeg")
|
|
return err == nil
|
|
}
|
|
|
|
func CleanupDownloadsDir() {
|
|
downloadsDir := os.Getenv("DOWNLOAD_DIR")
|
|
if downloadsDir == "" {
|
|
downloadsDir = "downloads"
|
|
}
|
|
filepath.Walk(downloadsDir, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
if path == downloadsDir {
|
|
return nil
|
|
}
|
|
if time.Since(info.ModTime()) > 30*time.Minute {
|
|
if info.IsDir() {
|
|
os.RemoveAll(path)
|
|
} else {
|
|
os.Remove(path)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func StartDownloadsCleanup() {
|
|
ticker := time.NewTicker(1 * time.Hour)
|
|
go func() {
|
|
for {
|
|
CleanupDownloadsDir()
|
|
<-ticker.C
|
|
}
|
|
}()
|
|
}
|