Files
iptvc/app/utils/utils.go

75 lines
1.9 KiB
Go

/*
* Copyright (c) 2025, Антон Аксенов
* This file is part of iptvc project
* MIT License: https://git.axenov.dev/IPTV/iptvc/src/branch/master/LICENSE
*/
package utils
import (
"crypto/md5"
"crypto/tls"
"encoding/hex"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strings"
"time"
)
// ExpandPath возвращает полный путь к файлу, где ~ заменяется на домашнюю папку пользователя
func ExpandPath(path string) (string, error) {
homepath, err := os.UserHomeDir()
if err != nil {
return "", err
}
newpath, err := filepath.Abs(strings.Replace(path, "~", homepath, 1))
return newpath, err
}
// ArrayUnique возвращает массив уникальных элементов
func ArrayUnique(arr []string) []string {
size := len(arr)
result := make([]string, 0, size)
temp := map[string]struct{}{}
for i := 0; i < size; i++ {
if _, ok := temp[arr[i]]; ok != true {
temp[arr[i]] = struct{}{}
result = append(result, arr[i])
}
}
return result
}
// Fetch выполняет GET запрос и возвращает результат в виде массива байт
func Fetch(url string) ([]byte, error) {
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
}
req.Header.Set("User-Agent", "Mozilla/5.0 WINK/1.31.1 (AndroidTV/9) HlsWinkPlayer")
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
httpClient := http.Client{Timeout: 10 * time.Second}
resp, err := httpClient.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("HTTP status %d", resp.StatusCode)
}
return io.ReadAll(resp.Body)
}
// Md5str возвращает хэш строки в виде строки
func Md5str(str string) string {
hash := md5.Sum([]byte(str))
return hex.EncodeToString(hash[:])
}