1
0
mirror of https://github.com/nitezs/sub2clash.git synced 2024-12-24 13:34:41 -05:00
sub2clash/common/sub.go

87 lines
1.9 KiB
Go
Raw Normal View History

package common
2023-09-12 06:40:24 -04:00
import (
2023-09-17 03:52:37 -04:00
"crypto/sha256"
2023-09-12 06:40:24 -04:00
"encoding/hex"
"fmt"
"io"
"net/http"
2023-09-12 06:40:24 -04:00
"os"
"path/filepath"
2023-09-13 01:47:22 -04:00
"sub2clash/config"
2023-09-12 12:46:17 -04:00
"sync"
2023-09-12 06:40:24 -04:00
"time"
)
var subsDir = "subs"
2023-09-12 12:46:17 -04:00
var fileLock sync.RWMutex
2023-09-12 06:40:24 -04:00
func LoadSubscription(url string, refresh bool, userAgent string) ([]byte, error) {
2023-09-12 06:40:24 -04:00
if refresh {
return FetchSubscriptionFromAPI(url, userAgent)
2023-09-12 06:40:24 -04:00
}
2023-09-17 03:52:37 -04:00
hash := sha256.Sum224([]byte(url))
2023-09-12 12:46:17 -04:00
fileName := filepath.Join(subsDir, hex.EncodeToString(hash[:]))
2023-09-12 06:40:24 -04:00
stat, err := os.Stat(fileName)
if err != nil {
if !os.IsNotExist(err) {
return nil, err
}
return FetchSubscriptionFromAPI(url, userAgent)
2023-09-12 06:40:24 -04:00
}
lastGetTime := stat.ModTime().Unix() // 单位是秒
2023-09-13 01:47:22 -04:00
if lastGetTime+config.Default.CacheExpire > time.Now().Unix() {
2023-09-12 06:40:24 -04:00
file, err := os.Open(fileName)
if err != nil {
return nil, err
}
defer func(file *os.File) {
2023-09-22 13:31:04 -04:00
if file != nil {
_ = file.Close()
}
2023-09-12 06:40:24 -04:00
}(file)
2023-09-12 12:46:17 -04:00
fileLock.RLock()
defer fileLock.RUnlock()
2023-09-12 06:40:24 -04:00
subContent, err := io.ReadAll(file)
if err != nil {
return nil, err
}
return subContent, nil
}
return FetchSubscriptionFromAPI(url, userAgent)
2023-09-12 06:40:24 -04:00
}
func FetchSubscriptionFromAPI(url string, userAgent string) ([]byte, error) {
2023-09-17 03:52:37 -04:00
hash := sha256.Sum224([]byte(url))
2023-09-12 12:46:17 -04:00
fileName := filepath.Join(subsDir, hex.EncodeToString(hash[:]))
resp, err := Get(url, WithUserAgent(userAgent))
2023-09-12 06:40:24 -04:00
if err != nil {
return nil, err
}
defer func(resp *http.Response) {
if resp != nil && resp.Body != nil {
_ = resp.Body.Close()
2023-09-22 13:31:04 -04:00
}
}(resp)
2023-09-12 06:40:24 -04:00
data, err := io.ReadAll(resp.Body)
if err != nil {
return nil, fmt.Errorf("failed to read response body: %w", err)
}
file, err := os.Create(fileName)
if err != nil {
return nil, err
}
defer func(file *os.File) {
2023-09-22 13:31:04 -04:00
if file != nil {
_ = file.Close()
}
2023-09-12 06:40:24 -04:00
}(file)
2023-09-12 12:46:17 -04:00
fileLock.Lock()
defer fileLock.Unlock()
2023-09-12 06:40:24 -04:00
_, err = file.Write(data)
if err != nil {
return nil, fmt.Errorf("failed to write to sub.yaml: %w", err)
}
return data, nil
}