feat: 🎉 init commit

Author: kierankihn

.gitignore

@@ -0,0 +1,25 @@
+# If you prefer the allow list template instead of the deny list, see community template:
+# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
+#
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Go workspace file
+go.work
+go.work.sum
+
+# env file
+.env

config.yaml

@@ -0,0 +1,5 @@
+last_download_time: 1970-01-01T00:00:00Z # 最后下载时间，避免重复下载
+downloads: 
+  - name: name     # 番剧名称
+    path: path     # 存储路径
+    filter: filter # 正则表达式匹配

go.mod

@@ -0,0 +1,25 @@
+module go-ani-rss
+
+go 1.22.2
+
+require github.com/mmcdole/gofeed v1.3.0
+
+require (
+	github.com/PuerkitoBio/goquery v1.8.0 // indirect
+	github.com/andybalholm/cascadia v1.3.1 // indirect
+	github.com/docker/go-units v0.5.0 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/mitchellh/colorstring v0.0.0-20190213212951-d06e56a500db // indirect
+	github.com/mmcdole/goxpp v1.1.1-0.20240225020742-a0c311522b23 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/rivo/uniseg v0.4.7 // indirect
+	github.com/schollz/progressbar v1.0.0 // indirect
+	github.com/schollz/progressbar/v3 v3.14.6 // indirect
+	golang.org/x/net v0.29.0 // indirect
+	golang.org/x/sys v0.25.0 // indirect
+	golang.org/x/term v0.24.0 // indirect
+	golang.org/x/text v0.18.0 // indirect
+	golang.org/x/tools v0.25.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,59 @@
+github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U=
+github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI=
+github.com/andybalholm/cascadia v1.3.1 h1:nhxRkql1kdYCc8Snf7D5/D3spOX+dBgjA6u8x004T2c=
+github.com/andybalholm/cascadia v1.3.1/go.mod h1:R4bJ1UQfqADjvDa4P6HZHLh/3OxWWEqc0Sk8XGwHqvA=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4=
+github.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
+github.com/k0kubun/go-ansi v0.0.0-20180517002512-3bf9e2903213/go.mod h1:vNUNkEQ1e29fT/6vq2aBdFsgNPmy8qMdSay1npru+Sw=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mitchellh/colorstring v0.0.0-20190213212951-d06e56a500db h1:62I3jR2EmQ4l5rM/4FEfDWcRD+abF5XlKShorW5LRoQ=
+github.com/mitchellh/colorstring v0.0.0-20190213212951-d06e56a500db/go.mod h1:l0dey0ia/Uv7NcFFVbCLtqEBQbrT4OCwCSKTEv6enCw=
+github.com/mmcdole/gofeed v1.3.0 h1:5yn+HeqlcvjMeAI4gu6T+crm7d0anY85+M+v6fIFNG4=
+github.com/mmcdole/gofeed v1.3.0/go.mod h1:9TGv2LcJhdXePDzxiuMnukhV2/zb6VtnZt1mS+SjkLE=
+github.com/mmcdole/goxpp v1.1.1-0.20240225020742-a0c311522b23 h1:Zr92CAlFhy2gL+V1F+EyIuzbQNbSgP4xhTODZtrXUtk=
+github.com/mmcdole/goxpp v1.1.1-0.20240225020742-a0c311522b23/go.mod h1:v+25+lT2ViuQ7mVxcncQ8ch1URund48oH+jhjiwEgS8=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
+github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
+github.com/schollz/progressbar v1.0.0 h1:gbyFReLHDkZo8mxy/dLWMr+Mpb1MokGJ1FqCiqacjZM=
+github.com/schollz/progressbar v1.0.0/go.mod h1:/l9I7PC3L3erOuz54ghIRKUEFcosiWfLvJv+Eq26UMs=
+github.com/schollz/progressbar/v3 v3.14.6 h1:GyjwcWBAf+GFDMLziwerKvpuS7ZF+mNTAXIB2aspiZs=
+github.com/schollz/progressbar/v3 v3.14.6/go.mod h1:Nrzpuw3Nl0srLY0VlTvC4V6RL50pcEymjy6qyJAaLa0=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+golang.org/x/net v0.0.0-20210916014120-12bc252f5db8/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
+golang.org/x/net v0.4.0 h1:Q5QPcMlvfxFTAPV0+07Xz/MpK9NTXu2VDUuy0FeMfaU=
+golang.org/x/net v0.4.0/go.mod h1:MBQ8lrhLObU/6UmLb4fmbmk5OcyYmqtbGd/9yIeKjEE=
+golang.org/x/net v0.29.0 h1:5ORfpBpCs4HzDYoodCDBbwHzdR5UrLBZ3sOnUJmFoHo=
+golang.org/x/net v0.29.0/go.mod h1:gLkgy8jTGERgjzMic6DS9+SP0ajcu6Xu3Orq/SpETg0=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34=
+golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.22.0/go.mod h1:F3qCibpT5AMpCRfhfT53vVJwhLtIVHhB9XDjfFvnMI4=
+golang.org/x/term v0.24.0 h1:Mh5cbb+Zk2hqqXNO7S1iTjEphVL+jb8ZWaqh/g+JWkM=
+golang.org/x/term v0.24.0/go.mod h1:lOBK/LVxemqiMij05LGJ0tzNr8xlmwBRJ81PX6wVLH8=
+golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.5.0 h1:OLmvp0KP+FVG99Ct/qFiL/Fhk4zp4QQnZ7b2U+5piUM=
+golang.org/x/text v0.5.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
+golang.org/x/text v0.18.0 h1:XvMDiNzPAl0jr17s6W9lcaIhGUfUORdGCNsuLmPG224=
+golang.org/x/text v0.18.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.25.0 h1:oFU9pkj/iJgs+0DT+VMHrx+oBKs/LJMV+Uvg78sl+fE=
+golang.org/x/tools v0.25.0/go.mod h1:/vtpO8WL1N9cQC3FN5zPqb//fRXskFHbLKk4OW1Q7rg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"fmt"
+	"go-ani-rss/src/download"
+	"go-ani-rss/src/settings"
+	"time"
+
+	"github.com/mmcdole/gofeed"
+)
+
+var (
+	rssFeedUrl   string
+	feedProvider *gofeed.Parser
+)
+
+func init() {
+	rssFeedUrl = "https://api.ani.rip/ani-download.xml"
+
+	feedProvider = gofeed.NewParser()
+}
+
+func main() {
+	err := settings.ParserSettings()
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	for {
+		feed, err := feedProvider.ParseURL(rssFeedUrl)
+		if err != nil {
+			fmt.Println(err)
+		}
+
+		err = download.ProceedRssItems(feed)
+		if err != nil {
+			fmt.Println(err)
+		}
+
+		time.Sleep(60 * time.Second)
+	}
+}

src/download/download.go

@@ -0,0 +1,103 @@
+package download
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"regexp"
+	"strings"
+
+	"github.com/docker/go-units"
+	"github.com/mmcdole/gofeed"
+	"github.com/schollz/progressbar/v3"
+
+	"go-ani-rss/src/format"
+	"go-ani-rss/src/settings"
+)
+
+func downloadFileFromUrl(url string, path string, size int64) error {
+	// create output file
+	outputDir := filepath.Dir(path)
+	err := os.MkdirAll(outputDir, 0755)
+	if err != nil {
+		return err
+	}
+	output, err := os.Create(path)
+	if err != nil {
+		return err
+	}
+	defer output.Close()
+
+	// make http request
+	request, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return err
+	}
+	response, err := http.DefaultClient.Do(request)
+	if err != nil {
+		return err
+	}
+	defer response.Body.Close()
+
+	// progress bar
+	bar := progressbar.DefaultBytes(size, "downloading")
+
+	// write into target file
+	io.Copy(io.MultiWriter(output, bar), response.Body)
+
+	return nil
+}
+
+func proceedRssItem(item *gofeed.Item, itemConfig settings.ItemConfig) error {
+	fmt.Printf("Started downloading %s\n", item.Title)
+
+	// convert file size
+	size, err := units.FromHumanSize(strings.Replace(item.Extensions["anime"]["size"][0].Value, " ", "", -1))
+	if err != nil {
+		return nil
+	}
+
+	// convert download path
+	path, err := format.FormatPath(item.Title, itemConfig.Name, itemConfig.Path)
+	if err != nil {
+		return err
+	}
+	// download file
+	err = downloadFileFromUrl(item.Link, path, size)
+	if err != nil {
+		fmt.Println(err)
+		return nil
+	}
+
+	fmt.Printf("\nDownloaded %s\n", item.Title)
+	return nil
+}
+
+func ProceedRssItems(feed *gofeed.Feed) error {
+	for _, item := range feed.Items {
+		for _, itemConfig := range settings.Config.ItemConfigs {
+			reg, err := regexp.Compile(itemConfig.Filter)
+			if err != nil {
+				return err
+			}
+
+			if reg.MatchString(item.Title) {
+				if item.PublishedParsed.After(settings.Config.LastDownloadTime) {
+					err = proceedRssItem(item, itemConfig)
+					if err != nil {
+						return err
+					}
+				}
+			}
+		}
+	}
+
+	err := settings.SaveSettings()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

src/format/format.go

@@ -0,0 +1,42 @@
+package format
+
+import (
+	"errors"
+	"regexp"
+	"strings"
+)
+
+func getSeasonId(name string) string {
+	chineseToDigit := map[string]string{"零": "00", "一": "01", "二": "02", "三": "03", "四": "04", "五": "05", "六": "06", "七": "07", "八": "08", "九": "09", "十": "10"}
+
+	const pattern = `第.季`
+	reg, _ := regexp.Compile(pattern)
+
+	matches := reg.FindStringSubmatch(name)
+
+	if len(matches) == 2 && matches[0] == name {
+		return chineseToDigit[matches[1]]
+	}
+
+	return "01"
+}
+
+func FormatPath(originName string, title string, format string) (string, error) {
+	// compile regexp
+	const pattern = `\[ANi\] (.*) - (\d{2}) [^\.]*\.(.*)`
+	reg, _ := regexp.Compile(pattern)
+
+	matches := reg.FindStringSubmatch(originName)
+
+	if len(matches) == 4 && matches[0] == originName {
+		name, seasonId, episodeId, extName := title, getSeasonId(matches[1]), matches[2], matches[3]
+
+		format = strings.ReplaceAll(format, `{name}`, name)
+		format = strings.ReplaceAll(format, `{season}`, seasonId)
+		format = strings.ReplaceAll(format, `{episode}`, episodeId)
+		format = strings.ReplaceAll(format, `{ext}`, extName)
+
+		return format, nil
+	}
+	return "", errors.New("origin name is not valid")
+}

src/settings/settings.go

@@ -0,0 +1,53 @@
+package settings
+
+import (
+	"os"
+	"time"
+
+	"gopkg.in/yaml.v3"
+)
+
+type ItemConfig struct {
+	Name   string `yaml:"name"`
+	Filter string `yaml:"filter"`
+	Path   string `yaml:"path"`
+}
+
+type AppConfig struct {
+	LastDownloadTime time.Time    `yaml:"last_download_time"`
+	ItemConfigs      []ItemConfig `yaml:"downloads"`
+}
+
+var (
+	Config AppConfig
+)
+
+func ParserSettings() error {
+	data, err := os.ReadFile("config.yaml")
+	if err != nil {
+		return err
+	}
+
+	err = yaml.Unmarshal(data, &Config)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func SaveSettings() error {
+	Config.LastDownloadTime = time.Now()
+
+	data, err := yaml.Marshal(&Config)
+	if err != nil {
+		return err
+	}
+
+	err = os.WriteFile("config.yaml", data, os.ModePerm)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}