AI intergration WIP

chore(deps): bump golang.org/x/crypto from 0.27.0 to 0.31.0 in the go_modules group across 1 directory

.github/workflows/go.yml

@@ -29,6 +29,19 @@ jobs:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
           password: ${{ secrets.GHCR_TOKEN }}
+
+      - name: Set up Flutter
+        uses: subosito/flutter-action@v2
+        with:
+          channel: stable
+          flutter-version: 3
+
+      - name: Build Web
+        run: |
+          cd ui
+          flutter pub get
+          flutter build web --no-web-resources-cdn
+          
       - name: Extract metadata (tags, labels) for Docker
         id: meta
         uses: docker/metadata-action@v5

.github/workflows/release.yml

@@ -37,12 +37,25 @@ jobs:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
           password: ${{ secrets.GHCR_TOKEN }}
+  
       - name: Extract metadata (tags, labels) for Docker
         id: meta
         uses: docker/metadata-action@v5
         with:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
 
+      - name: Set up Flutter
+        uses: subosito/flutter-action@v2
+        with:
+          channel: stable
+          flutter-version: 3
+
+      - name: Build Web
+        run: |
+          cd ui
+          flutter pub get
+          flutter build web --no-web-resources-cdn
+
       - name: Build and push
         id: push
         uses: docker/build-push-action@v6
@@ -50,9 +63,7 @@ jobs:
           context: .
           file: Dockerfile
           push: true
-          platforms: |
-            linux/amd64
-            linux/arm64
+          platforms: linux/amd64,linux/arm64,linux/arm/v7,linux/386,linux/s390x,linux/ppc64le
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
 

Dockerfile

@@ -1,12 +1,4 @@
-FROM instrumentisto/flutter:3 AS flutter
-WORKDIR /app
-COPY ./ui/pubspec.yaml ./ui/pubspec.lock ./
-RUN flutter pub get
-COPY ./ui/ ./
-RUN flutter build web --no-web-resources-cdn --web-renderer html
-
-# 打包依赖阶段使用golang作为基础镜像
-FROM golang:1.22 as builder
+FROM golang:1.23 as builder
 
 # 启用go module
 ENV GO111MODULE=on \
@@ -20,21 +12,24 @@ RUN go mod download
 
 COPY . .
 
-COPY --from=flutter /app/build/web ./ui/build/web/
 # 指定OS等，并go build
-RUN CGO_ENABLED=1 go build -o polaris -ldflags="-X polaris/db.Version=$(git describe --tags --long)"  ./cmd/ 
+RUN CGO_ENABLED=0 go build -o polaris -ldflags="-X polaris/db.Version=$(git describe --tags --long)"  ./cmd/ 
 
 FROM debian:stable-slim
-ENV TZ="Asia/Shanghai" GIN_MODE=release
 
 WORKDIR /app
-RUN apt-get update && apt-get -y install ca-certificates
+RUN apt-get update && apt-get -y install ca-certificates tzdata gosu tini locales && ln -sf /usr/share/zoneinfo/${TZ} /etc/localtime \ 
+    && echo "${TZ}" > /etc/timezone && apt-get clean && sed -i '/en_US.UTF-8/s/^# //g' /etc/locale.gen && locale-gen
+
+ENV TZ="Asia/Shanghai" GIN_MODE=release PUID=0 PGID=0 UMASK=0
+ENV LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8
 
 # 将上一个阶段publish文件夹下的所有文件复制进来
 COPY --from=builder /app/polaris .
+COPY --from=builder /app/entrypoint.sh .
+RUN chmod +x /app/entrypoint.sh
 
+VOLUME /app/data
 EXPOSE 8080
 
-#USER 1000:1000
-
-ENTRYPOINT ["./polaris"]
+ENTRYPOINT ["tini","./entrypoint.sh"]

README.md

@@ -1,4 +1,10 @@
-# Polaris
+
+![main_page](./doc/assets/main_page.png)
+
+
+<h1 align="center">Polaris</h1>
+
+<div align="center">
 
 ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/simon-ding/polaris/go.yml)
 ![GitHub Release](https://img.shields.io/github/v/release/simon-ding/polaris)
@@ -6,54 +12,75 @@
 ![GitHub go.mod Go version](https://img.shields.io/github/go-mod/go-version/simon-ding/polaris)
 
 
-Polaris 是一个电视剧和电影的追踪软件。配置好了之后，当剧集或者电影播出后，会第一时间下载对应的资源。支持本地存储或者webdav。
+**Polaris 是一个电视剧和电影的追踪下载软件。对美剧动漫日剧都有良好的匹配，支持多种存储方式（webdav、alist、本地存储）。**
+
+</div>
 
-![main_page](./doc/assets/main_page.png)
-![detail_page](./doc/assets/detail_page.png)
-![anime](./doc/assets/anime_match.png)
 
 交流群： https://t.me/+8R2nzrlSs2JhMDgx
 
 ## 快速开始
 
-使用此程序参考 [【快速开始】](https://simonding.gitbook.io/polaris/quick_start)
+使用此程序参考 [【快速开始】](./doc/quick_start.md)
 
 ## Features
 
 - [x] 电视剧自动追踪下载
 - [x] 电影自动追踪下载
-- [x] webdav 存储支持，配合 [alist](https://github.com/alist-org/alist) 或阿里云等实现更多功能
+- [x] 本地、webdav、  [alist](https://github.com/alist-org/alist) 存储支持，使用 alist 存储支持秒传功能
 - [x] 事件通知推送，目前支持 Pushover和 Bark，还在扩充中
-- [x] 后台代理支持
+- [x] TMDB 代理支持
 - [x] 用户认证
 - [x] plex 刮削支持
 - [x] NFO 刮削文件支持
 - [x] BT/PT 支持
+- [x] qbittorrent/transmission客户端支持
+- [x] 支持导入plex watchlist，plex里标记，自动导入polaris
 - [x] and more...
 
+## 支持的平台
+
+ - linux/amd64
+ - linux/arm64
+ - linux/arm/v7
+ - linux/386
+ - linux/s390x
+ - linux/ppc64le
+
 ## Todos
 
-- [ ] qbittorrent客户端支持
+
 - [ ] 更多通知客户端支持
-- [ ] 第三方watchlist导入支持
+- [ ] 更多第三方watchlist导入支持
 - [ ] 手机客户端
 
 
-## 原理
+## 截图
 
-本程序不提供任何视频相关资源，所有的资源都通过 jackett/prowlarr 所对接的BT/PT站点提供。
-    
-1. 此程序通过调用 jackett/prowlarr API搜索相关资源，然后匹配上对应的剧集
-2. 把搜索到的资源送到下载器下载
-3. 下载完成后归入对应的路径
+![main_page](./doc/assets/main_page.png)
+![detail_page](./doc/assets/detail_page.png)
+![anime](./doc/assets/anime_match.png)
 
-## 对比 sonarr/radarr
+
+## 运行原理
+
+Polaris本身不提供任何资源，要使其能正常工作，需要同时安装一个下载客户端（transmission）和一个索引客户端（jackett）。
+
+Polaris通过索引客户端查询相关的BT/PT站点，然后把查到的资源送到下载器，等待下载器下载完成对应资源。下载完成后，根据下载的对应资源，整理到对应的文件夹里去
+
+![](./doc/assets/yuanli.png)
+
+<!-- ## 对比 sonarr/radarr
 * 更好的中文支持
 * 对于动漫、日剧的良好支持，配合国内站点基本能匹配上对应资源
 * 支持 webdav 后端存储，可以配合 alist 或者阿里云来实现下载后实时传到云上的功能。这样外出就可以不依靠家里的宽带来看电影了，或者实现个轻 NAS 功能，下载功能放在本地，数据放在云盘
 * golang 实现后端，相比于 .NET 更节省资源
 * 一个程序同时实现了电影、电视剧功能，不需要装两个程序
-* 当然 sonarr/radarr 也是非常优秀的开源项目，目前 Polaris 功能还没有 sonarr/radarr 丰富
+* 当然 sonarr/radarr 也是非常优秀的开源项目，目前 Polaris 功能还没有 sonarr/radarr 丰富 -->
+
+
+## Stargazers over time
+[![Stargazers over time](https://starchart.cc/simon-ding/polaris.svg?variant=adaptive)](https://starchart.cc/simon-ding/polaris)
 
 -------------
 

cmd/main.go

@@ -3,28 +3,26 @@ package main
 import (
 	"polaris/db"
 	"polaris/log"
-	"polaris/pkg/utils"
 	"polaris/server"
-	"time"
 )
 
 func main() {
 	log.Infof("------------------- Starting Polaris ---------------------")
 
-	utils.MaxPermission()
+	//utils.MaxPermission()
 
 	dbClient, err := db.Open()
 	if err != nil {
 		log.Panicf("init db error: %v", err)
 	}
 
-	go func() {
-		time.Sleep(2 * time.Second)
-		if err := utils.OpenURL("http://127.0.0.1:8080"); err != nil {
-			log.Errorf("open url error: %v", err)
-		}
+	// go func() {
+	// 	time.Sleep(2 * time.Second)
+	// 	if err := utils.OpenURL("http://127.0.0.1:8080"); err != nil {
+	// 		log.Errorf("open url error: %v", err)
+	// 	}
 
-	}()
+	// }()
 	s := server.NewServer(dbClient)
 	if err := s.Serve(); err != nil {
 		log.Errorf("server start error: %v", err)

db/const.go

@@ -1,5 +1,7 @@
 package db
 
+import "polaris/ent/media"
+
 var Version = "undefined"
 
 const (
@@ -14,9 +16,16 @@ const (
 	SettingNfoSupportEnabled      = "nfo_support_enabled"
 	SettingAllowQiangban          = "filter_qiangban"
 	SettingEnableTmdbAdultContent = "tmdb_adult_content"
-	SetttingSizeLimiter           = "size_limiter"
 	SettingTvNamingFormat         = "tv_naming_format"
 	SettingMovieNamingFormat      = "movie_naming_format"
+	SettingProwlarrInfo           = "prowlarr_info"
+
+	SettingTvSizeLimiter           = "tv_size_limiter"
+	SettingMovieSizeLimiter        = "movie_size_limiter"
+	SettingAcceptedVideoFormats    = "accepted_video_formats"
+	SettingAcceptedSubtitleFormats = "accepted_subtitle_formats"
+
+	SettingAIConfig = "ai_config"
 )
 
 const (
@@ -39,6 +48,18 @@ const (
 
 const DefaultNamingFormat = "{{.NameCN}} {{.NameEN}} {{if .Year}} ({{.Year}}) {{end}}"
 
+// https://en.wikipedia.org/wiki/Video_file_format
+var defaultAcceptedVideoFormats = []string{
+	".webm", ".mkv", ".flv", ".vob", ".ogv", ".ogg", ".drc", ".mng", ".avi", ".mts", ".m2ts", ".ts",
+	".mov", ".qt", ".wmv", ".yuv", ".rm", ".rmvb", ".viv", ".amv", ".mp4", ".m4p", ".m4v",
+	".mpg", ".mp2", ".mpeg", ".mpe", ".mpv", ".m2v", ".m4v",
+	".svi", ".3gp", ".3g2", ".nsv",
+}
+
+var defaultAcceptedSubtitleFormats = []string{
+	".ass", ".srt", ".vtt", ".webvtt", ".sub", ".idx",
+}
+
 type NamingInfo struct {
 	NameCN string
 	NameEN string
@@ -50,13 +71,37 @@ type ResolutionType string
 
 const JwtSerectKey = "jwt_secrect_key"
 
-type SizeLimiter struct {
-	R720p  Limiter `json:"720p"`
-	R1080p Limiter `json:"1080p"`
-	R2160p Limiter `json:"2160p"`
+type MediaSizeLimiter struct {
+	P720p SizeLimiter `json:"720p"`
+	P1080 SizeLimiter `json:"1080p"`
+	P2160 SizeLimiter `json:"2160p"`
 }
 
-type Limiter struct {
-	Max int `json:"max"`
-	Min int `json:"min"`
+func (m *MediaSizeLimiter) GetLimiter(r media.Resolution) SizeLimiter {
+	if r == media.Resolution1080p {
+		return m.P1080
+	} else if r == media.Resolution720p {
+		return m.P720p
+	} else if r == media.Resolution2160p {
+		return m.P2160
+	}
+	return SizeLimiter{}
+}
+
+type SizeLimiter struct {
+	MaxSIze    int64 `json:"max_size"`
+	MinSize    int64 `json:"min_size"`
+	PreferSIze int64 `json:"prefer_size"`
+}
+
+type ProwlarrSetting struct {
+	Disabled bool   `json:"disabled"`
+	ApiKey   string `json:"api_key"`
+	URL      string `json:"url"`
+}
+
+type AIConfig struct {
+	Enabled         bool   `json:"enabled"`
+	GeminiApiKey    string `json:"gemini_api_key"`
+	GeminiModelName string `json:"gemini_model_name"`
 }

db/db.go

@@ -69,9 +69,9 @@ func (c *Client) init() {
 	if tr := c.GetAllDonloadClients(); len(tr) == 0 {
 		log.Warnf("no download client, set default download client")
 		c.SaveDownloader(&ent.DownloadClients{
-			Name: "transmission",
+			Name:           "transmission",
 			Implementation: downloadclients.ImplementationTransmission,
-			URL: "http://transmission:9091",
+			URL:            "http://transmission:9091",
 		})
 	}
 }
@@ -157,6 +157,7 @@ func (c *Client) AddMediaWatchlist(m *ent.Media, episodes []int) (*ent.Media, er
 		SetDownloadHistoryEpisodes(m.DownloadHistoryEpisodes).
 		SetLimiter(m.Limiter).
 		SetExtras(m.Extras).
+		SetAlternativeTitles(m.AlternativeTitles).
 		AddEpisodeIDs(episodes...).
 		Save(context.TODO())
 	return r, err
@@ -251,6 +252,7 @@ func (c *Client) SaveEposideDetail2(d *ent.Episode) (int, error) {
 		SetMediaID(d.MediaID).
 		SetStatus(d.Status).
 		SetOverview(d.Overview).
+		SetMonitored(d.Monitored).
 		SetTitle(d.Title).Save(context.TODO())
 
 	return ep.ID, err
@@ -339,7 +341,6 @@ func (c *Client) SaveDownloader(downloader *ent.DownloadClients) error {
 	return err
 }
 
-
 func (c *Client) GetAllDonloadClients() []*ent.DownloadClients {
 	cc, err := c.ent.DownloadClients.Query().Order(ent.Asc(downloadclients.FieldPriority1)).All(context.TODO())
 	if err != nil {
@@ -375,6 +376,15 @@ func (s *StorageInfo) ToWebDavSetting() WebdavSetting {
 	}
 }
 
+func (s *StorageInfo) ToAlistSetting() WebdavSetting {
+	return WebdavSetting{
+		URL:            s.Settings["url"],
+		User:           s.Settings["user"],
+		Password:       s.Settings["password"],
+		ChangeFileHash: s.Settings["change_file_hash"],
+	}
+}
+
 type WebdavSetting struct {
 	URL            string `json:"url"`
 	User           string `json:"user"`
@@ -433,7 +443,7 @@ type Storage struct {
 }
 
 func (s *Storage) ToWebDavSetting() WebdavSetting {
-	if s.Implementation != storage.ImplementationWebdav {
+	if s.Implementation != storage.ImplementationWebdav && s.Implementation != storage.ImplementationAlist {
 		panic("not webdav storage")
 	}
 	var webdavSetting WebdavSetting
@@ -474,9 +484,18 @@ func (c *Client) SetDefaultStorageByName(name string) error {
 }
 
 func (c *Client) SaveHistoryRecord(h ent.History) (*ent.History, error) {
+	if h.Link != "" {
+		r, err := utils.Link2Magnet(h.Link)
+		if err != nil {
+			log.Warnf("convert link to magnet error, link %v, error: %v", h.Link, err)
+		} else {
+			h.Link = r
+		}
+	}
 	return c.ent.History.Create().SetMediaID(h.MediaID).SetEpisodeID(h.EpisodeID).SetDate(time.Now()).
 		SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetIndexerID(h.IndexerID).
-		SetDownloadClientID(h.DownloadClientID).SetSize(h.Size).SetSaved(h.Saved).Save(context.TODO())
+		SetDownloadClientID(h.DownloadClientID).SetSize(h.Size).SetSaved(h.Saved).SetSeasonNum(h.SeasonNum).
+		SetEpisodeNums(h.EpisodeNums).SetLink(h.Link).Save(context.TODO())
 }
 
 func (c *Client) SetHistoryStatus(id int, status history.Status) error {
@@ -614,19 +633,38 @@ func (c *Client) DeleteImportlist(id int) error {
 	return c.ent.ImportList.DeleteOneID(id).Exec(context.TODO())
 }
 
-func (c *Client) GetSizeLimiter() (*SizeLimiter, error) {
-	v := c.GetSetting(SetttingSizeLimiter)
-	var limiter SizeLimiter
+func (c *Client) GetSizeLimiter(mediaType string) (*MediaSizeLimiter, error) {
+	var v string
+	if mediaType == "tv" {
+		v = c.GetSetting(SettingTvSizeLimiter)
+	} else if mediaType == "movie" {
+		v = c.GetSetting(SettingMovieSizeLimiter)
+	} else {
+		return nil, errors.Errorf("media type not supported: %v", mediaType)
+	}
+
+	var limiter MediaSizeLimiter
+	if v == "" {
+		return &limiter, nil
+	}
+
 	err := json.Unmarshal([]byte(v), &limiter)
 	return &limiter, err
 }
 
-func (c *Client) SetSizeLimiter(limiter *SizeLimiter) error {
+func (c *Client) SetSizeLimiter(mediaType string, limiter *MediaSizeLimiter) error {
 	data, err := json.Marshal(limiter)
 	if err != nil {
 		return err
 	}
-	return c.SetSetting(SetttingSizeLimiter, string(data))
+	if mediaType == "tv" {
+		return c.SetSetting(SettingTvSizeLimiter, string(data))
+	} else if mediaType == "movie" {
+		return c.SetSetting(SettingMovieSizeLimiter, string(data))
+	} else {
+		return errors.Errorf("media type not supported: %v", mediaType)
+	}
+
 }
 
 func (c *Client) GetTvNamingFormat() string {
@@ -649,3 +687,98 @@ func (c *Client) CleanAllDanglingEpisodes() error {
 	_, err := c.ent.Episode.Delete().Where(episode.Not(episode.HasMedia())).Exec(context.Background())
 	return err
 }
+
+func (c *Client) AddBlacklistItem(item *ent.Blacklist) error {
+	return c.ent.Blacklist.Create().SetType(item.Type).SetValue(item.Value).SetNotes(item.Notes).Exec(context.Background())
+}
+
+func (c *Client) GetProwlarrSetting() (*ProwlarrSetting, error) {
+	s := c.GetSetting(SettingProwlarrInfo)
+	if s == "" {
+		return nil, errors.New("prowlarr setting not set")
+	}
+	var se ProwlarrSetting
+	if err := json.Unmarshal([]byte(s), &se); err != nil {
+		return nil, err
+	}
+	return &se, nil
+}
+
+func (c *Client) SaveProwlarrSetting(se *ProwlarrSetting) error {
+	data, err := json.Marshal(se)
+	if err != nil {
+		return err
+	}
+	return c.SetSetting(SettingProwlarrInfo, string(data))
+}
+
+
+func (c *Client) getAcceptedFormats(key string) ([]string, error) {
+	v := c.GetSetting(key)
+	if v == "" {
+		return nil, nil
+	}
+	var res []string
+
+	err := json.Unmarshal([]byte(v), &res)
+	return res, err
+}
+
+func (c *Client) setAcceptedFormats(key string, v []string) error {
+	data, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+	return c.SetSetting(key, string(data))
+} 
+
+func (c *Client) GetAcceptedVideoFormats() ([]string, error) {
+	res, err := c.getAcceptedFormats(SettingAcceptedVideoFormats)
+	if err != nil {
+		return nil, err
+	}
+	if res == nil {
+		return defaultAcceptedVideoFormats, nil
+	}
+	return res, nil
+}
+
+func (c *Client) SetAcceptedVideoFormats(key string, v []string) error {
+	return c.setAcceptedFormats(SettingAcceptedVideoFormats, v)
+}
+
+func (c *Client) GetAcceptedSubtitleFormats() ([]string, error) {
+	res, err := c.getAcceptedFormats(SettingAcceptedSubtitleFormats)
+	if err != nil {
+		return nil, err
+	}
+	if res== nil {
+		return defaultAcceptedSubtitleFormats, nil
+	}
+	return res, nil
+}
+
+func (c *Client) SetAcceptedSubtitleFormats(key string, v []string) error {
+	return c.setAcceptedFormats(SettingAcceptedSubtitleFormats, v)
+}
+
+func (c *Client) GetAIConfig() (AIConfig, error) {
+	cfg := c.GetSetting(SettingAIConfig)
+	var ai AIConfig
+	if cfg == "" {
+		return ai, nil
+	}
+	err := json.Unmarshal([]byte(cfg), &ai)
+	if err != nil {
+		return AIConfig{}, err
+	}
+	return ai, nil
+}
+
+func (c *Client) SetAIConfig(cfg *AIConfig) error {
+	if data, err := json.Marshal(cfg); err != nil {
+		return err
+	} else {
+		return c.SetSetting(SettingAIConfig, string(data))
+	}
+}

doc/alist.md

@@ -0,0 +1,13 @@
+# alist 对接
+
+> 本程序可以把alist作为一个存储后台使用，下载完成的电影电视剧上传到alist对应的文件夹。配合阿里云、夸克云盘等实现云盘NAS功能。目前支持两种对接方式，webdav和直接对接
+
+## webdav
+
+使用webdav形式对接本程序，支持程序所有功能，但是不支持秒传，上传会比较慢
+
+## alist 直接对接
+
+存储设置里选择 alist，填入对应的信息即可。
+
+优点：支持秒传，上传速度快。缺点：部分功能无法使用（plex和nfo文件刮削）

doc/configuration.md

@@ -2,44 +2,28 @@
 
 要正确使用此程序，需要配置好以下设置：
 
-### TMDB设置
+### 1. TMDB设置
+
 1. 因为此程序需要使用到 TMDB 的数据，使用此程序首先要申请一个 TMDB 的 Api Key. 申请教程请 google [tmdb api key申请](https://www.google.com/search?q=tmdb+api+key%E7%94%B3%E8%AF%B7)
 
 2. 拿到 TMDB Api Key之后，请填到 *设置 -> 常规设置 -> TMDB Api Key里*
 
-### 索引器
+**注意：** TMDB可能需要翻墙才能使用，参考 [TMDB 访问问题](./tmdb.md)
 
-索引器是资源提供者，目前支持 torznab 协议，意味着 polarr 或者 jackett 都可以支持。请自行部署相关程序，或者使用的 docker compose 配置一起拉起
+### 2. 索引器
 
-推荐使用 linuxserver 的镜像：https://docs.linuxserver.io/images/docker-jackett/
+使用配置页面索引器配置或者prowlarr设置，其中一个即可。
 
-#### 索引器配置
+#### jackett配置参考 [jackett](./jackett.md)
 
-索引器配置这里以 jackett 为例。使用默认 docker compose 配置拉起后以 http://< ip >:9117 可访问 jackett 的主页。
+#### prowlarr设置
 
-1. 打开 jackett 主页后，点击页面上面的 Add indexer，会出现 BT/PT 站点列表，选择你需要的站点点击+号添加。如果是PT，请自行配置好相关配置
-
-![add indexer](./assets/add_indexer.png)
-
-![search add](./assets/search_add.png)
-
-
-2. 添加后主页即会显示相应的BT/PT站点，点击 *Copy Torznab Feed* 即得到了我们需要的地址
-
-![copy feed](./assets/copy_feed.png)
-
-3. 回到我们的主程序 Polaris 当中，点击 *设置 -> 索引器设置* -> 点击+号增加新的索引器，输入一个名称，拷贝我们第2步得到的地址到地址栏
-
-![polaris add indexer](./assets/polaris_add_indexer.png)
-
-4. 选相框中我们可以看到，还需要一个 API Key，我们回到 Jackett 中，在页面右上角，复制我们需要的 API Key：
-![api key](./assets/jackett_api_key.png)
-
-5. 恭喜！你已经成功完成了索引器配置。如需要更多的站点，请重复相同的操作完成配置
+1) 取得prowlarr的url和api key， api key在 *Prowlarr 设置 -> 通用 -> API 密钥* 处取得
+2) 对应参数填到 polaris程序，*设置->prowlarr设置*当中
 
 ### 下载器
 
-资源下载器，目前可支持 tansmission，请配置好对应配置
+资源下载器，目前可支持 tansmission/qbittorrent，请配置好对应配置
 
 ![transmission](./assets/downloader.png)
 

doc/jackett.md

@@ -0,0 +1,21 @@
+## jackett 索引器配置
+
+1. 打开 jackett 主页后，点击页面上面的 Add indexer，会出现 BT/PT 站点列表，选择你需要的站点点击+号添加。如果是PT，请自行配置好相关配置
+
+![add indexer](./assets/add_indexer.png)
+
+![search add](./assets/search_add.png)
+
+
+2. 添加后主页即会显示相应的BT/PT站点，点击 *Copy Torznab Feed* 即得到了我们需要的地址
+
+![copy feed](./assets/copy_feed.png)
+
+3. 回到我们的主程序 Polaris 当中，点击 *设置 -> 索引器设置* -> 点击+号增加新的索引器，输入一个名称，拷贝我们第2步得到的地址到地址栏
+
+![polaris add indexer](./assets/polaris_add_indexer.png)
+
+4. 选相框中我们可以看到，还需要一个 API Key，我们回到 Jackett 中，在页面右上角，复制我们需要的 API Key：
+![api key](./assets/jackett_api_key.png)
+
+5. 恭喜！你已经成功完成了索引器配置。如需要更多的站点，请重复相同的操作完成配置

doc/quick_start.md

@@ -1,8 +1,83 @@
-## 快速开始
+# 快速开始
 
-最简单部署 Polaris 的方式是使用 docker compose，Polaris要完整运行另外需要一个索引客户端和一个下载客户端。索引客户端支持 polarr 或 jackett，下载客户端目前只支持 transmission。
+## 1. 安装 Polaris
 
-下面是一个示例 docker-compose 配置，为了简单起见，一起拉起了 transmission 和 jackett，你也可选择单独安装
+### 1.1 Docker Compose 方式安装
+
+最简单使用本程序的方式是通过docker compose，下面内容保存成 docker-compose.yml，然后执行 docker compose up -d, 即可拉起程序。
+
+```yaml
+services:
+  polaris:
+    image: ghcr.io/simon-ding/polaris:latest 
+    restart: always
+    environment:
+      - PUID=99 #程序运行的用户UID
+      - PGID=100 #程序运行的用户GID
+      - TZ=Asia/Shanghai #时区
+    volumes:
+      - <配置文件路径>:/app/data #程序配置文件路径
+      - <下载路径>:/downloads #下载路径，需要和下载客户端配置一致
+      - <媒体文件路径>:/data #媒体数据存储路径，也可以启动自己配置webdav存储
+    ports:
+      - 8080:8080 #端口映射，冒号前的端口可自行改为需要的
+```
+
+> latest为发布版本，如果你追求新功能，且能接受bug，可以使用main tag
+
+### 1.2 Docker 方式安装
+
+也可以通过原始 docker 命令的方式安装 Polaris：
+
+```bash
+docker run -d \
+    -v <配置文件路径>:/app/data \
+    -v <下载路径>:/downloads \
+    -v <媒体文件路径>:/data \
+    -e PUID=99 \
+    -e PGID=100 \
+    -e TZ=Asia/Shanghai \
+    -p 8080:8080 \
+    --restart always \
+    ghcr.io/simon-ding/polaris:latest
+```
+
+### 1.3 Unraid 安装
+
+参考下图进行配置
+
+![](./assets/unraid.png)
+
+### 1.4 访问
+拉起之后访问 http://< ip >:8080 即可访问 Polaris 的主页：
+
+![](./assets/main_page.png)
+
+
+## 2. 安装下载客户端
+
+Polaris 需要下载客户端的配合使用，目前支持 Transmission 和 Qbittorrent。推荐使用linuxserver镜像进行安装
+
+ * [linuxserver/transmission](https://docs.linuxserver.io/images/docker-transmission)
+
+ * [linuxserver/qbittorrent](https://docs.linuxserver.io/images/docker-qbittorrent/)
+
+需要注意的是下载客户端内 /downloads 路径的映射地址要和 Polaris的/downloads路径映射保持一致。也就是说他俩都要映射到同一路径。
+
+## 3. 安装 Jackett/Prowlarr
+
+Polaris 如果要正常工作，还需要一个索引客户端的支持，目前支持jackett/prowlarr索引客户端。推荐使用prowlarr，设置更简单
+
+安装方式见：
+
+ * [linuxserver/jackett](https://docs.linuxserver.io/images/docker-jackett/)
+ * [linuxserver/prowlarr](https://docs.linuxserver.io/images/docker-prowlarr/)
+
+
+
+## 4. 联合安装
+
+如果觉得一个个安装麻烦，也可以使用下面docker compose文件，一键拉起所有组件
 
  **注意：** transmission 的下载路径映射要和 polaris 保持一致，如果您不知道怎么做，请保持默认设置。
 
@@ -11,6 +86,10 @@ services:
   polaris:
     image: ghcr.io/simon-ding/polaris:latest
     restart: always
+    environment:
+      - PUID=1000
+      - PGID=1000
+      - TZ=Asia/Shanghai
     volumes:
       - ./config/polaris:/app/data #程序配置文件路径
       - /downloads:/downloads #下载路径，需要和下载客户端配置一致
@@ -43,27 +122,16 @@ services:
     restart: unless-stopped
 ```
 
-拉起之后访问 http://< ip >:8080 的形式访问
+复制上面文件保存成 docker-compose.yml 文件，然后执行下面命令
+
+```bash
+docker compose up -d
+```
 
 
-![](./assets/main_page.png)
-
-## 配置
+## 5. 配置
 
 详细配置请看 [配置篇](./configuration.md)
 
 
-## 开始使用
-
-1. 完成配置之后，我们就可以在右上角的搜索按钮里输入我们想看的电影、电视剧。
-  ![search](./assets/search_series.png)
-
-2. 找到对应电影电视剧后，点击加入想看列表
-  ![add](./assets/add_series.png)
-
-3. 当电影有资源、或者电视剧有更新时，程序就会自动下载对应资源到指定的存储。对于剧集，您也可以进入剧集的详细页面，点击搜索按钮来自己搜索对应集的资源。
-
-
-到此，您已经基本掌握了此程序的使用方式，请尽情体验吧！
-
 

doc/tmdb.md

@@ -0,0 +1,6 @@
+# TMDB 访问问题
+
+由于国内特殊的网络环境，访问 tmdb 可能需要翻墙代理，有两种解决方案
+
+1. 设置路由器全局代理，openwrt等第三方路由固件可以实现这个功能
+2. 使用程序代理功能，*常规->代理地址* 中设置，格式如 http://192.168.0.10:1080

ent/blacklist.go

@@ -3,34 +3,40 @@
 package ent
 
 import (
+	"encoding/json"
 	"fmt"
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
+	"polaris/ent/schema"
 	"strings"
 
 	"entgo.io/ent"
 	"entgo.io/ent/dialect/sql"
 )
 
-// Blocklist is the model entity for the Blocklist schema.
-type Blocklist struct {
+// Blacklist is the model entity for the Blacklist schema.
+type Blacklist struct {
 	config `json:"-"`
 	// ID of the ent.
 	ID int `json:"id,omitempty"`
 	// Type holds the value of the "type" field.
-	Type blocklist.Type `json:"type,omitempty"`
+	Type blacklist.Type `json:"type,omitempty"`
 	// Value holds the value of the "value" field.
-	Value        string `json:"value,omitempty"`
+	Value schema.BlacklistValue `json:"value,omitempty"`
+	// Notes holds the value of the "notes" field.
+	Notes        string `json:"notes,omitempty"`
 	selectValues sql.SelectValues
 }
 
 // scanValues returns the types for scanning values from sql.Rows.
-func (*Blocklist) scanValues(columns []string) ([]any, error) {
+func (*Blacklist) scanValues(columns []string) ([]any, error) {
 	values := make([]any, len(columns))
 	for i := range columns {
 		switch columns[i] {
-		case blocklist.FieldID:
+		case blacklist.FieldValue:
+			values[i] = new([]byte)
+		case blacklist.FieldID:
 			values[i] = new(sql.NullInt64)
-		case blocklist.FieldType, blocklist.FieldValue:
+		case blacklist.FieldType, blacklist.FieldNotes:
 			values[i] = new(sql.NullString)
 		default:
 			values[i] = new(sql.UnknownType)
@@ -40,30 +46,38 @@ func (*Blocklist) scanValues(columns []string) ([]any, error) {
 }
 
 // assignValues assigns the values that were returned from sql.Rows (after scanning)
-// to the Blocklist fields.
-func (b *Blocklist) assignValues(columns []string, values []any) error {
+// to the Blacklist fields.
+func (b *Blacklist) assignValues(columns []string, values []any) error {
 	if m, n := len(values), len(columns); m < n {
 		return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
 	}
 	for i := range columns {
 		switch columns[i] {
-		case blocklist.FieldID:
+		case blacklist.FieldID:
 			value, ok := values[i].(*sql.NullInt64)
 			if !ok {
 				return fmt.Errorf("unexpected type %T for field id", value)
 			}
 			b.ID = int(value.Int64)
-		case blocklist.FieldType:
+		case blacklist.FieldType:
 			if value, ok := values[i].(*sql.NullString); !ok {
 				return fmt.Errorf("unexpected type %T for field type", values[i])
 			} else if value.Valid {
-				b.Type = blocklist.Type(value.String)
+				b.Type = blacklist.Type(value.String)
 			}
-		case blocklist.FieldValue:
-			if value, ok := values[i].(*sql.NullString); !ok {
+		case blacklist.FieldValue:
+			if value, ok := values[i].(*[]byte); !ok {
 				return fmt.Errorf("unexpected type %T for field value", values[i])
+			} else if value != nil && len(*value) > 0 {
+				if err := json.Unmarshal(*value, &b.Value); err != nil {
+					return fmt.Errorf("unmarshal field value: %w", err)
+				}
+			}
+		case blacklist.FieldNotes:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field notes", values[i])
 			} else if value.Valid {
-				b.Value = value.String
+				b.Notes = value.String
 			}
 		default:
 			b.selectValues.Set(columns[i], values[i])
@@ -72,43 +86,46 @@ func (b *Blocklist) assignValues(columns []string, values []any) error {
 	return nil
 }
 
-// GetValue returns the ent.Value that was dynamically selected and assigned to the Blocklist.
+// GetValue returns the ent.Value that was dynamically selected and assigned to the Blacklist.
 // This includes values selected through modifiers, order, etc.
-func (b *Blocklist) GetValue(name string) (ent.Value, error) {
+func (b *Blacklist) GetValue(name string) (ent.Value, error) {
 	return b.selectValues.Get(name)
 }
 
-// Update returns a builder for updating this Blocklist.
-// Note that you need to call Blocklist.Unwrap() before calling this method if this Blocklist
+// Update returns a builder for updating this Blacklist.
+// Note that you need to call Blacklist.Unwrap() before calling this method if this Blacklist
 // was returned from a transaction, and the transaction was committed or rolled back.
-func (b *Blocklist) Update() *BlocklistUpdateOne {
-	return NewBlocklistClient(b.config).UpdateOne(b)
+func (b *Blacklist) Update() *BlacklistUpdateOne {
+	return NewBlacklistClient(b.config).UpdateOne(b)
 }
 
-// Unwrap unwraps the Blocklist entity that was returned from a transaction after it was closed,
+// Unwrap unwraps the Blacklist entity that was returned from a transaction after it was closed,
 // so that all future queries will be executed through the driver which created the transaction.
-func (b *Blocklist) Unwrap() *Blocklist {
+func (b *Blacklist) Unwrap() *Blacklist {
 	_tx, ok := b.config.driver.(*txDriver)
 	if !ok {
-		panic("ent: Blocklist is not a transactional entity")
+		panic("ent: Blacklist is not a transactional entity")
 	}
 	b.config.driver = _tx.drv
 	return b
 }
 
 // String implements the fmt.Stringer.
-func (b *Blocklist) String() string {
+func (b *Blacklist) String() string {
 	var builder strings.Builder
-	builder.WriteString("Blocklist(")
+	builder.WriteString("Blacklist(")
 	builder.WriteString(fmt.Sprintf("id=%v, ", b.ID))
 	builder.WriteString("type=")
 	builder.WriteString(fmt.Sprintf("%v", b.Type))
 	builder.WriteString(", ")
 	builder.WriteString("value=")
-	builder.WriteString(b.Value)
+	builder.WriteString(fmt.Sprintf("%v", b.Value))
+	builder.WriteString(", ")
+	builder.WriteString("notes=")
+	builder.WriteString(b.Notes)
 	builder.WriteByte(')')
 	return builder.String()
 }
 
-// Blocklists is a parsable slice of Blocklist.
-type Blocklists []*Blocklist
+// Blacklists is a parsable slice of Blacklist.
+type Blacklists []*Blacklist

ent/blacklist/blacklist.go

@@ -1,31 +1,35 @@
 // Code generated by ent, DO NOT EDIT.
 
-package blocklist
+package blacklist
 
 import (
 	"fmt"
+	"polaris/ent/schema"
 
 	"entgo.io/ent/dialect/sql"
 )
 
 const (
-	// Label holds the string label denoting the blocklist type in the database.
-	Label = "blocklist"
+	// Label holds the string label denoting the blacklist type in the database.
+	Label = "blacklist"
 	// FieldID holds the string denoting the id field in the database.
 	FieldID = "id"
 	// FieldType holds the string denoting the type field in the database.
 	FieldType = "type"
 	// FieldValue holds the string denoting the value field in the database.
 	FieldValue = "value"
-	// Table holds the table name of the blocklist in the database.
-	Table = "blocklists"
+	// FieldNotes holds the string denoting the notes field in the database.
+	FieldNotes = "notes"
+	// Table holds the table name of the blacklist in the database.
+	Table = "blacklists"
 )
 
-// Columns holds all SQL columns for blocklist fields.
+// Columns holds all SQL columns for blacklist fields.
 var Columns = []string{
 	FieldID,
 	FieldType,
 	FieldValue,
+	FieldNotes,
 }
 
 // ValidColumn reports if the column name is valid (part of the table columns).
@@ -38,6 +42,11 @@ func ValidColumn(column string) bool {
 	return false
 }
 
+var (
+	// DefaultValue holds the default value on creation for the "value" field.
+	DefaultValue schema.BlacklistValue
+)
+
 // Type defines the type for the "type" enum field.
 type Type string
 
@@ -57,11 +66,11 @@ func TypeValidator(_type Type) error {
 	case TypeMedia, TypeTorrent:
 		return nil
 	default:
-		return fmt.Errorf("blocklist: invalid enum value for type field: %q", _type)
+		return fmt.Errorf("blacklist: invalid enum value for type field: %q", _type)
 	}
 }
 
-// OrderOption defines the ordering options for the Blocklist queries.
+// OrderOption defines the ordering options for the Blacklist queries.
 type OrderOption func(*sql.Selector)
 
 // ByID orders the results by the id field.
@@ -74,7 +83,7 @@ func ByType(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldType, opts...).ToFunc()
 }
 
-// ByValue orders the results by the value field.
-func ByValue(opts ...sql.OrderTermOption) OrderOption {
-	return sql.OrderByField(FieldValue, opts...).ToFunc()
+// ByNotes orders the results by the notes field.
+func ByNotes(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldNotes, opts...).ToFunc()
 }

ent/blacklist/where.go

@@ -0,0 +1,169 @@
+// Code generated by ent, DO NOT EDIT.
+
+package blacklist
+
+import (
+	"polaris/ent/predicate"
+
+	"entgo.io/ent/dialect/sql"
+)
+
+// ID filters vertices based on their ID field.
+func ID(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEQ(FieldID, id))
+}
+
+// IDEQ applies the EQ predicate on the ID field.
+func IDEQ(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEQ(FieldID, id))
+}
+
+// IDNEQ applies the NEQ predicate on the ID field.
+func IDNEQ(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNEQ(FieldID, id))
+}
+
+// IDIn applies the In predicate on the ID field.
+func IDIn(ids ...int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldIn(FieldID, ids...))
+}
+
+// IDNotIn applies the NotIn predicate on the ID field.
+func IDNotIn(ids ...int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNotIn(FieldID, ids...))
+}
+
+// IDGT applies the GT predicate on the ID field.
+func IDGT(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldGT(FieldID, id))
+}
+
+// IDGTE applies the GTE predicate on the ID field.
+func IDGTE(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldGTE(FieldID, id))
+}
+
+// IDLT applies the LT predicate on the ID field.
+func IDLT(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldLT(FieldID, id))
+}
+
+// IDLTE applies the LTE predicate on the ID field.
+func IDLTE(id int) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldLTE(FieldID, id))
+}
+
+// Notes applies equality check predicate on the "notes" field. It's identical to NotesEQ.
+func Notes(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEQ(FieldNotes, v))
+}
+
+// TypeEQ applies the EQ predicate on the "type" field.
+func TypeEQ(v Type) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEQ(FieldType, v))
+}
+
+// TypeNEQ applies the NEQ predicate on the "type" field.
+func TypeNEQ(v Type) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNEQ(FieldType, v))
+}
+
+// TypeIn applies the In predicate on the "type" field.
+func TypeIn(vs ...Type) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldIn(FieldType, vs...))
+}
+
+// TypeNotIn applies the NotIn predicate on the "type" field.
+func TypeNotIn(vs ...Type) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNotIn(FieldType, vs...))
+}
+
+// NotesEQ applies the EQ predicate on the "notes" field.
+func NotesEQ(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEQ(FieldNotes, v))
+}
+
+// NotesNEQ applies the NEQ predicate on the "notes" field.
+func NotesNEQ(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNEQ(FieldNotes, v))
+}
+
+// NotesIn applies the In predicate on the "notes" field.
+func NotesIn(vs ...string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldIn(FieldNotes, vs...))
+}
+
+// NotesNotIn applies the NotIn predicate on the "notes" field.
+func NotesNotIn(vs ...string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNotIn(FieldNotes, vs...))
+}
+
+// NotesGT applies the GT predicate on the "notes" field.
+func NotesGT(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldGT(FieldNotes, v))
+}
+
+// NotesGTE applies the GTE predicate on the "notes" field.
+func NotesGTE(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldGTE(FieldNotes, v))
+}
+
+// NotesLT applies the LT predicate on the "notes" field.
+func NotesLT(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldLT(FieldNotes, v))
+}
+
+// NotesLTE applies the LTE predicate on the "notes" field.
+func NotesLTE(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldLTE(FieldNotes, v))
+}
+
+// NotesContains applies the Contains predicate on the "notes" field.
+func NotesContains(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldContains(FieldNotes, v))
+}
+
+// NotesHasPrefix applies the HasPrefix predicate on the "notes" field.
+func NotesHasPrefix(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldHasPrefix(FieldNotes, v))
+}
+
+// NotesHasSuffix applies the HasSuffix predicate on the "notes" field.
+func NotesHasSuffix(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldHasSuffix(FieldNotes, v))
+}
+
+// NotesIsNil applies the IsNil predicate on the "notes" field.
+func NotesIsNil() predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldIsNull(FieldNotes))
+}
+
+// NotesNotNil applies the NotNil predicate on the "notes" field.
+func NotesNotNil() predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldNotNull(FieldNotes))
+}
+
+// NotesEqualFold applies the EqualFold predicate on the "notes" field.
+func NotesEqualFold(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldEqualFold(FieldNotes, v))
+}
+
+// NotesContainsFold applies the ContainsFold predicate on the "notes" field.
+func NotesContainsFold(v string) predicate.Blacklist {
+	return predicate.Blacklist(sql.FieldContainsFold(FieldNotes, v))
+}
+
+// And groups predicates with the AND operator between them.
+func And(predicates ...predicate.Blacklist) predicate.Blacklist {
+	return predicate.Blacklist(sql.AndPredicates(predicates...))
+}
+
+// Or groups predicates with the OR operator between them.
+func Or(predicates ...predicate.Blacklist) predicate.Blacklist {
+	return predicate.Blacklist(sql.OrPredicates(predicates...))
+}
+
+// Not applies the not operator on the given predicate.
+func Not(p predicate.Blacklist) predicate.Blacklist {
+	return predicate.Blacklist(sql.NotPredicates(p))
+}

ent/blacklist_create.go

@@ -6,43 +6,67 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
+	"polaris/ent/schema"
 
 	"entgo.io/ent/dialect/sql/sqlgraph"
 	"entgo.io/ent/schema/field"
 )
 
-// BlocklistCreate is the builder for creating a Blocklist entity.
-type BlocklistCreate struct {
+// BlacklistCreate is the builder for creating a Blacklist entity.
+type BlacklistCreate struct {
 	config
-	mutation *BlocklistMutation
+	mutation *BlacklistMutation
 	hooks    []Hook
 }
 
 // SetType sets the "type" field.
-func (bc *BlocklistCreate) SetType(b blocklist.Type) *BlocklistCreate {
+func (bc *BlacklistCreate) SetType(b blacklist.Type) *BlacklistCreate {
 	bc.mutation.SetType(b)
 	return bc
 }
 
 // SetValue sets the "value" field.
-func (bc *BlocklistCreate) SetValue(s string) *BlocklistCreate {
-	bc.mutation.SetValue(s)
+func (bc *BlacklistCreate) SetValue(sv schema.BlacklistValue) *BlacklistCreate {
+	bc.mutation.SetValue(sv)
 	return bc
 }
 
-// Mutation returns the BlocklistMutation object of the builder.
-func (bc *BlocklistCreate) Mutation() *BlocklistMutation {
+// SetNillableValue sets the "value" field if the given value is not nil.
+func (bc *BlacklistCreate) SetNillableValue(sv *schema.BlacklistValue) *BlacklistCreate {
+	if sv != nil {
+		bc.SetValue(*sv)
+	}
+	return bc
+}
+
+// SetNotes sets the "notes" field.
+func (bc *BlacklistCreate) SetNotes(s string) *BlacklistCreate {
+	bc.mutation.SetNotes(s)
+	return bc
+}
+
+// SetNillableNotes sets the "notes" field if the given value is not nil.
+func (bc *BlacklistCreate) SetNillableNotes(s *string) *BlacklistCreate {
+	if s != nil {
+		bc.SetNotes(*s)
+	}
+	return bc
+}
+
+// Mutation returns the BlacklistMutation object of the builder.
+func (bc *BlacklistCreate) Mutation() *BlacklistMutation {
 	return bc.mutation
 }
 
-// Save creates the Blocklist in the database.
-func (bc *BlocklistCreate) Save(ctx context.Context) (*Blocklist, error) {
+// Save creates the Blacklist in the database.
+func (bc *BlacklistCreate) Save(ctx context.Context) (*Blacklist, error) {
+	bc.defaults()
 	return withHooks(ctx, bc.sqlSave, bc.mutation, bc.hooks)
 }
 
 // SaveX calls Save and panics if Save returns an error.
-func (bc *BlocklistCreate) SaveX(ctx context.Context) *Blocklist {
+func (bc *BlacklistCreate) SaveX(ctx context.Context) *Blacklist {
 	v, err := bc.Save(ctx)
 	if err != nil {
 		panic(err)
@@ -51,35 +75,43 @@ func (bc *BlocklistCreate) SaveX(ctx context.Context) *Blocklist {
 }
 
 // Exec executes the query.
-func (bc *BlocklistCreate) Exec(ctx context.Context) error {
+func (bc *BlacklistCreate) Exec(ctx context.Context) error {
 	_, err := bc.Save(ctx)
 	return err
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (bc *BlocklistCreate) ExecX(ctx context.Context) {
+func (bc *BlacklistCreate) ExecX(ctx context.Context) {
 	if err := bc.Exec(ctx); err != nil {
 		panic(err)
 	}
 }
 
+// defaults sets the default values of the builder before save.
+func (bc *BlacklistCreate) defaults() {
+	if _, ok := bc.mutation.Value(); !ok {
+		v := blacklist.DefaultValue
+		bc.mutation.SetValue(v)
+	}
+}
+
 // check runs all checks and user-defined validators on the builder.
-func (bc *BlocklistCreate) check() error {
+func (bc *BlacklistCreate) check() error {
 	if _, ok := bc.mutation.GetType(); !ok {
-		return &ValidationError{Name: "type", err: errors.New(`ent: missing required field "Blocklist.type"`)}
+		return &ValidationError{Name: "type", err: errors.New(`ent: missing required field "Blacklist.type"`)}
 	}
 	if v, ok := bc.mutation.GetType(); ok {
-		if err := blocklist.TypeValidator(v); err != nil {
-			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blocklist.type": %w`, err)}
+		if err := blacklist.TypeValidator(v); err != nil {
+			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blacklist.type": %w`, err)}
 		}
 	}
 	if _, ok := bc.mutation.Value(); !ok {
-		return &ValidationError{Name: "value", err: errors.New(`ent: missing required field "Blocklist.value"`)}
+		return &ValidationError{Name: "value", err: errors.New(`ent: missing required field "Blacklist.value"`)}
 	}
 	return nil
 }
 
-func (bc *BlocklistCreate) sqlSave(ctx context.Context) (*Blocklist, error) {
+func (bc *BlacklistCreate) sqlSave(ctx context.Context) (*Blacklist, error) {
 	if err := bc.check(); err != nil {
 		return nil, err
 	}
@@ -97,42 +129,47 @@ func (bc *BlocklistCreate) sqlSave(ctx context.Context) (*Blocklist, error) {
 	return _node, nil
 }
 
-func (bc *BlocklistCreate) createSpec() (*Blocklist, *sqlgraph.CreateSpec) {
+func (bc *BlacklistCreate) createSpec() (*Blacklist, *sqlgraph.CreateSpec) {
 	var (
-		_node = &Blocklist{config: bc.config}
-		_spec = sqlgraph.NewCreateSpec(blocklist.Table, sqlgraph.NewFieldSpec(blocklist.FieldID, field.TypeInt))
+		_node = &Blacklist{config: bc.config}
+		_spec = sqlgraph.NewCreateSpec(blacklist.Table, sqlgraph.NewFieldSpec(blacklist.FieldID, field.TypeInt))
 	)
 	if value, ok := bc.mutation.GetType(); ok {
-		_spec.SetField(blocklist.FieldType, field.TypeEnum, value)
+		_spec.SetField(blacklist.FieldType, field.TypeEnum, value)
 		_node.Type = value
 	}
 	if value, ok := bc.mutation.Value(); ok {
-		_spec.SetField(blocklist.FieldValue, field.TypeString, value)
+		_spec.SetField(blacklist.FieldValue, field.TypeJSON, value)
 		_node.Value = value
 	}
+	if value, ok := bc.mutation.Notes(); ok {
+		_spec.SetField(blacklist.FieldNotes, field.TypeString, value)
+		_node.Notes = value
+	}
 	return _node, _spec
 }
 
-// BlocklistCreateBulk is the builder for creating many Blocklist entities in bulk.
-type BlocklistCreateBulk struct {
+// BlacklistCreateBulk is the builder for creating many Blacklist entities in bulk.
+type BlacklistCreateBulk struct {
 	config
 	err      error
-	builders []*BlocklistCreate
+	builders []*BlacklistCreate
 }
 
-// Save creates the Blocklist entities in the database.
-func (bcb *BlocklistCreateBulk) Save(ctx context.Context) ([]*Blocklist, error) {
+// Save creates the Blacklist entities in the database.
+func (bcb *BlacklistCreateBulk) Save(ctx context.Context) ([]*Blacklist, error) {
 	if bcb.err != nil {
 		return nil, bcb.err
 	}
 	specs := make([]*sqlgraph.CreateSpec, len(bcb.builders))
-	nodes := make([]*Blocklist, len(bcb.builders))
+	nodes := make([]*Blacklist, len(bcb.builders))
 	mutators := make([]Mutator, len(bcb.builders))
 	for i := range bcb.builders {
 		func(i int, root context.Context) {
 			builder := bcb.builders[i]
+			builder.defaults()
 			var mut Mutator = MutateFunc(func(ctx context.Context, m Mutation) (Value, error) {
-				mutation, ok := m.(*BlocklistMutation)
+				mutation, ok := m.(*BlacklistMutation)
 				if !ok {
 					return nil, fmt.Errorf("unexpected mutation type %T", m)
 				}
@@ -179,7 +216,7 @@ func (bcb *BlocklistCreateBulk) Save(ctx context.Context) ([]*Blocklist, error)
 }
 
 // SaveX is like Save, but panics if an error occurs.
-func (bcb *BlocklistCreateBulk) SaveX(ctx context.Context) []*Blocklist {
+func (bcb *BlacklistCreateBulk) SaveX(ctx context.Context) []*Blacklist {
 	v, err := bcb.Save(ctx)
 	if err != nil {
 		panic(err)
@@ -188,13 +225,13 @@ func (bcb *BlocklistCreateBulk) SaveX(ctx context.Context) []*Blocklist {
 }
 
 // Exec executes the query.
-func (bcb *BlocklistCreateBulk) Exec(ctx context.Context) error {
+func (bcb *BlacklistCreateBulk) Exec(ctx context.Context) error {
 	_, err := bcb.Save(ctx)
 	return err
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (bcb *BlocklistCreateBulk) ExecX(ctx context.Context) {
+func (bcb *BlacklistCreateBulk) ExecX(ctx context.Context) {
 	if err := bcb.Exec(ctx); err != nil {
 		panic(err)
 	}

ent/blacklist_delete.go

@@ -4,7 +4,7 @@ package ent
 
 import (
 	"context"
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
 	"polaris/ent/predicate"
 
 	"entgo.io/ent/dialect/sql"
@@ -12,26 +12,26 @@ import (
 	"entgo.io/ent/schema/field"
 )
 
-// BlocklistDelete is the builder for deleting a Blocklist entity.
-type BlocklistDelete struct {
+// BlacklistDelete is the builder for deleting a Blacklist entity.
+type BlacklistDelete struct {
 	config
 	hooks    []Hook
-	mutation *BlocklistMutation
+	mutation *BlacklistMutation
 }
 
-// Where appends a list predicates to the BlocklistDelete builder.
-func (bd *BlocklistDelete) Where(ps ...predicate.Blocklist) *BlocklistDelete {
+// Where appends a list predicates to the BlacklistDelete builder.
+func (bd *BlacklistDelete) Where(ps ...predicate.Blacklist) *BlacklistDelete {
 	bd.mutation.Where(ps...)
 	return bd
 }
 
 // Exec executes the deletion query and returns how many vertices were deleted.
-func (bd *BlocklistDelete) Exec(ctx context.Context) (int, error) {
+func (bd *BlacklistDelete) Exec(ctx context.Context) (int, error) {
 	return withHooks(ctx, bd.sqlExec, bd.mutation, bd.hooks)
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (bd *BlocklistDelete) ExecX(ctx context.Context) int {
+func (bd *BlacklistDelete) ExecX(ctx context.Context) int {
 	n, err := bd.Exec(ctx)
 	if err != nil {
 		panic(err)
@@ -39,8 +39,8 @@ func (bd *BlocklistDelete) ExecX(ctx context.Context) int {
 	return n
 }
 
-func (bd *BlocklistDelete) sqlExec(ctx context.Context) (int, error) {
-	_spec := sqlgraph.NewDeleteSpec(blocklist.Table, sqlgraph.NewFieldSpec(blocklist.FieldID, field.TypeInt))
+func (bd *BlacklistDelete) sqlExec(ctx context.Context) (int, error) {
+	_spec := sqlgraph.NewDeleteSpec(blacklist.Table, sqlgraph.NewFieldSpec(blacklist.FieldID, field.TypeInt))
 	if ps := bd.mutation.predicates; len(ps) > 0 {
 		_spec.Predicate = func(selector *sql.Selector) {
 			for i := range ps {
@@ -56,32 +56,32 @@ func (bd *BlocklistDelete) sqlExec(ctx context.Context) (int, error) {
 	return affected, err
 }
 
-// BlocklistDeleteOne is the builder for deleting a single Blocklist entity.
-type BlocklistDeleteOne struct {
-	bd *BlocklistDelete
+// BlacklistDeleteOne is the builder for deleting a single Blacklist entity.
+type BlacklistDeleteOne struct {
+	bd *BlacklistDelete
 }
 
-// Where appends a list predicates to the BlocklistDelete builder.
-func (bdo *BlocklistDeleteOne) Where(ps ...predicate.Blocklist) *BlocklistDeleteOne {
+// Where appends a list predicates to the BlacklistDelete builder.
+func (bdo *BlacklistDeleteOne) Where(ps ...predicate.Blacklist) *BlacklistDeleteOne {
 	bdo.bd.mutation.Where(ps...)
 	return bdo
 }
 
 // Exec executes the deletion query.
-func (bdo *BlocklistDeleteOne) Exec(ctx context.Context) error {
+func (bdo *BlacklistDeleteOne) Exec(ctx context.Context) error {
 	n, err := bdo.bd.Exec(ctx)
 	switch {
 	case err != nil:
 		return err
 	case n == 0:
-		return &NotFoundError{blocklist.Label}
+		return &NotFoundError{blacklist.Label}
 	default:
 		return nil
 	}
 }
 
 // ExecX is like Exec, but panics if an error occurs.
-func (bdo *BlocklistDeleteOne) ExecX(ctx context.Context) {
+func (bdo *BlacklistDeleteOne) ExecX(ctx context.Context) {
 	if err := bdo.Exec(ctx); err != nil {
 		panic(err)
 	}

ent/blacklist_query.go

@@ -6,7 +6,7 @@ import (
 	"context"
 	"fmt"
 	"math"
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
 	"polaris/ent/predicate"
 
 	"entgo.io/ent/dialect/sql"
@@ -14,64 +14,64 @@ import (
 	"entgo.io/ent/schema/field"
 )
 
-// BlocklistQuery is the builder for querying Blocklist entities.
-type BlocklistQuery struct {
+// BlacklistQuery is the builder for querying Blacklist entities.
+type BlacklistQuery struct {
 	config
 	ctx        *QueryContext
-	order      []blocklist.OrderOption
+	order      []blacklist.OrderOption
 	inters     []Interceptor
-	predicates []predicate.Blocklist
+	predicates []predicate.Blacklist
 	// intermediate query (i.e. traversal path).
 	sql  *sql.Selector
 	path func(context.Context) (*sql.Selector, error)
 }
 
-// Where adds a new predicate for the BlocklistQuery builder.
-func (bq *BlocklistQuery) Where(ps ...predicate.Blocklist) *BlocklistQuery {
+// Where adds a new predicate for the BlacklistQuery builder.
+func (bq *BlacklistQuery) Where(ps ...predicate.Blacklist) *BlacklistQuery {
 	bq.predicates = append(bq.predicates, ps...)
 	return bq
 }
 
 // Limit the number of records to be returned by this query.
-func (bq *BlocklistQuery) Limit(limit int) *BlocklistQuery {
+func (bq *BlacklistQuery) Limit(limit int) *BlacklistQuery {
 	bq.ctx.Limit = &limit
 	return bq
 }
 
 // Offset to start from.
-func (bq *BlocklistQuery) Offset(offset int) *BlocklistQuery {
+func (bq *BlacklistQuery) Offset(offset int) *BlacklistQuery {
 	bq.ctx.Offset = &offset
 	return bq
 }
 
 // Unique configures the query builder to filter duplicate records on query.
 // By default, unique is set to true, and can be disabled using this method.
-func (bq *BlocklistQuery) Unique(unique bool) *BlocklistQuery {
+func (bq *BlacklistQuery) Unique(unique bool) *BlacklistQuery {
 	bq.ctx.Unique = &unique
 	return bq
 }
 
 // Order specifies how the records should be ordered.
-func (bq *BlocklistQuery) Order(o ...blocklist.OrderOption) *BlocklistQuery {
+func (bq *BlacklistQuery) Order(o ...blacklist.OrderOption) *BlacklistQuery {
 	bq.order = append(bq.order, o...)
 	return bq
 }
 
-// First returns the first Blocklist entity from the query.
-// Returns a *NotFoundError when no Blocklist was found.
-func (bq *BlocklistQuery) First(ctx context.Context) (*Blocklist, error) {
+// First returns the first Blacklist entity from the query.
+// Returns a *NotFoundError when no Blacklist was found.
+func (bq *BlacklistQuery) First(ctx context.Context) (*Blacklist, error) {
 	nodes, err := bq.Limit(1).All(setContextOp(ctx, bq.ctx, "First"))
 	if err != nil {
 		return nil, err
 	}
 	if len(nodes) == 0 {
-		return nil, &NotFoundError{blocklist.Label}
+		return nil, &NotFoundError{blacklist.Label}
 	}
 	return nodes[0], nil
 }
 
 // FirstX is like First, but panics if an error occurs.
-func (bq *BlocklistQuery) FirstX(ctx context.Context) *Blocklist {
+func (bq *BlacklistQuery) FirstX(ctx context.Context) *Blacklist {
 	node, err := bq.First(ctx)
 	if err != nil && !IsNotFound(err) {
 		panic(err)
@@ -79,22 +79,22 @@ func (bq *BlocklistQuery) FirstX(ctx context.Context) *Blocklist {
 	return node
 }
 
-// FirstID returns the first Blocklist ID from the query.
-// Returns a *NotFoundError when no Blocklist ID was found.
-func (bq *BlocklistQuery) FirstID(ctx context.Context) (id int, err error) {
+// FirstID returns the first Blacklist ID from the query.
+// Returns a *NotFoundError when no Blacklist ID was found.
+func (bq *BlacklistQuery) FirstID(ctx context.Context) (id int, err error) {
 	var ids []int
 	if ids, err = bq.Limit(1).IDs(setContextOp(ctx, bq.ctx, "FirstID")); err != nil {
 		return
 	}
 	if len(ids) == 0 {
-		err = &NotFoundError{blocklist.Label}
+		err = &NotFoundError{blacklist.Label}
 		return
 	}
 	return ids[0], nil
 }
 
 // FirstIDX is like FirstID, but panics if an error occurs.
-func (bq *BlocklistQuery) FirstIDX(ctx context.Context) int {
+func (bq *BlacklistQuery) FirstIDX(ctx context.Context) int {
 	id, err := bq.FirstID(ctx)
 	if err != nil && !IsNotFound(err) {
 		panic(err)
@@ -102,10 +102,10 @@ func (bq *BlocklistQuery) FirstIDX(ctx context.Context) int {
 	return id
 }
 
-// Only returns a single Blocklist entity found by the query, ensuring it only returns one.
-// Returns a *NotSingularError when more than one Blocklist entity is found.
-// Returns a *NotFoundError when no Blocklist entities are found.
-func (bq *BlocklistQuery) Only(ctx context.Context) (*Blocklist, error) {
+// Only returns a single Blacklist entity found by the query, ensuring it only returns one.
+// Returns a *NotSingularError when more than one Blacklist entity is found.
+// Returns a *NotFoundError when no Blacklist entities are found.
+func (bq *BlacklistQuery) Only(ctx context.Context) (*Blacklist, error) {
 	nodes, err := bq.Limit(2).All(setContextOp(ctx, bq.ctx, "Only"))
 	if err != nil {
 		return nil, err
@@ -114,14 +114,14 @@ func (bq *BlocklistQuery) Only(ctx context.Context) (*Blocklist, error) {
 	case 1:
 		return nodes[0], nil
 	case 0:
-		return nil, &NotFoundError{blocklist.Label}
+		return nil, &NotFoundError{blacklist.Label}
 	default:
-		return nil, &NotSingularError{blocklist.Label}
+		return nil, &NotSingularError{blacklist.Label}
 	}
 }
 
 // OnlyX is like Only, but panics if an error occurs.
-func (bq *BlocklistQuery) OnlyX(ctx context.Context) *Blocklist {
+func (bq *BlacklistQuery) OnlyX(ctx context.Context) *Blacklist {
 	node, err := bq.Only(ctx)
 	if err != nil {
 		panic(err)
@@ -129,10 +129,10 @@ func (bq *BlocklistQuery) OnlyX(ctx context.Context) *Blocklist {
 	return node
 }
 
-// OnlyID is like Only, but returns the only Blocklist ID in the query.
-// Returns a *NotSingularError when more than one Blocklist ID is found.
+// OnlyID is like Only, but returns the only Blacklist ID in the query.
+// Returns a *NotSingularError when more than one Blacklist ID is found.
 // Returns a *NotFoundError when no entities are found.
-func (bq *BlocklistQuery) OnlyID(ctx context.Context) (id int, err error) {
+func (bq *BlacklistQuery) OnlyID(ctx context.Context) (id int, err error) {
 	var ids []int
 	if ids, err = bq.Limit(2).IDs(setContextOp(ctx, bq.ctx, "OnlyID")); err != nil {
 		return
@@ -141,15 +141,15 @@ func (bq *BlocklistQuery) OnlyID(ctx context.Context) (id int, err error) {
 	case 1:
 		id = ids[0]
 	case 0:
-		err = &NotFoundError{blocklist.Label}
+		err = &NotFoundError{blacklist.Label}
 	default:
-		err = &NotSingularError{blocklist.Label}
+		err = &NotSingularError{blacklist.Label}
 	}
 	return
 }
 
 // OnlyIDX is like OnlyID, but panics if an error occurs.
-func (bq *BlocklistQuery) OnlyIDX(ctx context.Context) int {
+func (bq *BlacklistQuery) OnlyIDX(ctx context.Context) int {
 	id, err := bq.OnlyID(ctx)
 	if err != nil {
 		panic(err)
@@ -157,18 +157,18 @@ func (bq *BlocklistQuery) OnlyIDX(ctx context.Context) int {
 	return id
 }
 
-// All executes the query and returns a list of Blocklists.
-func (bq *BlocklistQuery) All(ctx context.Context) ([]*Blocklist, error) {
+// All executes the query and returns a list of Blacklists.
+func (bq *BlacklistQuery) All(ctx context.Context) ([]*Blacklist, error) {
 	ctx = setContextOp(ctx, bq.ctx, "All")
 	if err := bq.prepareQuery(ctx); err != nil {
 		return nil, err
 	}
-	qr := querierAll[[]*Blocklist, *BlocklistQuery]()
-	return withInterceptors[[]*Blocklist](ctx, bq, qr, bq.inters)
+	qr := querierAll[[]*Blacklist, *BlacklistQuery]()
+	return withInterceptors[[]*Blacklist](ctx, bq, qr, bq.inters)
 }
 
 // AllX is like All, but panics if an error occurs.
-func (bq *BlocklistQuery) AllX(ctx context.Context) []*Blocklist {
+func (bq *BlacklistQuery) AllX(ctx context.Context) []*Blacklist {
 	nodes, err := bq.All(ctx)
 	if err != nil {
 		panic(err)
@@ -176,20 +176,20 @@ func (bq *BlocklistQuery) AllX(ctx context.Context) []*Blocklist {
 	return nodes
 }
 
-// IDs executes the query and returns a list of Blocklist IDs.
-func (bq *BlocklistQuery) IDs(ctx context.Context) (ids []int, err error) {
+// IDs executes the query and returns a list of Blacklist IDs.
+func (bq *BlacklistQuery) IDs(ctx context.Context) (ids []int, err error) {
 	if bq.ctx.Unique == nil && bq.path != nil {
 		bq.Unique(true)
 	}
 	ctx = setContextOp(ctx, bq.ctx, "IDs")
-	if err = bq.Select(blocklist.FieldID).Scan(ctx, &ids); err != nil {
+	if err = bq.Select(blacklist.FieldID).Scan(ctx, &ids); err != nil {
 		return nil, err
 	}
 	return ids, nil
 }
 
 // IDsX is like IDs, but panics if an error occurs.
-func (bq *BlocklistQuery) IDsX(ctx context.Context) []int {
+func (bq *BlacklistQuery) IDsX(ctx context.Context) []int {
 	ids, err := bq.IDs(ctx)
 	if err != nil {
 		panic(err)
@@ -198,16 +198,16 @@ func (bq *BlocklistQuery) IDsX(ctx context.Context) []int {
 }
 
 // Count returns the count of the given query.
-func (bq *BlocklistQuery) Count(ctx context.Context) (int, error) {
+func (bq *BlacklistQuery) Count(ctx context.Context) (int, error) {
 	ctx = setContextOp(ctx, bq.ctx, "Count")
 	if err := bq.prepareQuery(ctx); err != nil {
 		return 0, err
 	}
-	return withInterceptors[int](ctx, bq, querierCount[*BlocklistQuery](), bq.inters)
+	return withInterceptors[int](ctx, bq, querierCount[*BlacklistQuery](), bq.inters)
 }
 
 // CountX is like Count, but panics if an error occurs.
-func (bq *BlocklistQuery) CountX(ctx context.Context) int {
+func (bq *BlacklistQuery) CountX(ctx context.Context) int {
 	count, err := bq.Count(ctx)
 	if err != nil {
 		panic(err)
@@ -216,7 +216,7 @@ func (bq *BlocklistQuery) CountX(ctx context.Context) int {
 }
 
 // Exist returns true if the query has elements in the graph.
-func (bq *BlocklistQuery) Exist(ctx context.Context) (bool, error) {
+func (bq *BlacklistQuery) Exist(ctx context.Context) (bool, error) {
 	ctx = setContextOp(ctx, bq.ctx, "Exist")
 	switch _, err := bq.FirstID(ctx); {
 	case IsNotFound(err):
@@ -229,7 +229,7 @@ func (bq *BlocklistQuery) Exist(ctx context.Context) (bool, error) {
 }
 
 // ExistX is like Exist, but panics if an error occurs.
-func (bq *BlocklistQuery) ExistX(ctx context.Context) bool {
+func (bq *BlacklistQuery) ExistX(ctx context.Context) bool {
 	exist, err := bq.Exist(ctx)
 	if err != nil {
 		panic(err)
@@ -237,18 +237,18 @@ func (bq *BlocklistQuery) ExistX(ctx context.Context) bool {
 	return exist
 }
 
-// Clone returns a duplicate of the BlocklistQuery builder, including all associated steps. It can be
+// Clone returns a duplicate of the BlacklistQuery builder, including all associated steps. It can be
 // used to prepare common query builders and use them differently after the clone is made.
-func (bq *BlocklistQuery) Clone() *BlocklistQuery {
+func (bq *BlacklistQuery) Clone() *BlacklistQuery {
 	if bq == nil {
 		return nil
 	}
-	return &BlocklistQuery{
+	return &BlacklistQuery{
 		config:     bq.config,
 		ctx:        bq.ctx.Clone(),
-		order:      append([]blocklist.OrderOption{}, bq.order...),
+		order:      append([]blacklist.OrderOption{}, bq.order...),
 		inters:     append([]Interceptor{}, bq.inters...),
-		predicates: append([]predicate.Blocklist{}, bq.predicates...),
+		predicates: append([]predicate.Blacklist{}, bq.predicates...),
 		// clone intermediate query.
 		sql:  bq.sql.Clone(),
 		path: bq.path,
@@ -261,19 +261,19 @@ func (bq *BlocklistQuery) Clone() *BlocklistQuery {
 // Example:
 //
 //	var v []struct {
-//		Type blocklist.Type `json:"type,omitempty"`
+//		Type blacklist.Type `json:"type,omitempty"`
 //		Count int `json:"count,omitempty"`
 //	}
 //
-//	client.Blocklist.Query().
-//		GroupBy(blocklist.FieldType).
+//	client.Blacklist.Query().
+//		GroupBy(blacklist.FieldType).
 //		Aggregate(ent.Count()).
 //		Scan(ctx, &v)
-func (bq *BlocklistQuery) GroupBy(field string, fields ...string) *BlocklistGroupBy {
+func (bq *BlacklistQuery) GroupBy(field string, fields ...string) *BlacklistGroupBy {
 	bq.ctx.Fields = append([]string{field}, fields...)
-	grbuild := &BlocklistGroupBy{build: bq}
+	grbuild := &BlacklistGroupBy{build: bq}
 	grbuild.flds = &bq.ctx.Fields
-	grbuild.label = blocklist.Label
+	grbuild.label = blacklist.Label
 	grbuild.scan = grbuild.Scan
 	return grbuild
 }
@@ -284,26 +284,26 @@ func (bq *BlocklistQuery) GroupBy(field string, fields ...string) *BlocklistGrou
 // Example:
 //
 //	var v []struct {
-//		Type blocklist.Type `json:"type,omitempty"`
+//		Type blacklist.Type `json:"type,omitempty"`
 //	}
 //
-//	client.Blocklist.Query().
-//		Select(blocklist.FieldType).
+//	client.Blacklist.Query().
+//		Select(blacklist.FieldType).
 //		Scan(ctx, &v)
-func (bq *BlocklistQuery) Select(fields ...string) *BlocklistSelect {
+func (bq *BlacklistQuery) Select(fields ...string) *BlacklistSelect {
 	bq.ctx.Fields = append(bq.ctx.Fields, fields...)
-	sbuild := &BlocklistSelect{BlocklistQuery: bq}
-	sbuild.label = blocklist.Label
+	sbuild := &BlacklistSelect{BlacklistQuery: bq}
+	sbuild.label = blacklist.Label
 	sbuild.flds, sbuild.scan = &bq.ctx.Fields, sbuild.Scan
 	return sbuild
 }
 
-// Aggregate returns a BlocklistSelect configured with the given aggregations.
-func (bq *BlocklistQuery) Aggregate(fns ...AggregateFunc) *BlocklistSelect {
+// Aggregate returns a BlacklistSelect configured with the given aggregations.
+func (bq *BlacklistQuery) Aggregate(fns ...AggregateFunc) *BlacklistSelect {
 	return bq.Select().Aggregate(fns...)
 }
 
-func (bq *BlocklistQuery) prepareQuery(ctx context.Context) error {
+func (bq *BlacklistQuery) prepareQuery(ctx context.Context) error {
 	for _, inter := range bq.inters {
 		if inter == nil {
 			return fmt.Errorf("ent: uninitialized interceptor (forgotten import ent/runtime?)")
@@ -315,7 +315,7 @@ func (bq *BlocklistQuery) prepareQuery(ctx context.Context) error {
 		}
 	}
 	for _, f := range bq.ctx.Fields {
-		if !blocklist.ValidColumn(f) {
+		if !blacklist.ValidColumn(f) {
 			return &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
 		}
 	}
@@ -329,16 +329,16 @@ func (bq *BlocklistQuery) prepareQuery(ctx context.Context) error {
 	return nil
 }
 
-func (bq *BlocklistQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Blocklist, error) {
+func (bq *BlacklistQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Blacklist, error) {
 	var (
-		nodes = []*Blocklist{}
+		nodes = []*Blacklist{}
 		_spec = bq.querySpec()
 	)
 	_spec.ScanValues = func(columns []string) ([]any, error) {
-		return (*Blocklist).scanValues(nil, columns)
+		return (*Blacklist).scanValues(nil, columns)
 	}
 	_spec.Assign = func(columns []string, values []any) error {
-		node := &Blocklist{config: bq.config}
+		node := &Blacklist{config: bq.config}
 		nodes = append(nodes, node)
 		return node.assignValues(columns, values)
 	}
@@ -354,7 +354,7 @@ func (bq *BlocklistQuery) sqlAll(ctx context.Context, hooks ...queryHook) ([]*Bl
 	return nodes, nil
 }
 
-func (bq *BlocklistQuery) sqlCount(ctx context.Context) (int, error) {
+func (bq *BlacklistQuery) sqlCount(ctx context.Context) (int, error) {
 	_spec := bq.querySpec()
 	_spec.Node.Columns = bq.ctx.Fields
 	if len(bq.ctx.Fields) > 0 {
@@ -363,8 +363,8 @@ func (bq *BlocklistQuery) sqlCount(ctx context.Context) (int, error) {
 	return sqlgraph.CountNodes(ctx, bq.driver, _spec)
 }
 
-func (bq *BlocklistQuery) querySpec() *sqlgraph.QuerySpec {
-	_spec := sqlgraph.NewQuerySpec(blocklist.Table, blocklist.Columns, sqlgraph.NewFieldSpec(blocklist.FieldID, field.TypeInt))
+func (bq *BlacklistQuery) querySpec() *sqlgraph.QuerySpec {
+	_spec := sqlgraph.NewQuerySpec(blacklist.Table, blacklist.Columns, sqlgraph.NewFieldSpec(blacklist.FieldID, field.TypeInt))
 	_spec.From = bq.sql
 	if unique := bq.ctx.Unique; unique != nil {
 		_spec.Unique = *unique
@@ -373,9 +373,9 @@ func (bq *BlocklistQuery) querySpec() *sqlgraph.QuerySpec {
 	}
 	if fields := bq.ctx.Fields; len(fields) > 0 {
 		_spec.Node.Columns = make([]string, 0, len(fields))
-		_spec.Node.Columns = append(_spec.Node.Columns, blocklist.FieldID)
+		_spec.Node.Columns = append(_spec.Node.Columns, blacklist.FieldID)
 		for i := range fields {
-			if fields[i] != blocklist.FieldID {
+			if fields[i] != blacklist.FieldID {
 				_spec.Node.Columns = append(_spec.Node.Columns, fields[i])
 			}
 		}
@@ -403,12 +403,12 @@ func (bq *BlocklistQuery) querySpec() *sqlgraph.QuerySpec {
 	return _spec
 }
 
-func (bq *BlocklistQuery) sqlQuery(ctx context.Context) *sql.Selector {
+func (bq *BlacklistQuery) sqlQuery(ctx context.Context) *sql.Selector {
 	builder := sql.Dialect(bq.driver.Dialect())
-	t1 := builder.Table(blocklist.Table)
+	t1 := builder.Table(blacklist.Table)
 	columns := bq.ctx.Fields
 	if len(columns) == 0 {
-		columns = blocklist.Columns
+		columns = blacklist.Columns
 	}
 	selector := builder.Select(t1.Columns(columns...)...).From(t1)
 	if bq.sql != nil {
@@ -435,28 +435,28 @@ func (bq *BlocklistQuery) sqlQuery(ctx context.Context) *sql.Selector {
 	return selector
 }
 
-// BlocklistGroupBy is the group-by builder for Blocklist entities.
-type BlocklistGroupBy struct {
+// BlacklistGroupBy is the group-by builder for Blacklist entities.
+type BlacklistGroupBy struct {
 	selector
-	build *BlocklistQuery
+	build *BlacklistQuery
 }
 
 // Aggregate adds the given aggregation functions to the group-by query.
-func (bgb *BlocklistGroupBy) Aggregate(fns ...AggregateFunc) *BlocklistGroupBy {
+func (bgb *BlacklistGroupBy) Aggregate(fns ...AggregateFunc) *BlacklistGroupBy {
 	bgb.fns = append(bgb.fns, fns...)
 	return bgb
 }
 
 // Scan applies the selector query and scans the result into the given value.
-func (bgb *BlocklistGroupBy) Scan(ctx context.Context, v any) error {
+func (bgb *BlacklistGroupBy) Scan(ctx context.Context, v any) error {
 	ctx = setContextOp(ctx, bgb.build.ctx, "GroupBy")
 	if err := bgb.build.prepareQuery(ctx); err != nil {
 		return err
 	}
-	return scanWithInterceptors[*BlocklistQuery, *BlocklistGroupBy](ctx, bgb.build, bgb, bgb.build.inters, v)
+	return scanWithInterceptors[*BlacklistQuery, *BlacklistGroupBy](ctx, bgb.build, bgb, bgb.build.inters, v)
 }
 
-func (bgb *BlocklistGroupBy) sqlScan(ctx context.Context, root *BlocklistQuery, v any) error {
+func (bgb *BlacklistGroupBy) sqlScan(ctx context.Context, root *BlacklistQuery, v any) error {
 	selector := root.sqlQuery(ctx).Select()
 	aggregation := make([]string, 0, len(bgb.fns))
 	for _, fn := range bgb.fns {
@@ -483,28 +483,28 @@ func (bgb *BlocklistGroupBy) sqlScan(ctx context.Context, root *BlocklistQuery,
 	return sql.ScanSlice(rows, v)
 }
 
-// BlocklistSelect is the builder for selecting fields of Blocklist entities.
-type BlocklistSelect struct {
-	*BlocklistQuery
+// BlacklistSelect is the builder for selecting fields of Blacklist entities.
+type BlacklistSelect struct {
+	*BlacklistQuery
 	selector
 }
 
 // Aggregate adds the given aggregation functions to the selector query.
-func (bs *BlocklistSelect) Aggregate(fns ...AggregateFunc) *BlocklistSelect {
+func (bs *BlacklistSelect) Aggregate(fns ...AggregateFunc) *BlacklistSelect {
 	bs.fns = append(bs.fns, fns...)
 	return bs
 }
 
 // Scan applies the selector query and scans the result into the given value.
-func (bs *BlocklistSelect) Scan(ctx context.Context, v any) error {
+func (bs *BlacklistSelect) Scan(ctx context.Context, v any) error {
 	ctx = setContextOp(ctx, bs.ctx, "Select")
 	if err := bs.prepareQuery(ctx); err != nil {
 		return err
 	}
-	return scanWithInterceptors[*BlocklistQuery, *BlocklistSelect](ctx, bs.BlocklistQuery, bs, bs.inters, v)
+	return scanWithInterceptors[*BlacklistQuery, *BlacklistSelect](ctx, bs.BlacklistQuery, bs, bs.inters, v)
 }
 
-func (bs *BlocklistSelect) sqlScan(ctx context.Context, root *BlocklistQuery, v any) error {
+func (bs *BlacklistSelect) sqlScan(ctx context.Context, root *BlacklistQuery, v any) error {
 	selector := root.sqlQuery(ctx)
 	aggregation := make([]string, 0, len(bs.fns))
 	for _, fn := range bs.fns {

ent/blacklist_update.go

@@ -0,0 +1,322 @@
+// Code generated by ent, DO NOT EDIT.
+
+package ent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"polaris/ent/blacklist"
+	"polaris/ent/predicate"
+	"polaris/ent/schema"
+
+	"entgo.io/ent/dialect/sql"
+	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/schema/field"
+)
+
+// BlacklistUpdate is the builder for updating Blacklist entities.
+type BlacklistUpdate struct {
+	config
+	hooks    []Hook
+	mutation *BlacklistMutation
+}
+
+// Where appends a list predicates to the BlacklistUpdate builder.
+func (bu *BlacklistUpdate) Where(ps ...predicate.Blacklist) *BlacklistUpdate {
+	bu.mutation.Where(ps...)
+	return bu
+}
+
+// SetType sets the "type" field.
+func (bu *BlacklistUpdate) SetType(b blacklist.Type) *BlacklistUpdate {
+	bu.mutation.SetType(b)
+	return bu
+}
+
+// SetNillableType sets the "type" field if the given value is not nil.
+func (bu *BlacklistUpdate) SetNillableType(b *blacklist.Type) *BlacklistUpdate {
+	if b != nil {
+		bu.SetType(*b)
+	}
+	return bu
+}
+
+// SetValue sets the "value" field.
+func (bu *BlacklistUpdate) SetValue(sv schema.BlacklistValue) *BlacklistUpdate {
+	bu.mutation.SetValue(sv)
+	return bu
+}
+
+// SetNillableValue sets the "value" field if the given value is not nil.
+func (bu *BlacklistUpdate) SetNillableValue(sv *schema.BlacklistValue) *BlacklistUpdate {
+	if sv != nil {
+		bu.SetValue(*sv)
+	}
+	return bu
+}
+
+// SetNotes sets the "notes" field.
+func (bu *BlacklistUpdate) SetNotes(s string) *BlacklistUpdate {
+	bu.mutation.SetNotes(s)
+	return bu
+}
+
+// SetNillableNotes sets the "notes" field if the given value is not nil.
+func (bu *BlacklistUpdate) SetNillableNotes(s *string) *BlacklistUpdate {
+	if s != nil {
+		bu.SetNotes(*s)
+	}
+	return bu
+}
+
+// ClearNotes clears the value of the "notes" field.
+func (bu *BlacklistUpdate) ClearNotes() *BlacklistUpdate {
+	bu.mutation.ClearNotes()
+	return bu
+}
+
+// Mutation returns the BlacklistMutation object of the builder.
+func (bu *BlacklistUpdate) Mutation() *BlacklistMutation {
+	return bu.mutation
+}
+
+// Save executes the query and returns the number of nodes affected by the update operation.
+func (bu *BlacklistUpdate) Save(ctx context.Context) (int, error) {
+	return withHooks(ctx, bu.sqlSave, bu.mutation, bu.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (bu *BlacklistUpdate) SaveX(ctx context.Context) int {
+	affected, err := bu.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return affected
+}
+
+// Exec executes the query.
+func (bu *BlacklistUpdate) Exec(ctx context.Context) error {
+	_, err := bu.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (bu *BlacklistUpdate) ExecX(ctx context.Context) {
+	if err := bu.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (bu *BlacklistUpdate) check() error {
+	if v, ok := bu.mutation.GetType(); ok {
+		if err := blacklist.TypeValidator(v); err != nil {
+			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blacklist.type": %w`, err)}
+		}
+	}
+	return nil
+}
+
+func (bu *BlacklistUpdate) sqlSave(ctx context.Context) (n int, err error) {
+	if err := bu.check(); err != nil {
+		return n, err
+	}
+	_spec := sqlgraph.NewUpdateSpec(blacklist.Table, blacklist.Columns, sqlgraph.NewFieldSpec(blacklist.FieldID, field.TypeInt))
+	if ps := bu.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := bu.mutation.GetType(); ok {
+		_spec.SetField(blacklist.FieldType, field.TypeEnum, value)
+	}
+	if value, ok := bu.mutation.Value(); ok {
+		_spec.SetField(blacklist.FieldValue, field.TypeJSON, value)
+	}
+	if value, ok := bu.mutation.Notes(); ok {
+		_spec.SetField(blacklist.FieldNotes, field.TypeString, value)
+	}
+	if bu.mutation.NotesCleared() {
+		_spec.ClearField(blacklist.FieldNotes, field.TypeString)
+	}
+	if n, err = sqlgraph.UpdateNodes(ctx, bu.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{blacklist.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return 0, err
+	}
+	bu.mutation.done = true
+	return n, nil
+}
+
+// BlacklistUpdateOne is the builder for updating a single Blacklist entity.
+type BlacklistUpdateOne struct {
+	config
+	fields   []string
+	hooks    []Hook
+	mutation *BlacklistMutation
+}
+
+// SetType sets the "type" field.
+func (buo *BlacklistUpdateOne) SetType(b blacklist.Type) *BlacklistUpdateOne {
+	buo.mutation.SetType(b)
+	return buo
+}
+
+// SetNillableType sets the "type" field if the given value is not nil.
+func (buo *BlacklistUpdateOne) SetNillableType(b *blacklist.Type) *BlacklistUpdateOne {
+	if b != nil {
+		buo.SetType(*b)
+	}
+	return buo
+}
+
+// SetValue sets the "value" field.
+func (buo *BlacklistUpdateOne) SetValue(sv schema.BlacklistValue) *BlacklistUpdateOne {
+	buo.mutation.SetValue(sv)
+	return buo
+}
+
+// SetNillableValue sets the "value" field if the given value is not nil.
+func (buo *BlacklistUpdateOne) SetNillableValue(sv *schema.BlacklistValue) *BlacklistUpdateOne {
+	if sv != nil {
+		buo.SetValue(*sv)
+	}
+	return buo
+}
+
+// SetNotes sets the "notes" field.
+func (buo *BlacklistUpdateOne) SetNotes(s string) *BlacklistUpdateOne {
+	buo.mutation.SetNotes(s)
+	return buo
+}
+
+// SetNillableNotes sets the "notes" field if the given value is not nil.
+func (buo *BlacklistUpdateOne) SetNillableNotes(s *string) *BlacklistUpdateOne {
+	if s != nil {
+		buo.SetNotes(*s)
+	}
+	return buo
+}
+
+// ClearNotes clears the value of the "notes" field.
+func (buo *BlacklistUpdateOne) ClearNotes() *BlacklistUpdateOne {
+	buo.mutation.ClearNotes()
+	return buo
+}
+
+// Mutation returns the BlacklistMutation object of the builder.
+func (buo *BlacklistUpdateOne) Mutation() *BlacklistMutation {
+	return buo.mutation
+}
+
+// Where appends a list predicates to the BlacklistUpdate builder.
+func (buo *BlacklistUpdateOne) Where(ps ...predicate.Blacklist) *BlacklistUpdateOne {
+	buo.mutation.Where(ps...)
+	return buo
+}
+
+// Select allows selecting one or more fields (columns) of the returned entity.
+// The default is selecting all fields defined in the entity schema.
+func (buo *BlacklistUpdateOne) Select(field string, fields ...string) *BlacklistUpdateOne {
+	buo.fields = append([]string{field}, fields...)
+	return buo
+}
+
+// Save executes the query and returns the updated Blacklist entity.
+func (buo *BlacklistUpdateOne) Save(ctx context.Context) (*Blacklist, error) {
+	return withHooks(ctx, buo.sqlSave, buo.mutation, buo.hooks)
+}
+
+// SaveX is like Save, but panics if an error occurs.
+func (buo *BlacklistUpdateOne) SaveX(ctx context.Context) *Blacklist {
+	node, err := buo.Save(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return node
+}
+
+// Exec executes the query on the entity.
+func (buo *BlacklistUpdateOne) Exec(ctx context.Context) error {
+	_, err := buo.Save(ctx)
+	return err
+}
+
+// ExecX is like Exec, but panics if an error occurs.
+func (buo *BlacklistUpdateOne) ExecX(ctx context.Context) {
+	if err := buo.Exec(ctx); err != nil {
+		panic(err)
+	}
+}
+
+// check runs all checks and user-defined validators on the builder.
+func (buo *BlacklistUpdateOne) check() error {
+	if v, ok := buo.mutation.GetType(); ok {
+		if err := blacklist.TypeValidator(v); err != nil {
+			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blacklist.type": %w`, err)}
+		}
+	}
+	return nil
+}
+
+func (buo *BlacklistUpdateOne) sqlSave(ctx context.Context) (_node *Blacklist, err error) {
+	if err := buo.check(); err != nil {
+		return _node, err
+	}
+	_spec := sqlgraph.NewUpdateSpec(blacklist.Table, blacklist.Columns, sqlgraph.NewFieldSpec(blacklist.FieldID, field.TypeInt))
+	id, ok := buo.mutation.ID()
+	if !ok {
+		return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "Blacklist.id" for update`)}
+	}
+	_spec.Node.ID.Value = id
+	if fields := buo.fields; len(fields) > 0 {
+		_spec.Node.Columns = make([]string, 0, len(fields))
+		_spec.Node.Columns = append(_spec.Node.Columns, blacklist.FieldID)
+		for _, f := range fields {
+			if !blacklist.ValidColumn(f) {
+				return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
+			}
+			if f != blacklist.FieldID {
+				_spec.Node.Columns = append(_spec.Node.Columns, f)
+			}
+		}
+	}
+	if ps := buo.mutation.predicates; len(ps) > 0 {
+		_spec.Predicate = func(selector *sql.Selector) {
+			for i := range ps {
+				ps[i](selector)
+			}
+		}
+	}
+	if value, ok := buo.mutation.GetType(); ok {
+		_spec.SetField(blacklist.FieldType, field.TypeEnum, value)
+	}
+	if value, ok := buo.mutation.Value(); ok {
+		_spec.SetField(blacklist.FieldValue, field.TypeJSON, value)
+	}
+	if value, ok := buo.mutation.Notes(); ok {
+		_spec.SetField(blacklist.FieldNotes, field.TypeString, value)
+	}
+	if buo.mutation.NotesCleared() {
+		_spec.ClearField(blacklist.FieldNotes, field.TypeString)
+	}
+	_node = &Blacklist{config: buo.config}
+	_spec.Assign = _node.assignValues
+	_spec.ScanValues = _node.scanValues
+	if err = sqlgraph.UpdateNode(ctx, buo.driver, _spec); err != nil {
+		if _, ok := err.(*sqlgraph.NotFoundError); ok {
+			err = &NotFoundError{blacklist.Label}
+		} else if sqlgraph.IsConstraintError(err) {
+			err = &ConstraintError{msg: err.Error(), wrap: err}
+		}
+		return nil, err
+	}
+	buo.mutation.done = true
+	return _node, nil
+}

ent/blocklist/where.go

@@ -1,159 +0,0 @@
-// Code generated by ent, DO NOT EDIT.
-
-package blocklist
-
-import (
-	"polaris/ent/predicate"
-
-	"entgo.io/ent/dialect/sql"
-)
-
-// ID filters vertices based on their ID field.
-func ID(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEQ(FieldID, id))
-}
-
-// IDEQ applies the EQ predicate on the ID field.
-func IDEQ(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEQ(FieldID, id))
-}
-
-// IDNEQ applies the NEQ predicate on the ID field.
-func IDNEQ(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNEQ(FieldID, id))
-}
-
-// IDIn applies the In predicate on the ID field.
-func IDIn(ids ...int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldIn(FieldID, ids...))
-}
-
-// IDNotIn applies the NotIn predicate on the ID field.
-func IDNotIn(ids ...int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNotIn(FieldID, ids...))
-}
-
-// IDGT applies the GT predicate on the ID field.
-func IDGT(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldGT(FieldID, id))
-}
-
-// IDGTE applies the GTE predicate on the ID field.
-func IDGTE(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldGTE(FieldID, id))
-}
-
-// IDLT applies the LT predicate on the ID field.
-func IDLT(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldLT(FieldID, id))
-}
-
-// IDLTE applies the LTE predicate on the ID field.
-func IDLTE(id int) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldLTE(FieldID, id))
-}
-
-// Value applies equality check predicate on the "value" field. It's identical to ValueEQ.
-func Value(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEQ(FieldValue, v))
-}
-
-// TypeEQ applies the EQ predicate on the "type" field.
-func TypeEQ(v Type) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEQ(FieldType, v))
-}
-
-// TypeNEQ applies the NEQ predicate on the "type" field.
-func TypeNEQ(v Type) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNEQ(FieldType, v))
-}
-
-// TypeIn applies the In predicate on the "type" field.
-func TypeIn(vs ...Type) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldIn(FieldType, vs...))
-}
-
-// TypeNotIn applies the NotIn predicate on the "type" field.
-func TypeNotIn(vs ...Type) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNotIn(FieldType, vs...))
-}
-
-// ValueEQ applies the EQ predicate on the "value" field.
-func ValueEQ(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEQ(FieldValue, v))
-}
-
-// ValueNEQ applies the NEQ predicate on the "value" field.
-func ValueNEQ(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNEQ(FieldValue, v))
-}
-
-// ValueIn applies the In predicate on the "value" field.
-func ValueIn(vs ...string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldIn(FieldValue, vs...))
-}
-
-// ValueNotIn applies the NotIn predicate on the "value" field.
-func ValueNotIn(vs ...string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldNotIn(FieldValue, vs...))
-}
-
-// ValueGT applies the GT predicate on the "value" field.
-func ValueGT(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldGT(FieldValue, v))
-}
-
-// ValueGTE applies the GTE predicate on the "value" field.
-func ValueGTE(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldGTE(FieldValue, v))
-}
-
-// ValueLT applies the LT predicate on the "value" field.
-func ValueLT(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldLT(FieldValue, v))
-}
-
-// ValueLTE applies the LTE predicate on the "value" field.
-func ValueLTE(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldLTE(FieldValue, v))
-}
-
-// ValueContains applies the Contains predicate on the "value" field.
-func ValueContains(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldContains(FieldValue, v))
-}
-
-// ValueHasPrefix applies the HasPrefix predicate on the "value" field.
-func ValueHasPrefix(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldHasPrefix(FieldValue, v))
-}
-
-// ValueHasSuffix applies the HasSuffix predicate on the "value" field.
-func ValueHasSuffix(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldHasSuffix(FieldValue, v))
-}
-
-// ValueEqualFold applies the EqualFold predicate on the "value" field.
-func ValueEqualFold(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldEqualFold(FieldValue, v))
-}
-
-// ValueContainsFold applies the ContainsFold predicate on the "value" field.
-func ValueContainsFold(v string) predicate.Blocklist {
-	return predicate.Blocklist(sql.FieldContainsFold(FieldValue, v))
-}
-
-// And groups predicates with the AND operator between them.
-func And(predicates ...predicate.Blocklist) predicate.Blocklist {
-	return predicate.Blocklist(sql.AndPredicates(predicates...))
-}
-
-// Or groups predicates with the OR operator between them.
-func Or(predicates ...predicate.Blocklist) predicate.Blocklist {
-	return predicate.Blocklist(sql.OrPredicates(predicates...))
-}
-
-// Not applies the not operator on the given predicate.
-func Not(p predicate.Blocklist) predicate.Blocklist {
-	return predicate.Blocklist(sql.NotPredicates(p))
-}

ent/blocklist_update.go

@@ -1,269 +0,0 @@
-// Code generated by ent, DO NOT EDIT.
-
-package ent
-
-import (
-	"context"
-	"errors"
-	"fmt"
-	"polaris/ent/blocklist"
-	"polaris/ent/predicate"
-
-	"entgo.io/ent/dialect/sql"
-	"entgo.io/ent/dialect/sql/sqlgraph"
-	"entgo.io/ent/schema/field"
-)
-
-// BlocklistUpdate is the builder for updating Blocklist entities.
-type BlocklistUpdate struct {
-	config
-	hooks    []Hook
-	mutation *BlocklistMutation
-}
-
-// Where appends a list predicates to the BlocklistUpdate builder.
-func (bu *BlocklistUpdate) Where(ps ...predicate.Blocklist) *BlocklistUpdate {
-	bu.mutation.Where(ps...)
-	return bu
-}
-
-// SetType sets the "type" field.
-func (bu *BlocklistUpdate) SetType(b blocklist.Type) *BlocklistUpdate {
-	bu.mutation.SetType(b)
-	return bu
-}
-
-// SetNillableType sets the "type" field if the given value is not nil.
-func (bu *BlocklistUpdate) SetNillableType(b *blocklist.Type) *BlocklistUpdate {
-	if b != nil {
-		bu.SetType(*b)
-	}
-	return bu
-}
-
-// SetValue sets the "value" field.
-func (bu *BlocklistUpdate) SetValue(s string) *BlocklistUpdate {
-	bu.mutation.SetValue(s)
-	return bu
-}
-
-// SetNillableValue sets the "value" field if the given value is not nil.
-func (bu *BlocklistUpdate) SetNillableValue(s *string) *BlocklistUpdate {
-	if s != nil {
-		bu.SetValue(*s)
-	}
-	return bu
-}
-
-// Mutation returns the BlocklistMutation object of the builder.
-func (bu *BlocklistUpdate) Mutation() *BlocklistMutation {
-	return bu.mutation
-}
-
-// Save executes the query and returns the number of nodes affected by the update operation.
-func (bu *BlocklistUpdate) Save(ctx context.Context) (int, error) {
-	return withHooks(ctx, bu.sqlSave, bu.mutation, bu.hooks)
-}
-
-// SaveX is like Save, but panics if an error occurs.
-func (bu *BlocklistUpdate) SaveX(ctx context.Context) int {
-	affected, err := bu.Save(ctx)
-	if err != nil {
-		panic(err)
-	}
-	return affected
-}
-
-// Exec executes the query.
-func (bu *BlocklistUpdate) Exec(ctx context.Context) error {
-	_, err := bu.Save(ctx)
-	return err
-}
-
-// ExecX is like Exec, but panics if an error occurs.
-func (bu *BlocklistUpdate) ExecX(ctx context.Context) {
-	if err := bu.Exec(ctx); err != nil {
-		panic(err)
-	}
-}
-
-// check runs all checks and user-defined validators on the builder.
-func (bu *BlocklistUpdate) check() error {
-	if v, ok := bu.mutation.GetType(); ok {
-		if err := blocklist.TypeValidator(v); err != nil {
-			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blocklist.type": %w`, err)}
-		}
-	}
-	return nil
-}
-
-func (bu *BlocklistUpdate) sqlSave(ctx context.Context) (n int, err error) {
-	if err := bu.check(); err != nil {
-		return n, err
-	}
-	_spec := sqlgraph.NewUpdateSpec(blocklist.Table, blocklist.Columns, sqlgraph.NewFieldSpec(blocklist.FieldID, field.TypeInt))
-	if ps := bu.mutation.predicates; len(ps) > 0 {
-		_spec.Predicate = func(selector *sql.Selector) {
-			for i := range ps {
-				ps[i](selector)
-			}
-		}
-	}
-	if value, ok := bu.mutation.GetType(); ok {
-		_spec.SetField(blocklist.FieldType, field.TypeEnum, value)
-	}
-	if value, ok := bu.mutation.Value(); ok {
-		_spec.SetField(blocklist.FieldValue, field.TypeString, value)
-	}
-	if n, err = sqlgraph.UpdateNodes(ctx, bu.driver, _spec); err != nil {
-		if _, ok := err.(*sqlgraph.NotFoundError); ok {
-			err = &NotFoundError{blocklist.Label}
-		} else if sqlgraph.IsConstraintError(err) {
-			err = &ConstraintError{msg: err.Error(), wrap: err}
-		}
-		return 0, err
-	}
-	bu.mutation.done = true
-	return n, nil
-}
-
-// BlocklistUpdateOne is the builder for updating a single Blocklist entity.
-type BlocklistUpdateOne struct {
-	config
-	fields   []string
-	hooks    []Hook
-	mutation *BlocklistMutation
-}
-
-// SetType sets the "type" field.
-func (buo *BlocklistUpdateOne) SetType(b blocklist.Type) *BlocklistUpdateOne {
-	buo.mutation.SetType(b)
-	return buo
-}
-
-// SetNillableType sets the "type" field if the given value is not nil.
-func (buo *BlocklistUpdateOne) SetNillableType(b *blocklist.Type) *BlocklistUpdateOne {
-	if b != nil {
-		buo.SetType(*b)
-	}
-	return buo
-}
-
-// SetValue sets the "value" field.
-func (buo *BlocklistUpdateOne) SetValue(s string) *BlocklistUpdateOne {
-	buo.mutation.SetValue(s)
-	return buo
-}
-
-// SetNillableValue sets the "value" field if the given value is not nil.
-func (buo *BlocklistUpdateOne) SetNillableValue(s *string) *BlocklistUpdateOne {
-	if s != nil {
-		buo.SetValue(*s)
-	}
-	return buo
-}
-
-// Mutation returns the BlocklistMutation object of the builder.
-func (buo *BlocklistUpdateOne) Mutation() *BlocklistMutation {
-	return buo.mutation
-}
-
-// Where appends a list predicates to the BlocklistUpdate builder.
-func (buo *BlocklistUpdateOne) Where(ps ...predicate.Blocklist) *BlocklistUpdateOne {
-	buo.mutation.Where(ps...)
-	return buo
-}
-
-// Select allows selecting one or more fields (columns) of the returned entity.
-// The default is selecting all fields defined in the entity schema.
-func (buo *BlocklistUpdateOne) Select(field string, fields ...string) *BlocklistUpdateOne {
-	buo.fields = append([]string{field}, fields...)
-	return buo
-}
-
-// Save executes the query and returns the updated Blocklist entity.
-func (buo *BlocklistUpdateOne) Save(ctx context.Context) (*Blocklist, error) {
-	return withHooks(ctx, buo.sqlSave, buo.mutation, buo.hooks)
-}
-
-// SaveX is like Save, but panics if an error occurs.
-func (buo *BlocklistUpdateOne) SaveX(ctx context.Context) *Blocklist {
-	node, err := buo.Save(ctx)
-	if err != nil {
-		panic(err)
-	}
-	return node
-}
-
-// Exec executes the query on the entity.
-func (buo *BlocklistUpdateOne) Exec(ctx context.Context) error {
-	_, err := buo.Save(ctx)
-	return err
-}
-
-// ExecX is like Exec, but panics if an error occurs.
-func (buo *BlocklistUpdateOne) ExecX(ctx context.Context) {
-	if err := buo.Exec(ctx); err != nil {
-		panic(err)
-	}
-}
-
-// check runs all checks and user-defined validators on the builder.
-func (buo *BlocklistUpdateOne) check() error {
-	if v, ok := buo.mutation.GetType(); ok {
-		if err := blocklist.TypeValidator(v); err != nil {
-			return &ValidationError{Name: "type", err: fmt.Errorf(`ent: validator failed for field "Blocklist.type": %w`, err)}
-		}
-	}
-	return nil
-}
-
-func (buo *BlocklistUpdateOne) sqlSave(ctx context.Context) (_node *Blocklist, err error) {
-	if err := buo.check(); err != nil {
-		return _node, err
-	}
-	_spec := sqlgraph.NewUpdateSpec(blocklist.Table, blocklist.Columns, sqlgraph.NewFieldSpec(blocklist.FieldID, field.TypeInt))
-	id, ok := buo.mutation.ID()
-	if !ok {
-		return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "Blocklist.id" for update`)}
-	}
-	_spec.Node.ID.Value = id
-	if fields := buo.fields; len(fields) > 0 {
-		_spec.Node.Columns = make([]string, 0, len(fields))
-		_spec.Node.Columns = append(_spec.Node.Columns, blocklist.FieldID)
-		for _, f := range fields {
-			if !blocklist.ValidColumn(f) {
-				return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
-			}
-			if f != blocklist.FieldID {
-				_spec.Node.Columns = append(_spec.Node.Columns, f)
-			}
-		}
-	}
-	if ps := buo.mutation.predicates; len(ps) > 0 {
-		_spec.Predicate = func(selector *sql.Selector) {
-			for i := range ps {
-				ps[i](selector)
-			}
-		}
-	}
-	if value, ok := buo.mutation.GetType(); ok {
-		_spec.SetField(blocklist.FieldType, field.TypeEnum, value)
-	}
-	if value, ok := buo.mutation.Value(); ok {
-		_spec.SetField(blocklist.FieldValue, field.TypeString, value)
-	}
-	_node = &Blocklist{config: buo.config}
-	_spec.Assign = _node.assignValues
-	_spec.ScanValues = _node.scanValues
-	if err = sqlgraph.UpdateNode(ctx, buo.driver, _spec); err != nil {
-		if _, ok := err.(*sqlgraph.NotFoundError); ok {
-			err = &NotFoundError{blocklist.Label}
-		} else if sqlgraph.IsConstraintError(err) {
-			err = &ConstraintError{msg: err.Error(), wrap: err}
-		}
-		return nil, err
-	}
-	buo.mutation.done = true
-	return _node, nil
-}

ent/client.go

@@ -11,7 +11,7 @@ import (
 
 	"polaris/ent/migrate"
 
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
 	"polaris/ent/downloadclients"
 	"polaris/ent/episode"
 	"polaris/ent/history"
@@ -33,8 +33,8 @@ type Client struct {
 	config
 	// Schema is the client for creating, migrating and dropping schema.
 	Schema *migrate.Schema
-	// Blocklist is the client for interacting with the Blocklist builders.
-	Blocklist *BlocklistClient
+	// Blacklist is the client for interacting with the Blacklist builders.
+	Blacklist *BlacklistClient
 	// DownloadClients is the client for interacting with the DownloadClients builders.
 	DownloadClients *DownloadClientsClient
 	// Episode is the client for interacting with the Episode builders.
@@ -64,7 +64,7 @@ func NewClient(opts ...Option) *Client {
 
 func (c *Client) init() {
 	c.Schema = migrate.NewSchema(c.driver)
-	c.Blocklist = NewBlocklistClient(c.config)
+	c.Blacklist = NewBlacklistClient(c.config)
 	c.DownloadClients = NewDownloadClientsClient(c.config)
 	c.Episode = NewEpisodeClient(c.config)
 	c.History = NewHistoryClient(c.config)
@@ -166,7 +166,7 @@ func (c *Client) Tx(ctx context.Context) (*Tx, error) {
 	return &Tx{
 		ctx:                ctx,
 		config:             cfg,
-		Blocklist:          NewBlocklistClient(cfg),
+		Blacklist:          NewBlacklistClient(cfg),
 		DownloadClients:    NewDownloadClientsClient(cfg),
 		Episode:            NewEpisodeClient(cfg),
 		History:            NewHistoryClient(cfg),
@@ -195,7 +195,7 @@ func (c *Client) BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error)
 	return &Tx{
 		ctx:                ctx,
 		config:             cfg,
-		Blocklist:          NewBlocklistClient(cfg),
+		Blacklist:          NewBlacklistClient(cfg),
 		DownloadClients:    NewDownloadClientsClient(cfg),
 		Episode:            NewEpisodeClient(cfg),
 		History:            NewHistoryClient(cfg),
@@ -211,7 +211,7 @@ func (c *Client) BeginTx(ctx context.Context, opts *sql.TxOptions) (*Tx, error)
 // Debug returns a new debug-client. It's used to get verbose logging on specific operations.
 //
 //	client.Debug().
-//		Blocklist.
+//		Blacklist.
 //		Query().
 //		Count(ctx)
 func (c *Client) Debug() *Client {
@@ -234,7 +234,7 @@ func (c *Client) Close() error {
 // In order to add hooks to a specific client, call: `client.Node.Use(...)`.
 func (c *Client) Use(hooks ...Hook) {
 	for _, n := range []interface{ Use(...Hook) }{
-		c.Blocklist, c.DownloadClients, c.Episode, c.History, c.ImportList, c.Indexers,
+		c.Blacklist, c.DownloadClients, c.Episode, c.History, c.ImportList, c.Indexers,
 		c.Media, c.NotificationClient, c.Settings, c.Storage,
 	} {
 		n.Use(hooks...)
@@ -245,7 +245,7 @@ func (c *Client) Use(hooks ...Hook) {
 // In order to add interceptors to a specific client, call: `client.Node.Intercept(...)`.
 func (c *Client) Intercept(interceptors ...Interceptor) {
 	for _, n := range []interface{ Intercept(...Interceptor) }{
-		c.Blocklist, c.DownloadClients, c.Episode, c.History, c.ImportList, c.Indexers,
+		c.Blacklist, c.DownloadClients, c.Episode, c.History, c.ImportList, c.Indexers,
 		c.Media, c.NotificationClient, c.Settings, c.Storage,
 	} {
 		n.Intercept(interceptors...)
@@ -255,8 +255,8 @@ func (c *Client) Intercept(interceptors ...Interceptor) {
 // Mutate implements the ent.Mutator interface.
 func (c *Client) Mutate(ctx context.Context, m Mutation) (Value, error) {
 	switch m := m.(type) {
-	case *BlocklistMutation:
-		return c.Blocklist.mutate(ctx, m)
+	case *BlacklistMutation:
+		return c.Blacklist.mutate(ctx, m)
 	case *DownloadClientsMutation:
 		return c.DownloadClients.mutate(ctx, m)
 	case *EpisodeMutation:
@@ -280,107 +280,107 @@ func (c *Client) Mutate(ctx context.Context, m Mutation) (Value, error) {
 	}
 }
 
-// BlocklistClient is a client for the Blocklist schema.
-type BlocklistClient struct {
+// BlacklistClient is a client for the Blacklist schema.
+type BlacklistClient struct {
 	config
 }
 
-// NewBlocklistClient returns a client for the Blocklist from the given config.
-func NewBlocklistClient(c config) *BlocklistClient {
-	return &BlocklistClient{config: c}
+// NewBlacklistClient returns a client for the Blacklist from the given config.
+func NewBlacklistClient(c config) *BlacklistClient {
+	return &BlacklistClient{config: c}
 }
 
 // Use adds a list of mutation hooks to the hooks stack.
-// A call to `Use(f, g, h)` equals to `blocklist.Hooks(f(g(h())))`.
-func (c *BlocklistClient) Use(hooks ...Hook) {
-	c.hooks.Blocklist = append(c.hooks.Blocklist, hooks...)
+// A call to `Use(f, g, h)` equals to `blacklist.Hooks(f(g(h())))`.
+func (c *BlacklistClient) Use(hooks ...Hook) {
+	c.hooks.Blacklist = append(c.hooks.Blacklist, hooks...)
 }
 
 // Intercept adds a list of query interceptors to the interceptors stack.
-// A call to `Intercept(f, g, h)` equals to `blocklist.Intercept(f(g(h())))`.
-func (c *BlocklistClient) Intercept(interceptors ...Interceptor) {
-	c.inters.Blocklist = append(c.inters.Blocklist, interceptors...)
+// A call to `Intercept(f, g, h)` equals to `blacklist.Intercept(f(g(h())))`.
+func (c *BlacklistClient) Intercept(interceptors ...Interceptor) {
+	c.inters.Blacklist = append(c.inters.Blacklist, interceptors...)
 }
 
-// Create returns a builder for creating a Blocklist entity.
-func (c *BlocklistClient) Create() *BlocklistCreate {
-	mutation := newBlocklistMutation(c.config, OpCreate)
-	return &BlocklistCreate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+// Create returns a builder for creating a Blacklist entity.
+func (c *BlacklistClient) Create() *BlacklistCreate {
+	mutation := newBlacklistMutation(c.config, OpCreate)
+	return &BlacklistCreate{config: c.config, hooks: c.Hooks(), mutation: mutation}
 }
 
-// CreateBulk returns a builder for creating a bulk of Blocklist entities.
-func (c *BlocklistClient) CreateBulk(builders ...*BlocklistCreate) *BlocklistCreateBulk {
-	return &BlocklistCreateBulk{config: c.config, builders: builders}
+// CreateBulk returns a builder for creating a bulk of Blacklist entities.
+func (c *BlacklistClient) CreateBulk(builders ...*BlacklistCreate) *BlacklistCreateBulk {
+	return &BlacklistCreateBulk{config: c.config, builders: builders}
 }
 
 // MapCreateBulk creates a bulk creation builder from the given slice. For each item in the slice, the function creates
 // a builder and applies setFunc on it.
-func (c *BlocklistClient) MapCreateBulk(slice any, setFunc func(*BlocklistCreate, int)) *BlocklistCreateBulk {
+func (c *BlacklistClient) MapCreateBulk(slice any, setFunc func(*BlacklistCreate, int)) *BlacklistCreateBulk {
 	rv := reflect.ValueOf(slice)
 	if rv.Kind() != reflect.Slice {
-		return &BlocklistCreateBulk{err: fmt.Errorf("calling to BlocklistClient.MapCreateBulk with wrong type %T, need slice", slice)}
+		return &BlacklistCreateBulk{err: fmt.Errorf("calling to BlacklistClient.MapCreateBulk with wrong type %T, need slice", slice)}
 	}
-	builders := make([]*BlocklistCreate, rv.Len())
+	builders := make([]*BlacklistCreate, rv.Len())
 	for i := 0; i < rv.Len(); i++ {
 		builders[i] = c.Create()
 		setFunc(builders[i], i)
 	}
-	return &BlocklistCreateBulk{config: c.config, builders: builders}
+	return &BlacklistCreateBulk{config: c.config, builders: builders}
 }
 
-// Update returns an update builder for Blocklist.
-func (c *BlocklistClient) Update() *BlocklistUpdate {
-	mutation := newBlocklistMutation(c.config, OpUpdate)
-	return &BlocklistUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation}
+// Update returns an update builder for Blacklist.
+func (c *BlacklistClient) Update() *BlacklistUpdate {
+	mutation := newBlacklistMutation(c.config, OpUpdate)
+	return &BlacklistUpdate{config: c.config, hooks: c.Hooks(), mutation: mutation}
 }
 
 // UpdateOne returns an update builder for the given entity.
-func (c *BlocklistClient) UpdateOne(b *Blocklist) *BlocklistUpdateOne {
-	mutation := newBlocklistMutation(c.config, OpUpdateOne, withBlocklist(b))
-	return &BlocklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+func (c *BlacklistClient) UpdateOne(b *Blacklist) *BlacklistUpdateOne {
+	mutation := newBlacklistMutation(c.config, OpUpdateOne, withBlacklist(b))
+	return &BlacklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
 }
 
 // UpdateOneID returns an update builder for the given id.
-func (c *BlocklistClient) UpdateOneID(id int) *BlocklistUpdateOne {
-	mutation := newBlocklistMutation(c.config, OpUpdateOne, withBlocklistID(id))
-	return &BlocklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
+func (c *BlacklistClient) UpdateOneID(id int) *BlacklistUpdateOne {
+	mutation := newBlacklistMutation(c.config, OpUpdateOne, withBlacklistID(id))
+	return &BlacklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: mutation}
 }
 
-// Delete returns a delete builder for Blocklist.
-func (c *BlocklistClient) Delete() *BlocklistDelete {
-	mutation := newBlocklistMutation(c.config, OpDelete)
-	return &BlocklistDelete{config: c.config, hooks: c.Hooks(), mutation: mutation}
+// Delete returns a delete builder for Blacklist.
+func (c *BlacklistClient) Delete() *BlacklistDelete {
+	mutation := newBlacklistMutation(c.config, OpDelete)
+	return &BlacklistDelete{config: c.config, hooks: c.Hooks(), mutation: mutation}
 }
 
 // DeleteOne returns a builder for deleting the given entity.
-func (c *BlocklistClient) DeleteOne(b *Blocklist) *BlocklistDeleteOne {
+func (c *BlacklistClient) DeleteOne(b *Blacklist) *BlacklistDeleteOne {
 	return c.DeleteOneID(b.ID)
 }
 
 // DeleteOneID returns a builder for deleting the given entity by its id.
-func (c *BlocklistClient) DeleteOneID(id int) *BlocklistDeleteOne {
-	builder := c.Delete().Where(blocklist.ID(id))
+func (c *BlacklistClient) DeleteOneID(id int) *BlacklistDeleteOne {
+	builder := c.Delete().Where(blacklist.ID(id))
 	builder.mutation.id = &id
 	builder.mutation.op = OpDeleteOne
-	return &BlocklistDeleteOne{builder}
+	return &BlacklistDeleteOne{builder}
 }
 
-// Query returns a query builder for Blocklist.
-func (c *BlocklistClient) Query() *BlocklistQuery {
-	return &BlocklistQuery{
+// Query returns a query builder for Blacklist.
+func (c *BlacklistClient) Query() *BlacklistQuery {
+	return &BlacklistQuery{
 		config: c.config,
-		ctx:    &QueryContext{Type: TypeBlocklist},
+		ctx:    &QueryContext{Type: TypeBlacklist},
 		inters: c.Interceptors(),
 	}
 }
 
-// Get returns a Blocklist entity by its id.
-func (c *BlocklistClient) Get(ctx context.Context, id int) (*Blocklist, error) {
-	return c.Query().Where(blocklist.ID(id)).Only(ctx)
+// Get returns a Blacklist entity by its id.
+func (c *BlacklistClient) Get(ctx context.Context, id int) (*Blacklist, error) {
+	return c.Query().Where(blacklist.ID(id)).Only(ctx)
 }
 
 // GetX is like Get, but panics if an error occurs.
-func (c *BlocklistClient) GetX(ctx context.Context, id int) *Blocklist {
+func (c *BlacklistClient) GetX(ctx context.Context, id int) *Blacklist {
 	obj, err := c.Get(ctx, id)
 	if err != nil {
 		panic(err)
@@ -389,27 +389,27 @@ func (c *BlocklistClient) GetX(ctx context.Context, id int) *Blocklist {
 }
 
 // Hooks returns the client hooks.
-func (c *BlocklistClient) Hooks() []Hook {
-	return c.hooks.Blocklist
+func (c *BlacklistClient) Hooks() []Hook {
+	return c.hooks.Blacklist
 }
 
 // Interceptors returns the client interceptors.
-func (c *BlocklistClient) Interceptors() []Interceptor {
-	return c.inters.Blocklist
+func (c *BlacklistClient) Interceptors() []Interceptor {
+	return c.inters.Blacklist
 }
 
-func (c *BlocklistClient) mutate(ctx context.Context, m *BlocklistMutation) (Value, error) {
+func (c *BlacklistClient) mutate(ctx context.Context, m *BlacklistMutation) (Value, error) {
 	switch m.Op() {
 	case OpCreate:
-		return (&BlocklistCreate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+		return (&BlacklistCreate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
 	case OpUpdate:
-		return (&BlocklistUpdate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+		return (&BlacklistUpdate{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
 	case OpUpdateOne:
-		return (&BlocklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
+		return (&BlacklistUpdateOne{config: c.config, hooks: c.Hooks(), mutation: m}).Save(ctx)
 	case OpDelete, OpDeleteOne:
-		return (&BlocklistDelete{config: c.config, hooks: c.Hooks(), mutation: m}).Exec(ctx)
+		return (&BlacklistDelete{config: c.config, hooks: c.Hooks(), mutation: m}).Exec(ctx)
 	default:
-		return nil, fmt.Errorf("ent: unknown Blocklist mutation op: %q", m.Op())
+		return nil, fmt.Errorf("ent: unknown Blacklist mutation op: %q", m.Op())
 	}
 }
 
@@ -1645,11 +1645,11 @@ func (c *StorageClient) mutate(ctx context.Context, m *StorageMutation) (Value,
 // hooks and interceptors per client, for fast access.
 type (
 	hooks struct {
-		Blocklist, DownloadClients, Episode, History, ImportList, Indexers, Media,
+		Blacklist, DownloadClients, Episode, History, ImportList, Indexers, Media,
 		NotificationClient, Settings, Storage []ent.Hook
 	}
 	inters struct {
-		Blocklist, DownloadClients, Episode, History, ImportList, Indexers, Media,
+		Blacklist, DownloadClients, Episode, History, ImportList, Indexers, Media,
 		NotificationClient, Settings, Storage []ent.Interceptor
 	}
 )

ent/ent.go

@@ -6,7 +6,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"polaris/ent/blocklist"
+	"polaris/ent/blacklist"
 	"polaris/ent/downloadclients"
 	"polaris/ent/episode"
 	"polaris/ent/history"
@@ -82,7 +82,7 @@ var (
 func checkColumn(table, column string) error {
 	initCheck.Do(func() {
 		columnCheck = sql.NewColumnCheck(map[string]func(string) bool{
-			blocklist.Table:          blocklist.ValidColumn,
+			blacklist.Table:          blacklist.ValidColumn,
 			downloadclients.Table:    downloadclients.ValidColumn,
 			episode.Table:            episode.ValidColumn,
 			history.Table:            history.ValidColumn,

ent/history.go

@@ -3,6 +3,7 @@
 package ent
 
 import (
+	"encoding/json"
 	"fmt"
 	"polaris/ent/history"
 	"strings"
@@ -19,8 +20,12 @@ type History struct {
 	ID int `json:"id,omitempty"`
 	// MediaID holds the value of the "media_id" field.
 	MediaID int `json:"media_id,omitempty"`
-	// EpisodeID holds the value of the "episode_id" field.
+	// deprecated
 	EpisodeID int `json:"episode_id,omitempty"`
+	// EpisodeNums holds the value of the "episode_nums" field.
+	EpisodeNums []int `json:"episode_nums,omitempty"`
+	// SeasonNum holds the value of the "season_num" field.
+	SeasonNum int `json:"season_num,omitempty"`
 	// SourceTitle holds the value of the "source_title" field.
 	SourceTitle string `json:"source_title,omitempty"`
 	// Date holds the value of the "date" field.
@@ -33,9 +38,11 @@ type History struct {
 	DownloadClientID int `json:"download_client_id,omitempty"`
 	// IndexerID holds the value of the "indexer_id" field.
 	IndexerID int `json:"indexer_id,omitempty"`
+	// Link holds the value of the "link" field.
+	Link string `json:"link,omitempty"`
 	// Status holds the value of the "status" field.
 	Status history.Status `json:"status,omitempty"`
-	// Saved holds the value of the "saved" field.
+	// deprecated
 	Saved        string `json:"saved,omitempty"`
 	selectValues sql.SelectValues
 }
@@ -45,9 +52,11 @@ func (*History) scanValues(columns []string) ([]any, error) {
 	values := make([]any, len(columns))
 	for i := range columns {
 		switch columns[i] {
-		case history.FieldID, history.FieldMediaID, history.FieldEpisodeID, history.FieldSize, history.FieldDownloadClientID, history.FieldIndexerID:
+		case history.FieldEpisodeNums:
+			values[i] = new([]byte)
+		case history.FieldID, history.FieldMediaID, history.FieldEpisodeID, history.FieldSeasonNum, history.FieldSize, history.FieldDownloadClientID, history.FieldIndexerID:
 			values[i] = new(sql.NullInt64)
-		case history.FieldSourceTitle, history.FieldTargetDir, history.FieldStatus, history.FieldSaved:
+		case history.FieldSourceTitle, history.FieldTargetDir, history.FieldLink, history.FieldStatus, history.FieldSaved:
 			values[i] = new(sql.NullString)
 		case history.FieldDate:
 			values[i] = new(sql.NullTime)
@@ -84,6 +93,20 @@ func (h *History) assignValues(columns []string, values []any) error {
 			} else if value.Valid {
 				h.EpisodeID = int(value.Int64)
 			}
+		case history.FieldEpisodeNums:
+			if value, ok := values[i].(*[]byte); !ok {
+				return fmt.Errorf("unexpected type %T for field episode_nums", values[i])
+			} else if value != nil && len(*value) > 0 {
+				if err := json.Unmarshal(*value, &h.EpisodeNums); err != nil {
+					return fmt.Errorf("unmarshal field episode_nums: %w", err)
+				}
+			}
+		case history.FieldSeasonNum:
+			if value, ok := values[i].(*sql.NullInt64); !ok {
+				return fmt.Errorf("unexpected type %T for field season_num", values[i])
+			} else if value.Valid {
+				h.SeasonNum = int(value.Int64)
+			}
 		case history.FieldSourceTitle:
 			if value, ok := values[i].(*sql.NullString); !ok {
 				return fmt.Errorf("unexpected type %T for field source_title", values[i])
@@ -120,6 +143,12 @@ func (h *History) assignValues(columns []string, values []any) error {
 			} else if value.Valid {
 				h.IndexerID = int(value.Int64)
 			}
+		case history.FieldLink:
+			if value, ok := values[i].(*sql.NullString); !ok {
+				return fmt.Errorf("unexpected type %T for field link", values[i])
+			} else if value.Valid {
+				h.Link = value.String
+			}
 		case history.FieldStatus:
 			if value, ok := values[i].(*sql.NullString); !ok {
 				return fmt.Errorf("unexpected type %T for field status", values[i])
@@ -174,6 +203,12 @@ func (h *History) String() string {
 	builder.WriteString("episode_id=")
 	builder.WriteString(fmt.Sprintf("%v", h.EpisodeID))
 	builder.WriteString(", ")
+	builder.WriteString("episode_nums=")
+	builder.WriteString(fmt.Sprintf("%v", h.EpisodeNums))
+	builder.WriteString(", ")
+	builder.WriteString("season_num=")
+	builder.WriteString(fmt.Sprintf("%v", h.SeasonNum))
+	builder.WriteString(", ")
 	builder.WriteString("source_title=")
 	builder.WriteString(h.SourceTitle)
 	builder.WriteString(", ")
@@ -192,6 +227,9 @@ func (h *History) String() string {
 	builder.WriteString("indexer_id=")
 	builder.WriteString(fmt.Sprintf("%v", h.IndexerID))
 	builder.WriteString(", ")
+	builder.WriteString("link=")
+	builder.WriteString(h.Link)
+	builder.WriteString(", ")
 	builder.WriteString("status=")
 	builder.WriteString(fmt.Sprintf("%v", h.Status))
 	builder.WriteString(", ")

ent/history/history.go

@@ -17,6 +17,10 @@ const (
 	FieldMediaID = "media_id"
 	// FieldEpisodeID holds the string denoting the episode_id field in the database.
 	FieldEpisodeID = "episode_id"
+	// FieldEpisodeNums holds the string denoting the episode_nums field in the database.
+	FieldEpisodeNums = "episode_nums"
+	// FieldSeasonNum holds the string denoting the season_num field in the database.
+	FieldSeasonNum = "season_num"
 	// FieldSourceTitle holds the string denoting the source_title field in the database.
 	FieldSourceTitle = "source_title"
 	// FieldDate holds the string denoting the date field in the database.
@@ -29,6 +33,8 @@ const (
 	FieldDownloadClientID = "download_client_id"
 	// FieldIndexerID holds the string denoting the indexer_id field in the database.
 	FieldIndexerID = "indexer_id"
+	// FieldLink holds the string denoting the link field in the database.
+	FieldLink = "link"
 	// FieldStatus holds the string denoting the status field in the database.
 	FieldStatus = "status"
 	// FieldSaved holds the string denoting the saved field in the database.
@@ -42,12 +48,15 @@ var Columns = []string{
 	FieldID,
 	FieldMediaID,
 	FieldEpisodeID,
+	FieldEpisodeNums,
+	FieldSeasonNum,
 	FieldSourceTitle,
 	FieldDate,
 	FieldTargetDir,
 	FieldSize,
 	FieldDownloadClientID,
 	FieldIndexerID,
+	FieldLink,
 	FieldStatus,
 	FieldSaved,
 }
@@ -111,6 +120,11 @@ func ByEpisodeID(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldEpisodeID, opts...).ToFunc()
 }
 
+// BySeasonNum orders the results by the season_num field.
+func BySeasonNum(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldSeasonNum, opts...).ToFunc()
+}
+
 // BySourceTitle orders the results by the source_title field.
 func BySourceTitle(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldSourceTitle, opts...).ToFunc()
@@ -141,6 +155,11 @@ func ByIndexerID(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldIndexerID, opts...).ToFunc()
 }
 
+// ByLink orders the results by the link field.
+func ByLink(opts ...sql.OrderTermOption) OrderOption {
+	return sql.OrderByField(FieldLink, opts...).ToFunc()
+}
+
 // ByStatus orders the results by the status field.
 func ByStatus(opts ...sql.OrderTermOption) OrderOption {
 	return sql.OrderByField(FieldStatus, opts...).ToFunc()

ent/history/where.go

@@ -64,6 +64,11 @@ func EpisodeID(v int) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldEpisodeID, v))
 }
 
+// SeasonNum applies equality check predicate on the "season_num" field. It's identical to SeasonNumEQ.
+func SeasonNum(v int) predicate.History {
+	return predicate.History(sql.FieldEQ(FieldSeasonNum, v))
+}
+
 // SourceTitle applies equality check predicate on the "source_title" field. It's identical to SourceTitleEQ.
 func SourceTitle(v string) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldSourceTitle, v))
@@ -94,6 +99,11 @@ func IndexerID(v int) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldIndexerID, v))
 }
 
+// Link applies equality check predicate on the "link" field. It's identical to LinkEQ.
+func Link(v string) predicate.History {
+	return predicate.History(sql.FieldEQ(FieldLink, v))
+}
+
 // Saved applies equality check predicate on the "saved" field. It's identical to SavedEQ.
 func Saved(v string) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldSaved, v))
@@ -189,6 +199,66 @@ func EpisodeIDNotNil() predicate.History {
 	return predicate.History(sql.FieldNotNull(FieldEpisodeID))
 }
 
+// EpisodeNumsIsNil applies the IsNil predicate on the "episode_nums" field.
+func EpisodeNumsIsNil() predicate.History {
+	return predicate.History(sql.FieldIsNull(FieldEpisodeNums))
+}
+
+// EpisodeNumsNotNil applies the NotNil predicate on the "episode_nums" field.
+func EpisodeNumsNotNil() predicate.History {
+	return predicate.History(sql.FieldNotNull(FieldEpisodeNums))
+}
+
+// SeasonNumEQ applies the EQ predicate on the "season_num" field.
+func SeasonNumEQ(v int) predicate.History {
+	return predicate.History(sql.FieldEQ(FieldSeasonNum, v))
+}
+
+// SeasonNumNEQ applies the NEQ predicate on the "season_num" field.
+func SeasonNumNEQ(v int) predicate.History {
+	return predicate.History(sql.FieldNEQ(FieldSeasonNum, v))
+}
+
+// SeasonNumIn applies the In predicate on the "season_num" field.
+func SeasonNumIn(vs ...int) predicate.History {
+	return predicate.History(sql.FieldIn(FieldSeasonNum, vs...))
+}
+
+// SeasonNumNotIn applies the NotIn predicate on the "season_num" field.
+func SeasonNumNotIn(vs ...int) predicate.History {
+	return predicate.History(sql.FieldNotIn(FieldSeasonNum, vs...))
+}
+
+// SeasonNumGT applies the GT predicate on the "season_num" field.
+func SeasonNumGT(v int) predicate.History {
+	return predicate.History(sql.FieldGT(FieldSeasonNum, v))
+}
+
+// SeasonNumGTE applies the GTE predicate on the "season_num" field.
+func SeasonNumGTE(v int) predicate.History {
+	return predicate.History(sql.FieldGTE(FieldSeasonNum, v))
+}
+
+// SeasonNumLT applies the LT predicate on the "season_num" field.
+func SeasonNumLT(v int) predicate.History {
+	return predicate.History(sql.FieldLT(FieldSeasonNum, v))
+}
+
+// SeasonNumLTE applies the LTE predicate on the "season_num" field.
+func SeasonNumLTE(v int) predicate.History {
+	return predicate.History(sql.FieldLTE(FieldSeasonNum, v))
+}
+
+// SeasonNumIsNil applies the IsNil predicate on the "season_num" field.
+func SeasonNumIsNil() predicate.History {
+	return predicate.History(sql.FieldIsNull(FieldSeasonNum))
+}
+
+// SeasonNumNotNil applies the NotNil predicate on the "season_num" field.
+func SeasonNumNotNil() predicate.History {
+	return predicate.History(sql.FieldNotNull(FieldSeasonNum))
+}
+
 // SourceTitleEQ applies the EQ predicate on the "source_title" field.
 func SourceTitleEQ(v string) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldSourceTitle, v))
@@ -499,6 +569,81 @@ func IndexerIDNotNil() predicate.History {
 	return predicate.History(sql.FieldNotNull(FieldIndexerID))
 }
 
+// LinkEQ applies the EQ predicate on the "link" field.
+func LinkEQ(v string) predicate.History {
+	return predicate.History(sql.FieldEQ(FieldLink, v))
+}
+
+// LinkNEQ applies the NEQ predicate on the "link" field.
+func LinkNEQ(v string) predicate.History {
+	return predicate.History(sql.FieldNEQ(FieldLink, v))
+}
+
+// LinkIn applies the In predicate on the "link" field.
+func LinkIn(vs ...string) predicate.History {
+	return predicate.History(sql.FieldIn(FieldLink, vs...))
+}
+
+// LinkNotIn applies the NotIn predicate on the "link" field.
+func LinkNotIn(vs ...string) predicate.History {
+	return predicate.History(sql.FieldNotIn(FieldLink, vs...))
+}
+
+// LinkGT applies the GT predicate on the "link" field.
+func LinkGT(v string) predicate.History {
+	return predicate.History(sql.FieldGT(FieldLink, v))
+}
+
+// LinkGTE applies the GTE predicate on the "link" field.
+func LinkGTE(v string) predicate.History {
+	return predicate.History(sql.FieldGTE(FieldLink, v))
+}
+
+// LinkLT applies the LT predicate on the "link" field.
+func LinkLT(v string) predicate.History {
+	return predicate.History(sql.FieldLT(FieldLink, v))
+}
+
+// LinkLTE applies the LTE predicate on the "link" field.
+func LinkLTE(v string) predicate.History {
+	return predicate.History(sql.FieldLTE(FieldLink, v))
+}
+
+// LinkContains applies the Contains predicate on the "link" field.
+func LinkContains(v string) predicate.History {
+	return predicate.History(sql.FieldContains(FieldLink, v))
+}
+
+// LinkHasPrefix applies the HasPrefix predicate on the "link" field.
+func LinkHasPrefix(v string) predicate.History {
+	return predicate.History(sql.FieldHasPrefix(FieldLink, v))
+}
+
+// LinkHasSuffix applies the HasSuffix predicate on the "link" field.
+func LinkHasSuffix(v string) predicate.History {
+	return predicate.History(sql.FieldHasSuffix(FieldLink, v))
+}
+
+// LinkIsNil applies the IsNil predicate on the "link" field.
+func LinkIsNil() predicate.History {
+	return predicate.History(sql.FieldIsNull(FieldLink))
+}
+
+// LinkNotNil applies the NotNil predicate on the "link" field.
+func LinkNotNil() predicate.History {
+	return predicate.History(sql.FieldNotNull(FieldLink))
+}
+
+// LinkEqualFold applies the EqualFold predicate on the "link" field.
+func LinkEqualFold(v string) predicate.History {
+	return predicate.History(sql.FieldEqualFold(FieldLink, v))
+}
+
+// LinkContainsFold applies the ContainsFold predicate on the "link" field.
+func LinkContainsFold(v string) predicate.History {
+	return predicate.History(sql.FieldContainsFold(FieldLink, v))
+}
+
 // StatusEQ applies the EQ predicate on the "status" field.
 func StatusEQ(v Status) predicate.History {
 	return predicate.History(sql.FieldEQ(FieldStatus, v))

ent/history_create.go

@@ -40,6 +40,26 @@ func (hc *HistoryCreate) SetNillableEpisodeID(i *int) *HistoryCreate {
 	return hc
 }
 
+// SetEpisodeNums sets the "episode_nums" field.
+func (hc *HistoryCreate) SetEpisodeNums(i []int) *HistoryCreate {
+	hc.mutation.SetEpisodeNums(i)
+	return hc
+}
+
+// SetSeasonNum sets the "season_num" field.
+func (hc *HistoryCreate) SetSeasonNum(i int) *HistoryCreate {
+	hc.mutation.SetSeasonNum(i)
+	return hc
+}
+
+// SetNillableSeasonNum sets the "season_num" field if the given value is not nil.
+func (hc *HistoryCreate) SetNillableSeasonNum(i *int) *HistoryCreate {
+	if i != nil {
+		hc.SetSeasonNum(*i)
+	}
+	return hc
+}
+
 // SetSourceTitle sets the "source_title" field.
 func (hc *HistoryCreate) SetSourceTitle(s string) *HistoryCreate {
 	hc.mutation.SetSourceTitle(s)
@@ -100,6 +120,20 @@ func (hc *HistoryCreate) SetNillableIndexerID(i *int) *HistoryCreate {
 	return hc
 }
 
+// SetLink sets the "link" field.
+func (hc *HistoryCreate) SetLink(s string) *HistoryCreate {
+	hc.mutation.SetLink(s)
+	return hc
+}
+
+// SetNillableLink sets the "link" field if the given value is not nil.
+func (hc *HistoryCreate) SetNillableLink(s *string) *HistoryCreate {
+	if s != nil {
+		hc.SetLink(*s)
+	}
+	return hc
+}
+
 // SetStatus sets the "status" field.
 func (hc *HistoryCreate) SetStatus(h history.Status) *HistoryCreate {
 	hc.mutation.SetStatus(h)
@@ -220,6 +254,14 @@ func (hc *HistoryCreate) createSpec() (*History, *sqlgraph.CreateSpec) {
 		_spec.SetField(history.FieldEpisodeID, field.TypeInt, value)
 		_node.EpisodeID = value
 	}
+	if value, ok := hc.mutation.EpisodeNums(); ok {
+		_spec.SetField(history.FieldEpisodeNums, field.TypeJSON, value)
+		_node.EpisodeNums = value
+	}
+	if value, ok := hc.mutation.SeasonNum(); ok {
+		_spec.SetField(history.FieldSeasonNum, field.TypeInt, value)
+		_node.SeasonNum = value
+	}
 	if value, ok := hc.mutation.SourceTitle(); ok {
 		_spec.SetField(history.FieldSourceTitle, field.TypeString, value)
 		_node.SourceTitle = value
@@ -244,6 +286,10 @@ func (hc *HistoryCreate) createSpec() (*History, *sqlgraph.CreateSpec) {
 		_spec.SetField(history.FieldIndexerID, field.TypeInt, value)
 		_node.IndexerID = value
 	}
+	if value, ok := hc.mutation.Link(); ok {
+		_spec.SetField(history.FieldLink, field.TypeString, value)
+		_node.Link = value
+	}
 	if value, ok := hc.mutation.Status(); ok {
 		_spec.SetField(history.FieldStatus, field.TypeEnum, value)
 		_node.Status = value

ent/history_update.go

@@ -12,6 +12,7 @@ import (
 
 	"entgo.io/ent/dialect/sql"
 	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/dialect/sql/sqljson"
 	"entgo.io/ent/schema/field"
 )
 
@@ -76,6 +77,51 @@ func (hu *HistoryUpdate) ClearEpisodeID() *HistoryUpdate {
 	return hu
 }
 
+// SetEpisodeNums sets the "episode_nums" field.
+func (hu *HistoryUpdate) SetEpisodeNums(i []int) *HistoryUpdate {
+	hu.mutation.SetEpisodeNums(i)
+	return hu
+}
+
+// AppendEpisodeNums appends i to the "episode_nums" field.
+func (hu *HistoryUpdate) AppendEpisodeNums(i []int) *HistoryUpdate {
+	hu.mutation.AppendEpisodeNums(i)
+	return hu
+}
+
+// ClearEpisodeNums clears the value of the "episode_nums" field.
+func (hu *HistoryUpdate) ClearEpisodeNums() *HistoryUpdate {
+	hu.mutation.ClearEpisodeNums()
+	return hu
+}
+
+// SetSeasonNum sets the "season_num" field.
+func (hu *HistoryUpdate) SetSeasonNum(i int) *HistoryUpdate {
+	hu.mutation.ResetSeasonNum()
+	hu.mutation.SetSeasonNum(i)
+	return hu
+}
+
+// SetNillableSeasonNum sets the "season_num" field if the given value is not nil.
+func (hu *HistoryUpdate) SetNillableSeasonNum(i *int) *HistoryUpdate {
+	if i != nil {
+		hu.SetSeasonNum(*i)
+	}
+	return hu
+}
+
+// AddSeasonNum adds i to the "season_num" field.
+func (hu *HistoryUpdate) AddSeasonNum(i int) *HistoryUpdate {
+	hu.mutation.AddSeasonNum(i)
+	return hu
+}
+
+// ClearSeasonNum clears the value of the "season_num" field.
+func (hu *HistoryUpdate) ClearSeasonNum() *HistoryUpdate {
+	hu.mutation.ClearSeasonNum()
+	return hu
+}
+
 // SetSourceTitle sets the "source_title" field.
 func (hu *HistoryUpdate) SetSourceTitle(s string) *HistoryUpdate {
 	hu.mutation.SetSourceTitle(s)
@@ -193,6 +239,26 @@ func (hu *HistoryUpdate) ClearIndexerID() *HistoryUpdate {
 	return hu
 }
 
+// SetLink sets the "link" field.
+func (hu *HistoryUpdate) SetLink(s string) *HistoryUpdate {
+	hu.mutation.SetLink(s)
+	return hu
+}
+
+// SetNillableLink sets the "link" field if the given value is not nil.
+func (hu *HistoryUpdate) SetNillableLink(s *string) *HistoryUpdate {
+	if s != nil {
+		hu.SetLink(*s)
+	}
+	return hu
+}
+
+// ClearLink clears the value of the "link" field.
+func (hu *HistoryUpdate) ClearLink() *HistoryUpdate {
+	hu.mutation.ClearLink()
+	return hu
+}
+
 // SetStatus sets the "status" field.
 func (hu *HistoryUpdate) SetStatus(h history.Status) *HistoryUpdate {
 	hu.mutation.SetStatus(h)
@@ -296,6 +362,26 @@ func (hu *HistoryUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if hu.mutation.EpisodeIDCleared() {
 		_spec.ClearField(history.FieldEpisodeID, field.TypeInt)
 	}
+	if value, ok := hu.mutation.EpisodeNums(); ok {
+		_spec.SetField(history.FieldEpisodeNums, field.TypeJSON, value)
+	}
+	if value, ok := hu.mutation.AppendedEpisodeNums(); ok {
+		_spec.AddModifier(func(u *sql.UpdateBuilder) {
+			sqljson.Append(u, history.FieldEpisodeNums, value)
+		})
+	}
+	if hu.mutation.EpisodeNumsCleared() {
+		_spec.ClearField(history.FieldEpisodeNums, field.TypeJSON)
+	}
+	if value, ok := hu.mutation.SeasonNum(); ok {
+		_spec.SetField(history.FieldSeasonNum, field.TypeInt, value)
+	}
+	if value, ok := hu.mutation.AddedSeasonNum(); ok {
+		_spec.AddField(history.FieldSeasonNum, field.TypeInt, value)
+	}
+	if hu.mutation.SeasonNumCleared() {
+		_spec.ClearField(history.FieldSeasonNum, field.TypeInt)
+	}
 	if value, ok := hu.mutation.SourceTitle(); ok {
 		_spec.SetField(history.FieldSourceTitle, field.TypeString, value)
 	}
@@ -329,6 +415,12 @@ func (hu *HistoryUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if hu.mutation.IndexerIDCleared() {
 		_spec.ClearField(history.FieldIndexerID, field.TypeInt)
 	}
+	if value, ok := hu.mutation.Link(); ok {
+		_spec.SetField(history.FieldLink, field.TypeString, value)
+	}
+	if hu.mutation.LinkCleared() {
+		_spec.ClearField(history.FieldLink, field.TypeString)
+	}
 	if value, ok := hu.mutation.Status(); ok {
 		_spec.SetField(history.FieldStatus, field.TypeEnum, value)
 	}
@@ -406,6 +498,51 @@ func (huo *HistoryUpdateOne) ClearEpisodeID() *HistoryUpdateOne {
 	return huo
 }
 
+// SetEpisodeNums sets the "episode_nums" field.
+func (huo *HistoryUpdateOne) SetEpisodeNums(i []int) *HistoryUpdateOne {
+	huo.mutation.SetEpisodeNums(i)
+	return huo
+}
+
+// AppendEpisodeNums appends i to the "episode_nums" field.
+func (huo *HistoryUpdateOne) AppendEpisodeNums(i []int) *HistoryUpdateOne {
+	huo.mutation.AppendEpisodeNums(i)
+	return huo
+}
+
+// ClearEpisodeNums clears the value of the "episode_nums" field.
+func (huo *HistoryUpdateOne) ClearEpisodeNums() *HistoryUpdateOne {
+	huo.mutation.ClearEpisodeNums()
+	return huo
+}
+
+// SetSeasonNum sets the "season_num" field.
+func (huo *HistoryUpdateOne) SetSeasonNum(i int) *HistoryUpdateOne {
+	huo.mutation.ResetSeasonNum()
+	huo.mutation.SetSeasonNum(i)
+	return huo
+}
+
+// SetNillableSeasonNum sets the "season_num" field if the given value is not nil.
+func (huo *HistoryUpdateOne) SetNillableSeasonNum(i *int) *HistoryUpdateOne {
+	if i != nil {
+		huo.SetSeasonNum(*i)
+	}
+	return huo
+}
+
+// AddSeasonNum adds i to the "season_num" field.
+func (huo *HistoryUpdateOne) AddSeasonNum(i int) *HistoryUpdateOne {
+	huo.mutation.AddSeasonNum(i)
+	return huo
+}
+
+// ClearSeasonNum clears the value of the "season_num" field.
+func (huo *HistoryUpdateOne) ClearSeasonNum() *HistoryUpdateOne {
+	huo.mutation.ClearSeasonNum()
+	return huo
+}
+
 // SetSourceTitle sets the "source_title" field.
 func (huo *HistoryUpdateOne) SetSourceTitle(s string) *HistoryUpdateOne {
 	huo.mutation.SetSourceTitle(s)
@@ -523,6 +660,26 @@ func (huo *HistoryUpdateOne) ClearIndexerID() *HistoryUpdateOne {
 	return huo
 }
 
+// SetLink sets the "link" field.
+func (huo *HistoryUpdateOne) SetLink(s string) *HistoryUpdateOne {
+	huo.mutation.SetLink(s)
+	return huo
+}
+
+// SetNillableLink sets the "link" field if the given value is not nil.
+func (huo *HistoryUpdateOne) SetNillableLink(s *string) *HistoryUpdateOne {
+	if s != nil {
+		huo.SetLink(*s)
+	}
+	return huo
+}
+
+// ClearLink clears the value of the "link" field.
+func (huo *HistoryUpdateOne) ClearLink() *HistoryUpdateOne {
+	huo.mutation.ClearLink()
+	return huo
+}
+
 // SetStatus sets the "status" field.
 func (huo *HistoryUpdateOne) SetStatus(h history.Status) *HistoryUpdateOne {
 	huo.mutation.SetStatus(h)
@@ -656,6 +813,26 @@ func (huo *HistoryUpdateOne) sqlSave(ctx context.Context) (_node *History, err e
 	if huo.mutation.EpisodeIDCleared() {
 		_spec.ClearField(history.FieldEpisodeID, field.TypeInt)
 	}
+	if value, ok := huo.mutation.EpisodeNums(); ok {
+		_spec.SetField(history.FieldEpisodeNums, field.TypeJSON, value)
+	}
+	if value, ok := huo.mutation.AppendedEpisodeNums(); ok {
+		_spec.AddModifier(func(u *sql.UpdateBuilder) {
+			sqljson.Append(u, history.FieldEpisodeNums, value)
+		})
+	}
+	if huo.mutation.EpisodeNumsCleared() {
+		_spec.ClearField(history.FieldEpisodeNums, field.TypeJSON)
+	}
+	if value, ok := huo.mutation.SeasonNum(); ok {
+		_spec.SetField(history.FieldSeasonNum, field.TypeInt, value)
+	}
+	if value, ok := huo.mutation.AddedSeasonNum(); ok {
+		_spec.AddField(history.FieldSeasonNum, field.TypeInt, value)
+	}
+	if huo.mutation.SeasonNumCleared() {
+		_spec.ClearField(history.FieldSeasonNum, field.TypeInt)
+	}
 	if value, ok := huo.mutation.SourceTitle(); ok {
 		_spec.SetField(history.FieldSourceTitle, field.TypeString, value)
 	}
@@ -689,6 +866,12 @@ func (huo *HistoryUpdateOne) sqlSave(ctx context.Context) (_node *History, err e
 	if huo.mutation.IndexerIDCleared() {
 		_spec.ClearField(history.FieldIndexerID, field.TypeInt)
 	}
+	if value, ok := huo.mutation.Link(); ok {
+		_spec.SetField(history.FieldLink, field.TypeString, value)
+	}
+	if huo.mutation.LinkCleared() {
+		_spec.ClearField(history.FieldLink, field.TypeString)
+	}
 	if value, ok := huo.mutation.Status(); ok {
 		_spec.SetField(history.FieldStatus, field.TypeEnum, value)
 	}

ent/hook/hook.go

@@ -8,16 +8,16 @@ import (
 	"polaris/ent"
 )
 
-// The BlocklistFunc type is an adapter to allow the use of ordinary
-// function as Blocklist mutator.
-type BlocklistFunc func(context.Context, *ent.BlocklistMutation) (ent.Value, error)
+// The BlacklistFunc type is an adapter to allow the use of ordinary
+// function as Blacklist mutator.
+type BlacklistFunc func(context.Context, *ent.BlacklistMutation) (ent.Value, error)
 
 // Mutate calls f(ctx, m).
-func (f BlocklistFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) {
-	if mv, ok := m.(*ent.BlocklistMutation); ok {
+func (f BlacklistFunc) Mutate(ctx context.Context, m ent.Mutation) (ent.Value, error) {
+	if mv, ok := m.(*ent.BlacklistMutation); ok {
 		return f(ctx, mv)
 	}
-	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.BlocklistMutation", m)
+	return nil, fmt.Errorf("unexpected mutation type %T. expect *ent.BlacklistMutation", m)
 }
 
 // The DownloadClientsFunc type is an adapter to allow the use of ordinary

ent/media.go

@@ -49,6 +49,8 @@ type Media struct {
 	Limiter schema.MediaLimiter `json:"limiter,omitempty"`
 	// Extras holds the value of the "extras" field.
 	Extras schema.MediaExtras `json:"extras,omitempty"`
+	// AlternativeTitles holds the value of the "alternative_titles" field.
+	AlternativeTitles []schema.AlternativeTilte `json:"alternative_titles,omitempty"`
 	// Edges holds the relations/edges for other nodes in the graph.
 	// The values are being populated by the MediaQuery when eager-loading is set.
 	Edges        MediaEdges `json:"edges"`
@@ -78,7 +80,7 @@ func (*Media) scanValues(columns []string) ([]any, error) {
 	values := make([]any, len(columns))
 	for i := range columns {
 		switch columns[i] {
-		case media.FieldLimiter, media.FieldExtras:
+		case media.FieldLimiter, media.FieldExtras, media.FieldAlternativeTitles:
 			values[i] = new([]byte)
 		case media.FieldDownloadHistoryEpisodes:
 			values[i] = new(sql.NullBool)
@@ -203,6 +205,14 @@ func (m *Media) assignValues(columns []string, values []any) error {
 					return fmt.Errorf("unmarshal field extras: %w", err)
 				}
 			}
+		case media.FieldAlternativeTitles:
+			if value, ok := values[i].(*[]byte); !ok {
+				return fmt.Errorf("unexpected type %T for field alternative_titles", values[i])
+			} else if value != nil && len(*value) > 0 {
+				if err := json.Unmarshal(*value, &m.AlternativeTitles); err != nil {
+					return fmt.Errorf("unmarshal field alternative_titles: %w", err)
+				}
+			}
 		default:
 			m.selectValues.Set(columns[i], values[i])
 		}
@@ -288,6 +298,9 @@ func (m *Media) String() string {
 	builder.WriteString(", ")
 	builder.WriteString("extras=")
 	builder.WriteString(fmt.Sprintf("%v", m.Extras))
+	builder.WriteString(", ")
+	builder.WriteString("alternative_titles=")
+	builder.WriteString(fmt.Sprintf("%v", m.AlternativeTitles))
 	builder.WriteByte(')')
 	return builder.String()
 }

ent/media/media.go

@@ -45,6 +45,8 @@ const (
 	FieldLimiter = "limiter"
 	// FieldExtras holds the string denoting the extras field in the database.
 	FieldExtras = "extras"
+	// FieldAlternativeTitles holds the string denoting the alternative_titles field in the database.
+	FieldAlternativeTitles = "alternative_titles"
 	// EdgeEpisodes holds the string denoting the episodes edge name in mutations.
 	EdgeEpisodes = "episodes"
 	// Table holds the table name of the media in the database.
@@ -76,6 +78,7 @@ var Columns = []string{
 	FieldDownloadHistoryEpisodes,
 	FieldLimiter,
 	FieldExtras,
+	FieldAlternativeTitles,
 }
 
 // ValidColumn reports if the column name is valid (part of the table columns).

ent/media/where.go

@@ -795,6 +795,16 @@ func ExtrasNotNil() predicate.Media {
 	return predicate.Media(sql.FieldNotNull(FieldExtras))
 }
 
+// AlternativeTitlesIsNil applies the IsNil predicate on the "alternative_titles" field.
+func AlternativeTitlesIsNil() predicate.Media {
+	return predicate.Media(sql.FieldIsNull(FieldAlternativeTitles))
+}
+
+// AlternativeTitlesNotNil applies the NotNil predicate on the "alternative_titles" field.
+func AlternativeTitlesNotNil() predicate.Media {
+	return predicate.Media(sql.FieldNotNull(FieldAlternativeTitles))
+}
+
 // HasEpisodes applies the HasEdge predicate on the "episodes" edge.
 func HasEpisodes() predicate.Media {
 	return predicate.Media(func(s *sql.Selector) {

ent/media_create.go

@@ -184,6 +184,12 @@ func (mc *MediaCreate) SetNillableExtras(se *schema.MediaExtras) *MediaCreate {
 	return mc
 }
 
+// SetAlternativeTitles sets the "alternative_titles" field.
+func (mc *MediaCreate) SetAlternativeTitles(st []schema.AlternativeTilte) *MediaCreate {
+	mc.mutation.SetAlternativeTitles(st)
+	return mc
+}
+
 // AddEpisodeIDs adds the "episodes" edge to the Episode entity by IDs.
 func (mc *MediaCreate) AddEpisodeIDs(ids ...int) *MediaCreate {
 	mc.mutation.AddEpisodeIDs(ids...)
@@ -377,6 +383,10 @@ func (mc *MediaCreate) createSpec() (*Media, *sqlgraph.CreateSpec) {
 		_spec.SetField(media.FieldExtras, field.TypeJSON, value)
 		_node.Extras = value
 	}
+	if value, ok := mc.mutation.AlternativeTitles(); ok {
+		_spec.SetField(media.FieldAlternativeTitles, field.TypeJSON, value)
+		_node.AlternativeTitles = value
+	}
 	if nodes := mc.mutation.EpisodesIDs(); len(nodes) > 0 {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,

ent/media_update.go

@@ -14,6 +14,7 @@ import (
 
 	"entgo.io/ent/dialect/sql"
 	"entgo.io/ent/dialect/sql/sqlgraph"
+	"entgo.io/ent/dialect/sql/sqljson"
 	"entgo.io/ent/schema/field"
 )
 
@@ -290,6 +291,24 @@ func (mu *MediaUpdate) ClearExtras() *MediaUpdate {
 	return mu
 }
 
+// SetAlternativeTitles sets the "alternative_titles" field.
+func (mu *MediaUpdate) SetAlternativeTitles(st []schema.AlternativeTilte) *MediaUpdate {
+	mu.mutation.SetAlternativeTitles(st)
+	return mu
+}
+
+// AppendAlternativeTitles appends st to the "alternative_titles" field.
+func (mu *MediaUpdate) AppendAlternativeTitles(st []schema.AlternativeTilte) *MediaUpdate {
+	mu.mutation.AppendAlternativeTitles(st)
+	return mu
+}
+
+// ClearAlternativeTitles clears the value of the "alternative_titles" field.
+func (mu *MediaUpdate) ClearAlternativeTitles() *MediaUpdate {
+	mu.mutation.ClearAlternativeTitles()
+	return mu
+}
+
 // AddEpisodeIDs adds the "episodes" edge to the Episode entity by IDs.
 func (mu *MediaUpdate) AddEpisodeIDs(ids ...int) *MediaUpdate {
 	mu.mutation.AddEpisodeIDs(ids...)
@@ -454,6 +473,17 @@ func (mu *MediaUpdate) sqlSave(ctx context.Context) (n int, err error) {
 	if mu.mutation.ExtrasCleared() {
 		_spec.ClearField(media.FieldExtras, field.TypeJSON)
 	}
+	if value, ok := mu.mutation.AlternativeTitles(); ok {
+		_spec.SetField(media.FieldAlternativeTitles, field.TypeJSON, value)
+	}
+	if value, ok := mu.mutation.AppendedAlternativeTitles(); ok {
+		_spec.AddModifier(func(u *sql.UpdateBuilder) {
+			sqljson.Append(u, media.FieldAlternativeTitles, value)
+		})
+	}
+	if mu.mutation.AlternativeTitlesCleared() {
+		_spec.ClearField(media.FieldAlternativeTitles, field.TypeJSON)
+	}
 	if mu.mutation.EpisodesCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,
@@ -779,6 +809,24 @@ func (muo *MediaUpdateOne) ClearExtras() *MediaUpdateOne {
 	return muo
 }
 
+// SetAlternativeTitles sets the "alternative_titles" field.
+func (muo *MediaUpdateOne) SetAlternativeTitles(st []schema.AlternativeTilte) *MediaUpdateOne {
+	muo.mutation.SetAlternativeTitles(st)
+	return muo
+}
+
+// AppendAlternativeTitles appends st to the "alternative_titles" field.
+func (muo *MediaUpdateOne) AppendAlternativeTitles(st []schema.AlternativeTilte) *MediaUpdateOne {
+	muo.mutation.AppendAlternativeTitles(st)
+	return muo
+}
+
+// ClearAlternativeTitles clears the value of the "alternative_titles" field.
+func (muo *MediaUpdateOne) ClearAlternativeTitles() *MediaUpdateOne {
+	muo.mutation.ClearAlternativeTitles()
+	return muo
+}
+
 // AddEpisodeIDs adds the "episodes" edge to the Episode entity by IDs.
 func (muo *MediaUpdateOne) AddEpisodeIDs(ids ...int) *MediaUpdateOne {
 	muo.mutation.AddEpisodeIDs(ids...)
@@ -973,6 +1021,17 @@ func (muo *MediaUpdateOne) sqlSave(ctx context.Context) (_node *Media, err error
 	if muo.mutation.ExtrasCleared() {
 		_spec.ClearField(media.FieldExtras, field.TypeJSON)
 	}
+	if value, ok := muo.mutation.AlternativeTitles(); ok {
+		_spec.SetField(media.FieldAlternativeTitles, field.TypeJSON, value)
+	}
+	if value, ok := muo.mutation.AppendedAlternativeTitles(); ok {
+		_spec.AddModifier(func(u *sql.UpdateBuilder) {
+			sqljson.Append(u, media.FieldAlternativeTitles, value)
+		})
+	}
+	if muo.mutation.AlternativeTitlesCleared() {
+		_spec.ClearField(media.FieldAlternativeTitles, field.TypeJSON)
+	}
 	if muo.mutation.EpisodesCleared() {
 		edge := &sqlgraph.EdgeSpec{
 			Rel:     sqlgraph.O2M,

ent/migrate/schema.go

@@ -8,17 +8,18 @@ import (
 )
 
 var (
-	// BlocklistsColumns holds the columns for the "blocklists" table.
-	BlocklistsColumns = []*schema.Column{
+	// BlacklistsColumns holds the columns for the "blacklists" table.
+	BlacklistsColumns = []*schema.Column{
 		{Name: "id", Type: field.TypeInt, Increment: true},
 		{Name: "type", Type: field.TypeEnum, Enums: []string{"media", "torrent"}},
-		{Name: "value", Type: field.TypeString},
+		{Name: "value", Type: field.TypeJSON},
+		{Name: "notes", Type: field.TypeString, Nullable: true},
 	}
-	// BlocklistsTable holds the schema information for the "blocklists" table.
-	BlocklistsTable = &schema.Table{
-		Name:       "blocklists",
-		Columns:    BlocklistsColumns,
-		PrimaryKey: []*schema.Column{BlocklistsColumns[0]},
+	// BlacklistsTable holds the schema information for the "blacklists" table.
+	BlacklistsTable = &schema.Table{
+		Name:       "blacklists",
+		Columns:    BlacklistsColumns,
+		PrimaryKey: []*schema.Column{BlacklistsColumns[0]},
 	}
 	// DownloadClientsColumns holds the columns for the "download_clients" table.
 	DownloadClientsColumns = []*schema.Column{
@@ -73,12 +74,15 @@ var (
 		{Name: "id", Type: field.TypeInt, Increment: true},
 		{Name: "media_id", Type: field.TypeInt},
 		{Name: "episode_id", Type: field.TypeInt, Nullable: true},
+		{Name: "episode_nums", Type: field.TypeJSON, Nullable: true},
+		{Name: "season_num", Type: field.TypeInt, Nullable: true},
 		{Name: "source_title", Type: field.TypeString},
 		{Name: "date", Type: field.TypeTime},
 		{Name: "target_dir", Type: field.TypeString},
 		{Name: "size", Type: field.TypeInt, Default: 0},
 		{Name: "download_client_id", Type: field.TypeInt, Nullable: true},
 		{Name: "indexer_id", Type: field.TypeInt, Nullable: true},
+		{Name: "link", Type: field.TypeString, Nullable: true},
 		{Name: "status", Type: field.TypeEnum, Enums: []string{"running", "success", "fail", "uploading", "seeding"}},
 		{Name: "saved", Type: field.TypeString, Nullable: true},
 	}
@@ -139,6 +143,7 @@ var (
 		{Name: "download_history_episodes", Type: field.TypeBool, Nullable: true, Default: false},
 		{Name: "limiter", Type: field.TypeJSON, Nullable: true},
 		{Name: "extras", Type: field.TypeJSON, Nullable: true},
+		{Name: "alternative_titles", Type: field.TypeJSON, Nullable: true},
 	}
 	// MediaTable holds the schema information for the "media" table.
 	MediaTable = &schema.Table{
@@ -176,7 +181,7 @@ var (
 	StoragesColumns = []*schema.Column{
 		{Name: "id", Type: field.TypeInt, Increment: true},
 		{Name: "name", Type: field.TypeString, Unique: true},
-		{Name: "implementation", Type: field.TypeEnum, Enums: []string{"webdav", "local"}},
+		{Name: "implementation", Type: field.TypeEnum, Enums: []string{"webdav", "local", "alist"}},
 		{Name: "tv_path", Type: field.TypeString, Nullable: true},
 		{Name: "movie_path", Type: field.TypeString, Nullable: true},
 		{Name: "settings", Type: field.TypeString, Nullable: true},
@@ -191,7 +196,7 @@ var (
 	}
 	// Tables holds all the tables in the schema.
 	Tables = []*schema.Table{
-		BlocklistsTable,
+		BlacklistsTable,
 		DownloadClientsTable,
 		EpisodesTable,
 		HistoriesTable,

ent/predicate/predicate.go

@@ -6,8 +6,8 @@ import (
 	"entgo.io/ent/dialect/sql"
 )
 
-// Blocklist is the predicate function for blocklist builders.
-type Blocklist func(*sql.Selector)
+// Blacklist is the predicate function for blacklist builders.
+type Blacklist func(*sql.Selector)
 
 // DownloadClients is the predicate function for downloadclients builders.
 type DownloadClients func(*sql.Selector)

ent/runtime.go

@@ -3,6 +3,7 @@
 package ent
 
 import (
+	"polaris/ent/blacklist"
 	"polaris/ent/downloadclients"
 	"polaris/ent/episode"
 	"polaris/ent/history"
@@ -18,6 +19,12 @@ import (
 // (default values, validators, hooks and policies) and stitches it
 // to their package variables.
 func init() {
+	blacklistFields := schema.Blacklist{}.Fields()
+	_ = blacklistFields
+	// blacklistDescValue is the schema descriptor for value field.
+	blacklistDescValue := blacklistFields[1].Descriptor()
+	// blacklist.DefaultValue holds the default value on creation for the value field.
+	blacklist.DefaultValue = blacklistDescValue.Default.(schema.BlacklistValue)
 	downloadclientsFields := schema.DownloadClients{}.Fields()
 	_ = downloadclientsFields
 	// downloadclientsDescUser is the schema descriptor for user field.
@@ -59,7 +66,7 @@ func init() {
 	historyFields := schema.History{}.Fields()
 	_ = historyFields
 	// historyDescSize is the schema descriptor for size field.
-	historyDescSize := historyFields[5].Descriptor()
+	historyDescSize := historyFields[7].Descriptor()
 	// history.DefaultSize holds the default value on creation for the size field.
 	history.DefaultSize = historyDescSize.Default.(int)
 	indexersFields := schema.Indexers{}.Fields()

ent/schema/blacklist.go

@@ -0,0 +1,30 @@
+package schema
+
+import (
+	"entgo.io/ent"
+	"entgo.io/ent/schema/field"
+)
+
+// Blacklist holds the schema definition for the Blacklist entity.
+type Blacklist struct {
+	ent.Schema
+}
+
+// Fields of the Blacklist.
+func (Blacklist) Fields() []ent.Field {
+	return []ent.Field{
+		field.Enum("type").Values("media", "torrent"),
+		field.JSON("value", BlacklistValue{}).Default(BlacklistValue{}),
+		field.String("notes").Optional(),
+	}
+}
+
+// Edges of the Blacklist.
+func (Blacklist) Edges() []ent.Edge {
+	return nil
+}
+
+type BlacklistValue struct {
+	TmdbID      int    `json:"tmdb_id"`
+	TorrentHash string `json:"torrent_hash"`
+}

ent/schema/blocklist.go

@@ -1,24 +0,0 @@
-package schema
-
-import (
-	"entgo.io/ent"
-	"entgo.io/ent/schema/field"
-)
-
-// Blocklist holds the schema definition for the Blocklist entity.
-type Blocklist struct {
-	ent.Schema
-}
-
-// Fields of the Blocklist.
-func (Blocklist) Fields() []ent.Field {
-	return  []ent.Field{
-		field.Enum("type").Values("media", "torrent"),
-		field.String("value"),
-	}
-}
-
-// Edges of the Blocklist.
-func (Blocklist) Edges() []ent.Edge {
-	return nil
-}

ent/schema/history.go

@@ -14,15 +14,18 @@ type History struct {
 func (History) Fields() []ent.Field {
 	return []ent.Field{
 		field.Int("media_id"),
-		field.Int("episode_id").Optional(),
+		field.Int("episode_id").Optional().Comment("deprecated"),
+		field.Ints("episode_nums").Optional(),
+		field.Int("season_num").Optional(),
 		field.String("source_title"),
 		field.Time("date"),
 		field.String("target_dir"),
 		field.Int("size").Default(0),
 		field.Int("download_client_id").Optional(),
 		field.Int("indexer_id").Optional(),
+		field.String("link").Optional(), //should be magnet link
 		field.Enum("status").Values("running", "success", "fail", "uploading", "seeding"),
-		field.String("saved").Optional(),
+		field.String("saved").Optional().Comment("deprecated"), //deprecated
 	}
 }
 

ent/schema/media.go

@@ -31,6 +31,7 @@ func (Media) Fields() []ent.Field {
 		field.Bool("download_history_episodes").Optional().Default(false).Comment("tv series only"),
 		field.JSON("limiter", MediaLimiter{}).Optional(),
 		field.JSON("extras", MediaExtras{}).Optional(),
+		field.JSON("alternative_titles", []AlternativeTilte{}).Optional(),
 	}
 }
 
@@ -41,9 +42,16 @@ func (Media) Edges() []ent.Edge {
 	}
 }
 
+type AlternativeTilte struct {
+	Iso3166_1 string `json:"iso_3166_1"`
+	Title     string `json:"title"`
+	Type      string `json:"type"`
+}
+
 type MediaLimiter struct {
-	SizeMin int `json:"size_min"` //in B
-	SizeMax int `json:"size_max"` //in B
+	SizeMin    int64 `json:"size_min"` //in B
+	SizeMax    int64 `json:"size_max"` //in B
+	PreferSize int64 `json:"prefer_max"`
 }
 
 type MediaExtras struct {
@@ -51,7 +59,7 @@ type MediaExtras struct {
 	JavId        string `json:"javid"`
 	//OriginCountry    []string `json:"origin_country"`
 	OriginalLanguage string `json:"original_language"`
-	Genres []struct {
+	Genres           []struct {
 		ID   int64  `json:"id"`
 		Name string `json:"name"`
 	} `json:"genres"`

ent/schema/storage.go

@@ -14,7 +14,7 @@ type Storage struct {
 func (Storage) Fields() []ent.Field {
 	return []ent.Field{
 		field.String("name").Unique(),
-		field.Enum("implementation").Values("webdav", "local"),
+		field.Enum("implementation").Values("webdav", "local", "alist"),
 		field.String("tv_path").Optional(),
 		field.String("movie_path").Optional(),
 		field.String("settings").Optional(),

ent/storage/storage.go

@@ -67,6 +67,7 @@ type Implementation string
 const (
 	ImplementationWebdav Implementation = "webdav"
 	ImplementationLocal  Implementation = "local"
+	ImplementationAlist  Implementation = "alist"
 )
 
 func (i Implementation) String() string {
@@ -76,7 +77,7 @@ func (i Implementation) String() string {
 // ImplementationValidator is a validator for the "implementation" field enum values. It is called by the builders before save.
 func ImplementationValidator(i Implementation) error {
 	switch i {
-	case ImplementationWebdav, ImplementationLocal:
+	case ImplementationWebdav, ImplementationLocal, ImplementationAlist:
 		return nil
 	default:
 		return fmt.Errorf("storage: invalid enum value for implementation field: %q", i)

ent/tx.go

@@ -12,8 +12,8 @@ import (
 // Tx is a transactional client that is created by calling Client.Tx().
 type Tx struct {
 	config
-	// Blocklist is the client for interacting with the Blocklist builders.
-	Blocklist *BlocklistClient
+	// Blacklist is the client for interacting with the Blacklist builders.
+	Blacklist *BlacklistClient
 	// DownloadClients is the client for interacting with the DownloadClients builders.
 	DownloadClients *DownloadClientsClient
 	// Episode is the client for interacting with the Episode builders.
@@ -163,7 +163,7 @@ func (tx *Tx) Client() *Client {
 }
 
 func (tx *Tx) init() {
-	tx.Blocklist = NewBlocklistClient(tx.config)
+	tx.Blacklist = NewBlacklistClient(tx.config)
 	tx.DownloadClients = NewDownloadClientsClient(tx.config)
 	tx.Episode = NewEpisodeClient(tx.config)
 	tx.History = NewHistoryClient(tx.config)
@@ -182,7 +182,7 @@ func (tx *Tx) init() {
 // of them in order to commit or rollback the transaction.
 //
 // If a closed transaction is embedded in one of the generated entities, and the entity
-// applies a query, for example: Blocklist.QueryXXX(), the query will be executed
+// applies a query, for example: Blacklist.QueryXXX(), the query will be executed
 // through the driver which created this transaction.
 //
 // Note that txDriver is not goroutine safe.

entrypoint.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+groupadd -g ${PGID} abc1
+useradd abc1 -u ${PUID} -g ${PGID} -m -s /bin/bash
+
+## 重设权限
+chown -R "${PUID}:${PGID}" /app/data
+
+umask ${UMASK:-022}
+
+cd /app
+exec gosu "${PUID}:${PGID}" /app/polaris

go.mod

@@ -1,6 +1,8 @@
 module polaris
 
-go 1.22.4
+go 1.23
+
+toolchain go1.23.1
 
 require (
 	entgo.io/ent v0.13.1
@@ -8,34 +10,74 @@ require (
 	github.com/mattn/go-sqlite3 v1.14.22 // indirect
 	github.com/robfig/cron v1.2.0
 	go.uber.org/zap v1.27.0
-	golang.org/x/net v0.27.0
+	golang.org/x/net v0.33.0
 )
 
 require (
 	github.com/PuerkitoBio/goquery v1.9.2
+	github.com/anacrolix/torrent v1.57.1
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc
 	github.com/gin-contrib/zap v1.1.3
 	github.com/ncruces/go-sqlite3 v0.18.4
 	github.com/nikoksr/notify v1.0.0
 	github.com/stretchr/testify v1.9.0
+	golift.io/starr v1.0.0
 )
 
 require (
+	cloud.google.com/go v0.115.0 // indirect
+	cloud.google.com/go/ai v0.8.0 // indirect
+	cloud.google.com/go/auth v0.7.1 // indirect
+	cloud.google.com/go/auth/oauth2adapt v0.2.3 // indirect
+	cloud.google.com/go/compute/metadata v0.5.0 // indirect
+	cloud.google.com/go/longrunning v0.5.10 // indirect
 	github.com/BurntSushi/toml v1.4.0 // indirect
 	github.com/DATA-DOG/go-sqlmock v1.5.2 // indirect
+	github.com/anacrolix/generics v0.0.3-0.20240902042256-7fb2702ef0ca // indirect
+	github.com/anacrolix/missinggo v1.3.0 // indirect
+	github.com/anacrolix/missinggo/v2 v2.7.4 // indirect
 	github.com/andybalholm/cascadia v1.3.2 // indirect
 	github.com/blinkbean/dingtalk v1.1.3 // indirect
+	github.com/bradfitz/iter v0.0.0-20191230175014-e8f45d346db8 // indirect
+	github.com/felixge/httpsnoop v1.0.4 // indirect
+	github.com/go-logr/logr v1.4.2 // indirect
+	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible // indirect
 	github.com/go-test/deep v1.0.4 // indirect
+	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
+	github.com/golang/protobuf v1.5.4 // indirect
+	github.com/google/generative-ai-go v0.19.0 // indirect
+	github.com/google/s2a-go v0.1.7 // indirect
+	github.com/googleapis/enterprise-certificate-proxy v0.3.2 // indirect
+	github.com/googleapis/gax-go/v2 v2.12.5 // indirect
 	github.com/gregdel/pushover v1.3.1 // indirect
+	github.com/huandu/xstrings v1.3.2 // indirect
+	github.com/minio/sha256-simd v1.0.0 // indirect
+	github.com/mr-tron/base58 v1.2.0 // indirect
+	github.com/multiformats/go-multihash v0.2.3 // indirect
+	github.com/multiformats/go-varint v0.0.6 // indirect
 	github.com/ncruces/julianday v1.0.0 // indirect
 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
+	github.com/spaolacci/murmur3 v1.1.0 // indirect
 	github.com/stretchr/objx v0.5.2 // indirect
 	github.com/technoweenie/multipartstreamer v1.0.1 // indirect
 	github.com/tetratelabs/wazero v1.8.0 // indirect
-	golang.org/x/sync v0.8.0 // indirect
+	go.opencensus.io v0.24.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.53.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.53.0 // indirect
+	go.opentelemetry.io/otel v1.28.0 // indirect
+	go.opentelemetry.io/otel/metric v1.28.0 // indirect
+	go.opentelemetry.io/otel/trace v1.28.0 // indirect
+	golang.org/x/oauth2 v0.21.0 // indirect
+	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/time v0.5.0 // indirect
+	google.golang.org/api v0.188.0 // indirect
+	google.golang.org/genproto/googleapis/api v0.0.0-20240711142825-46eb208f015d // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20240711142825-46eb208f015d // indirect
+	google.golang.org/grpc v1.65.0 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
+	lukechampine.com/blake3 v1.1.6 // indirect
 )
 
 require (
@@ -80,11 +122,11 @@ require (
 	github.com/ugorji/go/codec v1.2.12 // indirect
 	github.com/zclconf/go-cty v1.8.0 // indirect
 	golang.org/x/arch v0.8.0 // indirect
-	golang.org/x/crypto v0.27.0
-	golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56
-	golang.org/x/mod v0.19.0 // indirect
-	golang.org/x/sys v0.25.0
-	golang.org/x/text v0.18.0 // indirect
+	golang.org/x/crypto v0.31.0
+	golang.org/x/exp v0.0.0-20240823005443-9b4947da3948
+	golang.org/x/mod v0.20.0 // indirect
+	golang.org/x/sys v0.28.0
+	golang.org/x/text v0.21.0 // indirect
 	google.golang.org/protobuf v1.34.2 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect

go.sum

@@ -1,37 +1,115 @@
 ariga.io/atlas v0.19.1-0.20240203083654-5948b60a8e43 h1:GwdJbXydHCYPedeeLt4x/lrlIISQ4JTH1mRWuE5ZZ14=
 ariga.io/atlas v0.19.1-0.20240203083654-5948b60a8e43/go.mod h1:uj3pm+hUTVN/X5yfdBexHlZv+1Xu5u5ZbZx7+CDavNU=
+cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
+cloud.google.com/go v0.34.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
+cloud.google.com/go v0.115.0 h1:CnFSK6Xo3lDYRoBKEcAtia6VSC837/ZkJuRduSFnr14=
+cloud.google.com/go v0.115.0/go.mod h1:8jIM5vVgoAEoiVxQ/O4BFTfHqulPZgs/ufEzMcFMdWU=
+cloud.google.com/go/ai v0.8.0 h1:rXUEz8Wp2OlrM8r1bfmpF2+VKqc1VJpafE3HgzRnD/w=
+cloud.google.com/go/ai v0.8.0/go.mod h1:t3Dfk4cM61sytiggo2UyGsDVW3RF1qGZaUKDrZFyqkE=
+cloud.google.com/go/auth v0.7.1 h1:Iv1bbpzJ2OIg16m94XI9/tlzZZl3cdeR3nGVGj78N7s=
+cloud.google.com/go/auth v0.7.1/go.mod h1:VEc4p5NNxycWQTMQEDQF0bd6aTMb6VgYDXEwiJJQAbs=
+cloud.google.com/go/auth/oauth2adapt v0.2.3 h1:MlxF+Pd3OmSudg/b1yZ5lJwoXCEaeedAguodky1PcKI=
+cloud.google.com/go/auth/oauth2adapt v0.2.3/go.mod h1:tMQXOfZzFuNuUxOypHlQEXgdfX5cuhwU+ffUuXRJE8I=
+cloud.google.com/go/compute v1.24.0 h1:phWcR2eWzRJaL/kOiJwfFsPs4BaKq1j6vnpZrc1YlVg=
+cloud.google.com/go/compute/metadata v0.5.0 h1:Zr0eK8JbFv6+Wi4ilXAR8FJ3wyNdpxHKJNPos6LTZOY=
+cloud.google.com/go/compute/metadata v0.5.0/go.mod h1:aHnloV2TPI38yx4s9+wAZhHykWvVCfu7hQbF+9CWoiY=
+cloud.google.com/go/longrunning v0.5.10 h1:eB/BniENNRKhjz/xgiillrdcH3G74TGSl3BXinGlI7E=
+cloud.google.com/go/longrunning v0.5.10/go.mod h1:tljz5guTr5oc/qhlUjBlk7UAIFMOGuPNxkNDZXlLics=
+crawshaw.io/iox v0.0.0-20181124134642-c51c3df30797/go.mod h1:sXBiorCo8c46JlQV3oXPKINnZ8mcqnye1EkVkqsectk=
+crawshaw.io/sqlite v0.3.2/go.mod h1:igAO5JulrQ1DbdZdtVq48mnZUBAPOeFzer7VhDWNtW4=
 entgo.io/ent v0.13.1 h1:uD8QwN1h6SNphdCCzmkMN3feSUzNnVvV/WIkHKMbzOE=
 entgo.io/ent v0.13.1/go.mod h1:qCEmo+biw3ccBn9OyL4ZK5dfpwg++l1Gxwac5B1206A=
+github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/BurntSushi/toml v1.4.0 h1:kuoIxZQy2WRRk1pttg9asf+WVv6tWQuBNVmK8+nqPr0=
 github.com/BurntSushi/toml v1.4.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
 github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU=
 github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU=
 github.com/PuerkitoBio/goquery v1.9.2 h1:4/wZksC3KgkQw7SQgkKotmKljk0M6V8TUvA8Wb4yPeE=
 github.com/PuerkitoBio/goquery v1.9.2/go.mod h1:GHPCaP0ODyyxqcNoFGYlAprUFH81NuRPd0GX3Zu2Mvk=
+github.com/RoaringBitmap/roaring v0.4.7/go.mod h1:8khRDP4HmeXns4xIj9oGrKSz7XTQiJx2zgh7AcNke4w=
+github.com/RoaringBitmap/roaring v0.4.17/go.mod h1:D3qVegWTmfCaX4Bl5CrBE9hfrSrrXIr8KVNvRsDi1NI=
+github.com/RoaringBitmap/roaring v0.4.23/go.mod h1:D0gp8kJQgE1A4LQ5wFLggQEyvDi06Mq5mKs52e1TwOo=
+github.com/Shopify/sarama v1.19.0/go.mod h1:FVkBWblsNy7DGZRfXLU0O9RCGt5g3g3yEuWXgklEdEo=
+github.com/Shopify/toxiproxy v2.1.4+incompatible/go.mod h1:OXgGpZ6Cli1/URJOF1DMxUHB2q5Ap20/P/eIdh4G0pI=
 github.com/agext/levenshtein v1.2.1 h1:QmvMAjj2aEICytGiWzmxoE0x2KZvE0fvmqMOfy2tjT8=
 github.com/agext/levenshtein v1.2.1/go.mod h1:JEDfjyjHDjOF/1e4FlBE/PkbqA9OfWu2ki2W0IB5558=
+github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
+github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
+github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
+github.com/alecthomas/units v0.0.0-20190717042225-c3de453c63f4/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
+github.com/anacrolix/dht/v2 v2.19.2-0.20221121215055-066ad8494444 h1:8V0K09lrGoeT2KRJNOtspA7q+OMxGwQqK/Ug0IiaaRE=
+github.com/anacrolix/dht/v2 v2.19.2-0.20221121215055-066ad8494444/go.mod h1:MctKM1HS5YYDb3F30NGJxLE+QPuqWoT5ReW/4jt8xew=
+github.com/anacrolix/envpprof v0.0.0-20180404065416-323002cec2fa/go.mod h1:KgHhUaQMc8cC0+cEflSgCFNFbKwi5h54gqtVn8yhP7c=
+github.com/anacrolix/envpprof v1.0.0/go.mod h1:KgHhUaQMc8cC0+cEflSgCFNFbKwi5h54gqtVn8yhP7c=
+github.com/anacrolix/envpprof v1.1.0/go.mod h1:My7T5oSqVfEn4MD4Meczkw/f5lSIndGAKu/0SM/rkf4=
+github.com/anacrolix/generics v0.0.3-0.20240902042256-7fb2702ef0ca h1:aiiGqSQWjtVNdi8zUMfA//IrM8fPkv2bWwZVPbDe0wg=
+github.com/anacrolix/generics v0.0.3-0.20240902042256-7fb2702ef0ca/go.mod h1:MN3ve08Z3zSV/rTuX/ouI4lNdlfTxgdafQJiLzyNRB8=
+github.com/anacrolix/log v0.3.0/go.mod h1:lWvLTqzAnCWPJA08T2HCstZi0L1y2Wyvm3FJgwU9jwU=
+github.com/anacrolix/log v0.6.0/go.mod h1:lWvLTqzAnCWPJA08T2HCstZi0L1y2Wyvm3FJgwU9jwU=
+github.com/anacrolix/missinggo v1.1.0/go.mod h1:MBJu3Sk/k3ZfGYcS7z18gwfu72Ey/xopPFJJbTi5yIo=
+github.com/anacrolix/missinggo v1.1.2-0.20190815015349-b888af804467/go.mod h1:MBJu3Sk/k3ZfGYcS7z18gwfu72Ey/xopPFJJbTi5yIo=
+github.com/anacrolix/missinggo v1.2.1/go.mod h1:J5cMhif8jPmFoC3+Uvob3OXXNIhOUikzMt+uUjeM21Y=
+github.com/anacrolix/missinggo v1.3.0 h1:06HlMsudotL7BAELRZs0yDZ4yVXsHXGi323QBjAVASw=
+github.com/anacrolix/missinggo v1.3.0/go.mod h1:bqHm8cE8xr+15uVfMG3BFui/TxyB6//H5fwlq/TeqMc=
+github.com/anacrolix/missinggo/perf v1.0.0/go.mod h1:ljAFWkBuzkO12MQclXzZrosP5urunoLS0Cbvb4V0uMQ=
+github.com/anacrolix/missinggo/v2 v2.2.0/go.mod h1:o0jgJoYOyaoYQ4E2ZMISVa9c88BbUBVQQW4QeRkNCGY=
+github.com/anacrolix/missinggo/v2 v2.5.1/go.mod h1:WEjqh2rmKECd0t1VhQkLGTdIWXO6f6NLjp5GlMZ+6FA=
+github.com/anacrolix/missinggo/v2 v2.7.4 h1:47h5OXoPV8JbA/ACA+FLwKdYbAinuDO8osc2Cu9xkxg=
+github.com/anacrolix/missinggo/v2 v2.7.4/go.mod h1:vVO5FEziQm+NFmJesc7StpkquZk+WJFCaL0Wp//2sa0=
+github.com/anacrolix/stm v0.2.0/go.mod h1:zoVQRvSiGjGoTmbM0vSLIiaKjWtNPeTvXUSdJQA4hsg=
+github.com/anacrolix/tagflag v0.0.0-20180109131632-2146c8d41bf0/go.mod h1:1m2U/K6ZT+JZG0+bdMK6qauP49QT4wE5pmhJXOKKCHw=
+github.com/anacrolix/tagflag v1.0.0/go.mod h1:1m2U/K6ZT+JZG0+bdMK6qauP49QT4wE5pmhJXOKKCHw=
+github.com/anacrolix/tagflag v1.1.0/go.mod h1:Scxs9CV10NQatSmbyjqmqmeQNwGzlNe0CMUMIxqHIG8=
+github.com/anacrolix/torrent v1.57.1 h1:CS8rYfC2Oe15NPBhwCNs/3WBY6HiBCPDFpY+s9aFHbA=
+github.com/anacrolix/torrent v1.57.1/go.mod h1:NNBg4lP2/us9Hp5+cLNcZRILM69cNoKIkqMGqr9AuR0=
 github.com/andybalholm/cascadia v1.3.2 h1:3Xi6Dw5lHF15JtdcmAHD3i1+T8plmv7BQ/nsViSLyss=
 github.com/andybalholm/cascadia v1.3.2/go.mod h1:7gtRlve5FxPPgIgX36uWBX58OdBsSS6lUvCFb+h7KvU=
+github.com/apache/thrift v0.12.0/go.mod h1:cp2SuWMxlEZw2r+iP2GNCdIi4C1qmUzdZFSVb+bacwQ=
 github.com/apparentlymart/go-textseg/v13 v13.0.0 h1:Y+KvPE1NYz0xl601PVImeQfFyEy6iT90AvPUL1NNfNw=
 github.com/apparentlymart/go-textseg/v13 v13.0.0/go.mod h1:ZK2fH7c4NqDTLtiYLvIkEghdlcqw7yxLeM89kiTRPUo=
+github.com/benbjohnson/immutable v0.2.0/go.mod h1:uc6OHo6PN2++n98KHLxW8ef4W42ylHiQSENghE1ezxI=
+github.com/beorn7/perks v0.0.0-20180321164747-3a771d992973/go.mod h1:Dwedo/Wpr24TaqPxmxbtue+5NUziq4I4S80YR8gNf3Q=
+github.com/beorn7/perks v1.0.0/go.mod h1:KWe93zE9D1o94FZ5RNwFwVgaQK1VOXiVxmqh+CedLV8=
+github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
 github.com/blinkbean/dingtalk v1.1.3 h1:MbidFZYom7DTFHD/YIs+eaI7kRy52kmWE/sy0xjo6E4=
 github.com/blinkbean/dingtalk v1.1.3/go.mod h1:9BaLuGSBqY3vT5hstValh48DbsKO7vaHaJnG9pXwbto=
+github.com/bradfitz/iter v0.0.0-20140124041915-454541ec3da2/go.mod h1:PyRFw1Lt2wKX4ZVSQ2mk+PeDa1rxyObEDlApuIsUKuo=
+github.com/bradfitz/iter v0.0.0-20190303215204-33e6a9893b0c/go.mod h1:PyRFw1Lt2wKX4ZVSQ2mk+PeDa1rxyObEDlApuIsUKuo=
+github.com/bradfitz/iter v0.0.0-20191230175014-e8f45d346db8 h1:GKTyiRCL6zVf5wWaqKnf+7Qs6GbEPfd4iMOitWzXJx8=
+github.com/bradfitz/iter v0.0.0-20191230175014-e8f45d346db8/go.mod h1:spo1JLcs67NmW1aVLEgtA8Yy1elc+X8y5SRW1sFW4Og=
 github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=
 github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=
 github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=
 github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
+github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
+github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=
 github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
 github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
 github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
 github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
+github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=
 github.com/cyruzin/golang-tmdb v1.6.3 h1:TKK9h+uuwiDOaFlsVispG1KxqhsSM5Y4ZELnUF3GlqU=
 github.com/cyruzin/golang-tmdb v1.6.3/go.mod h1:ZSryJLCcY+9TiKU+LbouXKns++YBrM8Tizannr05c+I=
 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/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/docopt/docopt-go v0.0.0-20180111231733-ee0de3bc6815/go.mod h1:WwZ+bS3ebgob9U8Nd0kOddGdZWjyMGR8Wziv+TBNwSE=
+github.com/dustin/go-humanize v0.0.0-20180421182945-02af3965c54e/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
+github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
+github.com/eapache/go-resiliency v1.1.0/go.mod h1:kFI+JgMyC7bLPUVY133qvEBtVayf5mFgVsvEsIPBvNs=
+github.com/eapache/go-xerial-snappy v0.0.0-20180814174437-776d5712da21/go.mod h1:+020luEh2TKB4/GOp8oxxtq0Daoen/Cii55CzbTV6DU=
+github.com/eapache/queue v1.1.0/go.mod h1:6eCeP0CKFpHLu8blIFXhExK/dRa7WDZfr6jVFPTqq+I=
+github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
+github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
+github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=
+github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
+github.com/felixge/httpsnoop v1.0.4 h1:NFTV2Zj1bL4mc9sqWACXbQFVBBg2W3GPvqp8/ESS2Wg=
+github.com/felixge/httpsnoop v1.0.4/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
+github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
 github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=
 github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=
 github.com/gabriel-vasile/mimetype v1.4.4 h1:QjV6pZ7/XZ7ryI2KuyeEDE8wnh7fHP9YnQy+R0LnH8I=
@@ -44,6 +122,21 @@ github.com/gin-contrib/zap v1.1.3 h1:9e/U9fYd4/OBfmSEBs5hHZq114uACn7bpuzvCkcJySA
 github.com/gin-contrib/zap v1.1.3/go.mod h1:+BD/6NYZKJyUpqVoJEvgeq9GLz8pINEQvak9LHNOTSE=
 github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
 github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
+github.com/glycerine/go-unsnap-stream v0.0.0-20180323001048-9f0cb55181dd/go.mod h1:/20jfyN9Y5QPEAprSgKAUr+glWDY39ZiUEAYOEv5dsE=
+github.com/glycerine/go-unsnap-stream v0.0.0-20181221182339-f9677308dec2/go.mod h1:/20jfyN9Y5QPEAprSgKAUr+glWDY39ZiUEAYOEv5dsE=
+github.com/glycerine/go-unsnap-stream v0.0.0-20190901134440-81cf024a9e0a/go.mod h1:/20jfyN9Y5QPEAprSgKAUr+glWDY39ZiUEAYOEv5dsE=
+github.com/glycerine/goconvey v0.0.0-20180728074245-46e3a41ad493/go.mod h1:Ogl1Tioa0aV7gstGFO7KhffUsb9M4ydbEbbxpcEDc24=
+github.com/glycerine/goconvey v0.0.0-20190315024820-982ee783a72e/go.mod h1:Ogl1Tioa0aV7gstGFO7KhffUsb9M4ydbEbbxpcEDc24=
+github.com/glycerine/goconvey v0.0.0-20190410193231-58a59202ab31/go.mod h1:Ogl1Tioa0aV7gstGFO7KhffUsb9M4ydbEbbxpcEDc24=
+github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
+github.com/go-kit/kit v0.9.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
+github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=
+github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=
+github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A=
+github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
+github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
+github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
+github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
 github.com/go-openapi/inflect v0.19.0 h1:9jCH9scKIbHeV9m12SmPilScz6krDxKRasNNSNPXu/4=
 github.com/go-openapi/inflect v0.19.0/go.mod h1:lHpZVlpIQqLyKwJ4N+YSc9hchQy/i12fJykb83CRBH4=
 github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
@@ -54,26 +147,71 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=
 github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
+github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
 github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible h1:2cauKuaELYAEARXRkq2LrJ0yDDv1rW7+wrTEdVL3uaU=
 github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM=
 github.com/go-test/deep v1.0.4 h1:u2CU3YKy9I2pmu9pX0eq50wCgjfGIt539SqR7FbHiho=
 github.com/go-test/deep v1.0.4/go.mod h1:wGDj63lr65AM2AQyKZd/NYHGb0R+1RLqB8NKt3aSFNA=
 github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
 github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=
+github.com/gogo/protobuf v1.2.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=
 github.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk=
 github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
+github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
+github.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
+github.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
+github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=
+github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
+github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
+github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
+github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.3.4/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw=
+github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=
+github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=
+github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=
+github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=
+github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=
+github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=
+github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
+github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
+github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/google/btree v0.0.0-20180124185431-e89373fe6b4a/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
+github.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
+github.com/google/generative-ai-go v0.19.0 h1:R71szggh8wHMCUlEMsW2A/3T+5LdEIkiaHSYgSpUgdg=
+github.com/google/generative-ai-go v0.19.0/go.mod h1:JYolL13VG7j79kM5BtHz4qwONHkeJQzOCkKXnpqtS/E=
+github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
+github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
 github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
+github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/google/s2a-go v0.1.7 h1:60BLSyTrOV4/haCDW4zb1guZItoSq8foHCXrAnjBo/o=
+github.com/google/s2a-go v0.1.7/go.mod h1:50CgR4k1jNlWBu4UfS4AcfhVe1r6pdZPygJ3R8F0Qdw=
+github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/googleapis/enterprise-certificate-proxy v0.3.2 h1:Vie5ybvEvT75RniqhfFxPRy3Bf7vr3h0cechB90XaQs=
+github.com/googleapis/enterprise-certificate-proxy v0.3.2/go.mod h1:VLSiSSBs/ksPL8kq3OBOQ6WRI2QnaFynd1DCjZ62+V0=
+github.com/googleapis/gax-go/v2 v2.12.5 h1:8gw9KZK8TiVKB6q3zHY3SBzLnrGp6HQjyfYBYGmXdxA=
+github.com/googleapis/gax-go/v2 v2.12.5/go.mod h1:BUDKcWo+RaKq5SC9vVYL0wLADa3VcfswbOMMRmB9H3E=
+github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
+github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
+github.com/gopherjs/gopherjs v0.0.0-20190309154008-847fc94819f9/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
+github.com/gopherjs/gopherjs v0.0.0-20190910122728-9d188e94fb99/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
+github.com/gorilla/context v1.1.1/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
+github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
 github.com/gregdel/pushover v1.3.1 h1:4bMLITOZ15+Zpi6qqoGqOPuVHCwSUvMCgVnN5Xhilfo=
 github.com/gregdel/pushover v1.3.1/go.mod h1:EcaO66Nn1StkpEm1iKtBTV3d2A16SoMsVER1PthX7to=
 github.com/hashicorp/go-cleanhttp v0.5.2 h1:035FKYIWjmULyFRBKPs8TBQoi0x6d9G4xc9neXJWAZQ=
 github.com/hashicorp/go-cleanhttp v0.5.2/go.mod h1:kO/YDlP8L1346E6Sodw+PrpBSV4/SoxCXGY6BqNFT48=
+github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
 github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
 github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
 github.com/hashicorp/hcl/v2 v2.13.0 h1:0Apadu1w6M11dyGFxWnmhhcMjkbAiKCv7G1r/2QgCNc=
@@ -82,15 +220,30 @@ github.com/hekmon/cunits/v2 v2.1.0 h1:k6wIjc4PlacNOHwKEMBgWV2/c8jyD4eRMs5mR1BBhI
 github.com/hekmon/cunits/v2 v2.1.0/go.mod h1:9r1TycXYXaTmEWlAIfFV8JT+Xo59U96yUJAYHxzii2M=
 github.com/hekmon/transmissionrpc/v3 v3.0.0 h1:0Fb11qE0IBh4V4GlOwHNYpqpjcYDp5GouolwrpmcUDQ=
 github.com/hekmon/transmissionrpc/v3 v3.0.0/go.mod h1:38SlNhFzinVUuY87wGj3acOmRxeYZAZfrj6Re7UgCDg=
+github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
+github.com/huandu/xstrings v1.0.0/go.mod h1:4qWG/gcEcfX4z/mBDHJ++3ReCw9ibxbsNJbcucJdbSo=
+github.com/huandu/xstrings v1.2.0/go.mod h1:DvyZB1rfVYsBIigL8HwpZgxHwXozlTgGqn63UyNX5k4=
+github.com/huandu/xstrings v1.3.1/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=
+github.com/huandu/xstrings v1.3.2 h1:L18LIDzqlW6xN2rEkpdV8+oL/IXWJ1APd+vsdYy4Wdw=
+github.com/huandu/xstrings v1.3.2/go.mod h1:y5/lhBue+AyNmUVz9RLU9xbLR0o4KIIExikq4ovT0aE=
 github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible h1:jdpOPRN1zP63Td1hDQbZW73xKmzDvZHzVdNYxhnTMDA=
 github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible/go.mod h1:1c7szIrayyPPB/987hsnvNzLushdWf4o/79s3P08L8A=
+github.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=
+github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
 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/jtolds/gls v4.2.1+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
+github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
+github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=
+github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
 github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE=
+github.com/klauspost/cpuid/v2 v2.0.4/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
 github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
 github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM=
 github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
 github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
+github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
+github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
@@ -108,6 +261,9 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
 github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
+github.com/minio/sha256-simd v1.0.0 h1:v1ta+49hkWZyvaKwrQB8elexRqm6Y0aMLjCNsrYxo6g=
+github.com/minio/sha256-simd v1.0.0/go.mod h1:OuYzVNI5vcoYIAmbIvHPl3N3jUzVedXbKy5RFepssQM=
 github.com/mitchellh/go-wordwrap v0.0.0-20150314170334-ad45545899c7 h1:DpOJ2HYzCv8LZP15IdmG+YdwD2luVPHITV96TkirNBM=
 github.com/mitchellh/go-wordwrap v0.0.0-20150314170334-ad45545899c7/go.mod h1:ZXFpozHsX6DPmq2I0TCekCxypsnAUbP2oI0UX1GXzOo=
 github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
@@ -115,8 +271,19 @@ github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RR
 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 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
+github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
 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/mr-tron/base58 v1.2.0 h1:T/HDJBh4ZCPbU39/+c3rRvE0uKBQlU27+QI8LJ4t64o=
+github.com/mr-tron/base58 v1.2.0/go.mod h1:BinMc/sQntlIE1frQmRFPUoPA1Zkr8VRgBdjWI2mNwc=
+github.com/mschoch/smat v0.0.0-20160514031455-90eadee771ae/go.mod h1:qAyveg+e4CE+eKJXWVjKXM4ck2QobLqTDytGJbLLhJg=
+github.com/mschoch/smat v0.2.0/go.mod h1:kc9mz7DoBKqDyiRL7VZN8KvXQMWeTaVnttLRXOlotKw=
+github.com/multiformats/go-multihash v0.2.3 h1:7Lyc8XfX/IY2jWb/gI7JP+o7JEq9hOa7BFvVU9RSh+U=
+github.com/multiformats/go-multihash v0.2.3/go.mod h1:dXgKXCXjBzdscBLk9JkjINiEsCKRVch90MdaGiKsvSM=
+github.com/multiformats/go-varint v0.0.6 h1:gk85QWKxh3TazbLxED/NlDVv8+q+ReFJk7Y2W/KhfNY=
+github.com/multiformats/go-varint v0.0.6/go.mod h1:3Ls8CIEsrijN6+B7PbrXRPxHRPuXSrVKRY101jdMZYE=
+github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
 github.com/natefinch/lumberjack v2.0.0+incompatible h1:4QJd3OLAMgj7ph+yZTuX13Ld4UpgHp07nNdFX7mqFfM=
 github.com/natefinch/lumberjack v2.0.0+incompatible/go.mod h1:Wi9p2TTF5DG5oU+6YfsmYQpsTIOm0B1VNzQg9Mw6nPk=
 github.com/ncruces/go-sqlite3 v0.18.4 h1:Je8o3y33MDwPYY/Cacas8yCsuoUzpNY/AgoSlN2ekyE=
@@ -125,25 +292,60 @@ github.com/ncruces/julianday v1.0.0 h1:fH0OKwa7NWvniGQtxdJRxAgkBMolni2BjDHaWTxqt
 github.com/ncruces/julianday v1.0.0/go.mod h1:Dusn2KvZrrovOMJuOt0TNXL6tB7U2E8kvza5fFc9G7g=
 github.com/nikoksr/notify v1.0.0 h1:qe9/6FRsWdxBgQgWcpvQ0sv8LRGJZDpRB4TkL2uNdO8=
 github.com/nikoksr/notify v1.0.0/go.mod h1:hPaaDt30d6LAA7/5nb0e48Bp/MctDfycCSs8VEgN29I=
+github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/openzipkin/zipkin-go v0.1.6/go.mod h1:QgAqvLzwWbR/WpD4A3cGpPtJrZXNIiJc5AZX7/PBEpw=
 github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
 github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
+github.com/philhofer/fwd v1.0.0/go.mod h1:gk3iGcWd9+svBvR0sR+KPcfE+RNWozjowpeBVG3ZVNU=
+github.com/pierrec/lz4 v2.0.5+incompatible/go.mod h1:pdkljMzZIN41W+lC3N2tnIh5sFi+IEE17M5jbnwPHcY=
+github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
+github.com/prometheus/client_golang v0.9.3-0.20190127221311-3c4408c8b829/go.mod h1:p2iRAGwDERtqlqzRXnrOVns+ignqQo//hLXqYxZYVNs=
+github.com/prometheus/client_golang v1.0.0/go.mod h1:db9x61etRT2tGnBNRi70OPL5FsnadC4Ky3P0J6CfImo=
+github.com/prometheus/client_golang v1.5.1/go.mod h1:e9GMxYsXl05ICDXkRhurwBS4Q3OK1iX/F2sw+iXX5zU=
+github.com/prometheus/client_model v0.0.0-20180712105110-5c3871d89910/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=
+github.com/prometheus/client_model v0.0.0-20190115171406-56726106282f/go.mod h1:MbSGuTsp3dbXC40dX6PRTWyKYBIrTGTE9sqQNg2J8bo=
+github.com/prometheus/client_model v0.0.0-20190129233127-fd36f4220a90/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
+github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
+github.com/prometheus/client_model v0.2.0/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
+github.com/prometheus/common v0.2.0/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=
+github.com/prometheus/common v0.4.1/go.mod h1:TNfzLD0ON7rHzMJeJkieUDPYmFC7Snx/y86RQel1bk4=
+github.com/prometheus/common v0.9.1/go.mod h1:yhUN8i9wzaXS3w1O07YhxHEBxD+W35wd8bs7vj7HSQ4=
+github.com/prometheus/procfs v0.0.0-20181005140218-185b4288413d/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=
+github.com/prometheus/procfs v0.0.0-20190117184657-bf6a532e95b1/go.mod h1:c3At6R/oaqEKCNdg8wHV1ftS6bRYblBhIjjI8uT2IGk=
+github.com/prometheus/procfs v0.0.2/go.mod h1:TjEm7ze935MbeOT/UhFTIMYKhuLP4wbCsTZCD3I8kEA=
+github.com/prometheus/procfs v0.0.8/go.mod h1:7Qr8sr6344vo1JqZ6HhLceV9o3AJ1Ff+GxbHq6oeK9A=
+github.com/prometheus/procfs v0.0.11/go.mod h1:lV6e/gmhEcM9IjHGsFOCxxuZ+z1YqCvr4OA4YeYWdaU=
+github.com/rcrowley/go-metrics v0.0.0-20181016184325-3113b8401b8a/go.mod h1:bCqnVzQkZxMG4s8nGwiZ5l3QUCyqpo9Y+/ZMZ9VjZe4=
 github.com/robfig/cron v1.2.0 h1:ZjScXvvxeQ63Dbyxy76Fj3AT3Ut0aKsyd2/tl3DTMuQ=
 github.com/robfig/cron v1.2.0/go.mod h1:JGuDeoQd7Z6yL4zQhZ3OPEVHB7fL6Ka6skscFHfmt2k=
 github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
 github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/ryszard/goskiplist v0.0.0-20150312221310-2dfbae5fcf46/go.mod h1:uAQ5PCi+MFsC7HjREoAz1BU+Mq60+05gifQSsHSDG/8=
 github.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6keLGt6kNQ=
 github.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4=
 github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
 github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
 github.com/sergi/go-diff v1.0.0 h1:Kpca3qRNrduNnOQeazBd0ysaKrUJiIuISHxogkT9RPQ=
 github.com/sergi/go-diff v1.0.0/go.mod h1:0CfEIISq7TuYL3j771MWULgwwjU+GofnZX9QAmXWZgo=
+github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
+github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=
+github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
+github.com/smartystreets/assertions v0.0.0-20190215210624-980c5ac6f3ac/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
+github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s=
+github.com/smartystreets/goconvey v0.0.0-20190306220146-200a235640ff/go.mod h1:KSQcGKpxUMHk3nbYzs/tIBAM2iDooCn0BmttHOJEbLs=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
+github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
+github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
 github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=
 github.com/spf13/afero v1.11.0/go.mod h1:GH9Y3pIexgf1MTIWtNGyogA5MwRIDXGUr+hbWNoBjkY=
 github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=
@@ -153,11 +355,15 @@ github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An
 github.com/spf13/viper v1.19.0 h1:RWq5SEjt8o25SROyN3z2OrDB9l7RPd3lwTWU8EcEdcI=
 github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+Ntkg=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
 github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
 github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
 github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
+github.com/stretchr/testify v1.2.1/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
@@ -171,15 +377,35 @@ github.com/technoweenie/multipartstreamer v1.0.1 h1:XRztA5MXiR1TIRHxH2uNxXxaIkKQ
 github.com/technoweenie/multipartstreamer v1.0.1/go.mod h1:jNVxdtShOxzAsukZwTSw6MDx5eUJoiEBsSvzDU9uzog=
 github.com/tetratelabs/wazero v1.8.0 h1:iEKu0d4c2Pd+QSRieYbnQC9yiFlMS9D+Jr0LsRmcF4g=
 github.com/tetratelabs/wazero v1.8.0/go.mod h1:yAI0XTsMBhREkM/YDAK/zNou3GoiAce1P6+rp/wQhjs=
+github.com/tinylib/msgp v1.0.2/go.mod h1:+d+yLhGm8mzTaHzB+wgMYrodPfmZrzkirds8fDWklFE=
+github.com/tinylib/msgp v1.1.0/go.mod h1:+d+yLhGm8mzTaHzB+wgMYrodPfmZrzkirds8fDWklFE=
+github.com/tinylib/msgp v1.1.2/go.mod h1:+d+yLhGm8mzTaHzB+wgMYrodPfmZrzkirds8fDWklFE=
 github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
 github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
 github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
 github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
 github.com/vmihailenco/msgpack/v4 v4.3.12/go.mod h1:gborTTJjAo/GWTqqRjrLCn9pgNN+NXzzngzBKDPIqw4=
 github.com/vmihailenco/tagparser v0.1.1/go.mod h1:OeAg3pn3UbLjkWt+rN9oFYB6u/cQgqMEUPoW2WPyhdI=
+github.com/willf/bitset v1.1.9/go.mod h1:RjeCKbqT1RxIR/KWY6phxZiaY1IyutSBfGjNPySAYV4=
+github.com/willf/bitset v1.1.10/go.mod h1:RjeCKbqT1RxIR/KWY6phxZiaY1IyutSBfGjNPySAYV4=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 github.com/zclconf/go-cty v1.8.0 h1:s4AvqaeQzJIu3ndv4gVIhplVD0krU+bgrcLSVUnaWuA=
 github.com/zclconf/go-cty v1.8.0/go.mod h1:vVKLxnk3puL4qRAv72AO+W99LUD4da90g3uUAzyuvAk=
+go.opencensus.io v0.20.1/go.mod h1:6WKK9ahsWS3RSO+PY9ZHZUfv2irvY6gN279GOPZjmmk=
+go.opencensus.io v0.20.2/go.mod h1:6WKK9ahsWS3RSO+PY9ZHZUfv2irvY6gN279GOPZjmmk=
+go.opencensus.io v0.22.3/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=
+go.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=
+go.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=
+go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.53.0 h1:9G6E0TXzGFVfTnawRzrPl83iHOAV7L8NJiR8RSGYV1g=
+go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.53.0/go.mod h1:azvtTADFQJA8mX80jIH/akaE7h+dbm/sVuaHqN13w74=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.53.0 h1:4K4tsIXefpVJtvA/8srF4V4y0akAoPHkIslgAkjixJA=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.53.0/go.mod h1:jjdQuTGVsXV4vSs+CJ2qYDeDPf9yIJV23qlIzBm73Vg=
+go.opentelemetry.io/otel v1.28.0 h1:/SqNcYk+idO0CxKEUOtKQClMK/MimZihKYMruSMViUo=
+go.opentelemetry.io/otel v1.28.0/go.mod h1:q68ijF8Fc8CnMHKyzqL6akLO46ePnjkgfIMIjUIX9z4=
+go.opentelemetry.io/otel/metric v1.28.0 h1:f0HGvSl1KRAU1DLgLGFjrwVyismPlnuU6JD6bOeuA5Q=
+go.opentelemetry.io/otel/metric v1.28.0/go.mod h1:Fb1eVBFZmLVTMb6PPohq3TO9IIhUisDsbJoL/+uQW4s=
+go.opentelemetry.io/otel/trace v1.28.0 h1:GhQ9cUuQGmNDd5BTCP2dAvv75RdMxEfTmYejp+lkx9g=
+go.opentelemetry.io/otel/trace v1.28.0/go.mod h1:jPyXzNPg6da9+38HEwElrQiHlVMTnVfM3/yv2OlIHaI=
 go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
 go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
@@ -189,31 +415,70 @@ go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
 golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
 golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
 golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
-golang.org/x/crypto v0.27.0 h1:GXm2NjJrPaiv/h1tb2UH8QfgC/hOf/+z0p6PT8o1w7A=
-golang.org/x/crypto v0.27.0/go.mod h1:1Xngt8kV6Dvbssa53Ziq6Eqn0HqbZi5Z6R0ZpwQzt70=
-golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 h1:2dVuKD2vS7b0QIHQbpyTISPd0LeHDbnYEryqj5Q1ug8=
-golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56/go.mod h1:M4RDyNAINzryxdtnbRXRL/OHtkFuWGRjvuhBJpk2IlY=
+golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
+golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
+golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
+golang.org/x/exp v0.0.0-20240823005443-9b4947da3948 h1:kx6Ds3MlpiUHKj7syVnbp57++8WpuKPcR5yjLBjvLEA=
+golang.org/x/exp v0.0.0-20240823005443-9b4947da3948/go.mod h1:akd2r19cwCdwSwWeIdzYQGa/EZZyqcOdwWiwj5L5eKQ=
+golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
+golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
+golang.org/x/lint v0.0.0-20190301231843-5614ed5bae6f/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
+golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
 golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
 golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
-golang.org/x/mod v0.19.0 h1:fEdghXQSo20giMthA7cd28ZC+jts4amQ3YMXiP5oMQ8=
-golang.org/x/mod v0.19.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
+golang.org/x/mod v0.20.0 h1:utOm6MM3R3dnawAiJgn0y+xvuYRsm1RKM/4giyfDgV0=
+golang.org/x/mod v0.20.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
+golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190108225652-1e06a53dbb7e/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190125091013-d26f9f9a57f3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
 golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
+golang.org/x/net v0.0.0-20190613194153-d28f0bde5980/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
 golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
 golang.org/x/net v0.0.0-20200301022130-244492dfa37a/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
 golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
 golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
 golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
 golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns=
-golang.org/x/net v0.27.0 h1:5K3Njcw06/l2y9vpGCSdcxWOYHOUk3dVNGDXN+FvAys=
-golang.org/x/net v0.27.0/go.mod h1:dDi0PyhWNoiUOrAS8uXv/vnScO4wnHQO4mj9fn/RytE=
+golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
+golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
+golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
+golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
+golang.org/x/oauth2 v0.21.0 h1:tsimM75w1tF/uws5rbeHzIWxEqElMehnc+iW793zsZs=
+golang.org/x/oauth2 v0.21.0/go.mod h1:XYTD2NtWslqkgxebSiOHnXEap4TF09sJSc7H1sXbhtI=
+golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
-golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ=
+golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200106162015-b016eb3dc98e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200122134326-e047566fdf82/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200413165638-669c56c373c4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
@@ -221,8 +486,8 @@ golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBc
 golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-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/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
+golang.org/x/sys v0.28.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.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
@@ -234,28 +499,85 @@ golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
 golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
-golang.org/x/text v0.18.0 h1:XvMDiNzPAl0jr17s6W9lcaIhGUfUORdGCNsuLmPG224=
-golang.org/x/text v0.18.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
+golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
+golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
+golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
+golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
+golang.org/x/tools v0.0.0-20180828015842-6cd1fcedba52/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
+golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
+golang.org/x/tools v0.0.0-20190312170243-e65039ee4138/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
+golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
 golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
 golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
 golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+golift.io/starr v1.0.0 h1:IDSaSL+ZYxdLT/Lg//dg/iwZ39LHO3D5CmbLCOgSXbI=
+golift.io/starr v1.0.0/go.mod h1:xnUwp4vK62bDvozW9QHUYc08m6kjwaZnGw3Db65fQHw=
+google.golang.org/api v0.3.1/go.mod h1:6wY9I6uQWHQ8EM57III9mq/AjF+i8G65rmVagqKMtkk=
+google.golang.org/api v0.188.0 h1:51y8fJ/b1AaaBRJr4yWm96fPcuxSo0JcegXE3DaHQHw=
+google.golang.org/api v0.188.0/go.mod h1:VR0d+2SIiWOYG3r/jdm7adPW9hI2aRv9ETOSCQ9Beag=
+google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
+google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
 google.golang.org/appengine v1.6.5/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=
+google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
+google.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
+google.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
+google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=
+google.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=
+google.golang.org/genproto v0.0.0-20240711142825-46eb208f015d h1:/hmn0Ku5kWij/kjGsrcJeC1T/MrJi2iNWwgAqrihFwc=
+google.golang.org/genproto/googleapis/api v0.0.0-20240711142825-46eb208f015d h1:kHjw/5UfflP/L5EbledDrcG4C2597RtymmGRZvHiCuY=
+google.golang.org/genproto/googleapis/api v0.0.0-20240711142825-46eb208f015d/go.mod h1:mw8MG/Qz5wfgYr6VqVCiZcHe/GJEfI+oGGDCohaVgB0=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20240711142825-46eb208f015d h1:JU0iKnSg02Gmb5ZdV8nYsKEKsP6o/FGVWTrw4i1DA9A=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20240711142825-46eb208f015d/go.mod h1:Ue6ibwXGpU+dqIcODieyLOcgj7z8+IcskoNIgZxtrFY=
+google.golang.org/grpc v1.17.0/go.mod h1:6QZJwpn2B+Zp71q/5VxRsJ6NXXVCE5NRUHRo+f3cWCs=
+google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=
+google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=
+google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=
+google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=
+google.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=
+google.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc=
+google.golang.org/grpc v1.65.0 h1:bs/cUb4lp1G5iImFFd3u5ixQzweKizoZJAwBNLR42lc=
+google.golang.org/grpc v1.65.0/go.mod h1:WgYC2ypjlB0EiQi6wdKixMqukr6lBc0Vo+oOgjrM5ZQ=
+google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=
+google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=
+google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=
+google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=
+google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=
+google.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
+google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
+google.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=
+google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=
 google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=
 google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=
+gopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
 gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=
 gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
 gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
 gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
+gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
 gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+honnef.co/go/tools v0.0.0-20180728063816-88497007e858/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
+honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
+honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
+lukechampine.com/blake3 v1.1.6 h1:H3cROdztr7RCfoaTpGZFQsrqvweFLrqS73j7L7cmR5c=
+lukechampine.com/blake3 v1.1.6/go.mod h1:tkKEOtDkNtklkXtLNEOGNq5tcV90tJiA1vAA12R78LA=
 nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
 rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

pkg/alist/alist.go

@@ -0,0 +1,196 @@
+package alist
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/pkg/errors"
+)
+
+type Resposne[T any] struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    T      `json:"data"`
+}
+
+type Config struct {
+	Username string
+	Password string
+	URL      string
+}
+
+func New(cfg *Config) *Client {
+	cfg.URL = strings.Trim(cfg.URL, "/")
+	return &Client{
+		cfg:  cfg,
+		http: http.DefaultClient,
+	}
+}
+
+type Client struct {
+	cfg   *Config
+	http  *http.Client
+	token string
+}
+
+func (c *Client) Login() (string, error) {
+	p := map[string]string{
+		"username": c.cfg.Username,
+		"password": c.cfg.Password,
+	}
+	data, _ := json.Marshal(p)
+	resp, err := c.http.Post(c.cfg.URL+loginUrl, "application/json", bytes.NewBuffer(data))
+	if err != nil {
+		return "", errors.Wrap(err, "login")
+	}
+	defer resp.Body.Close()
+	d1, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", errors.Wrap(err, "read body")
+	}
+	var rp Resposne[map[string]string]
+
+	err = json.Unmarshal(d1, &rp)
+	if err != nil {
+		return "", errors.Wrap(err, "json")
+	}
+	if rp.Code != 200 {
+		return "", errors.Errorf("alist error: code %d, %s", rp.Code, rp.Message)
+	}
+	c.token = rp.Data["token"]
+	return c.token, nil
+}
+
+type LsInfo struct {
+	Content []struct {
+		Name     string    `json:"name"`
+		Size     int       `json:"size"`
+		IsDir    bool      `json:"is_dir"`
+		Modified time.Time `json:"modified"`
+		Created  time.Time `json:"created"`
+		Sign     string    `json:"sign"`
+		Thumb    string    `json:"thumb"`
+		Type     int       `json:"type"`
+		Hashinfo string    `json:"hashinfo"`
+		HashInfo any       `json:"hash_info"`
+	} `json:"content"`
+	Total    int    `json:"total"`
+	Readme   string `json:"readme"`
+	Header   string `json:"header"`
+	Write    bool   `json:"write"`
+	Provider string `json:"provider"`
+}
+
+func (c *Client) Ls(dir string) (*LsInfo, error) {
+	in := map[string]string{
+		"path": dir,
+	}
+
+	resp, err := c.post(c.cfg.URL+lsUrl, in)
+	if err != nil {
+		return nil, errors.Wrap(err, "http")
+	}
+
+	var out Resposne[LsInfo]
+	err = json.Unmarshal(resp, &out)
+	if err != nil {
+		return nil, err
+	}
+	if out.Code != 200 {
+		return nil, errors.Errorf("alist error: code %d, %s", out.Code, out.Message)
+	}
+	return &out.Data, nil
+}
+
+func (c *Client) Mkdir(dir string) error {
+	in := map[string]string{
+		"path": dir,
+	}
+	resp, err := c.post(c.cfg.URL+mkdirUrl, in)
+	if err != nil {
+		return errors.Wrap(err, "http")
+	}
+	var out Resposne[any]
+	err = json.Unmarshal(resp, &out)
+	if err != nil {
+		return err
+	}
+	if out.Code != 200 {
+		return errors.Errorf("alist error: code %d, %s", out.Code, out.Message)
+	}
+	return nil
+}
+
+func (c *Client) post(url string, body interface{}) ([]byte, error) {
+	data, err := json.Marshal(body)
+	if err != nil {
+		return nil, err
+	}
+	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(data))
+	if err != nil {
+		return nil, errors.Wrap(err, "new request")
+	}
+
+	req.Header.Add("Authorization", c.token)
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := c.http.Do(req)
+	if err != nil {
+		return nil, errors.Wrap(err, "http")
+	}
+	defer resp.Body.Close()
+	d1, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, errors.Wrap(err, "read body")
+	}
+	return d1, nil
+}
+
+type UploadStreamResponse struct {
+	Task struct {
+		ID       string `json:"id"`
+		Name     string `json:"name"`
+		State    int    `json:"state"`
+		Status   string `json:"status"`
+		Progress int    `json:"progress"`
+		Error    string `json:"error"`
+	} `json:"task"`
+}
+
+func (c *Client) UploadStream(reader io.Reader, size int64, toDir string) (*UploadStreamResponse, error) {
+	req, err := http.NewRequest(http.MethodPut, c.cfg.URL+streamUploadUrl, reader)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Add("Authorization", c.token)
+	req.Header.Add("File-Path", url.PathEscape(toDir))
+	req.Header.Add("As-Task", "true")
+	req.Header.Add("Content-Type", "application/octet-stream")
+	req.ContentLength = size
+	res, err := c.http.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+	d1, err := io.ReadAll(res.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	var out Resposne[UploadStreamResponse]
+	err = json.Unmarshal(d1, &out)
+	if err != nil {
+		return nil, err
+	}
+	if out.Code != 200 {
+		return nil, errors.Errorf("alist error: code %d, %s", out.Code, out.Message)
+	}
+
+	return &out.Data, nil
+}

pkg/alist/alist_test.go

@@ -0,0 +1,46 @@
+package alist
+
+import (
+	"os"
+	"polaris/log"
+	"testing"
+)
+
+func TestLogin(t *testing.T) {
+	c := New(&Config{
+		URL: "http://10.0.0.8:5244/",
+		Username: "",
+		Password: "",
+	})
+	cre, err := c.Login()
+	if err != nil {
+		log.Errorf("login fail: %v", err)
+		t.Fail()
+	} else {
+		log.Errorf("login success: %s", cre)
+	}
+	info, err := c.Ls("/aliyun")
+	if err != nil {
+		log.Errorf("ls fail: %v", err)
+		t.Fail()
+	} else {
+		log.Infof("ls results: %+v", info)
+	}
+
+	f, err := os.Open("/Users/simonding/Downloads/Steam Link_1.3.9_APKPure.apk")
+	if err != nil {
+		log.Errorf("openfile: %v", err)
+		t.Fail()
+	} else {
+		defer f.Close()
+		ss, _ := f.Stat()
+		log.Infof("upload file size %d", ss.Size())
+		info, err := c.UploadStream(f, ss.Size(), "/aliyun/Steam Link_1.3.9_APKPure.apk")
+		if err != nil {
+			log.Errorf("upload error: %v", err)
+			t.Fail()
+		} else {
+			log.Infof("upload success: %+v", info)
+		}
+	}
+}

pkg/alist/url.go

@@ -0,0 +1,8 @@
+package alist
+
+const (
+	loginUrl = "/api/auth/login"
+	lsUrl = "/api/fs/list"
+	mkdirUrl = "/api/fs/mkdir"
+	streamUploadUrl = "/api/fs/put"
+)

pkg/doc.go

@@ -6,11 +6,11 @@ type Torrent interface {
 	Stop() error
 	Start() error
 	Remove() error
-	Save() string
+	//Save() string
 	Exists() bool
 	SeedRatio() (float64, error)
 	GetHash() string
-	Reload() error
+	//Reload() error
 }
 
 type Downloader interface {

pkg/gemini/gemini.go

@@ -0,0 +1,266 @@
+package gemini
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"polaris/log"
+	"strings"
+
+	"github.com/google/generative-ai-go/genai"
+	"google.golang.org/api/option"
+)
+
+func NewClient(apiKey, modelName string) (*Client, error) {
+	ctx := context.Background()
+	client, err := genai.NewClient(ctx, option.WithAPIKey(apiKey))
+	if err != nil {
+		return nil, err
+	}
+	return &Client{apiKey: apiKey, modelName: modelName, c: client}, nil
+
+}
+
+type Client struct {
+	apiKey    string
+	modelName string
+	c         *genai.Client
+}
+
+type TvInfo struct {
+	TitleEnglish     string `json:"title_english"`
+	TitleChinses     string `json:"title_chinese"`
+	Season           int    `json:"season"`
+	StartEpisode     int    `json:"start_episode"`
+	EndEpisode       int    `json:"end_episode"`
+	Resolution       string `json:"resolution"`
+	Subtitle         string `json:"subtitle"`
+	ReleaseGroup     string `json:"release_group"`
+	Year             int    `json:"year"`
+	AudioLanguage    string `json:"audio_language"`
+	IsCompleteSeason bool   `json:"is_complete_season"`
+}
+
+func (c *Client) ParseTvInfo(q string) (*TvInfo, error) {
+	log.Info(q)
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type: genai.TypeObject,
+		Properties: map[string]*genai.Schema{
+			"title_english": {Type: genai.TypeString},
+			"title_chinese": {Type: genai.TypeString},
+			"season":        {Type: genai.TypeInteger, Description: "season number"},
+			"start_episode": {Type: genai.TypeInteger},
+			"end_episode":   {Type: genai.TypeInteger},
+			//"episodes":       {Type: genai.TypeString},
+			"resolution":         {Type: genai.TypeString},
+			"subtitle":           {Type: genai.TypeString},
+			"release_group":      {Type: genai.TypeString},
+			"year":               {Type: genai.TypeInteger},
+			"audio_language":     {Type: genai.TypeString},
+			"is_complete_season": {Type: genai.TypeBoolean},
+		},
+		Required: []string{"title_english", "title_chinese", "season", "start_episode", "resolution"},
+	}
+
+	resp, err := model.GenerateContent(ctx, genai.Text(q))
+	if err != nil {
+		return nil, err
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+			var info TvInfo
+			if err := json.Unmarshal([]byte(txt), &info); err != nil {
+				return nil, err
+			}
+			return &info, nil
+		}
+	}
+	return nil, fmt.Errorf("not found")
+
+}
+
+type MovieInfo struct {
+	TitleEnglish     string `json:"title_english"`
+	TitleChinses     string `json:"title_chinese"`
+	Resolution       string `json:"resolution"`
+	Subtitle         string `json:"subtitle"`
+	ReleaseGroup     string `json:"release_group"`
+	Year             int    `json:"year"`
+	AudioLanguage    string `json:"audio_language"`
+}
+
+func (c *Client) ParseMovieInfo(q string) (*MovieInfo, error) {
+	log.Info(q)
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type: genai.TypeObject,
+		Properties: map[string]*genai.Schema{
+			"title_english":  {Type: genai.TypeString},
+			"title_chinese":  {Type: genai.TypeString},
+			"resolution":     {Type: genai.TypeString},
+			"subtitle":       {Type: genai.TypeString},
+			"release_group":  {Type: genai.TypeString},
+			"year":           {Type: genai.TypeInteger},
+			"audio_language": {Type: genai.TypeString},
+		},
+		Required: []string{"title_english", "title_chinese", "resolution"},
+	}
+
+	resp, err := model.GenerateContent(ctx, genai.Text(q))
+	if err != nil {
+		return nil, err
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+			var info MovieInfo
+			if err := json.Unmarshal([]byte(txt), &info); err != nil {
+				return nil, err
+			}
+			return &info, nil
+		}
+	}
+	return nil, fmt.Errorf("not found")
+
+}
+
+func (c *Client) isTvSeries(q string) (bool, error) {
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type: genai.TypeBoolean, Nullable: true, Description: "whether the input text implies a tv series",
+	}
+
+	resp, err := model.GenerateContent(ctx, genai.Text(q))
+	if err != nil {
+		return false, err
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+			return strings.ToLower(string(txt)) == "true", nil
+		}
+	}
+	return false, fmt.Errorf("error")
+}
+
+func (c *Client) ImpliesSameTvOrMovie(torrentName, mediaName string) bool {
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type: genai.TypeBoolean, Nullable: true,
+	}
+	q := fmt.Sprintf("whether this file name \"%s\" implies the same TV series or movie with name \"%s\"?", torrentName, mediaName)
+	resp, err := model.GenerateContent(ctx, genai.Text(q))
+	if err != nil {
+		return false
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+			return strings.ToLower(string(txt)) == "true"
+		}
+	}
+	return false
+
+}
+
+func (c *Client) FilterTvOrMovies(resourcesNames []string, titles ...string) ([]string, error) {
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type:  genai.TypeArray,
+		Items: &genai.Schema{Type: genai.TypeString},
+	}
+	for i, s := range titles {
+		titles[i] = "\"" + s + "\""
+	}
+	p := &bytes.Buffer{}
+	p.WriteString(`the following list of file names, list all of which implies the same TV series or movie of name`)
+	p.WriteString(strings.Join(titles, " or "))
+	p.WriteString(":\n")
+
+	for _, r := range resourcesNames {
+		p.WriteString(" * ")
+		p.WriteString(r)
+		p.WriteString("\n")
+	}
+	log.Debugf("FilterTvOrMovies prompt is %s", p.String())
+
+	resp, err := model.GenerateContent(ctx, genai.Text(p.String()))
+	if err != nil {
+		return nil, err
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+
+			var s []string
+			if err := json.Unmarshal([]byte(txt), &s); err != nil {
+				return nil, err
+			}
+			return s, nil
+		}
+	}
+	return nil, fmt.Errorf("nothing found")
+
+}
+
+
+func (c *Client) FilterMovies(resourcesNames []string, year int, titles ...string) ([]string, error) {
+	ctx := context.Background()
+
+	model := c.c.GenerativeModel(c.modelName)
+
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type:  genai.TypeArray,
+		Items: &genai.Schema{Type: genai.TypeString},
+	}
+	for i, s := range titles {
+		titles[i] = "\"" + s + "\""
+	}
+	p := &bytes.Buffer{}
+	p.WriteString( fmt.Sprint("the following list of file names, list all of which match following criteria: 1. Is movie 2. Released in year %d 3. Have name of ", year))
+	p.WriteString(strings.Join(titles, " or "))
+	p.WriteString(":\n")
+
+	for _, r := range resourcesNames {
+		p.WriteString(" * ")
+		p.WriteString(r)
+		p.WriteString("\n")
+	}
+	log.Debugf("FilterTvOrMovies prompt is %s", p.String())
+
+	resp, err := model.GenerateContent(ctx, genai.Text(p.String()))
+	if err != nil {
+		return nil, err
+	}
+	for _, part := range resp.Candidates[0].Content.Parts {
+		if txt, ok := part.(genai.Text); ok {
+
+			var s []string
+			if err := json.Unmarshal([]byte(txt), &s); err != nil {
+				return nil, err
+			}
+			return s, nil
+		}
+	}
+	return nil, fmt.Errorf("nothing found")
+
+}

pkg/gemini/genimi_test.go

@@ -0,0 +1,8 @@
+package gemini
+
+import (
+	"testing"
+)
+
+func Test_any1(t *testing.T) {
+}

pkg/go-qbittorrent/qbt/api.go

@@ -469,7 +469,6 @@ func (client *Client) Torrents(opts TorrentsOptions) (torrentList []TorrentInfo,
 	return torrentList, nil
 }
 
-// deprecated
 // Torrent returns a specific torrent matching the hash
 func (client *Client) Torrent(hash string) (torrent Torrent, err error) {
 	var opts = map[string]string{"hash": strings.ToLower(hash)}

pkg/importlist/douban/douban.go

@@ -37,6 +37,7 @@ func ParseDoulist(doulistUrl string) (*importlist.Response, error) {
 	if err != nil {
 		return nil, err
 	}
+	var items []importlist.Item
 	doc.Find("div[class=doulist-item]").Each(func(i int, selection *goquery.Selection) {
 		titleDiv := selection.Find("div[class=title]")
 		link := titleDiv.Find("div>a")
@@ -64,18 +65,26 @@ func ParseDoulist(doulistUrl string) (*importlist.Response, error) {
 				}
 			}
 		}
+		_, err := parseDetailPage(strings.TrimSpace(href))
+		if err != nil {
+			log.Errorf("get detail page: %v", err)
+			return
+		}
 
 		item := importlist.Item{
 			Title: strings.TrimSpace(link.Text()),
 			Year:  year,
 		}
+		items = append(items, item)
 		_ = item
-		println(link.Text(), href)
+		//println(link.Text(), href)
 	})
-	return nil, nil
+
+	return &importlist.Response{Items: items}, nil
 }
 
 func parseDetailPage(url string) (string, error) {
+	println(url)
 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
 		return "", err
@@ -95,6 +104,14 @@ func parseDetailPage(url string) (string, error) {
 	if err != nil {
 		return "", err
 	}
+
+	doc.Find("div[class='subject clearfix']").Each(func(i int, se *goquery.Selection) {
+		println(se.Text())
+		se.Children().Get(1)
+		imdb := se.Find("div[class='info']").First().Children().Last()
+		println(imdb.Text())
+	})
+
 	_ = doc
 	return "", nil
 }

pkg/importlist/douban/douban_test.go

@@ -6,6 +6,6 @@ import (
 )
 
 func TestParseDoulist(t *testing.T) {
-	r, err := ParseDoulist("https://www.douban.com/doulist/166422/")
+	r, err := ParseDoulist("https://www.douban.com/doulist/81580/")
 	log.Info(r, err)
 }

pkg/metadata/movie.go

@@ -2,6 +2,7 @@ package metadata
 
 import (
 	"fmt"
+	"polaris/pkg/utils"
 	"regexp"
 	"strconv"
 	"strings"
@@ -21,6 +22,11 @@ func (m *MovieMetadata) IsAcceptable(names... string) bool {
 		name2 := re.ReplaceAllString(strings.ToLower(m.Name), " ")
 		name = strings.Join(strings.Fields(name), " ")
 		name2 = strings.Join(strings.Fields(name2), " ")
+		if utils.IsASCII(name) { //ascii name should match words
+			re := regexp.MustCompile(`\b` + name + `\b`)
+			return re.MatchString(name2)
+		}
+		
 		if  strings.Contains(name2, name) {
 			return true
 		}

pkg/metadata/tv.go

@@ -7,237 +7,104 @@ import (
 	"regexp"
 	"strconv"
 	"strings"
+	"time"
 )
 
-type Metadata struct {
+type Info struct {
 	NameEn       string
 	NameCn       string
+	Year         int
 	Season       int
-	Episode      int
+	StartEpisode int
+	EndEpisode   int
 	Resolution   string
 	IsSeasonPack bool
 }
 
-func (m *Metadata) IsAcceptable(names... string) bool {
+func (m *Info) ParseExtraDescription(desc string) {
+	if m.IsSeasonPack { //try to parse episode number with description
+		mm := ParseTv(desc)
+		if mm.StartEpisode > 0 { //sometimes they put episode info in desc text
+			m.IsSeasonPack = false
+			m.StartEpisode = mm.StartEpisode
+			m.EndEpisode = mm.EndEpisode
+		}
+	}
+}
+
+func (m *Info) IsAcceptable(names ...string) bool {
+	re := regexp.MustCompile(`[^\p{L}\w\s]`)
+
+	nameCN := re.ReplaceAllString(strings.ToLower(m.NameCn), " ")
+	nameEN := re.ReplaceAllString(strings.ToLower(m.NameEn), " ")
+	nameCN = strings.Join(strings.Fields(nameCN), " ")
+	nameEN = strings.Join(strings.Fields(nameEN), " ")
+
 	for _, name := range names {
-		re := regexp.MustCompile(`[^\p{L}\w\s]`)
 		name = re.ReplaceAllString(strings.ToLower(name), " ")
-		nameCN := re.ReplaceAllString(strings.ToLower(m.NameCn), " ")
-		nameEN := re.ReplaceAllString(strings.ToLower(m.NameEn), " ")
 		name = strings.Join(strings.Fields(name), " ")
-		nameCN = strings.Join(strings.Fields(nameCN), " ")
-		nameEN = strings.Join(strings.Fields(nameEN), " ")
-		if  strings.Contains(nameCN, name) || strings.Contains(nameEN, name) {
+		if utils.IsASCII(name) { //ascii name should match words
+			re := regexp.MustCompile(`\b` + name + `\b`)
+			if re.MatchString(nameCN) || re.MatchString(nameEN) {
+				return true
+			} else {
+				continue
+			}
+		}
+
+		if strings.Contains(nameCN, name) || strings.Contains(nameEN, name) {
 			return true
 		}
-	
+
 	}
 	return false
 }
 
-
-func ParseTv(name string) *Metadata {
+func ParseTv(name string) *Info {
 	name = strings.ToLower(name)
 	name = strings.ReplaceAll(name, "\u200b", "") //remove unicode hidden character
-	if utils.ContainsChineseChar(name) {
-		return parseChineseName(name)
-	}
-	return parseEnglishName(name)
+
+	return parseName(name)
 }
 
-func parseEnglishName(name string) *Metadata {
-	re := regexp.MustCompile(`[^\p{L}\w\s]`)
-	name = re.ReplaceAllString(strings.ToLower(name), " ")
-	newSplits := strings.Split(strings.TrimSpace(name), " ")
-
-	seasonRe := regexp.MustCompile(`^s\d{1,2}`)
-	resRe := regexp.MustCompile(`^\d{3,4}p`)
-	episodeRe := regexp.MustCompile(`e\d{1,3}`)
-
-	var seasonIndex = -1
-	var episodeIndex = -1
-	var resIndex = -1
-	for i, p := range newSplits {
-		p = strings.TrimSpace(p)
-		if p == "" {
-			continue
-		}
-		if seasonRe.MatchString(p) {
-			//season part
-			seasonIndex = i
-		} else if resRe.MatchString(p) {
-			resIndex = i
-		}
-		if i >= seasonIndex && episodeRe.MatchString(p) {
-			episodeIndex = i
+func adjacentNumber(s string, start int) (n1 int, l int) {
+	runes := []rune(s)
+	if start > len(runes)-1 { //out of bound
+		return -1, -1
+	}
+	var n []rune
+	for i := start; i < len(runes); i++ {
+		k := runes[i]
+		if (k < '0' || k > '9') && !chineseNum[k] { //not digit anymore
+			break
 		}
+		n = append(n, k)
 	}
-
-	meta := &Metadata{
-		Season:  -1,
-		Episode: -1,
+	if len(n) == 0 {
+		return -1, -1
 	}
-	if seasonIndex != -1 {
-		//season exists
-		ss := seasonRe.FindAllString(newSplits[seasonIndex], -1)
-		if len(ss) != 0 {
-			//season info
-
-			ssNum := strings.TrimLeft(ss[0], "s")
-			n, err := strconv.Atoi(ssNum)
-			if err != nil {
-				panic(fmt.Sprintf("convert %s error: %v", ssNum, err))
-			}
-			meta.Season = n
-		}
-	} else { //maybe like Season 1?
-		seasonRe := regexp.MustCompile(`season \d{1,2}`)
-		matches := seasonRe.FindAllString(name, -1)
-		if len(matches) > 0 {
-			for i, s := range newSplits {
-				if s == "season" {
-					seasonIndex = i
-				}
-			}
-			numRe := regexp.MustCompile(`\d{1,2}`)
-			seNum := numRe.FindAllString(matches[0], -1)[0]
-			n, err := strconv.Atoi(seNum)
-			if err != nil {
-				panic(fmt.Sprintf("convert %s error: %v", seNum, err))
-			}
-			meta.Season = n
-
-		}
+	m, err := strconv.Atoi(string(n))
+	if err != nil {
+		return chinese2Num[string(n)], len(n)
 	}
-	if episodeIndex != -1 {
-		ep := episodeRe.FindAllString(newSplits[episodeIndex], -1)
-		if len(ep) > 0 {
-			//episode info exists
-			epNum := strings.TrimLeft(ep[0], "e")
-			n, err := strconv.Atoi(epNum)
-			if err != nil {
-				panic(fmt.Sprintf("convert %s error: %v", epNum, err))
-			}
-			meta.Episode = n
-		}
-	} else { //no episode, maybe like  One Punch Man S2 - 08 [1080p].mkv
-
-		// numRe := regexp.MustCompile(`^\d{1,2}$`)
-		// for i, p := range newSplits {
-		// 	if numRe.MatchString(p) {
-		// 		if i > 0 && strings.Contains(newSplits[i-1], "season") { //last word cannot be season
-		// 			continue
-		// 		}
-		// 		if i < seasonIndex {
-		// 			//episode number most likely  should comes alfter season number
-		// 			continue
-		// 		}
-		// 		//episodeIndex = i
-		// 		n, err := strconv.Atoi(p)
-		// 		if err != nil {
-		// 			panic(fmt.Sprintf("convert %s error: %v", p, err))
-		// 		}
-		// 		meta.Episode = n
-
-		// 	}
-		// }
-
-	}
-	if resIndex != -1 {
-		//resolution exists
-		meta.Resolution = newSplits[resIndex]
-	}
-	if meta.Episode == -1 {
-		meta.Episode = -1
-		meta.IsSeasonPack = true
-	}
-
-	if seasonIndex > 0 {
-		//name exists
-		names := newSplits[0:seasonIndex]
-		meta.NameEn = strings.TrimSpace(strings.Join(names, " "))
-	} else {
-		meta.NameEn = name
-	}
-
-	return meta
+	return m, len(n)
 }
 
-func parseChineseName(name string) *Metadata {
-	var meta = parseEnglishName(name)
-	if meta.Season != -1 && (meta.Episode != -1 || meta.IsSeasonPack) {
-		return meta
-	}
-	meta = &Metadata{Season: 1}
-	//season pack
-	packRe := regexp.MustCompile(`(\d{1,2}-\d{1,2})|(全集)`)
-	if packRe.MatchString(name) {
-		meta.IsSeasonPack = true
-	}
-	//resolution
-	resRe := regexp.MustCompile(`\d{3,4}p`)
-	resMatches := resRe.FindAllString(name, -1)
-	if len(resMatches) != 0 {
-		meta.Resolution = resMatches[0]
-	} else {
-		if strings.Contains(name, "720") {
-			meta.Resolution = "720p"
-		} else if strings.Contains(name, "1080") {
-			meta.Resolution = "1080p"
-		}
-
-	}
-
-	//episode number
-	re1 := regexp.MustCompile(`\[\d{1,3}\]`)
-	episodeMatches1 := re1.FindAllString(name, -1)
-	if len(episodeMatches1) > 0 { //[11] [1080p]
-		epNum := strings.TrimRight(strings.TrimLeft(episodeMatches1[0], "["), "]")
-		n, err := strconv.Atoi(epNum)
-		if err != nil {
-			panic(fmt.Sprintf("convert %s error: %v", epNum, err))
-		}
-		meta.Episode = n
-	} else { //【第09話】
-		re2 := regexp.MustCompile(`第\d{1,4}(话|話|集)`)
-		episodeMatches1 := re2.FindAllString(name, -1)
-		if len(episodeMatches1) > 0 {
-			re := regexp.MustCompile(`\d{1,4}`)
-			epNum := re.FindAllString(episodeMatches1[0], -1)[0]
-			n, err := strconv.Atoi(epNum)
-			if err != nil {
-				panic(fmt.Sprintf("convert %s error: %v", epNum, err))
-			}
-			meta.Episode = n
-		} else { //SHY 靦腆英雄 / Shy -05 ( CR 1920x1080 AVC AAC MKV)
-			re3 := regexp.MustCompile(`[^\d\w]\d{1,2}[^\d\w]`)
-			epNums := re3.FindAllString(name, -1)
-			if len(epNums) > 0 {
-
-				re3 := regexp.MustCompile(`\d{1,2}`)
-				epNum := re3.FindAllString(epNums[0], -1)[0]
-				n, err := strconv.Atoi(epNum)
-				if err != nil {
-					panic(fmt.Sprintf("convert %s error: %v", epNum, err))
-				}
-				meta.Episode = n
-			}
-		}
-	}
-
+func findSeason(s string) (n int, p int) {
 	//season numner
 	seasonRe1 := regexp.MustCompile(`s\d{1,2}`)
-	seasonMatches := seasonRe1.FindAllString(name, -1)
+	seasonMatches := seasonRe1.FindAllString(s, -1)
 	if len(seasonMatches) > 0 {
 		seNum := seasonMatches[0][1:]
 		n, err := strconv.Atoi(seNum)
 		if err != nil {
 			panic(fmt.Sprintf("convert %s error: %v", seNum, err))
 		}
-		meta.Season = n
+
+		return n, strings.Index(s, seNum)
 	} else {
 		seasonRe1 := regexp.MustCompile(`season \d{1,2}`)
-		seasonMatches := seasonRe1.FindAllString(name, -1)
+		seasonMatches := seasonRe1.FindAllString(s, -1)
 		if len(seasonMatches) > 0 {
 			re3 := regexp.MustCompile(`\d{1,2}`)
 			seNum := re3.FindAllString(seasonMatches[0], -1)[0]
@@ -245,10 +112,10 @@ func parseChineseName(name string) *Metadata {
 			if err != nil {
 				panic(fmt.Sprintf("convert %s error: %v", seNum, err))
 			}
-			meta.Season = n
+			return n, strings.Index(s, seasonMatches[0])
 		} else {
 			seasonRe1 := regexp.MustCompile(`第.{1,2}季`)
-			seasonMatches := seasonRe1.FindAllString(name, -1)
+			seasonMatches := seasonRe1.FindAllString(s, -1)
 			if len(seasonMatches) > 0 {
 				m1 := []rune(seasonMatches[0])
 				seNum := m1[1 : len(m1)-1]
@@ -257,58 +124,364 @@ func parseChineseName(name string) *Metadata {
 					log.Warnf("parse season number %v error: %v, try to parse using chinese", seNum, err)
 					n = chinese2Num[string(seNum)]
 				}
-				meta.Season = n
+				return n, strings.Index(s, seasonMatches[0])
+			}
+		}
+	}
+	return -1, -1
+}
 
+func findEpisodes(s string) (start int, end int) {
+	var episodeCn = map[rune]bool{
+		'话': true,
+		'話': true,
+		'集': true,
+	}
+
+	rr := []rune(s)
+	for i := 0; i < len(rr); i++ {
+		r := rr[i]
+		if r == 'e' {
+			n, l := adjacentNumber(s, i+1)
+
+			if n > 0 {
+				foundDash := false
+				for j := i + l + 1; j < len(rr); j++ {
+					r1 := rr[j]
+					if r1 == '-' {
+						foundDash = true
+						continue
+					}
+					if r1 == ' ' || r1 == 'e' {
+						continue
+					}
+
+					if foundDash {
+						if r1 == 's' {
+							s1, l1 := adjacentNumber(s, j+1)
+							if s1 > 0 { //S01E01-S01E21
+								n1, _ := adjacentNumber(s, j+l1+2)
+								if n1 > 0 {
+									return n, n1
+								}
+							}
+						}
+						n1, _ := adjacentNumber(s, j)
+						if n1 > 0 {
+							return n, n1
+						}
+					} else {
+						break
+					}
+				}
+				return n, n
+			}
+		} else if r == '第' {
+			n, l := adjacentNumber(s, i+1)
+			if len(rr) > i+l+1 && episodeCn[rr[i+l+1]] {
+				return n, n
+			} else if len(rr) > i+l+1 {
+				if rr[i+l+1] == '-' {
+					n1, l1 := adjacentNumber(s, i+l+2)
+					if episodeCn[rr[i+l+2+l1]] {
+						return n, n1
+					}
+				}
+			}
+
+		}
+	}
+	//episode number
+	re1 := regexp.MustCompile(`\[\d{1,4}\]`)
+	episodeMatches1 := re1.FindAllString(s, -1)
+	if len(episodeMatches1) > 0 { //[11] [1080p], [2022][113][HEVC][GB][4K]
+		for _, m := range episodeMatches1 {
+			epNum := strings.TrimRight(strings.TrimLeft(m, "["), "]")
+			n, err := strconv.Atoi(epNum)
+			if err != nil {
+				log.Debugf("convert %s error: %v", epNum, err)
+				continue
+			}
+			nowYear := time.Now().Year()
+			if n > nowYear-50 { //high possibility is year number
+				continue
+			}
+			return n, n
+		}
+	} else { //【第09話】
+		re2 := regexp.MustCompile(`第\d{1,4}([话話集])`)
+		episodeMatches1 := re2.FindAllString(s, -1)
+		if len(episodeMatches1) > 0 {
+			re := regexp.MustCompile(`\d{1,4}`)
+			epNum := re.FindAllString(episodeMatches1[0], -1)[0]
+			n, err := strconv.Atoi(epNum)
+			if err != nil {
+				panic(fmt.Sprintf("convert %s error: %v", epNum, err))
+			}
+			return n, n
+		} else { //The Road Season 2 Episode 12 XviD-AFG
+			re3 := regexp.MustCompile(`episode \d{1,4}`)
+			epNums := re3.FindAllString(s, -1)
+			if len(epNums) > 0 {
+				re3 := regexp.MustCompile(`\d{1,4}`)
+				epNum := re3.FindAllString(epNums[0], -1)[0]
+				n, err := strconv.Atoi(epNum)
+				if err != nil {
+					panic(fmt.Sprintf("convert %s error: %v", epNum, err))
+				}
+				return n, n
+
+			} else { //SHY 靦腆英雄 / Shy -05 ( CR 1920x1080 AVC AAC MKV)
+				if maybeSeasonPack(s) { //avoid miss match, season pack not use this rule
+					return -1, -1
+				}
+				re3 := regexp.MustCompile(`[^(season)][^\d\w]\d{1,2}[^\d\w]`)
+				epNums := re3.FindAllString(s, -1)
+				if len(epNums) > 0 {
+
+					re3 := regexp.MustCompile(`\d{1,2}`)
+					epNum := re3.FindAllString(epNums[0], -1)[0]
+					n, err := strconv.Atoi(epNum)
+					if err != nil {
+						panic(fmt.Sprintf("convert %s error: %v", epNum, err))
+					}
+					return n, n
+				}
 			}
 		}
 	}
 
-	if meta.IsSeasonPack && meta.Episode != 0 {
-		meta.Season = meta.Episode
-		meta.Episode = -1
+	return -1, -1
+}
+
+func matchResolution(s string) string {
+	//resolution
+	resRe := regexp.MustCompile(`\d{3,4}p`)
+	resMatches := resRe.FindAllString(s, -1)
+	if len(resMatches) != 0 {
+		return resMatches[0]
+	} else {
+		if strings.Contains(s, "720") {
+			return "720p"
+		} else if strings.Contains(s, "1080") {
+			return "1080p"
+		}
 	}
+	return ""
+}
+
+func maybeSeasonPack(s string) bool {
+	//season pack
+	packRe := regexp.MustCompile(`((\d{1,2}-\d{1,2}))|(complete)|(全集)`)
+	if packRe.MatchString(s) {
+		return true
+	}
+	return false
+}
+
+//func parseEnglishName(name string) *Info {
+//	meta := &Info{
+//		//Season:  -1,
+//		Episode: -1,
+//	}
+//
+//	start, end := findEpisodes(name)
+//	if start > 0 && end > 0 {
+//		meta.Episode = start
+//	}
+//
+//	re := regexp.MustCompile(`[^\p{L}\w\s]`)
+//	name = re.ReplaceAllString(strings.ToLower(name), " ")
+//	newSplits := strings.Split(strings.TrimSpace(name), " ")
+//
+//	seasonRe := regexp.MustCompile(`^s\d{1,2}`)
+//	resRe := regexp.MustCompile(`^\d{3,4}p`)
+//	episodeRe := regexp.MustCompile(`e\d{1,3}`)
+//
+//	var seasonIndex = -1
+//	var episodeIndex = -1
+//	var resIndex = -1
+//	for i, p := range newSplits {
+//		p = strings.TrimSpace(p)
+//		if p == "" {
+//			continue
+//		}
+//		if seasonRe.MatchString(p) {
+//			//season part
+//			seasonIndex = i
+//		} else if resRe.MatchString(p) {
+//			resIndex = i
+//		}
+//		if i >= seasonIndex && episodeRe.MatchString(p) {
+//			episodeIndex = i
+//		}
+//	}
+//
+//	if seasonIndex != -1 {
+//		//season exists
+//		ss := seasonRe.FindAllString(newSplits[seasonIndex], -1)
+//		if len(ss) != 0 {
+//			//season info
+//
+//			ssNum := strings.TrimLeft(ss[0], "s")
+//			n, err := strconv.Atoi(ssNum)
+//			if err != nil {
+//				panic(fmt.Sprintf("convert %s error: %v", ssNum, err))
+//			}
+//			meta.Season = n
+//		}
+//	} else { //maybe like Season 1?
+//		seasonRe := regexp.MustCompile(`season \d{1,2}`)
+//		matches := seasonRe.FindAllString(name, -1)
+//		if len(matches) > 0 {
+//			for i, s := range newSplits {
+//				if s == "season" {
+//					seasonIndex = i
+//				}
+//			}
+//			numRe := regexp.MustCompile(`\d{1,2}`)
+//			seNum := numRe.FindAllString(matches[0], -1)[0]
+//			n, err := strconv.Atoi(seNum)
+//			if err != nil {
+//				panic(fmt.Sprintf("convert %s error: %v", seNum, err))
+//			}
+//			meta.Season = n
+//
+//		}
+//	}
+//
+//	if episodeIndex != -1 {
+//		//	ep := episodeRe.FindAllString(newSplits[episodeIndex], -1)
+//		//if len(ep) > 0 {
+//		//	//episode info exists
+//		//	epNum := strings.TrimLeft(ep[0], "e")
+//		//	n, err := strconv.Atoi(epNum)
+//		//	if err != nil {
+//		//		panic(fmt.Sprintf("convert %s error: %v", epNum, err))
+//		//	}
+//		//	meta.Episode = n
+//		//}
+//	} else { //no episode, maybe like  One Punch Man S2 - 08 [1080p].mkv
+//
+//		// numRe := regexp.MustCompile(`^\d{1,2}$`)
+//		// for i, p := range newSplits {
+//		// 	if numRe.MatchString(p) {
+//		// 		if i > 0 && strings.Contains(newSplits[i-1], "season") { //last word cannot be season
+//		// 			continue
+//		// 		}
+//		// 		if i < seasonIndex {
+//		// 			//episode number most likely  should comes alfter season number
+//		// 			continue
+//		// 		}
+//		// 		//episodeIndex = i
+//		// 		n, err := strconv.Atoi(p)
+//		// 		if err != nil {
+//		// 			panic(fmt.Sprintf("convert %s error: %v", p, err))
+//		// 		}
+//		// 		meta.Episode = n
+//
+//		// 	}
+//		// }
+//
+//	}
+//	if resIndex != -1 {
+//		//resolution exists
+//		meta.Resolution = newSplits[resIndex]
+//	}
+//	if meta.Episode == -1 {
+//		meta.Episode = -1
+//		meta.IsSeasonPack = true
+//	}
+//
+//	if seasonIndex > 0 {
+//		//name exists
+//		names := newSplits[0:seasonIndex]
+//		meta.NameEn = strings.TrimSpace(strings.Join(names, " "))
+//	} else {
+//		meta.NameEn = name
+//	}
+//
+//	return meta
+//}
+
+func parseName(name string) *Info {
+	meta := &Info{Season: 1}
+	if strings.TrimSpace(name) == "" {
+		return meta
+	}
+
+	season, p := findSeason(name)
+	if season == -1 {
+		log.Debugf("not find season info: %s", name)
+		if !utils.IsASCII(name) {
+			season = 1
+		}
+		p = len(name) - 1
+	}
+	meta.Season = season
+
+	start, end := findEpisodes(name)
+	if start > 0 && end > 0 {
+		meta.StartEpisode = start
+		meta.EndEpisode = end
+	} else {
+		meta.IsSeasonPack = true
+	}
+
+	meta.Resolution = matchResolution(name)
+
+	//if meta.IsSeasonPack && meta.Episode != 0 {
+	//	meta.Season = meta.Episode
+	//	meta.Episode = -1
+	//}
 
 	//tv name
-
-	fields := strings.FieldsFunc(name, func(r rune) bool {
-		return r == '[' || r == ']' || r == '【' || r == '】'
-	})
-	titleCn := ""
-	title := ""
-	for _, p := range fields { //寻找匹配的最长的字符串，最有可能是名字
-		if utils.ContainsChineseChar(p) && len([]rune(p)) > len([]rune(titleCn)) { //最长含中文字符串
-			titleCn = p
-		}
-		if len([]rune(p)) > len([]rune(title)) { //最长字符串
-			title = p
-		}
-	}
-	re := regexp.MustCompile(`[^\p{L}\w\s]`)
-	title = re.ReplaceAllString(strings.TrimSpace(strings.ToLower(title)), "") //去除标点符号
-	titleCn = re.ReplaceAllString(strings.TrimSpace(strings.ToLower(titleCn)), "")
-
-	meta.NameCn = titleCn
-	cnRe := regexp.MustCompile(`\p{Han}.*\p{Han}`)
-	cnmatches := cnRe.FindAllString(titleCn, -1)
-
-	//titleCn中最长的中文字符
-	if len(cnmatches) > 0 {
-		for _, t := range cnmatches {
-			if len([]rune(t)) > len([]rune(meta.NameCn)) {
-				meta.NameCn = strings.ToLower(t)
+	if utils.IsASCII(name) && p < len(name) && p-1 > 0 {
+		meta.NameEn = strings.TrimSpace(name[:p-1])
+		meta.NameCn = meta.NameEn
+	} else {
+		fields := strings.FieldsFunc(name, func(r rune) bool {
+			return r == '[' || r == ']' || r == '【' || r == '】'
+		})
+		titleCn := ""
+		title := ""
+		for _, p := range fields { //寻找匹配的最长的字符串，最有可能是名字
+			if utils.ContainsChineseChar(p) && len([]rune(p)) > len([]rune(titleCn)) { //最长含中文字符串
+				titleCn = p
+			}
+			if len([]rune(p)) > len([]rune(title)) { //最长字符串
+				title = p
 			}
 		}
-	}
+		re := regexp.MustCompile(`[^\p{L}\w\s]`)
+		title = re.ReplaceAllString(strings.TrimSpace(strings.ToLower(title)), "") //去除标点符号
+		titleCn = re.ReplaceAllString(strings.TrimSpace(strings.ToLower(titleCn)), "")
 
-	//匹配title中最长拉丁字符串
-	enRe := regexp.MustCompile(`[[:ascii:]]*`)
-	enM := enRe.FindAllString(title, -1)
-	if len(enM) > 0 {
-		for _, t := range enM {
-			if len(t) > len(meta.NameEn) {
-				meta.NameEn = strings.TrimSpace(strings.ToLower(t))
+		meta.NameCn = titleCn
+		cnRe := regexp.MustCompile(`\p{Han}.*\p{Han}`)
+		cnmatches := cnRe.FindAllString(titleCn, -1)
+
+		//titleCn中最长的中文字符
+		if len(cnmatches) > 0 {
+			for _, t := range cnmatches {
+				if len([]rune(t)) > len([]rune(meta.NameCn)) {
+					meta.NameCn = strings.ToLower(t)
+				}
 			}
 		}
+		meta.NameEn = title
+
+		////匹配title中最长拉丁字符串
+		//enRe := regexp.MustCompile(`[[:ascii:]]*`)
+		//enM := enRe.FindAllString(title, -1)
+		//if len(enM) > 0 {
+		//	for _, t := range enM {
+		//		if len(t) > len(meta.NameEn) {
+		//			meta.NameEn = strings.TrimSpace(strings.ToLower(t))
+		//		}
+		//	}
+		//}
+
 	}
 
 	return meta
@@ -325,3 +498,15 @@ var chinese2Num = map[string]int{
 	"八": 8,
 	"九": 9,
 }
+
+var chineseNum = map[rune]bool{
+	'一': true,
+	'二': true,
+	'三': true,
+	'四': true,
+	'五': true,
+	'六': true,
+	'七': true,
+	'八': true,
+	'九': true,
+}

pkg/metadata/tv_test.go

@@ -21,7 +21,7 @@ func Test_ParseTV2(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 1)
-	assert.Equal(t, m.Episode, 4)
+	assert.Equal(t, m.StartEpisode, 4)
 	assert.Equal(t, m.IsSeasonPack, false)
 	assert.Equal(t, m.Resolution, "1080p")
 }
@@ -31,7 +31,7 @@ func Test_ParseTV3(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 37)
-	assert.Equal(t, m.Episode, 219)
+	assert.Equal(t, m.StartEpisode, 219)
 	assert.Equal(t, m.IsSeasonPack, false)
 	//assert.Equal(t, m.Resolution, "1080p")
 }
@@ -41,8 +41,8 @@ func Test_ParseTV4(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 2)
-	//assert.Equal(t, m.Episode, 219)
-	assert.Equal(t, m.IsSeasonPack, true)
+	assert.Equal(t, m.StartEpisode, 12)
+	assert.Equal(t, m.IsSeasonPack, false)
 	//assert.Equal(t, m.Resolution, "1080p")
 }
 
@@ -61,7 +61,7 @@ func Test_ParseTV6(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 1)
-	assert.Equal(t, m.Episode, 3)
+	assert.Equal(t, m.StartEpisode, 3)
 	assert.Equal(t, m.IsSeasonPack, false)
 	assert.Equal(t, m.Resolution, "1080p")
 }
@@ -71,7 +71,7 @@ func Test_ParseTV7(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 1)
-	assert.Equal(t, m.Episode, 1113)
+	assert.Equal(t, m.StartEpisode, 1113)
 	assert.Equal(t, m.IsSeasonPack, false)
 	assert.Equal(t, m.Resolution, "1080p")
 }
@@ -81,7 +81,7 @@ func Test_ParseTV8(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 1)
-	assert.Equal(t, m.Episode, 4)
+	assert.Equal(t, m.StartEpisode, 4)
 	assert.Equal(t, m.IsSeasonPack, false)
 	assert.Equal(t, m.Resolution, "1080p")
 }
@@ -91,7 +91,7 @@ func Test_ParseTV9(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, m.Season, 1)
-	assert.Equal(t, m.Episode, 16)
+	assert.Equal(t, m.StartEpisode, 16)
 	assert.Equal(t, m.IsSeasonPack, false)
 	assert.Equal(t, m.Resolution, "1080p")
 }
@@ -111,7 +111,7 @@ func Test_ParseTV11(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, 2, m.Season)
-	assert.Equal(t, 4, m.Episode)
+	assert.Equal(t, 4, m.StartEpisode)
 	assert.Equal(t, false, m.IsSeasonPack)
 	assert.Equal(t, "1080p", m.Resolution)
 }
@@ -121,7 +121,7 @@ func Test_ParseTV12(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, 2, m.Season)
-	assert.Equal(t, 4, m.Episode)
+	assert.Equal(t, 4, m.StartEpisode)
 	assert.Equal(t, false, m.IsSeasonPack)
 	assert.Equal(t, "1080p", m.Resolution)
 }
@@ -131,7 +131,7 @@ func Test_ParseTV13(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, 2, m.Season)
-	assert.Equal(t, 8, m.Episode)
+	assert.Equal(t, 8, m.StartEpisode)
 	assert.Equal(t, false, m.IsSeasonPack)
 	assert.Equal(t, "1080p", m.Resolution)
 }
@@ -141,7 +141,75 @@ func Test_ParseTV14(t *testing.T) {
 	m := ParseTv(s1)
 	log.Infof("results: %+v", m)
 	assert.Equal(t, 5, m.Season)
-	assert.Equal(t, 113, m.Episode)
+	assert.Equal(t, 113, m.StartEpisode)
 	assert.Equal(t, false, m.IsSeasonPack)
 	//assert.Equal(t, "720p", m.Resolution)
 }
+
+//
+
+func Test_ParseTV15(t *testing.T) {
+	s1 := "[7³ACG] 剧场版 回转企鹅罐 [前篇] 你的列车是生存战略/RE:cycle of the PENGUINDRUM  Zenpen [简繁字幕] BDrip 1080p x265 FLAC"
+	m := ParseTv(s1)
+	log.Infof("results: %+v", m)
+	b := m.IsAcceptable("The Penguin")
+	assert.False(t, b)
+	//assert.Equal(t, 1, m.Season)
+	//assert.Equal(t, 113, m.Episode)
+	//assert.Equal(t, false, m.IsSeasonPack)
+	//assert.Equal(t, "720p", m.Resolution)
+}
+
+func Test_ParseTV16(t *testing.T) {
+	s1 := "Romance in the Alley 2024 S01 E24-E25 1080p WEB-DL H.264 AAC-PTerWEB"
+	m := ParseTv(s1)
+	log.Infof("results: %+v", m)
+
+	assert.Equal(t, 1, m.Season)
+	assert.Equal(t, 24, m.StartEpisode)
+	assert.Equal(t, 25, m.EndEpisode)
+	//assert.Equal(t, false, m.IsSeasonPack)
+	//assert.Equal(t, "720p", m.Resolution)
+}
+
+func Test_ParseTV17(t *testing.T) {
+	s1 := "小巷人家/Romance in the Alley 第24-25集 "
+	m := ParseTv(s1)
+	log.Infof("results: %+v", m)
+
+	assert.Equal(t, 1, m.Season)
+	assert.Equal(t, 24, m.StartEpisode)
+	assert.Equal(t, 25, m.EndEpisode)
+	//assert.Equal(t, false, m.IsSeasonPack)
+	//assert.Equal(t, "720p", m.Resolution)
+}
+
+// Romance in the Alley 2024 S01E01-S01E21 2160p WEB-DL HEVC AAC-UBWEB
+func Test_ParseTV18(t *testing.T) {
+	s1 := "Romance in the Alley 2024 S01E01-S01E21 2160p WEB-DL HEVC AAC-UBWEB "
+	m := ParseTv(s1)
+	log.Infof("results: %+v", m)
+
+	assert.Equal(t, 1, m.Season)
+	assert.Equal(t, 1, m.StartEpisode)
+	assert.Equal(t, 21, m.EndEpisode)
+	//assert.Equal(t, false, m.IsSeasonPack)
+	//assert.Equal(t, "720p", m.Resolution)
+}
+
+// The Day of the Jackal (Season 1) WEB-DL 1080​p
+func Test_ParseTV19(t *testing.T) {
+	s1 := "The Day of the Jackal (Season 1) WEB-DL 1080​p "
+	m := ParseTv(s1)
+	log.Infof("results: %+v", m)
+
+	assert.Equal(t, 1, m.Season)
+	assert.Equal(t, true, m.IsSeasonPack)
+	//assert.Equal(t, "720p", m.Resolution)
+}
+
+func Test_Name(t *testing.T) {
+	m := Info{NameEn: "word кибердеревня новый год 2023 webrip 1080p"}
+	b := m.IsAcceptable("word")
+	assert.True(t, b)
+}

pkg/prowlarr/prowlarr.go

@@ -0,0 +1,77 @@
+package prowlarr
+
+import (
+	"encoding/json"
+	"fmt"
+	"polaris/db"
+	"polaris/ent"
+	"strings"
+	"time"
+
+	"golift.io/starr"
+	"golift.io/starr/prowlarr"
+)
+
+type ProwlarrSupportType string
+
+const (
+	TV    ProwlarrSupportType = "tv"
+	Movie ProwlarrSupportType = "movie"
+)
+
+type Client struct {
+	p      *prowlarr.Prowlarr
+	apiKey string
+	url    string
+}
+
+func New(apiKey, url string) *Client {
+	c := starr.New(apiKey, url, 10*time.Second)
+	p := prowlarr.New(c)
+	return &Client{p: p, apiKey: apiKey, url: url}
+}
+
+func (c *Client) GetIndexers(t ProwlarrSupportType) ([]*db.TorznabInfo, error) {
+	ins, err := c.p.GetIndexers()
+	if err != nil {
+		return nil, err
+	}
+	var indexers []*db.TorznabInfo
+	for _, in := range ins {
+		if !in.Enable {
+			continue
+		}
+		if t == "tv" && len(in.Capabilities.TvSearchParams) == 0 { //no tv resource in this indexer
+			continue
+		} else if t == "movie" && len(in.Capabilities.MovieSearchParams) == 0 { //no movie resource in this indexer
+			continue
+		}
+		seedRatio := 0.0
+		for _, f := range in.Fields {
+			if f.Name == "torrentBaseSettings.seedRatio" && f.Value != nil {
+				if r, ok := f.Value.(float64); ok {
+					seedRatio = r
+				}
+			}
+		}
+		setting := db.TorznabSetting{
+			URL:    fmt.Sprintf("%s/%d/api", strings.TrimSuffix(c.url, "/"), in.ID),
+			ApiKey: c.apiKey,
+		}
+		data, _ := json.Marshal(&setting)
+
+		entIndexer := ent.Indexers{
+			Name:           in.Name,
+			Implementation: "torznab",
+			Priority:       128 - int(in.Priority),
+			SeedRatio:      float32(seedRatio),
+			Settings:       string(data),
+		}
+
+		indexers = append(indexers, &db.TorznabInfo{
+			Indexers:       &entIndexer,
+			TorznabSetting: setting,
+		})
+	}
+	return indexers, nil
+}

pkg/prowlarr/prowlarr_test.go

@@ -0,0 +1,13 @@
+package prowlarr
+
+import (
+	"polaris/log"
+	"testing"
+)
+
+func Test111(t *testing.T) {
+	c := New("", "http://10.0.0.8:9696/")
+	apis , err := c.GetIndexers("tv")
+	log.Infof("errors: %v", err)
+	log.Infof("indexers: %+v", apis[0])
+}

pkg/qbittorrent/qbittorrent.go

@@ -1,15 +1,18 @@
 package qbittorrent
 
 import (
-	"encoding/json"
 	"fmt"
+	"os"
 	"polaris/pkg"
 	"polaris/pkg/go-qbittorrent/qbt"
-	"time"
+	"polaris/pkg/utils"
+	"strings"
 
 	"github.com/pkg/errors"
 )
 
+const btCategory = "polaris"
+
 type Info struct {
 	URL      string
 	User     string
@@ -17,7 +20,8 @@ type Info struct {
 }
 
 type Client struct {
-	c *qbt.Client
+	c        *qbt.Client
+	category string
 	Info
 }
 
@@ -35,20 +39,20 @@ func NewClient(url, user, pass string) (*Client, error) {
 		return nil, err
 	}
 
-	return &Client{c: qb, Info: Info{URL: url, User: user, Password: pass}}, nil
+	return &Client{c: qb, category: btCategory, Info: Info{URL: url, User: user, Password: pass}}, nil
 }
 
 func (c *Client) GetAll() ([]pkg.Torrent, error) {
-	tt, err :=c.c.Torrents(qbt.TorrentsOptions{})
+	tt, err := c.c.Torrents(qbt.TorrentsOptions{Category: &c.category})
 	if err != nil {
 		return nil, errors.Wrap(err, "get torrents")
 	}
 	var res []pkg.Torrent
 	for _, t := range tt {
 		t1 := &Torrent{
-			c: c.c,
-			Hash: t.Hash,
-			Info: c.Info,
+			c:    c.c,
+			hash: t.Hash,
+			//Info: c.Info,
 		}
 		res = append(res, t1)
 	}
@@ -56,71 +60,92 @@ func (c *Client) GetAll() ([]pkg.Torrent, error) {
 }
 
 func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
-	all, err := c.c.Torrents(qbt.TorrentsOptions{})
+	magnet, err := utils.Link2Magnet(link)
 	if err != nil {
-		return nil, errors.Wrap(err, "get old torrents")
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
 	}
-	allHash := make(map[string]bool, len(all))
-	for _, t := range all {
-		allHash[t.Hash] = true
+
+	hash, err := utils.MagnetHash(magnet)
+	if err != nil {
+		return nil, errors.Wrap(err, "get hash")
 	}
-	err = c.c.DownloadLinks([]string{link}, qbt.DownloadOptions{Savepath: &dir})
+	err = c.c.DownloadLinks([]string{magnet}, qbt.DownloadOptions{Savepath: &dir, Category: &c.category})
 	if err != nil {
 		return nil, errors.Wrap(err, "qbt download")
 	}
-	var newHash string
+	return &Torrent{hash: hash, c: c.c}, nil
 
-loop:
-	for i := 0; i < 10; i++ {
-		time.Sleep(1 * time.Second)
-		all, err = c.c.Torrents(qbt.TorrentsOptions{})
-		if err != nil {
-			return nil, errors.Wrap(err, "get new torrents")
-		}
+}
 
-		for _, t := range all {
-			if !allHash[t.Hash] {
-				newHash = t.Hash
-				break loop
-			}
-		}
+func NewTorrent(info Info, link string) (*Torrent, error) {
+	c, err := NewClient(info.URL, info.User, info.Password)
+	if err != nil {
+		return nil, err
+	}
+	magnet, err := utils.Link2Magnet(link)
+	if err != nil {
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
 	}
 
-	if newHash == "" {
-		return nil, fmt.Errorf("download torrent fail: timeout")
+	hash, err := utils.MagnetHash(magnet)
+	if err != nil {
+		return nil, err
 	}
-	return &Torrent{Hash: newHash, c: c.c, Info: c.Info}, nil
-
+	t := &Torrent{
+		c:    c.c,
+		hash: hash,
+	}
+	if !t.Exists() {
+		return nil, errors.Errorf("torrent not exist: %v", magnet)
+	}
+	return t, nil
 }
 
 type Torrent struct {
 	c    *qbt.Client
-	Hash string
-	Info
+	hash string
+	//info Info
 }
 
 func (t *Torrent) GetHash() string {
-	return t.Hash
+	return t.hash
 }
 
 func (t *Torrent) getTorrent() (*qbt.TorrentInfo, error) {
-	all, err := t.c.Torrents(qbt.TorrentsOptions{Hashes: []string{t.Hash}})
+	all, err := t.c.Torrents(qbt.TorrentsOptions{Hashes: []string{t.hash}})
 	if err != nil {
 		return nil, err
 	}
 	if len(all) == 0 {
-		return nil, fmt.Errorf("no such torrent: %v", t.Hash)
+		return nil, fmt.Errorf("no such torrent: %v", t.hash)
 	}
 	return &all[0], nil
 }
 
 func (t *Torrent) Name() (string, error) {
-	qb, err := t.getTorrent()
+	dir, err := t.getTorrentBaseNameOrDir()
+	if err != nil { //use torrent name
+		qb, err := t.getTorrent()
+		if err != nil {
+			return "", err
+		}
+		return qb.Name, nil
+	}
+	return dir, nil
+}
+
+// https://github.com/qbittorrent/qBittorrent/issues/13572
+func (t *Torrent) getTorrentBaseNameOrDir() (string, error) {
+	files, err := t.c.TorrentFiles(t.hash)
 	if err != nil {
 		return "", err
 	}
-
-	return qb.Name, nil
+	if len(files) == 0 {
+		return "", errors.Wrap(err, "no file")
+	}
+	name := files[0].Name
+	dir := strings.Split(name, string(os.PathSeparator))[0]
+	return dir, nil
 }
 
 func (t *Torrent) Progress() (int, error) {
@@ -140,11 +165,11 @@ func (t *Torrent) Progress() (int, error) {
 }
 
 func (t *Torrent) Stop() error {
-	return t.c.Pause([]string{t.Hash})
+	return t.c.Pause([]string{t.hash})
 }
 
 func (t *Torrent) Start() error {
-	ok, err := t.c.Resume([]string{t.Hash})
+	ok, err := t.c.Resume([]string{t.hash})
 	if err != nil {
 		return err
 	}
@@ -155,7 +180,7 @@ func (t *Torrent) Start() error {
 }
 
 func (t *Torrent) Remove() error {
-	ok, err := t.c.Delete([]string{t.Hash}, true)
+	ok, err := t.c.Delete([]string{t.hash}, true)
 	if err != nil {
 		return err
 	}
@@ -165,11 +190,6 @@ func (t *Torrent) Remove() error {
 	return nil
 }
 
-func (t *Torrent) Save() string {
-	data, _ := json.Marshal(t)
-	return string(data)
-}
-
 func (t *Torrent) Exists() bool {
 	_, err := t.getTorrent()
 	return err == nil
@@ -182,15 +202,3 @@ func (t *Torrent) SeedRatio() (float64, error) {
 	}
 	return qb.Ratio, nil
 }
-
-func (t *Torrent) Reload() error {
-	c, err := NewClient(t.URL, t.User, t.Password)
-	if err != nil {
-		return err
-	}
-	t.c = c.c
-	if !t.Exists() {
-		return errors.Errorf("torrent not exists: %v", t.Hash)
-	}
-	return nil
-}

pkg/qbittorrent/qbittorrent_test.go

@@ -0,0 +1,20 @@
+package qbittorrent
+
+import (
+	"polaris/log"
+	"testing"
+)
+
+func Test1(t *testing.T) {
+	c, err := NewClient("http://10.0.0.8:8081/", "", "")
+	if err != nil {
+		log.Errorf("new client error: %v", err)
+		t.Fail()
+	}
+	all, err := c.GetAll()
+	for _, t := range all {
+		name, _ := t.Name()
+		log.Infof("torrent: %+v", name)
+	}
+	
+}

pkg/storage/alist.go

@@ -0,0 +1,78 @@
+package storage
+
+import (
+	"io"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"polaris/pkg/alist"
+
+	"github.com/gabriel-vasile/mimetype"
+)
+
+func NewAlist(cfg *alist.Config, dir string, videoFormats []string, subtitleFormats []string) (*Alist, error) {
+	cl := alist.New(cfg)
+	_, err := cl.Login()
+	if err != nil {
+		return nil, err
+	}
+	return &Alist{baseDir: dir, cfg: cfg, client: cl, videoFormats: videoFormats, subtitleFormats: subtitleFormats}, nil
+}
+
+type Alist struct {
+	baseDir         string
+	cfg             *alist.Config
+	client          *alist.Client
+	progresser      func() float64
+	videoFormats    []string
+	subtitleFormats []string
+}
+
+func (a *Alist) Move(src, dest string) error {
+	if err := a.Copy(src, dest); err != nil {
+		return err
+	}
+	return os.RemoveAll(src)
+}
+
+func (a *Alist) Copy(src, dest string) error {
+	b, err := NewBase(src, a.videoFormats, a.subtitleFormats)
+	if err != nil {
+		return err
+	}
+	a.progresser = b.Progress
+
+	uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error {
+		_, err := a.client.UploadStream(srcReader, destInfo.Size(), destPath)
+		return err
+	}
+	mkdirFunc := func(dir string) error {
+		return a.client.Mkdir(dir)
+	}
+
+	baseDest := filepath.Join(a.baseDir, dest)
+	return b.Upload(baseDest, false, false, false, uploadFunc, mkdirFunc)
+}
+
+func (a *Alist) ReadDir(dir string) ([]fs.FileInfo, error) {
+	return nil, nil
+}
+
+func (a *Alist) ReadFile(s string) ([]byte, error) {
+	return nil, nil
+}
+
+func (a *Alist) WriteFile(s string, bytes []byte) error {
+	return nil
+}
+
+func (a *Alist) UploadProgress() float64 {
+	if a.progresser == nil {
+		return 0
+	}
+	return a.progresser()
+}
+
+func (a *Alist) RemoveAll(path string) error {
+	return nil
+}

pkg/storage/base.go

@@ -0,0 +1,182 @@
+package storage
+
+import (
+	"io"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"polaris/log"
+	"polaris/pkg/utils"
+	"strings"
+
+	"github.com/gabriel-vasile/mimetype"
+	"github.com/pkg/errors"
+)
+
+type Storage interface {
+	Move(src, dest string) error
+	Copy(src, dest string) error
+	ReadDir(dir string) ([]fs.FileInfo, error)
+	ReadFile(string) ([]byte, error)
+	WriteFile(string, []byte) error
+	UploadProgress() float64
+	RemoveAll(path string) error
+}
+
+type uploadFunc func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error
+
+type Base struct {
+	src             string
+	videoFormats    []string
+	subtitleFormats []string
+	totalSize       int64
+	uploadedSize    int64
+}
+
+func NewBase(src string, videoFormats []string, subtitleFormats []string) (*Base, error) {
+	b := &Base{src: src, videoFormats: videoFormats, subtitleFormats: subtitleFormats}
+	err := b.calculateSize()
+	return b, err
+}
+
+func (b *Base) checkVideoFilesExist() bool {
+	if len(b.videoFormats) == 0 { // do not check
+		return true
+	}
+	hasVideo := false
+	filepath.Walk(b.src, func(path string, info fs.FileInfo, err error) error {
+		ext := filepath.Ext(strings.ToLower(info.Name()))
+
+		for _, f := range b.videoFormats {
+			if f == ext {
+				hasVideo = true
+			}
+		}
+		return nil
+	})
+	return hasVideo
+}
+
+func (b *Base) isFileNeeded(name string) bool {
+	ext := filepath.Ext(strings.ToLower(name))
+	if len(b.videoFormats) == 0 {
+		return true
+	} else {
+		for _, f := range b.videoFormats {
+			if f == ext {
+				return true
+			}
+		}
+	}
+	if len(b.subtitleFormats) > 0 {
+		for _, f := range b.subtitleFormats {
+			if f == ext {
+				return true
+			}
+		}
+	}
+	return false
+
+}
+
+func (b *Base) Upload(destDir string, tryLink, detectMime, changeMediaHash bool, upload uploadFunc, mkdir func(string) error) error {
+	if !b.checkVideoFilesExist() {
+		return errors.Errorf("torrent has no video file(s)")
+	}
+	os.MkdirAll(destDir, os.ModePerm)
+
+	targetBase := filepath.Join(destDir, filepath.Base(b.src)) //文件的场景，要加上文件名, move filename ./dir/
+	info, err := os.Stat(b.src)
+	if err != nil {
+		return errors.Wrap(err, "read source dir")
+	}
+	if info.IsDir() { //如果是路径，则只移动路径里面的文件，不管当前路径, 行为类似 move dirname/* target_dir/
+		targetBase = destDir
+	}
+	log.Debugf("local storage target base dir is: %v", targetBase)
+
+	err = filepath.Walk(b.src, func(path string, info fs.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		rel, err := filepath.Rel(b.src, path)
+		if err != nil {
+			return errors.Wrapf(err, "relation between %s and %s", b.src, path)
+		}
+		destName := filepath.Join(targetBase, rel)
+
+		if info.IsDir() {
+			mkdir(destName)
+		} else { //is file
+			if !b.isFileNeeded(info.Name()) {
+				log.Debugf("file is not needed, skip: %s", info.Name())
+				return nil
+			}
+			if tryLink {
+				if err := os.Link(path, destName); err == nil {
+					return nil //link success
+				}
+				log.Warnf("hard link file error: %v, will try copy file, source: %s, dest: %s", err, path, destName)
+			}
+			if changeMediaHash {
+				if err := utils.ChangeFileHash(path); err != nil {
+					log.Errorf("change file %v hash error: %v", path, err)
+				}
+			}
+
+			if f, err := os.OpenFile(path, os.O_RDONLY, os.ModePerm); err != nil {
+				return errors.Wrapf(err, "read file %v", path)
+			} else { //open success
+				defer f.Close()
+				var mtype *mimetype.MIME
+				if detectMime {
+					mtype, err = mimetype.DetectFile(path)
+					if err != nil {
+						return errors.Wrap(err, "mime type error")
+					}
+				}
+				return upload(destName, info, &progressReader{R: f, Add: func(i int) {
+					b.uploadedSize += int64(i)
+				}}, mtype)
+			}
+
+		}
+		log.Infof("file copy complete: %v", destName)
+		return nil
+	})
+	if err != nil {
+		return errors.Wrap(err, "move file error")
+	}
+	return nil
+
+}
+
+func (b *Base) calculateSize() error {
+	var size int64
+	err := filepath.Walk(b.src, func(_ string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			size += info.Size()
+		}
+		return err
+	})
+	b.totalSize = size
+	return err
+}
+
+func (b *Base) Progress() float64 {
+	return float64(b.uploadedSize) / float64(b.totalSize)
+}
+
+type progressReader struct {
+	R   io.Reader
+	Add func(int)
+}
+
+func (pr *progressReader) Read(p []byte) (int, error) {
+	n, err := pr.R.Read(p)
+	pr.Add(n)
+	return n, err
+}

pkg/storage/local.go

@@ -6,81 +6,45 @@ import (
 	"io/ioutil"
 	"os"
 	"path/filepath"
-	"polaris/log"
 
+	"github.com/gabriel-vasile/mimetype"
 	"github.com/pkg/errors"
 )
 
-type Storage interface {
-	Move(src, dest string) error
-	Copy(src, dest string) error
-	ReadDir(dir string) ([]fs.FileInfo, error)
-	ReadFile(string) ([]byte, error)
-	WriteFile(string, []byte) error
-}
-
-func NewLocalStorage(dir string) (*LocalStorage, error) {
+func NewLocalStorage(dir string, videoFormats []string, subtitleFormats []string) (*LocalStorage, error) {
 	os.MkdirAll(dir, 0655)
-
-	return &LocalStorage{dir: dir}, nil
+	return &LocalStorage{dir: dir, videoFormats: videoFormats, subtitleFormats: subtitleFormats}, nil
 }
 
 type LocalStorage struct {
-	dir string
+	dir             string
+	videoFormats    []string
+	subtitleFormats []string
 }
 
 func (l *LocalStorage) Copy(src, destDir string) error {
-	os.MkdirAll(filepath.Join(l.dir, destDir), os.ModePerm)
-
-	targetBase := filepath.Join(l.dir, destDir, filepath.Base(src)) //文件的场景，要加上文件名, move filename ./dir/
-	info, err := os.Stat(src)
+	b, err := NewBase(src, l.videoFormats, l.subtitleFormats)
 	if err != nil {
-		return errors.Wrap(err, "read source dir")
+		return err
 	}
-	if info.IsDir() { //如果是路径，则只移动路径里面的文件，不管当前路径, 行为类似 move dirname/* target_dir/
-		targetBase = filepath.Join(l.dir, destDir)
-	}
-	log.Debugf("local storage target base dir is: %v", targetBase)
-
-	err = filepath.Walk(src, func(path string, info fs.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-		rel, err := filepath.Rel(src, path)
-		if err != nil {
-			return errors.Wrapf(err, "relation between %s and %s", src, path)
-		}
-		destName := filepath.Join(targetBase, rel)
-
-		if info.IsDir() {
-			os.Mkdir(destName, os.ModePerm)
-		} else { //is file
-			if err := os.Link(path, destName); err != nil {
-				log.Warnf("hard link file error: %v, will try copy file, source: %s, dest: %s", err, path, destName)
-				if writer, err := os.OpenFile(destName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil {
-					return errors.Wrapf(err, "create file %s", destName)
-				} else {
-					defer writer.Close()
-					if f, err := os.OpenFile(path, os.O_RDONLY, os.ModePerm); err != nil {
-						return errors.Wrapf(err, "read file %v", path)
-					} else { //open success
-						defer f.Close()
-						_, err := io.Copy(writer, f)
-						if err != nil {
-							return errors.Wrap(err, "transmitting data error")
-						}
-					}
-				}
 
+	baseDest := filepath.Join(l.dir, destDir)
+	uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error {
+		if writer, err := os.OpenFile(destPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil {
+			return errors.Wrapf(err, "create file %s", destPath)
+		} else {
+			defer writer.Close()
+			_, err := io.Copy(writer, srcReader)
+			if err != nil {
+				return errors.Wrap(err, "transmitting data error")
 			}
+
 		}
-		log.Infof("file copy complete: %v", destName)
 		return nil
-	})
-	if err != nil {
-		return errors.Wrap(err, "move file error")
 	}
-	return nil
+	return b.Upload(baseDest, true, false, false, uploadFunc, func(s string) error {
+		return os.Mkdir(s, os.ModePerm)
+	})
 }
 
 func (l *LocalStorage) Move(src, destDir string) error {
@@ -103,3 +67,11 @@ func (l *LocalStorage) WriteFile(name string, data []byte) error {
 	os.MkdirAll(filepath.Dir(path), os.ModePerm)
 	return os.WriteFile(path, data, os.ModePerm)
 }
+
+func (l *LocalStorage) UploadProgress() float64 {
+	return 0
+}
+
+func (l *LocalStorage) RemoveAll(path string) error {
+	return os.RemoveAll(filepath.Join(l.dir, path))
+}

pkg/storage/webdav.go

@@ -1,13 +1,12 @@
 package storage
 
 import (
+	"io"
 	"io/fs"
 	"net/http"
 	"os"
 	"path/filepath"
-	"polaris/log"
 	"polaris/pkg/gowebdav"
-	"polaris/pkg/utils"
 
 	"github.com/gabriel-vasile/mimetype"
 	"github.com/pkg/errors"
@@ -17,9 +16,12 @@ type WebdavStorage struct {
 	fs              *gowebdav.Client
 	dir             string
 	changeMediaHash bool
+	progresser      func() float64
+	videoFormats    []string
+	subtitleFormats []string
 }
 
-func NewWebdavStorage(url, user, password, path string, changeMediaHash bool) (*WebdavStorage, error) {
+func NewWebdavStorage(url, user, password, path string, changeMediaHash bool, videoFormats []string, subtitleFormats []string) (*WebdavStorage, error) {
 	c := gowebdav.NewClient(url, user, password)
 	if err := c.Connect(); err != nil {
 		return nil, errors.Wrap(err, "connect webdav")
@@ -27,71 +29,35 @@ func NewWebdavStorage(url, user, password, path string, changeMediaHash bool) (*
 	return &WebdavStorage{
 		fs:  c,
 		dir: path,
+		videoFormats: videoFormats,
+		subtitleFormats: subtitleFormats,
 	}, nil
 }
 
 func (w *WebdavStorage) Copy(local, remoteDir string) error {
-	remoteBase := filepath.Join(w.dir, remoteDir, filepath.Base(local))
-	info, err := os.Stat(local)
+	b, err := NewBase(local, w.videoFormats, w.subtitleFormats)
 	if err != nil {
-		return errors.Wrap(err, "read source dir")
-	}
-	if info.IsDir() { //如果是路径，则只移动路径里面的文件，不管当前路径, 行为类似 move dirname/* target_dir/
-		remoteBase = filepath.Join(w.dir, remoteDir)
+		return err
 	}
 
-	//log.Infof("remove all content in %s", remoteBase)
-	//w.fs.RemoveAll(remoteBase)
-	err = filepath.Walk(local, func(path string, info fs.FileInfo, err error) error {
-		if err != nil {
-			return errors.Wrapf(err, "read file %v", path)
+	w.progresser = b.Progress
+
+	uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mtype *mimetype.MIME) error {
+		callback := func(r *http.Request) {
+			r.Header.Set("Content-Type", mtype.String())
+			r.ContentLength = destInfo.Size()
 		}
 
-		rel, err := filepath.Rel(local, path)
-		if err != nil {
-			return errors.Wrap(err, "path relation")
+		if err := w.fs.WriteStream(destPath, srcReader, 0666, callback); err != nil {
+			return errors.Wrap(err, "transmitting data error")
 		}
-		remoteName := filepath.Join(remoteBase, rel)
+		return nil
 
-		if info.IsDir() {
-			log.Infof("skip dir %v, webdav will mkdir automatically", info.Name())
+	}
 
-			// if err := w.fs.Mkdir(remoteName, 0666); err != nil {
-			// 	return errors.Wrapf(err, "mkdir %v", remoteName)
-			// }
-
-		} else { //is file
-			if w.changeMediaHash {
-				if err := utils.ChangeFileHash(path); err != nil {
-					log.Errorf("change file %v hash error: %v", path, err)
-				}
-			}
-			if f, err := os.OpenFile(path, os.O_RDONLY, 0666); err != nil {
-				return errors.Wrapf(err, "read file %v", path)
-			} else { //open success
-				defer f.Close()
-				mtype, err := mimetype.DetectFile(path)
-				if err != nil {
-					return errors.Wrap(err, "mime type error")
-				}
-
-				callback := func(r *http.Request) {
-					r.Header.Set("Content-Type", mtype.String())
-					r.ContentLength = info.Size()
-				}
-
-				if err := w.fs.WriteStream(remoteName, f, 0666, callback); err != nil {
-					return errors.Wrap(err, "transmitting data error")
-				}
-			}
-		}
-		log.Infof("file copy complete: %v", remoteName)
+	return b.Upload(filepath.Join(w.dir, remoteDir), false, true, w.changeMediaHash, uploadFunc, func(s string) error {
 		return nil
 	})
-	if err != nil {
-		return errors.Wrap(err, "move file error")
-	}
-	return nil
 }
 
 func (w *WebdavStorage) Move(local, remoteDir string) error {
@@ -112,3 +78,14 @@ func (w *WebdavStorage) ReadFile(name string) ([]byte, error) {
 func (w *WebdavStorage) WriteFile(name string, data []byte) error {
 	return w.fs.Write(filepath.Join(w.dir, name), data, os.ModePerm)
 }
+
+func (w *WebdavStorage) UploadProgress() float64 {
+	if w.progresser == nil {
+		return 0
+	}
+	return w.progresser()
+}
+
+func (w *WebdavStorage) RemoveAll(path string) error {
+	return w.fs.RemoveAll(filepath.Join(w.dir, path))
+}

pkg/torznab/cache.go

@@ -1,8 +1,14 @@
 package torznab
 
 import (
+	"polaris/log"
 	"polaris/pkg/cache"
 	"time"
 )
 
-var cc = cache.NewCache[string, *Response](time.Minute * 30)
+var cc = cache.NewCache[string, []Result](time.Minute * 30)
+
+func CleanCache() {
+	log.Debugf("clean all torznab caches")
+	cc = cache.NewCache[string, []Result](time.Minute * 30)
+}

pkg/torznab/torznab.go

@@ -80,14 +80,26 @@ func (r *Response) ToResults(indexer *db.TorznabInfo) []Result {
 		if slices.Contains(item.Category, "3000") { //exclude audio files
 			continue
 		}
+		// link, err := utils.Link2Magnet(item.Link) //TODO time consuming operation
+		// if err != nil {
+		// 	log.Warnf("converting link to magnet error, error: %v, link: %v", err, item.Link)
+		// 	continue
+		// }
+		imdb := ""
+		if item.GetAttr("imdbid") != "" {
+			imdb = item.GetAttr("imdbid")
+		} else if item.GetAttr("imdb") != "" {
+			imdb = item.GetAttr("imdb")
+		}
 		r := Result{
 			Name:                 item.Title,
+			Description:          item.Description,
 			Link:                 item.Link,
 			Size:                 mustAtoI(item.Size),
-			Seeders:              mustAtoI(item.GetAttr("seeders")),
-			Peers:                mustAtoI(item.GetAttr("peers")),
-			Category:             mustAtoI(item.GetAttr("category")),
-			ImdbId:               item.GetAttr("imdbid"),
+			Seeders:              int(mustAtoI(item.GetAttr("seeders"))),
+			Peers:                int(mustAtoI(item.GetAttr("peers"))),
+			Category:             int(mustAtoI(item.GetAttr("category"))),
+			ImdbId:               imdb,
 			DownloadVolumeFactor: tryParseFloat(item.GetAttr("downloadvolumefactor")),
 			UploadVolumeFactor:   tryParseFloat(item.GetAttr("uploadvolumefactor")),
 			Source:               indexer.Name,
@@ -100,8 +112,8 @@ func (r *Response) ToResults(indexer *db.TorznabInfo) []Result {
 	return res
 }
 
-func mustAtoI(key string) int {
-	i, err := strconv.Atoi(key)
+func mustAtoI(key string) int64 {
+	i, err := strconv.ParseInt(key, 10, 64)
 	if err != nil {
 		log.Errorf("must atoi error: %v", err)
 		panic(err)
@@ -119,7 +131,7 @@ func tryParseFloat(s string) float32 {
 }
 
 func Search(indexer *db.TorznabInfo, keyWord string) ([]Result, error) {
-	ctx, cancel := context.WithTimeout(context.TODO(), 10*time.Second)
+	ctx, cancel := context.WithTimeout(context.TODO(), 30*time.Second)
 	defer cancel()
 
 	req, err := http.NewRequestWithContext(ctx, http.MethodGet, indexer.URL, nil)
@@ -135,15 +147,18 @@ func Search(indexer *db.TorznabInfo, keyWord string) ([]Result, error) {
 
 	cacheRes, ok := cc.Get(key)
 	if !ok {
+		log.Debugf("not found in cache, need query again: %v", key)
 		res, err := doRequest(req)
 		if err != nil {
-			cc.Set(key, &Response{})
+			cc.Set(key, nil)
 			return nil, errors.Wrap(err, "do http request")
 		}
-		cacheRes = res
+		cacheRes = res.ToResults(indexer)
 		cc.Set(key, cacheRes)
+	} else {
+		log.Debugf("found cache match for key: %v", key)
 	}
-	return cacheRes.ToResults(indexer), nil
+	return cacheRes, nil
 }
 
 func doRequest(req *http.Request) (*Response, error) {
@@ -166,8 +181,9 @@ func doRequest(req *http.Request) (*Response, error) {
 
 type Result struct {
 	Name                 string  `json:"name"`
+	Description          string  `json:"description"`
 	Link                 string  `json:"link"`
-	Size                 int     `json:"size"`
+	Size                 int64     `json:"size"`
 	Seeders              int     `json:"seeders"`
 	Peers                int     `json:"peers"`
 	Category             int     `json:"category"`

pkg/transmission/transmission.go

@@ -2,13 +2,11 @@ package transmission
 
 import (
 	"context"
-	"encoding/json"
 	"fmt"
-	"net/http"
 	"net/url"
 	"polaris/log"
 	"polaris/pkg"
-	"strings"
+	"polaris/pkg/utils"
 
 	"github.com/hekmon/transmissionrpc/v3"
 	"github.com/pkg/errors"
@@ -54,71 +52,74 @@ func (c *Client) GetAll() ([]pkg.Torrent, error) {
 	var torrents []pkg.Torrent
 	for _, t := range all {
 		torrents = append(torrents, &Torrent{
-			Hash:     *t.HashString,
-			c:      c.c,
-			Config: c.cfg,		
+			hash: *t.HashString,
+			c:    c.c,
+			//cfg: c.cfg,
 		})
 	}
 	return torrents, nil
 }
 
 func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
-	if strings.HasPrefix(link, "http") {
-		client := &http.Client{
-			CheckRedirect: func(req *http.Request, via []*http.Request) error {
-				return http.ErrUseLastResponse
-			},
-		}
-		resp, err:=client.Get(link)
-		if err == nil {
-			if resp.StatusCode == http.StatusFound {
-				loc, err := resp.Location()
-				if err == nil {
-					link = loc.String()
-					log.Warnf("transimision redirect to url: %v", link)
-				}
-			}
-	
-		}
-	
-	} 
-	t, err := c.c.TorrentAdd(context.TODO(), transmissionrpc.TorrentAddPayload{
-		Filename:    &link,
-		DownloadDir: &dir,
-	})
-	log.Infof("get torrent info: %+v", t)
-	if t.HashString == nil {
-		return nil, fmt.Errorf("download torrent error: %v", link)
+	magnet, err := utils.Link2Magnet(link)
+	if err != nil {
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
 	}
 
+	hash, err := utils.MagnetHash(magnet)
+	if err != nil {
+		return nil, errors.Wrap(err, "get hash")
+	}
+
+	t, err := c.c.TorrentAdd(context.TODO(), transmissionrpc.TorrentAddPayload{
+		Filename:    &magnet,
+		DownloadDir: &dir,
+	})
+	log.Debugf("get torrent info: %+v", t)
+
 	return &Torrent{
-		Hash:     *t.HashString,
-		c:      c.c,
-		Config: c.cfg,
+		hash: hash,
+		c:    c.c,
+		//cfg: c.cfg,
 	}, err
 }
 
+func NewTorrent(cfg Config, link string) (*Torrent, error) {
+	c, err := NewClient(cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	magnet, err := utils.Link2Magnet(link)
+	if err != nil {
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
+	}
+
+	hash, err := utils.MagnetHash(magnet)
+	if err != nil {
+		return nil, err
+	}
+
+	t := &Torrent{
+		c:    c.c,
+		hash: hash,
+		//cfg: cfg,
+	}
+	if !t.Exists() {
+		return nil, errors.Errorf("torrent not exist: %v", magnet)
+	}
+	return t, nil
+}
+
 type Torrent struct {
 	//t *transmissionrpc.Torrent
-	c  *transmissionrpc.Client
-	Hash string `json:"hash"`
-	Config
-}
-
-func (t *Torrent) Reload() error {
-	c, err := NewClient(t.Config)
-	if err != nil {
-		return err
-	}
-	t.c = c.c
-	if !t.Exists() {
-		return errors.Errorf("torrent not exists: %v", t.Hash)
-	}
-	return nil
+	c    *transmissionrpc.Client
+	hash string
+	//cfg Config
 }
 
 func (t *Torrent) getTorrent() (transmissionrpc.Torrent, error) {
-	r, err := t.c.TorrentGetAllForHashes(context.TODO(), []string{t.Hash})
+	r, err := t.c.TorrentGetAllForHashes(context.TODO(), []string{t.hash})
 	if err != nil {
 		log.Errorf("get torrent info for error: %v", err)
 	}
@@ -129,14 +130,13 @@ func (t *Torrent) getTorrent() (transmissionrpc.Torrent, error) {
 }
 
 func (t *Torrent) Exists() bool {
-	r, err := t.c.TorrentGetAllForHashes(context.TODO(), []string{t.Hash})
+	r, err := t.c.TorrentGetAllForHashes(context.TODO(), []string{t.hash})
 	if err != nil {
 		log.Errorf("get torrent info for error: %v", err)
 	}
 	return len(r) > 0
 }
 
-
 func (t *Torrent) Name() (string, error) {
 	tt, err := t.getTorrent()
 	if err != nil {
@@ -168,7 +168,7 @@ func (t *Torrent) Progress() (int, error) {
 }
 
 func (t *Torrent) Stop() error {
-	return t.c.TorrentStopHashes(context.TODO(), []string{t.Hash})
+	return t.c.TorrentStopHashes(context.TODO(), []string{t.hash})
 }
 
 func (t *Torrent) SeedRatio() (float64, error) {
@@ -183,7 +183,7 @@ func (t *Torrent) SeedRatio() (float64, error) {
 }
 
 func (t *Torrent) Start() error {
-	return t.c.TorrentStartHashes(context.TODO(), []string{t.Hash})
+	return t.c.TorrentStartHashes(context.TODO(), []string{t.hash})
 }
 
 func (t *Torrent) Remove() error {
@@ -205,12 +205,6 @@ func (t *Torrent) Size() (int, error) {
 	return int(tt.TotalSize.Byte()), nil
 }
 
-func (t *Torrent) Save() string {
-
-	d, _ := json.Marshal(*t)
-	return string(d)
-}
-
 func (t *Torrent) GetHash() string {
-	return t.Hash
+	return t.hash
 }

pkg/uploader/uploader.go

@@ -0,0 +1,86 @@
+package uploader
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"polaris/pkg/utils"
+	"sync/atomic"
+	"time"
+)
+
+type StreamWriter interface {
+	WriteStream(path string, stream io.Reader, _ os.FileMode) error
+}
+
+type Uploader struct {
+	sw       StreamWriter
+	progress atomic.Int64
+	dir      string
+	size     int64
+}
+
+func NewUploader(dir string, sw StreamWriter) (*Uploader, error) {
+	size, err := utils.DirSize(dir)
+	if err != nil {
+		return nil, err
+	}
+	return &Uploader{sw: sw, dir: dir, size: size, progress: atomic.Int64{}}, nil
+}
+
+func (u *Uploader) Upload() error {
+
+	return nil
+}
+
+type ProgressReader struct {
+	Reader   io.Reader
+	Progress atomic.Int64
+	Size     int64
+	Name     string
+	Once     bool
+	Done     atomic.Bool
+}
+
+func (progressReader *ProgressReader) NewLoop() {
+	ticker := time.NewTicker(time.Second)
+	var op int64
+	for range ticker.C {
+		p := progressReader.Progress.Load()
+		KB := (p - op) / 1024
+		var percent int64
+		if progressReader.Size != 0 {
+			percent = p * 100 / progressReader.Size
+		} else {
+			percent = 100
+		}
+		if KB < 1024 {
+			fmt.Printf("%s: %dKB/s %d%%\n", progressReader.Name, KB, percent)
+		} else {
+			fmt.Printf("%s: %.2fMB/s %d%%\n", progressReader.Name, float64(KB)/1024, percent)
+		}
+
+		if progressReader.Done.Load() {
+			ticker.Stop()
+			return
+		}
+	}
+}
+
+func (progressReader *ProgressReader) Read(p []byte) (int, error) {
+	n, err := progressReader.Reader.Read(p)
+	progressReader.Progress.Add(int64(n))
+	if !progressReader.Once {
+		progressReader.Once = true
+		go progressReader.NewLoop()
+	}
+	if err != nil {
+		progressReader.Done.Store(true)
+	}
+	return n, err
+}
+
+func (progressReader *ProgressReader) Close() error {
+	progressReader.Done.Store(true)
+	return nil
+}

pkg/utils/utils.go

@@ -2,14 +2,17 @@ package utils
 
 import (
 	"encoding/json"
+	"net/http"
 	"os"
 	"os/exec"
+	"path/filepath"
 	"regexp"
 	"runtime"
 	"strconv"
 	"strings"
 	"unicode"
 
+	"github.com/anacrolix/torrent/metainfo"
 	"github.com/pkg/errors"
 	"golang.org/x/crypto/bcrypt"
 	"golang.org/x/exp/rand"
@@ -218,3 +221,68 @@ func isWSL() bool {
 	}
 	return strings.Contains(strings.ToLower(string(releaseData)), "microsoft")
 }
+
+func Link2Magnet(link string) (string, error) {
+	if strings.HasPrefix(strings.ToLower(link), "magnet:") {
+		return link, nil
+	}
+	client := &http.Client{
+		CheckRedirect: func(req *http.Request, via []*http.Request) error {
+			return http.ErrUseLastResponse //do not follow redirects
+		},
+	}
+
+	resp, err := client.Get(link)
+	if err != nil {
+		return "", errors.Wrap(err, "get link")
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode >= 300 && resp.StatusCode < 400 {
+		//redirects
+		tourl := resp.Header.Get("Location")
+		return Link2Magnet(tourl)
+	}
+	info, err := metainfo.Load(resp.Body)
+	if err != nil {
+		return "", errors.Wrap(err, "parse response")
+	}
+	mg, err := info.MagnetV2()
+	if err != nil {
+		return "", errors.Wrap(err, "convert magnet")
+	}
+	return mg.String(), nil
+}
+
+func MagnetHash(link string) (string, error) {
+	if mi, err := metainfo.ParseMagnetV2Uri(link); err != nil {
+		return "", errors.Errorf("magnet link is not valid: %v", err)
+	} else {
+		hash := ""
+		if mi.InfoHash.Unwrap().HexString() != "" {
+			hash = mi.InfoHash.Unwrap().HexString()
+		} else {
+			btmh := mi.V2InfoHash.Unwrap()
+			if btmh.HexString() != "" {
+				hash = btmh.HexString()
+			}
+		}
+		if hash == "" {
+			return "", errors.Errorf("magnet has no info hash: %v", link)
+		}
+		return hash, nil
+	}
+}
+
+func DirSize(path string) (int64, error) {
+	var size int64
+	err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			size += info.Size()
+		}
+		return err
+	})
+	return size, err
+}

server/activity.go

@@ -3,8 +3,10 @@ package server
 import (
 	"fmt"
 	"polaris/ent"
+	"polaris/ent/blacklist"
 	"polaris/ent/episode"
 	"polaris/ent/history"
+	"polaris/ent/schema"
 	"polaris/log"
 	"polaris/pkg/utils"
 	"strconv"
@@ -15,8 +17,9 @@ import (
 
 type Activity struct {
 	*ent.History
-	Progress  int     `json:"progress"`
-	SeedRatio float64 `json:"seed_ratio"`
+	Progress       int     `json:"progress"`
+	SeedRatio      float64 `json:"seed_ratio"`
+	UploadProgress float64 `json:"upload_progress"`
 }
 
 func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
@@ -42,6 +45,9 @@ func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
 					} else {
 						a.SeedRatio = r
 					}
+					if task.UploadProgresser != nil {
+						a.UploadProgress = task.UploadProgresser()
+					}
 				}
 			}
 			activities = append(activities, a)
@@ -63,22 +69,35 @@ func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
 	return activities, nil
 }
 
+type removeActivityIn struct {
+	ID            int  `json:"id"`
+	Add2Blacklist bool `json:"add_2_blacklist"`
+}
+
 func (s *Server) RemoveActivity(c *gin.Context) (interface{}, error) {
-	ids := c.Param("id")
-	id, err := strconv.Atoi(ids)
-	if err != nil {
-		return nil, errors.Wrap(err, "convert")
+	var in removeActivityIn
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, errors.Wrap(err, "bind json")
 	}
-	his := s.db.GetHistory(id)
+
+	his := s.db.GetHistory(in.ID)
 	if his == nil {
-		log.Errorf("no record of id: %d", id)
+		log.Errorf("no record of id: %d", in.ID)
 		return nil, nil
 	}
+	if in.Add2Blacklist && his.Link != "" {
+		//should add to blacklist
+		if err := s.addTorrent2Blacklist(his.Link); err != nil {
+			return nil, errors.Errorf("add to blacklist: %v", err)
+		} else {
+			log.Infof("success add magnet link to blacklist: %v", his.Link)
+		}
+	}
 
 	if err := s.core.RemoveTaskAndTorrent(his.ID); err != nil {
 		return nil, errors.Wrap(err, "remove torrent")
 	}
-	err = s.db.DeleteHistory(id)
+	err := s.db.DeleteHistory(in.ID)
 	if err != nil {
 		return nil, errors.Wrap(err, "db")
 	}
@@ -102,6 +121,28 @@ func (s *Server) RemoveActivity(c *gin.Context) (interface{}, error) {
 	log.Infof("history record successful deleted: %v", his.SourceTitle)
 	return nil, nil
 }
+
+func (s *Server) addTorrent2Blacklist(link string) error {
+	if link == "" {
+		return nil
+	}
+	if hash, err := utils.MagnetHash(link); err != nil {
+		return err
+	} else {
+		item := ent.Blacklist{
+			Type: blacklist.TypeTorrent,
+			Value: schema.BlacklistValue{
+				TorrentHash: hash,
+			},
+		}
+		err := s.db.AddBlacklistItem(&item)
+		if err != nil {
+			return errors.Wrap(err, "add to db")
+		}
+	}
+	return nil
+}
+
 func (s *Server) GetMediaDownloadHistory(c *gin.Context) (interface{}, error) {
 	var ids = c.Param("id")
 	id, err := strconv.Atoi(ids)

server/core/client.go

@@ -1,7 +1,6 @@
 package core
 
 import (
-	"encoding/json"
 	"polaris/db"
 	"polaris/ent"
 	"polaris/ent/downloadclients"
@@ -27,69 +26,66 @@ func NewClient(db *db.Client, language string) *Client {
 
 type scheduler struct {
 	cron string
-	f func() error 
+	f    func() error
 }
 type Client struct {
-	db       *db.Client
-	cron     *cron.Cron
-	tasks    map[int]*Task
-	language string
+	db         *db.Client
+	cron       *cron.Cron
+	tasks      map[int]*Task
+	language   string
 	schedulers utils.Map[string, scheduler]
 }
 
 func (c *Client) registerCronJob(name string, cron string, f func() error) {
 	c.schedulers.Store(name, scheduler{
 		cron: cron,
-		f: f,
+		f:    f,
 	})
 }
 
 func (c *Client) Init() {
-	c.reloadTasks()
+	go c.reloadTasks()
 	c.addSysCron()
+	go c.checkW500PosterOnStartup()
 }
 
 func (c *Client) reloadTasks() {
 	allTasks := c.db.GetRunningHistories()
 	for _, t := range allTasks {
-		var torrent pkg.Torrent
-		if tt, err := c.reloadTransmiision(t.Saved); err == nil {
-			torrent = tt
-			log.Infof("load transmission task: %v", t.Saved)
-		} else if tt, err := c.reloadQbit(t.Saved); err == nil {
-			torrent = tt
-			log.Infof("load qbit task: %v", t.Saved)
-		} else {
-			log.Warnf("load task fail: %v", t.Saved)
+		dl, err := c.db.GetDownloadClient(t.DownloadClientID)
+		if err != nil {
+			log.Warnf("no download client related: %v", t.SourceTitle)
 			continue
 		}
-		c.tasks[t.ID] = &Task{Torrent: torrent}
-	}
-}
 
-func (c *Client) reloadTransmiision(s string) (pkg.Torrent, error) {
-	var t transmission.Torrent
-	if err := json.Unmarshal([]byte(s), &t); err != nil {
-		return nil, err
-	}
-	if err := t.Reload(); err != nil {
-		return nil, err
-	}
-	return &t, nil
-}
+		if dl.Implementation == downloadclients.ImplementationTransmission {
+			to, err := transmission.NewTorrent(transmission.Config{
+				URL:      dl.URL,
+				User:     dl.User,
+				Password: dl.Password,
+			}, t.Link)
+			if err != nil {
+				log.Warnf("get task error: %v", err)
+				continue
+			}
+			c.tasks[t.ID] = &Task{Torrent: to}
+		} else if dl.Implementation == downloadclients.ImplementationQbittorrent {
+			to, err := qbittorrent.NewTorrent(qbittorrent.Info{
+				URL:      dl.URL,
+				User:     dl.User,
+				Password: dl.Password,
+			}, t.Link)
+			if err != nil {
+				log.Warnf("get task error: %v", err)
+				continue
+			}
+			c.tasks[t.ID] = &Task{Torrent: to}
+		}
 
-func (c *Client) reloadQbit(s string) (pkg.Torrent, error) {
-	var t qbittorrent.Torrent
-	if err := json.Unmarshal([]byte(s), &t); err != nil {
-		return nil, err
 	}
-	if err := t.Reload(); err != nil {
-		return nil, err
-	}
-	return &t, nil
+	log.Infof("------ task reloading done ------")
 }
 
-
 func (c *Client) GetDownloadClient() (pkg.Downloader, *ent.DownloadClients, error) {
 	downloaders := c.db.GetAllDonloadClients()
 	for _, d := range downloaders {
@@ -107,7 +103,7 @@ func (c *Client) GetDownloadClient() (pkg.Downloader, *ent.DownloadClients, erro
 				continue
 			}
 			return trc, d, nil
-		
+
 		} else if d.Implementation == downloadclients.ImplementationQbittorrent {
 			qbt, err := qbittorrent.NewClient(d.URL, d.User, d.Password)
 			if err != nil {

server/core/fliters.go

@@ -0,0 +1,2 @@
+package core
+

server/core/importlist.go

@@ -76,6 +76,7 @@ func (c *Client) periodicallyUpdateImportlist() error {
 					if err != nil {
 						log.Errorf("[update_import_lists] add movie to watchlist error: %v", err)
 					} else {
+						c.sendMsg(fmt.Sprintf("成功监控电影：%v", d.Title))
 						log.Infof("[update_import_lists] add movie to watchlist success")
 					}
 				} else if len(tmdbRes.TvResults) > 0 {
@@ -95,6 +96,7 @@ func (c *Client) periodicallyUpdateImportlist() error {
 					if err != nil {
 						log.Errorf("[update_import_lists] add tv to watchlist error: %v", err)
 					} else {
+						c.sendMsg(fmt.Sprintf("成功监控电视剧：%v", d.Name))
 						log.Infof("[update_import_lists] add tv to watchlist success")
 					}
 
@@ -112,8 +114,9 @@ type AddWatchlistIn struct {
 	Resolution              string `json:"resolution" binding:"required"`
 	Folder                  string `json:"folder" binding:"required"`
 	DownloadHistoryEpisodes bool   `json:"download_history_episodes"` //for tv
-	SizeMin                 int    `json:"size_min"`
-	SizeMax                 int    `json:"size_max"`
+	SizeMin                 int64  `json:"size_min"`
+	SizeMax                 int64  `json:"size_max"`
+	PreferSize              int64  `json:"prefer_size"`
 }
 
 func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
@@ -137,7 +140,7 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 	}
 	log.Infof("find detail for tv id %d: %+v", in.TmdbID, detail)
 
-	 lastSeason := 0
+	lastSeason := 0
 	for _, season := range detail.Seasons {
 		if season.SeasonNumber > lastSeason && season.EpisodeCount > 0 { //如果最新一季已经有剧集信息，则以最新一季为准
 			lastSeason = season.SeasonNumber
@@ -146,6 +149,11 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 
 	log.Debugf("latest season is %v", lastSeason)
 
+	alterTitles, err := c.getAlterTitles(in.TmdbID, media.MediaTypeTv)
+	if err != nil {
+		return nil, errors.Wrap(err, "get alter titles")
+	}
+
 	var epIds []int
 	for _, season := range detail.Seasons {
 		seasonId := season.SeasonNumber
@@ -192,6 +200,7 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 			epIds = append(epIds, epid)
 		}
 	}
+	
 	m := &ent.Media{
 		TmdbID:                  int(detail.ID),
 		ImdbID:                  detail.IMDbID,
@@ -210,6 +219,7 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 			OriginalLanguage: detail.OriginalLanguage,
 			Genres:           detail.Genres,
 		},
+		AlternativeTitles: alterTitles,
 	}
 
 	r, err := c.db.AddMediaWatchlist(m, epIds)
@@ -220,6 +230,10 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 		if err := c.downloadPoster(detail.PosterPath, r.ID); err != nil {
 			log.Errorf("download poster error: %v", err)
 		}
+		if err := c.downloadW500Poster(detail.PosterPath, r.ID); err != nil {
+			log.Errorf("download w500 poster error: %v", err)
+		}
+
 		if err := c.downloadBackdrop(detail.BackdropPath, r.ID); err != nil {
 			log.Errorf("download poster error: %v", err)
 		}
@@ -233,6 +247,42 @@ func (c *Client) AddTv2Watchlist(in AddWatchlistIn) (interface{}, error) {
 	return nil, nil
 }
 
+func (c *Client) getAlterTitles(tmdbId int, mediaType media.MediaType) ([]schema.AlternativeTilte, error){
+	var titles []schema.AlternativeTilte
+
+	if mediaType == media.MediaTypeTv {
+		alterTitles, err := c.MustTMDB().GetTVAlternativeTitles(tmdbId, c.language)
+		if err != nil {
+			return nil, errors.Wrap(err, "tmdb")
+		}
+		
+		for _, t := range alterTitles.Results {
+			titles = append(titles, schema.AlternativeTilte{
+				Iso3166_1: t.Iso3166_1,
+				Title: t.Title,
+				Type: t.Type,
+			})
+		}
+	
+	} else if mediaType == media.MediaTypeMovie {
+		alterTitles, err := c.MustTMDB().GetMovieAlternativeTitles(tmdbId, c.language)
+		if err != nil {
+			return nil, errors.Wrap(err, "tmdb")
+		}
+		
+		for _, t := range alterTitles.Titles {
+			titles = append(titles, schema.AlternativeTilte{
+				Iso3166_1: t.Iso3166_1,
+				Title: t.Title,
+				Type: t.Type,
+			})
+		}
+	}
+	log.Debugf("get alternative titles: %+v", titles)
+
+	return titles, nil
+}
+
 func (c *Client) AddMovie2Watchlist(in AddWatchlistIn) (interface{}, error) {
 	log.Infof("add movie watchlist input: %+v", in)
 	detailCn, err := c.MustTMDB().GetMovieDetails(in.TmdbID, db.LanguageCN)
@@ -251,6 +301,12 @@ func (c *Client) AddMovie2Watchlist(in AddWatchlistIn) (interface{}, error) {
 	}
 	log.Infof("find detail for movie id %d: %v", in.TmdbID, detail)
 
+	alterTitles, err := c.getAlterTitles(in.TmdbID, media.MediaTypeMovie)
+	if err != nil {
+		return nil, errors.Wrap(err, "get alter titles")
+	}
+
+
 	epid, err := c.db.SaveEposideDetail(&ent.Episode{
 		SeasonNumber:  1,
 		EpisodeNumber: 1,
@@ -277,6 +333,7 @@ func (c *Client) AddMovie2Watchlist(in AddWatchlistIn) (interface{}, error) {
 		StorageID:    in.StorageID,
 		TargetDir:    in.Folder,
 		Limiter:      schema.MediaLimiter{SizeMin: in.SizeMin, SizeMax: in.SizeMax},
+		AlternativeTitles: alterTitles,
 	}
 
 	extras := schema.MediaExtras{
@@ -298,6 +355,10 @@ func (c *Client) AddMovie2Watchlist(in AddWatchlistIn) (interface{}, error) {
 		if err := c.downloadPoster(detail.PosterPath, r.ID); err != nil {
 			log.Errorf("download poster error: %v", err)
 		}
+		if err := c.downloadW500Poster(detail.PosterPath, r.ID); err != nil {
+			log.Errorf("download w500 poster error: %v", err)
+		}
+
 		if err := c.downloadBackdrop(detail.BackdropPath, r.ID); err != nil {
 			log.Errorf("download backdrop error: %v", err)
 		}
@@ -312,7 +373,7 @@ func (c *Client) AddMovie2Watchlist(in AddWatchlistIn) (interface{}, error) {
 }
 
 func (c *Client) checkMovieFolder(m *ent.Media) error {
-	var storageImpl, err = c.getStorage(m.StorageID, media.MediaTypeMovie)
+	var storageImpl, err = c.GetStorage(m.StorageID, media.MediaTypeMovie)
 	if err != nil {
 		return err
 	}
@@ -325,8 +386,8 @@ func (c *Client) checkMovieFolder(m *ent.Media) error {
 		return err
 	}
 
-	for _,f := range files {
-		if f.IsDir() || f.Size() < 100 * 1000 * 1000 /* 100M */{ //忽略路径和小于100M的文件
+	for _, f := range files {
+		if f.IsDir() || f.Size() < 100*1000*1000 /* 100M */ { //忽略路径和小于100M的文件
 			continue
 		}
 		meta := metadata.ParseMovie(f.Name())
@@ -369,6 +430,11 @@ func (c *Client) downloadPoster(path string, mediaID int) error {
 	return c.downloadImage(url, mediaID, "poster.jpg")
 }
 
+func (c *Client) downloadW500Poster(path string, mediaID int) error {
+	url := "https://image.tmdb.org/t/p/w500" + path
+	return c.downloadImage(url, mediaID, "poster_w500.jpg")
+}
+
 func (c *Client) downloadImage(url string, mediaID int, name string) error {
 
 	log.Infof("try to download image: %v", url)
@@ -394,6 +460,46 @@ func (c *Client) downloadImage(url string, mediaID int, name string) error {
 
 }
 
+func (c *Client) checkW500PosterOnStartup() {
+	log.Infof("check all w500 posters")
+	all := c.db.GetMediaWatchlist(media.MediaTypeTv)
+	movies := c.db.GetMediaWatchlist(media.MediaTypeMovie)
+	all = append(all, movies...)
+	for _, e := range all {
+		targetFile := filepath.Join(fmt.Sprintf("%v/%d", db.ImgPath, e.ID), "poster_w500.jpg")
+		if _, err := os.Stat(targetFile); err != nil {
+			log.Infof("poster_w500.jpg not exist for %s, will download it", e.NameEn)
+
+			if e.MediaType ==media.MediaTypeTv {
+				detail, err := c.MustTMDB().GetTvDetails(e.TmdbID, db.LanguageCN)
+				if err != nil {
+					log.Warnf("get tmdb detail for %s error: %v", e.NameEn, err)
+					continue
+				}
+	
+				if err := c.downloadW500Poster(detail.PosterPath, e.ID); err != nil {
+					log.Warnf("download w500 poster error: %v", err)
+					continue
+				}
+	
+			} else {
+				detail, err := c.MustTMDB().GetMovieDetails(e.TmdbID, db.LanguageCN)
+				if err != nil {
+					log.Warnf("get tmdb detail for %s error: %v", e.NameEn, err)
+					continue
+				}
+	
+				if err := c.downloadW500Poster(detail.PosterPath, e.ID); err != nil {
+					log.Warnf("download w500 poster error: %v", err)
+					continue
+				}
+	
+			}
+
+		}
+	}
+}
+
 func (c *Client) SuggestedMovieFolderName(tmdbId int) (string, error) {
 
 	d1, err := c.MustTMDB().GetMovieDetails(tmdbId, c.language)

server/core/integration.go

@@ -4,13 +4,13 @@ import (
 	"bytes"
 	"encoding/xml"
 	"fmt"
-	"github.com/pkg/errors"
 	"os"
 	"path/filepath"
 	"polaris/db"
 	"polaris/ent/media"
 	storage1 "polaris/ent/storage"
 	"polaris/log"
+	"polaris/pkg/alist"
 	"polaris/pkg/metadata"
 	"polaris/pkg/notifier"
 	"polaris/pkg/storage"
@@ -18,6 +18,8 @@ import (
 	"slices"
 	"strconv"
 	"strings"
+
+	"github.com/pkg/errors"
 )
 
 func (c *Client) writeNfoFile(historyId int) error {
@@ -33,7 +35,7 @@ func (c *Client) writeNfoFile(historyId int) error {
 	}
 
 	if md.MediaType == media.MediaTypeTv { //tvshow.nfo
-		st, err := c.getStorage(md.StorageID, media.MediaTypeTv)
+		st, err := c.GetStorage(md.StorageID, media.MediaTypeTv)
 		if err != nil {
 			return errors.Wrap(err, "get storage")
 		}
@@ -68,7 +70,7 @@ func (c *Client) writeNfoFile(historyId int) error {
 		}
 
 	} else if md.MediaType == media.MediaTypeMovie { //movie.nfo
-		st, err := c.getStorage(md.StorageID, media.MediaTypeMovie)
+		st, err := c.GetStorage(md.StorageID, media.MediaTypeMovie)
 		if err != nil {
 			return errors.Wrap(err, "get storage")
 		}
@@ -120,7 +122,7 @@ func (c *Client) writePlexmatch(historyId int) error {
 	if series.MediaType != media.MediaTypeTv { //.plexmatch only support tv series
 		return nil
 	}
-	st, err := c.getStorage(series.StorageID, media.MediaTypeTv)
+	st, err := c.GetStorage(series.StorageID, media.MediaTypeTv)
 	if err != nil {
 		return errors.Wrap(err, "get storage")
 	}
@@ -195,17 +197,26 @@ func (c *Client) nfoSupportEnabled() bool {
 	return c.db.GetSetting(db.SettingNfoSupportEnabled) == "true"
 }
 
-func (c *Client) getStorage(storageId int, mediaType media.MediaType) (storage.Storage, error) {
+func (c *Client) GetStorage(storageId int, mediaType media.MediaType) (storage.Storage, error) {
 	st := c.db.GetStorage(storageId)
 	targetPath := st.TvPath
 	if mediaType == media.MediaTypeMovie {
 		targetPath = st.MoviePath
 	}
+	videoFormats, err := c.db.GetAcceptedVideoFormats()
+	if err != nil {
+		log.Warnf("get accepted video format error: %v", err)
+	}
+	subtitleFormats, err := c.db.GetAcceptedSubtitleFormats()
+	if err != nil {
+		log.Warnf("get accepted subtitle format error: %v", err)
+	}
+
 
 	switch st.Implementation {
 	case storage1.ImplementationLocal:
 
-		storageImpl1, err := storage.NewLocalStorage(targetPath)
+		storageImpl1, err := storage.NewLocalStorage(targetPath, videoFormats, subtitleFormats)
 		if err != nil {
 			return nil, errors.Wrap(err, "new local")
 		}
@@ -213,11 +224,18 @@ func (c *Client) getStorage(storageId int, mediaType media.MediaType) (storage.S
 
 	case storage1.ImplementationWebdav:
 		ws := st.ToWebDavSetting()
-		storageImpl1, err := storage.NewWebdavStorage(ws.URL, ws.User, ws.Password, targetPath, ws.ChangeFileHash == "true")
+		storageImpl1, err := storage.NewWebdavStorage(ws.URL, ws.User, ws.Password, targetPath, ws.ChangeFileHash == "true", videoFormats, subtitleFormats)
 		if err != nil {
 			return nil, errors.Wrap(err, "new webdav")
 		}
 		return storageImpl1, nil
+	case storage1.ImplementationAlist:
+		cfg := st.ToWebDavSetting()
+		storageImpl1, err := storage.NewAlist(&alist.Config{URL: cfg.URL, Username: cfg.User, Password: cfg.Password}, targetPath, videoFormats, subtitleFormats)
+		if err != nil {
+			return nil, errors.Wrap(err, "alist")
+		}
+		return storageImpl1, nil
 	}
 	return nil, errors.New("no storage found")
 }
@@ -300,9 +318,9 @@ func (c *Client) findEpisodeFilesPreMoving(historyId int) error {
 			}
 
 			meta := metadata.ParseTv(f.Name())
-			if meta.Episode > 0 {
+			if meta.StartEpisode > 0 {
 				//episode exists
-				ep, err := c.db.GetEpisode(his.MediaID, seasonNum, meta.Episode)
+				ep, err := c.db.GetEpisode(his.MediaID, seasonNum, meta.StartEpisode)
 				if err != nil {
 					return err
 				}

server/core/resources.go

@@ -1,11 +1,13 @@
 package core
 
 import (
+	"bytes"
 	"fmt"
 	"polaris/ent"
 	"polaris/ent/episode"
 	"polaris/ent/history"
 	"polaris/log"
+	"polaris/pkg/metadata"
 	"polaris/pkg/notifier/message"
 	"polaris/pkg/torznab"
 	"polaris/pkg/utils"
@@ -13,7 +15,7 @@ import (
 	"github.com/pkg/errors"
 )
 
-func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum, episodeNum int) (*string, error) {
+func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum int, episodeNums ...int) (*string, error) {
 	trc, dlc, err := c.GetDownloadClient()
 	if err != nil {
 		return nil, errors.Wrap(err, "connect transmission")
@@ -31,83 +33,176 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
 		return nil, errors.New("no enough space")
 	}
 
-	var ep *ent.Episode
-	if episodeNum > 0 {
-		for _, e := range series.Episodes {
-			if e.SeasonNumber == seasonNum && e.EpisodeNumber == episodeNum {
-				ep = e
-			}
-		}
-		if ep == nil {
-			return nil, errors.Errorf("no episode of season %d episode %d", seasonNum, episodeNum)
-		}
-	} else { //season package download
-		ep = &ent.Episode{}
-	}
-	torrent, err := trc.Download(r1.Link, downloadDir)
+	magnet, err := utils.Link2Magnet(r1.Link)
 	if err != nil {
-		return nil, errors.Wrap(err, "downloading")
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", r1.Link, err)
 	}
-	torrent.Start()
 
 	dir := fmt.Sprintf("%s/Season %02d/", series.TargetDir, seasonNum)
 
+	if len(episodeNums) > 0 {
+		for _, epNum := range episodeNums {
+			var ep *ent.Episode
+			for _, e := range series.Episodes {
+				if e.SeasonNumber == seasonNum && e.EpisodeNumber == epNum {
+					ep = e
+				}
+			}
+			if ep == nil {
+				return nil, errors.Errorf("no episode of season %d episode %d", seasonNum, epNum)
+			}
+
+			if ep.Status == episode.StatusMissing {
+				c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
+			}
+
+		}
+	} else { //season package download
+		c.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
+
+	}
 	history, err := c.db.SaveHistoryRecord(ent.History{
-		MediaID:          seriesId,
-		EpisodeID:        ep.ID,
-		SourceTitle:      r1.Name,
-		TargetDir:        dir,
-		Status:           history.StatusRunning,
-		Size:             r1.Size,
-		Saved:            torrent.Save(),
+		MediaID:     seriesId,
+		EpisodeNums: episodeNums,
+		SeasonNum:   seasonNum,
+		SourceTitle: r1.Name,
+		TargetDir:   dir,
+		Status:      history.StatusRunning,
+		Size:        int(r1.Size),
+		//Saved:            torrent.Save(),
+		Link:             magnet,
 		DownloadClientID: dlc.ID,
 		IndexerID:        r1.IndexerId,
 	})
 	if err != nil {
 		return nil, errors.Wrap(err, "save record")
 	}
-	if episodeNum > 0 {
-		if ep.Status == episode.StatusMissing {
-			c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
-		}
-	} else {
-		c.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
+
+	torrent, err := trc.Download(magnet, downloadDir)
+	if err != nil {
+		return nil, errors.Wrap(err, "downloading")
 	}
+	torrent.Start()
 
 	c.tasks[history.ID] = &Task{Torrent: torrent}
-	c.sendMsg(fmt.Sprintf(message.BeginDownload, r1.Name))
+	name := r1.Name
+
+	if len(episodeNums) > 0 {
+		buff := &bytes.Buffer{}
+		for i, ep := range episodeNums {
+			if i != 0 {
+				buff.WriteString(",")
+
+			}
+			buff.WriteString(fmt.Sprint(ep))
+		}
+		name = fmt.Sprintf("第%s集 (%s)", buff.String(), name)
+	} else {
+		name = fmt.Sprintf("全集 (%s)", name)
+	}
+
+	c.sendMsg(fmt.Sprintf(message.BeginDownload, name))
 
 	log.Infof("success add %s to download task", r1.Name)
 	return &r1.Name, nil
-
 }
-func (c *Client) SearchAndDownload(seriesId, seasonNum, episodeNum int) (*string, error) {
-	var episodes []int
-	if episodeNum > 0 {
-		episodes = append(episodes, episodeNum)
+
+/*
+tmdb 校验获取的资源名，如果用资源名在tmdb搜索出来的结果能匹配上想要的资源，则认为资源有效，否则无效
+解决名称过于简单的影视会匹配过多资源的问题， 例如：梦魇绝镇 FROM
+*/
+func (c *Client) checkBtReourceWithTmdb(r *torznab.Result, seriesId int) bool {
+	m := metadata.ParseTv(r.Name)
+	se, err := c.MustTMDB().SearchMedia(m.NameEn, "", 1)
+	if err != nil {
+		log.Warnf("tmdb search error, consider this torrent ok: %v", err)
+		return true
+	} else {
+		if len(se.Results) == 0 {
+			log.Debugf("tmdb search no result, consider this torrent ok: %s", r.Name) //because tv name parse is not accurate
+			return true
+		}
+		series := c.db.GetMediaDetails(seriesId)
+
+		se0 := se.Results[0]
+		if se0.ID != int64(series.TmdbID) {
+			log.Warnf("bt reosurce name not match tmdb id: %s", r.Name)
+			return false
+		} else { //resource tmdb id match
+			return true
+		}
 	}
+}
+
+func (c *Client) SearchAndDownload(seriesId, seasonNum int, episodeNums ...int) ([]string, error) {
+
 	res, err := SearchTvSeries(c.db, &SearchParam{
 		MediaId:         seriesId,
 		SeasonNum:       seasonNum,
-		Episodes:        episodes,
+		Episodes:        episodeNums,
 		CheckFileSize:   true,
 		CheckResolution: true,
 	})
 	if err != nil {
 		return nil, err
 	}
-	r1 := res[0]
-	log.Infof("found resource to download: %+v", r1)
-	return c.DownloadEpisodeTorrent(r1, seriesId, seasonNum, episodeNum)
+	wanted := make(map[int]bool, len(episodeNums))
+	for _, ep := range episodeNums {
+		wanted[ep] = true
+	}
+	var torrentNames []string
+lo:
+	for _, r := range res {
+		if !c.checkBtReourceWithTmdb(&r, seriesId) {
+			continue
+		}
+		m := metadata.ParseTv(r.Name)
+		m.ParseExtraDescription(r.Description)
+		if len(episodeNums) == 0 { //want season pack
+			if m.IsSeasonPack {
+				name, err := c.DownloadEpisodeTorrent(r, seriesId, seasonNum)
+				if err != nil {
+					return nil, err
+				}
+				torrentNames = append(torrentNames, *name)
+				break lo
+			}
+		} else {
+			torrentEpisodes := make([]int, 0)
+			for i := m.StartEpisode; i <= m.EndEpisode; i++ {
+				if !wanted[i] { //torrent has episode not wanted
+					continue lo
+				}
+				torrentEpisodes = append(torrentEpisodes, i)
+			}
+			name, err := c.DownloadEpisodeTorrent(r, seriesId, seasonNum, torrentEpisodes...)
+			if err != nil {
+				return nil, err
+			}
+			torrentNames = append(torrentNames, *name)
+
+			for _, num := range torrentEpisodes {
+				delete(wanted, num) //delete downloaded episode from wanted
+			}
+		}
+	}
+	if len(wanted) > 0 {
+		log.Warnf("still wanted but not downloaded episodes: %v", wanted)
+	}
+	return torrentNames, nil
 }
 
-func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexerID int) (*string, error) {
+func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int64, indexerID int) (*string, error) {
 	trc, dlc, err := c.GetDownloadClient()
 	if err != nil {
 		return nil, errors.Wrap(err, "connect transmission")
 	}
+	magnet, err := utils.Link2Magnet(link)
+	if err != nil {
+		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
+	}
 
-	torrent, err := trc.Download(link, c.db.GetDownloadDir())
+	torrent, err := trc.Download(magnet, c.db.GetDownloadDir())
 	if err != nil {
 		return nil, errors.Wrap(err, "downloading")
 	}
@@ -119,13 +214,14 @@ func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexe
 	go func() {
 		ep, _ := c.db.GetMovieDummyEpisode(m.ID)
 		history, err := c.db.SaveHistoryRecord(ent.History{
-			MediaID:          m.ID,
-			EpisodeID:        ep.ID,
-			SourceTitle:      name,
-			TargetDir:        m.TargetDir,
-			Status:           history.StatusRunning,
-			Size:             size,
-			Saved:            torrent.Save(),
+			MediaID:     m.ID,
+			EpisodeID:   ep.ID,
+			SourceTitle: name,
+			TargetDir:   m.TargetDir,
+			Status:      history.StatusRunning,
+			Size:        int(size),
+			//Saved:            torrent.Save(),
+			Link:             magnet,
 			DownloadClientID: dlc.ID,
 			IndexerID:        indexerID,
 		})

server/core/scheduler.go

@@ -2,6 +2,7 @@ package core
 
 import (
 	"fmt"
+	"os"
 	"path/filepath"
 	"polaris/db"
 	"polaris/ent"
@@ -12,6 +13,7 @@ import (
 	"polaris/pkg"
 	"polaris/pkg/notifier/message"
 	"polaris/pkg/utils"
+	"time"
 
 	"github.com/pkg/errors"
 )
@@ -19,6 +21,10 @@ import (
 func (c *Client) addSysCron() {
 	c.registerCronJob("check_running_tasks", "@every 1m", c.checkTasks)
 	c.registerCronJob("check_available_medias_to_download", "0 0 * * * *", func() error {
+		v := os.Getenv("POLARIS_NO_AUTO_DOWNLOAD")
+		if v == "true" {
+			return nil
+		}
 		c.downloadAllTvSeries()
 		c.downloadAllMovies()
 		return nil
@@ -36,6 +42,7 @@ func (c *Client) addSysCron() {
 		return true
 	})
 	c.cron.Start()
+	log.Infof("--------- add cron jobs done --------")
 }
 
 func (c *Client) mustAddCron(spec string, cmd func()) {
@@ -113,6 +120,38 @@ func (c *Client) postTaskProcessing(id int) {
 	}
 }
 
+func getSeasonNum(h *ent.History) int {
+	if h.SeasonNum != 0 {
+		return h.SeasonNum
+	}
+	seasonNum, err := utils.SeasonId(h.TargetDir)
+	if err != nil {
+		log.Errorf("no season id: %v", h.TargetDir)
+		seasonNum = -1
+	}
+	return seasonNum
+}
+
+func (c *Client) GetEpisodeIds(r *ent.History) []int {
+	var episodeIds []int
+	seasonNum := getSeasonNum(r)
+
+	if r.EpisodeID > 0 {
+		episodeIds = append(episodeIds, r.EpisodeID)
+	}
+	if len(r.EpisodeNums) > 0 {
+		series := c.db.GetMediaDetails(r.MediaID)
+		for _, epNum := range r.EpisodeNums {
+			for _, ep := range series.Episodes {
+				if ep.SeasonNumber == seasonNum && ep.EpisodeNumber == epNum {
+					episodeIds = append(episodeIds, ep.ID)
+				}
+			}
+		}
+	}
+	return episodeIds
+}
+
 func (c *Client) moveCompletedTask(id int) (err1 error) {
 	torrent := c.tasks[id]
 	r := c.db.GetHistory(id)
@@ -121,11 +160,7 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
 		return nil
 	}
 	c.db.SetHistoryStatus(r.ID, history.StatusUploading)
-	seasonNum, err := utils.SeasonId(r.TargetDir)
-	if err != nil {
-		log.Errorf("no season id: %v", r.TargetDir)
-		seasonNum = -1
-	}
+
 	downloadclient, err := c.db.GetDownloadClient(r.DownloadClientID)
 	if err != nil {
 		log.Errorf("get task download client error: %v, use default one", err)
@@ -136,13 +171,19 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
 		return err
 	}
 
+	seasonNum := getSeasonNum(r)
+
+	episodeIds := c.GetEpisodeIds(r)
+
 	defer func() {
 
 		if err1 != nil {
 			c.db.SetHistoryStatus(r.ID, history.StatusFail)
-			if r.EpisodeID != 0 {
-				if !c.db.IsEpisodeDownloadingOrDownloaded(r.EpisodeID) {
-					c.db.SetEpisodeStatus(r.EpisodeID, episode.StatusMissing)
+			if len(episodeIds) > 0 {
+				for _, id := range episodeIds {
+					if !c.db.IsEpisodeDownloadingOrDownloaded(id) {
+						c.db.SetEpisodeStatus(id, episode.StatusMissing)
+					}
 				}
 			} else {
 				c.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusMissing)
@@ -162,7 +203,7 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
 	}
 	st := c.db.GetStorage(series.StorageID)
 	log.Infof("move task files to target dir: %v", r.TargetDir)
-	stImpl, err := c.getStorage(st.ID, series.MediaType)
+	stImpl, err := c.GetStorage(st.ID, series.MediaType)
 	if err != nil {
 		return err
 	}
@@ -171,10 +212,13 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
 	if err := stImpl.Copy(filepath.Join(c.db.GetDownloadDir(), torrentName), r.TargetDir); err != nil {
 		return errors.Wrap(err, "move file")
 	}
+	torrent.UploadProgresser = stImpl.UploadProgress
 
 	c.db.SetHistoryStatus(r.ID, history.StatusSeeding)
-	if r.EpisodeID != 0 {
-		c.db.SetEpisodeStatus(r.EpisodeID, episode.StatusDownloaded)
+	if len(episodeIds) > 0 {
+		for _, id := range episodeIds {
+			c.db.SetEpisodeStatus(id, episode.StatusDownloaded)
+		}
 	} else {
 		c.db.SetSeasonAllEpisodeStatus(r.MediaID, seasonNum, episode.StatusDownloaded)
 	}
@@ -199,7 +243,7 @@ func (c *Client) CheckDownloadedSeriesFiles(m *ent.Media) error {
 	}
 	log.Infof("check files in directory: %s", m.TargetDir)
 
-	var storageImpl, err = c.getStorage(m.StorageID, media.MediaTypeTv)
+	var storageImpl, err = c.GetStorage(m.StorageID, media.MediaTypeTv)
 	if err != nil {
 		return err
 	}
@@ -247,6 +291,7 @@ func (c *Client) CheckDownloadedSeriesFiles(m *ent.Media) error {
 type Task struct {
 	//Processing bool
 	pkg.Torrent
+	UploadProgresser func() float64
 }
 
 func (c *Client) DownloadSeriesAllEpisodes(id int) []string {
@@ -261,40 +306,53 @@ func (c *Client) DownloadSeriesAllEpisodes(id int) []string {
 			continue
 		}
 		wantedSeasonPack := true
+		seasonEpisodesWanted := make(map[int][]int, 0)
 		for _, ep := range epsides {
 			if !ep.Monitored {
 				wantedSeasonPack = false
+				continue
 			}
 			if ep.Status != episode.StatusMissing {
 				wantedSeasonPack = false
+				continue
 			}
+			if ep.AirDate != "" {
+				t, err := time.Parse("2006-01-02", ep.AirDate)
+				if err != nil {
+					continue
+				}
+				/*
+				 -------- now ------ t -----
+				 t - 1day < now 要检测的剧集
+				 提前一天开始检测
+				*/
+				if time.Now().Before(t.Add(-24 * time.Hour)) { //not aired
+					wantedSeasonPack = false
+					continue
+				}
+			}
+			seasonEpisodesWanted[ep.SeasonNumber] = append(seasonEpisodesWanted[ep.SeasonNumber], ep.EpisodeNumber)
 		}
 		if wantedSeasonPack {
-			name, err := c.SearchAndDownload(id, seasonNum, -1)
+			names, err := c.SearchAndDownload(id, seasonNum)
 			if err == nil {
-				allNames = append(allNames, *name)
-				log.Infof("begin download torrent resource: %v", name)
+				allNames = append(allNames, names...)
+				log.Infof("begin download torrent resource: %v", names)
 			} else {
 				log.Warnf("finding season pack error: %v", err)
 				wantedSeasonPack = false
 			}
-
 		}
 		if !wantedSeasonPack {
-			for _, ep := range epsides {
-				if !ep.Monitored {
-					continue
-				}
-				if ep.Status != episode.StatusMissing {
-					continue
-				}
-				name, err := c.SearchAndDownload(id, ep.SeasonNumber, ep.EpisodeNumber)
+
+			for se, eps := range seasonEpisodesWanted {
+				names, err := c.SearchAndDownload(id, se, eps...)
 				if err != nil {
-					log.Warnf("finding resoruces of season %d episode %d error: %v", ep.SeasonNumber, ep.EpisodeNumber, err)
+					log.Warnf("finding resoruces of season %d episode %v error: %v", se, eps, err)
 					continue
 				} else {
-					allNames = append(allNames, *name)
-					log.Infof("begin download torrent resource: %v", name)
+					allNames = append(allNames, names...)
+					log.Infof("begin download torrent resource: %v", names)
 				}
 			}
 
@@ -363,20 +421,27 @@ func (c *Client) downloadMovieSingleEpisode(ep *ent.Episode, targetDir string) (
 	}
 	r1 := res[0]
 	log.Infof("begin download torrent resource: %v", r1.Name)
-	torrent, err := trc.Download(r1.Link, c.db.GetDownloadDir())
+
+	magnet, err := utils.Link2Magnet(r1.Link)
+	if err != nil {
+		return "", errors.Errorf("converting link to magnet error, link: %v, error: %v", r1.Link, err)
+	}
+
+	torrent, err := trc.Download(magnet, c.db.GetDownloadDir())
 	if err != nil {
 		return "", errors.Wrap(err, "downloading")
 	}
 	torrent.Start()
 
 	history, err := c.db.SaveHistoryRecord(ent.History{
-		MediaID:          ep.MediaID,
-		EpisodeID:        ep.ID,
-		SourceTitle:      r1.Name,
-		TargetDir:        targetDir,
-		Status:           history.StatusRunning,
-		Size:             r1.Size,
-		Saved:            torrent.Save(),
+		MediaID:     ep.MediaID,
+		EpisodeID:   ep.ID,
+		SourceTitle: r1.Name,
+		TargetDir:   targetDir,
+		Status:      history.StatusRunning,
+		Size:        int(r1.Size),
+		//Saved:            torrent.Save(),
+		Link:             magnet,
 		DownloadClientID: dlc.ID,
 		IndexerID:        r1.IndexerId,
 	})
@@ -440,7 +505,7 @@ func (c *Client) checkSeiesNewSeason(media *ent.Media) error {
 	return nil
 }
 
-func (c *Client) isSeedRatioLimitReached(indexId int, t pkg.Torrent)(float64,bool)  {
+func (c *Client) isSeedRatioLimitReached(indexId int, t pkg.Torrent) (float64, bool) {
 	indexer, err := c.db.GetIndexer(indexId)
 	if err != nil {
 		return 0, true

server/core/torrent.go

@@ -3,9 +3,12 @@ package core
 import (
 	"fmt"
 	"polaris/db"
+	"polaris/ent"
 	"polaris/ent/media"
 	"polaris/log"
+	"polaris/pkg/gemini"
 	"polaris/pkg/metadata"
+	"polaris/pkg/prowlarr"
 	"polaris/pkg/torznab"
 	"slices"
 	"sort"
@@ -25,22 +28,128 @@ type SearchParam struct {
 	FilterQiangban  bool //for movie, 是否过滤枪版电影
 }
 
+func names2Query(media *ent.Media) []string {
+	var names = []string{media.NameEn}
+
+	if media.NameCn != "" {
+		hasName := false
+		for _, n := range names {
+			if media.NameCn == n {
+				hasName = true
+			}
+		}
+		if !hasName {
+			names = append(names, media.NameCn)
+		}
+
+	}
+	if media.OriginalName != "" {
+		hasName := false
+		for _, n := range names {
+			if media.OriginalName == n {
+				hasName = true
+			}
+		}
+		if !hasName {
+			names = append(names, media.OriginalName)
+		}
+
+	}
+
+	for _, t := range media.AlternativeTitles {
+		if (t.Iso3166_1 == "CN" || t.Iso3166_1 == "US") && t.Type == "" {
+			hasName := false
+			for _, n := range names {
+				if t.Title == n {
+					hasName = true
+				}
+			}
+			if !hasName {
+				names = append(names, t.Title)
+			}
+		}
+	}
+	log.Debugf("name to query %+v", names)
+	return names
+}
+
+func filterBasedOnGemini(cfg db.AIConfig, res []torznab.Result, names ...string) []torznab.Result {
+
+	var torrentNames []string
+	for _, r := range res {
+		torrentNames = append(torrentNames, r.Name)
+	}
+	g, err := gemini.NewClient(cfg.GeminiApiKey, cfg.GeminiModelName)
+	if err != nil {
+		log.Warnf("create gemini client: %v", err)
+		return res
+	}
+	resf, err := g.FilterTvOrMovies(torrentNames, names...)
+	if err != nil {
+		log.Warnf("filter with gemini: %v", err)
+		return res
+	}
+	var newRes []torznab.Result
+	for _, r := range res {
+		if slices.Contains(resf, r.Name) {
+			newRes = append(newRes, r)
+		}
+	}
+	return newRes
+}
+
+func filterBasedOnRules(res []torznab.Result, names ...string) []torznab.Result {
+	var filtered []torznab.Result
+	for _, r := range res {
+		meta := metadata.ParseTv(r.Name)
+		if meta.IsAcceptable(names...) {
+			filtered = append(filtered, r)
+		}
+	}
+	return filtered
+}
+
+func filterResourceNames(db1 *db.Client, res []torznab.Result, names ...string) []torznab.Result {
+	n1 := len(res)
+	cfg, err := db1.GetAIConfig()
+	if err != nil {
+		log.Warnf("get ai config: %v", err)
+	}
+	if cfg.Enabled {
+		res = filterBasedOnGemini(cfg, res, names...)
+	} else {
+		res = filterBasedOnRules(res, names...)
+	}
+	log.Infof("resource before name filtering length is %d, after filtering length is %d", n1, len(res))
+	return res
+
+}
+
 func SearchTvSeries(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 	series := db1.GetMediaDetails(param.MediaId)
 	if series == nil {
 		return nil, fmt.Errorf("no tv series of id %v", param.MediaId)
 	}
+	limiter, err := db1.GetSizeLimiter("tv")
+	if err != nil {
+		log.Warnf("get tv size limiter: %v", err)
+		limiter = &db.MediaSizeLimiter{}
+	}
 	log.Debugf("check tv series %s, season %d, episode %v", series.NameEn, param.SeasonNum, param.Episodes)
 
-	res := searchWithTorznab(db1, series.NameEn, series.NameCn, series.OriginalName)
+	names := names2Query(series.Media)
+
+	res := searchWithTorznab(db1, prowlarr.TV, names...)
+
+	res = filterResourceNames(db1, res, names...)
 
 	var filtered []torznab.Result
+lo:
 	for _, r := range res {
 		//log.Infof("torrent resource: %+v", r)
 		meta := metadata.ParseTv(r.Name)
-		if meta == nil { //cannot parse name
-			continue
-		}
+		meta.ParseExtraDescription(r.Description)
+
 		if isImdbidNotMatch(series.ImdbID, r.ImdbId) { //has imdb id and not match
 			continue
 		}
@@ -60,20 +169,26 @@ func SearchTvSeries(db1 *db.Client, param *SearchParam) ([]torznab.Result, error
 			continue
 		}
 
-		if len(param.Episodes) > 0 && !slices.Contains(param.Episodes, meta.Episode) { //not season pack, but episode number not equal
-			continue
-
+		if len(param.Episodes) > 0 { //not season pack, but episode number not equal
+			if meta.StartEpisode <= 0 {
+				continue lo
+			}
+			for i := meta.StartEpisode; i <= meta.EndEpisode; i++ {
+				if !slices.Contains(param.Episodes, i) {
+					continue lo
+				}
+			}
 		} else if len(param.Episodes) == 0 && !meta.IsSeasonPack { //want season pack, but not season pack
 			continue
 		}
 
-		if param.CheckResolution && 
-			series.Resolution != media.ResolutionAny && 
+		if param.CheckResolution &&
+			series.Resolution != media.ResolutionAny &&
 			meta.Resolution != series.Resolution.String() {
 			continue
 		}
 
-		if !torrentSizeOk(series, r.Size, param) {
+		if !torrentSizeOk(series, limiter, r.Size, meta.EndEpisode+1-meta.StartEpisode, param) {
 			continue
 		}
 
@@ -107,22 +222,42 @@ func imdbIDMatchExact(id1, id2 string) bool {
 	return id1 == id2
 }
 
-func torrentSizeOk(detail *db.MediaDetails, torrentSize int, param *SearchParam) bool {
-	if param.CheckFileSize {
-		multiplier := 1                                                        //大小倍数，正常为1，如果是季包则为季内集数
-		if detail.MediaType == media.MediaTypeTv && len(param.Episodes) == 0 { //tv season pack
+func torrentSizeOk(detail *db.MediaDetails, globalLimiter *db.MediaSizeLimiter, torrentSize int64,
+	torrentEpisodeNum int, param *SearchParam) bool {
+
+	multiplier := 1 //大小倍数，正常为1，如果是季包则为季内集数
+	if detail.MediaType == media.MediaTypeTv {
+		if len(param.Episodes) == 0 { //want tv season pack
 			multiplier = seasonEpisodeCount(detail, param.SeasonNum)
+		} else {
+			multiplier = torrentEpisodeNum
 		}
+	}
+
+	if param.CheckFileSize { //check file size when trigger automatic download
 
 		if detail.Limiter.SizeMin > 0 { //min size
-			sizeMin := detail.Limiter.SizeMin * multiplier
-			if torrentSize < sizeMin { //比最小要求的大小还要小
+			sizeMin := detail.Limiter.SizeMin * int64(multiplier)
+			if torrentSize < sizeMin { //比最小要求的大小还要小, min size not qualify
+				return false
+			}
+		} else if globalLimiter != nil {
+			resLimiter := globalLimiter.GetLimiter(detail.Resolution)
+			sizeMin := resLimiter.MinSize * int64(multiplier)
+			if torrentSize < sizeMin { //比最小要求的大小还要小, min size not qualify
 				return false
 			}
 		}
+
 		if detail.Limiter.SizeMax > 0 { //max size
-			sizeMax := detail.Limiter.SizeMax * multiplier
-			if torrentSize > sizeMax { //larger than max size wanted
+			sizeMax := detail.Limiter.SizeMax * int64(multiplier)
+			if torrentSize > sizeMax { //larger than max size wanted, max size not qualify
+				return false
+			}
+		} else if globalLimiter != nil {
+			resLimiter := globalLimiter.GetLimiter(detail.Resolution)
+			sizeMax := resLimiter.MaxSIze * int64(multiplier)
+			if torrentSize > sizeMax { //larger than max size wanted, max size not qualify
 				return false
 			}
 		}
@@ -161,9 +296,18 @@ func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 		return nil, errors.New("no media found of id")
 	}
 
-	res := searchWithTorznab(db1, movieDetail.NameEn, movieDetail.NameCn, movieDetail.OriginalName)
+	limiter, err := db1.GetSizeLimiter("movie")
+	if err != nil {
+		log.Warnf("get tv size limiter: %v", err)
+		limiter = &db.MediaSizeLimiter{}
+	}
+	names := names2Query(movieDetail.Media)
+
+	res := searchWithTorznab(db1, prowlarr.Movie, names...)
+	res = filterResourceNames(db1, res, names...)
+
 	if movieDetail.Extras.IsJav() {
-		res1 := searchWithTorznab(db1, movieDetail.Extras.JavId)
+		res1 := searchWithTorznab(db1, prowlarr.Movie, movieDetail.Extras.JavId)
 		res = append(res, res1...)
 	}
 
@@ -191,8 +335,8 @@ func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 			}
 		}
 
-		if param.CheckResolution && 
-			movieDetail.Resolution != media.ResolutionAny && 
+		if param.CheckResolution &&
+			movieDetail.Resolution != media.ResolutionAny &&
 			meta.Resolution != movieDetail.Resolution.String() {
 			continue
 		}
@@ -201,7 +345,7 @@ func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 			continue
 		}
 
-		if !torrentSizeOk(movieDetail, r.Size, param) {
+		if !torrentSizeOk(movieDetail, limiter, r.Size, 1, param) {
 			continue
 		}
 
@@ -217,10 +361,21 @@ func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 
 }
 
-func searchWithTorznab(db *db.Client, queries ...string) []torznab.Result {
+func searchWithTorznab(db *db.Client, t prowlarr.ProwlarrSupportType, queries ...string) []torznab.Result {
 
 	var res []torznab.Result
 	allTorznab := db.GetAllTorznabInfo()
+
+	p, err := db.GetProwlarrSetting()
+	if err == nil && !p.Disabled { //prowlarr exists
+		c := prowlarr.New(p.ApiKey, p.URL)
+		all, err := c.GetIndexers(t)
+		if err != nil {
+			log.Warnf("get prowlarr all indexer error: %v", err)
+		} else {
+			allTorznab = append(allTorznab, all...)
+		}
+	}
 	resChan := make(chan []torznab.Result)
 	var wg sync.WaitGroup
 
@@ -312,5 +467,7 @@ func torrentNameOk(detail *db.MediaDetails, tester NameTester) bool {
 	if detail.Extras.IsJav() && tester.IsAcceptable(detail.Extras.JavId) {
 		return true
 	}
-	return tester.IsAcceptable(detail.NameCn, detail.NameEn, detail.OriginalName)
+	names := names2Query(detail.Media)
+
+	return tester.IsAcceptable(names...)
 }

server/resources.go

@@ -28,7 +28,7 @@ func (s *Server) searchAndDownloadSeasonPackage(seriesId, seasonNum int) (*strin
 
 	r1 := res[0]
 	log.Infof("found resource to download: %+v", r1)
-	return s.core.DownloadEpisodeTorrent(r1, seriesId, seasonNum, -1)
+	return s.core.DownloadEpisodeTorrent(r1, seriesId, seasonNum)
 
 }
 
@@ -120,7 +120,7 @@ func (s *Server) SearchTvAndDownload(c *gin.Context) (interface{}, error) {
 		if err != nil {
 			return nil, errors.Wrap(err, "download")
 		}
-		name = *name1
+		name = name1[0]
 	}
 
 	return gin.H{
@@ -154,7 +154,7 @@ func (s *Server) DownloadTorrent(c *gin.Context) (interface{}, error) {
 				name = fmt.Sprintf("%v S%02d", m.OriginalName, in.Season)
 			}
 			res := torznab.Result{Name: name, Link: in.Link, Size: in.Size}
-			return s.core.DownloadEpisodeTorrent(res, in.MediaID, in.Season, -1)
+			return s.core.DownloadEpisodeTorrent(res, in.MediaID, in.Season)
 		}
 		name := in.Name
 		if name == "" {
@@ -175,6 +175,10 @@ func (s *Server) DownloadAll(c *gin.Context) (interface{}, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "convert")
 	}
+	return s.downloadAllEpisodes(id)
+}
+
+func (s *Server) downloadAllEpisodes(id int) (interface{}, error) {
 	m, err := s.db.GetMedia(id)
 	if err != nil {
 		return nil, errors.Wrap(err, "get media")
@@ -186,3 +190,27 @@ func (s *Server) DownloadAll(c *gin.Context) (interface{}, error) {
 
 	return []string{name}, err
 }
+
+func (s *Server) DownloadAllTv(c *gin.Context) (interface{}, error) {
+	tvs := s.db.GetMediaWatchlist(media.MediaTypeTv)
+	var allNames []string
+	for _, tv := range tvs {
+		names, err := s.downloadAllEpisodes(tv.ID)
+		if err == nil {
+			allNames = append(allNames, names.([]string)...)
+		}
+	}
+	return allNames, nil
+}
+
+func (s *Server) DownloadAllMovies(c *gin.Context) (interface{}, error) {
+	movies := s.db.GetMediaWatchlist(media.MediaTypeMovie)
+	var allNames []string
+	for _, mv := range movies {
+		names, err := s.downloadAllEpisodes(mv.ID)
+		if err == nil {
+			allNames = append(allNames, names.([]string)...)
+		}
+	}
+	return allNames, nil
+}

server/server.go

@@ -7,9 +7,11 @@ import (
 	"net/url"
 	"polaris/db"
 	"polaris/log"
+	"polaris/pkg/cache"
 	"polaris/pkg/tmdb"
 	"polaris/server/core"
 	"polaris/ui"
+	"time"
 
 	ginzap "github.com/gin-contrib/zap"
 
@@ -22,20 +24,24 @@ import (
 func NewServer(db *db.Client) *Server {
 	r := gin.Default()
 	s := &Server{
-		r:        r,
-		db:       db,
-		language: db.GetLanguage(),
+		r:                r,
+		db:               db,
+		language:         db.GetLanguage(),
+		monitorNumCache:  cache.NewCache[int, int](10 * time.Minute),
+		downloadNumCache: cache.NewCache[int, int](10 * time.Minute),
 	}
 	s.core = core.NewClient(db, s.language)
 	return s
 }
 
 type Server struct {
-	r         *gin.Engine
-	db        *db.Client
-	core      *core.Client
-	language  string
-	jwtSerect string
+	r                *gin.Engine
+	db               *db.Client
+	core             *core.Client
+	language         string
+	jwtSerect        string
+	monitorNumCache  *cache.Cache[int, int]
+	downloadNumCache *cache.Cache[int, int]
 }
 
 func (s *Server) Serve() error {
@@ -70,13 +76,19 @@ func (s *Server) Serve() error {
 		setting.POST("/parse/movie", HttpHandler(s.ParseMovie))
 		setting.POST("/monitoring", HttpHandler(s.ChangeEpisodeMonitoring))
 		setting.POST("/cron/trigger", HttpHandler(s.TriggerCronJob))
+		setting.GET("/prowlarr", HttpHandler(s.GetProwlarrSetting))
+		setting.POST("/prowlarr", HttpHandler(s.SaveProwlarrSetting))
+		setting.GET("/limiter", HttpHandler(s.GetSizeLimiter))
+		setting.POST("/limiter", HttpHandler(s.SetSizeLimiter))
+		setting.GET("/ai", HttpHandler(s.GetAIConfig))
+		setting.POST("/ai", HttpHandler(s.SetAIConfig))
 	}
 	activity := api.Group("/activity")
 	{
 		activity.GET("/", HttpHandler(s.GetAllActivities))
-		activity.DELETE("/:id", HttpHandler(s.RemoveActivity))
+		activity.POST("/delete", HttpHandler(s.RemoveActivity))
 		activity.GET("/media/:id", HttpHandler(s.GetMediaDownloadHistory))
-		activity.GET("/torrents", HttpHandler(s.GetAllTorrents))
+		//activity.GET("/torrents", HttpHandler(s.GetAllTorrents))
 	}
 
 	tv := api.Group("/media")
@@ -94,6 +106,8 @@ func (s *Server) Serve() error {
 		tv.GET("/suggest/tv/:tmdb_id", HttpHandler(s.SuggestedSeriesFolderName))
 		tv.GET("/suggest/movie/:tmdb_id", HttpHandler(s.SuggestedMovieFolderName))
 		tv.GET("/downloadall/:id", HttpHandler(s.DownloadAll))
+		tv.GET("/download/tv", HttpHandler(s.DownloadAllTv))
+		tv.GET("/download/movie", HttpHandler(s.DownloadAllMovies))
 	}
 	indexer := api.Group("/indexer")
 	{
@@ -128,6 +142,7 @@ func (s *Server) Serve() error {
 		importlist.POST("/add", HttpHandler(s.addImportlist))
 		importlist.DELETE("/delete", HttpHandler(s.deleteImportList))
 	}
+	log.Infof("----------- Polaris Server Successfully Started ------------")
 
 	return s.r.Run(":8080")
 }

server/setting.go

@@ -8,7 +8,9 @@ import (
 	"polaris/ent"
 	"polaris/ent/downloadclients"
 	"polaris/log"
+	"polaris/pkg/prowlarr"
 	"polaris/pkg/qbittorrent"
+	"polaris/pkg/torznab"
 	"polaris/pkg/transmission"
 	"polaris/pkg/utils"
 	"strconv"
@@ -69,7 +71,6 @@ func (s *Server) SetSetting(c *gin.Context) (interface{}, error) {
 		if _, err := template.New("test").Parse(in.MovieNamingFormat); err != nil {
 			return nil, errors.Wrap(err, "movie format")
 		}
-
 		s.db.SetSetting(db.SettingMovieNamingFormat, in.MovieNamingFormat)
 	} else {
 		s.db.SetSetting(db.SettingMovieNamingFormat, "")
@@ -171,6 +172,8 @@ func (s *Server) AddTorznabInfo(c *gin.Context) (interface{}, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "add ")
 	}
+
+	torznab.CleanCache() //need to clean exist cache, so next request will do actaul query
 	return nil, nil
 }
 
@@ -300,3 +303,82 @@ func (s *Server) TriggerCronJob(c *gin.Context) (interface{}, error) {
 	}
 	return "success", nil
 }
+
+func (s *Server) GetProwlarrSetting(c *gin.Context) (interface{}, error) {
+	se, err := s.db.GetProwlarrSetting()
+	if err != nil {
+		return &db.ProwlarrSetting{}, nil
+	}
+	return se, nil
+}
+func (s *Server) SaveProwlarrSetting(c *gin.Context) (interface{}, error) {
+	var in db.ProwlarrSetting
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, err
+	}
+	if !in.Disabled {
+		client := prowlarr.New(in.ApiKey, in.URL)
+		if _, err := client.GetIndexers(prowlarr.TV); err != nil {
+			return nil, errors.Wrap(err, "connect to prowlarr error")
+		}
+	}
+	err := s.db.SaveProwlarrSetting(&in)
+	if err != nil {
+		return nil, err
+	}
+	return "success", nil
+}
+
+type ResolutionSizeLimiter struct {
+	TvLimiter    *db.MediaSizeLimiter `json:"tv_limiter"`
+	MovieLimiter *db.MediaSizeLimiter `json:"movie_limiter"`
+}
+
+func (s *Server) GetSizeLimiter(c *gin.Context) (interface{}, error) {
+	tv, err := s.db.GetSizeLimiter("tv")
+	if err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	movie, err := s.db.GetSizeLimiter("movie")
+	if err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	r := ResolutionSizeLimiter{
+		TvLimiter:    tv,
+		MovieLimiter: movie,
+	}
+	return r, nil
+}
+
+func (s *Server) SetSizeLimiter(c *gin.Context) (interface{}, error) {
+	var in ResolutionSizeLimiter
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, err
+	}
+	if err := s.db.SetSizeLimiter("tv", in.TvLimiter); err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	if err := s.db.SetSizeLimiter("movie", in.MovieLimiter); err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	return "success", nil
+}
+
+func (s *Server) GetAIConfig(c *gin.Context) (interface{}, error) {
+	aiConfig, err := s.db.GetAIConfig()
+	if err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	return aiConfig, nil
+}
+
+func (s *Server) SetAIConfig(c *gin.Context) (interface{}, error) {
+	var in db.AIConfig
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, err
+	}
+	if err := s.db.SetAIConfig(&in); err != nil {
+		return nil, errors.Wrap(err, "db")
+	}
+	return "success", nil
+}

server/storage.go

@@ -2,9 +2,11 @@ package server
 
 import (
 	"fmt"
+	"os"
 	"polaris/db"
 
 	"polaris/log"
+	"polaris/pkg/alist"
 	"polaris/pkg/storage"
 	"polaris/pkg/utils"
 	"strconv"
@@ -28,7 +30,7 @@ func (s *Server) AddStorage(c *gin.Context) (interface{}, error) {
 	if in.Implementation == "webdav" {
 		//test webdav
 		wd := in.ToWebDavSetting()
-		st, err := storage.NewWebdavStorage(wd.URL, wd.User, wd.Password, in.TvPath, false)
+		st, err := storage.NewWebdavStorage(wd.URL, wd.User, wd.Password, in.TvPath, false, nil, nil)
 		if err != nil {
 			return nil, errors.Wrap(err, "new webdav")
 		}
@@ -39,6 +41,21 @@ func (s *Server) AddStorage(c *gin.Context) (interface{}, error) {
 		for _, f := range fs {
 			log.Infof("file name: %v", f.Name())
 		}
+	} else if in.Implementation == "alist" {
+		cfg := in.ToAlistSetting()
+		_, err := storage.NewAlist(&alist.Config{URL: cfg.URL, Username: cfg.User, Password: cfg.Password}, in.TvPath, nil, nil)
+		if err != nil {
+			return nil, errors.Wrap(err, "alist")
+		}
+	} else if in.Implementation == "local" {
+		_, err := os.Stat(in.TvPath)
+		if err != nil {
+			return nil, err
+		}
+		_, err = os.Stat(in.MoviePath)
+		if err != nil {
+			return nil, err
+		}
 	}
 	log.Infof("received add storage input: %v", in)
 	err := s.db.AddStorage(&in)

server/watchlist.go

@@ -10,6 +10,7 @@ import (
 	"polaris/log"
 	"polaris/server/core"
 	"strconv"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 	"github.com/pkg/errors"
@@ -102,17 +103,25 @@ func (s *Server) GetTvWatchlist(c *gin.Context) (interface{}, error) {
 			MonitoredNum:  0,
 			DownloadedNum: 0,
 		}
-
-		details := s.db.GetMediaDetails(item.ID)
-
-		for _, ep := range details.Episodes {
-			if ep.Monitored {
-				ms.MonitoredNum++
-				if ep.Status == episode.StatusDownloaded {
-					ms.DownloadedNum++
+		mon, ok1 := s.monitorNumCache.Get(item.ID)
+		dow, ok2 := s.downloadNumCache.Get(item.ID)
+		if ok1 && ok2 {
+			ms.MonitoredNum = mon
+			ms.DownloadedNum = dow
+		} else {
+			details := s.db.GetMediaDetails(item.ID)
+			for _, ep := range details.Episodes {
+				if ep.Monitored {
+					ms.MonitoredNum++
+					if ep.Status == episode.StatusDownloaded {
+						ms.DownloadedNum++
+					}
 				}
 			}
+			s.monitorNumCache.Set(item.ID, ms.MonitoredNum)
+			s.downloadNumCache.Set(item.ID, ms.DownloadedNum)
 		}
+
 		res[i] = ms
 	}
 	return res, nil
@@ -162,9 +171,32 @@ func (s *Server) DeleteFromWatchlist(c *gin.Context) (interface{}, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "convert")
 	}
+
+	deleteFiles := c.Query("delete_files")
+	if strings.ToLower(deleteFiles) == "true" {
+		//will delete local media file
+		log.Infof("will delete local media files for %d", id)
+		m, err := s.db.GetMedia(id)
+		if err != nil {
+			log.Warnf("get media: %v", err)
+		} else {
+			st, err := s.core.GetStorage(m.StorageID, m.MediaType)
+			if err != nil {
+				log.Warnf("get storage error: %v", err)
+			} else {
+				if err := st.RemoveAll(m.TargetDir); err != nil {
+					log.Warnf("remove all : %v", err)
+				} else {
+					log.Infof("delete media files success: %v", m.TargetDir)
+				}
+			}
+		}
+	}
+
 	if err := s.db.DeleteMedia(id); err != nil {
 		return nil, errors.Wrap(err, "delete db")
 	}
 	os.RemoveAll(filepath.Join(db.ImgPath, ids)) //delete image related
+
 	return "success", nil
 }

ui/.metadata

@@ -4,7 +4,7 @@
 # This file should be version controlled and should not be manually edited.
 
 version:
-  revision: "80c2e84975bbd28ecf5f8d4bd4ca5a2490bfc819"
+  revision: "dec2ee5c1f98f8e84a7d5380c05eb8a3d0a81668"
   channel: "stable"
 
 project_type: app
@@ -13,11 +13,11 @@ project_type: app
 migration:
   platforms:
     - platform: root
-      create_revision: 80c2e84975bbd28ecf5f8d4bd4ca5a2490bfc819
-      base_revision: 80c2e84975bbd28ecf5f8d4bd4ca5a2490bfc819
-    - platform: ios
-      create_revision: 80c2e84975bbd28ecf5f8d4bd4ca5a2490bfc819
-      base_revision: 80c2e84975bbd28ecf5f8d4bd4ca5a2490bfc819
+      create_revision: dec2ee5c1f98f8e84a7d5380c05eb8a3d0a81668
+      base_revision: dec2ee5c1f98f8e84a7d5380c05eb8a3d0a81668
+    - platform: macos
+      create_revision: dec2ee5c1f98f8e84a7d5380c05eb8a3d0a81668
+      base_revision: dec2ee5c1f98f8e84a7d5380c05eb8a3d0a81668
 
   # User provided section
 

ui/lib/activity.dart

@@ -138,7 +138,7 @@ class _ActivityPageState extends ConsumerState<ActivityPage>
                           trailing: selectedTab == 0
                               ? IconButton(
                                   tooltip: "删除任务",
-                                  onPressed: () => onDelete()(ac.id!.toString()),
+                                  onPressed: () => onDelete()(ac.id!),
                                   icon: const Icon(Icons.delete))
                               : const Text("-"),
                         ),
@@ -148,14 +148,14 @@ class _ActivityPageState extends ConsumerState<ActivityPage>
                   },
                 ));
               },
-              error: (err, trace) => Text("$err"),
+              error: (err, trace) => PoNetworkError(err: err),
               loading: () => const MyProgressIndicator());
         })
       ],
     );
   }
 
-  Function(String) onDelete() {
+  Function(int) onDelete() {
     return (id) {
       final f = ref
           .read(activitiesDataProvider("active").notifier)

ui/lib/init_wizard.dart

@@ -0,0 +1,118 @@
+import 'package:flutter/material.dart';
+import 'package:flutter_form_builder/flutter_form_builder.dart';
+import 'package:flutter_riverpod/flutter_riverpod.dart';
+import 'package:ui/settings/downloader.dart';
+import 'package:ui/settings/prowlarr.dart';
+import 'package:ui/settings/storage.dart';
+
+class InitWizard extends ConsumerStatefulWidget {
+  const InitWizard({super.key});
+  static final String route = "/init_wizard";
+  @override
+  ConsumerState<ConsumerStatefulWidget> createState() {
+    return _InitWizardState();
+  }
+}
+
+class _InitWizardState extends ConsumerState<InitWizard> {
+  @override
+  Widget build(BuildContext context) {
+    return Scaffold(
+      body: SelectionArea(
+          child: Container(
+        padding: EdgeInsets.all(50),
+        child: ListView(
+          children: [
+            Container(
+              alignment: Alignment.center,
+              child: Text(
+                "Polaris 影视追踪下载",
+                style: TextStyle(
+                    fontSize: 30,
+                    color: Theme.of(context).colorScheme.primary,
+                    fontWeight: FontWeight.bold),
+              ),
+            ),
+            Container(
+              padding: EdgeInsets.only(left: 10, top: 30, bottom: 30),
+              child: Text(
+                "设置向导",
+                style: TextStyle(
+                    fontSize: 20,
+                    fontWeight: FontWeight.bold,
+                    color: Theme.of(context).colorScheme.primary),
+              ),
+            ),
+            tmdbSetting(),
+            downloaderSetting(),
+            indexerSetting(),
+            storageSetting(),
+          ],
+        ),
+      )),
+    );
+  }
+
+  Widget tmdbSetting() {
+    return ExpansionTile(
+      title: Text(
+        "第一步：TMDB设置",
+        style: TextStyle(fontWeight: FontWeight.bold),
+      ),
+      childrenPadding: EdgeInsets.only(left: 100, right: 20),
+      initiallyExpanded: true,
+      children: [
+        Container(
+          alignment: Alignment.topLeft,
+          child: Text("TMDB API Key 设置，用来获取各种影视的信息，API Key获取方式参考官网"),
+        ),
+        FormBuilder(
+            child: Column(
+          children: [
+            FormBuilderTextField(
+              name: "tmdb",
+              decoration: InputDecoration(labelText: "TMDB API Key"),
+            ),
+            Center(
+                child: Padding(
+              padding: EdgeInsets.all(10),
+              child: ElevatedButton(onPressed: null, child: Text("保存")),
+            ))
+          ],
+        ))
+      ],
+    );
+  }
+
+  Widget indexerSetting() {
+    return ExpansionTile(
+      initiallyExpanded: true,
+      childrenPadding: EdgeInsets.only(left: 100, right: 20),
+      title: Text(
+        "第三步：Prowlarr设置",
+        style: TextStyle(fontWeight: FontWeight.bold),
+      ),
+      children: [ProwlarrSettingPage()],
+    );
+  }
+
+  Widget downloaderSetting() {
+    return ExpansionTile(
+      childrenPadding: EdgeInsets.only(left: 100, right: 20),
+      initiallyExpanded: true,
+      title: Text("第二步：下载客户端", style: TextStyle(fontWeight: FontWeight.bold)),
+      children: [
+        DownloaderSettings(),
+      ],
+    );
+  }
+
+  Widget storageSetting() {
+    return ExpansionTile(
+      childrenPadding: EdgeInsets.only(left: 100, right: 20),
+      title: Text("第四步：存储设置", style: TextStyle(fontWeight: FontWeight.bold)),
+      initiallyExpanded: true,
+      children: [StorageSettings()],
+    );
+  }
+}

ui/lib/main.dart

@@ -4,7 +4,7 @@ import 'package:flutter_riverpod/flutter_riverpod.dart';
 import 'package:go_router/go_router.dart';
 import 'package:intl/date_symbol_data_local.dart';
 import 'package:ui/activity.dart';
-import 'package:ui/calendar.dart';
+import 'package:ui/init_wizard.dart';
 import 'package:ui/login_page.dart';
 import 'package:ui/movie_watchlist.dart';
 import 'package:ui/providers/APIs.dart';
@@ -16,7 +16,8 @@ import 'package:ui/welcome_page.dart';
 import 'package:ui/widgets/utils.dart';
 
 void main() {
-  initializeDateFormatting().then((_) => runApp(MyApp()));
+  initializeDateFormatting()
+      .then((_) => runApp(const ProviderScope(child: MyApp())));
 }
 
 class MyApp extends ConsumerStatefulWidget {
@@ -45,114 +46,125 @@ CustomTransitionPage buildPageWithDefaultTransition<T>({
 
 class _MyAppState extends ConsumerState<MyApp> {
   // This widget is the root of your application.
+
   @override
   Widget build(BuildContext context) {
     // GoRouter configuration
-    final shellRoute = ShellRoute(
-      builder: (BuildContext context, GoRouterState state, Widget child) {
-        return SelectionArea(
-          child: MainSkeleton(
-            body: Padding(
-                padding: EdgeInsets.all(isSmallScreen(context) ? 5 : 20),
-                child: child),
-          ),
+    final shellRoute = StatefulShellRoute.indexedStack(
+      builder: (BuildContext context, GoRouterState state,
+          StatefulNavigationShell navigationShell) {
+        return MainSkeleton(
+          body: navigationShell,
         );
       },
-      routes: [
-        GoRoute(
-          path: "/",
-          redirect: (context, state) => WelcomePage.routeTv,
-        ),
-        GoRoute(
-          path: WelcomePage.routeTv,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context, state: state, child: const WelcomePage()),
-        ),
-        GoRoute(
-          path: WelcomePage.routeMoivie,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context, state: state, child: const WelcomePage()),
-        ),
-        GoRoute(
-          path: TvDetailsPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context,
-              state: state,
-              child: TvDetailsPage(seriesId: state.pathParameters['id']!)),
-        ),
-        GoRoute(
-          path: MovieDetailsPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context,
-              state: state,
-              child: MovieDetailsPage(id: state.pathParameters['id']!)),
-        ),
-        GoRoute(
-          path: SearchPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context,
-              state: state,
-              child: SearchPage(query: state.uri.queryParameters["query"])),
-        ),
-        GoRoute(
-          path: SystemSettingsPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context,
-              state: state,
-              child: const SystemSettingsPage()),
-        ),
-        GoRoute(
-          path: ActivityPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context, state: state, child: const ActivityPage()),
-        ),
-        GoRoute(
-          path: SystemPage.route,
-          pageBuilder: (context, state) => buildPageWithDefaultTransition(
-              context: context, state: state, child: const SystemPage()),
-        )
+      branches: [
+        StatefulShellBranch(initialLocation: WelcomePage.routeTv, routes: [
+          GoRoute(
+            path: WelcomePage.routeTv,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context, state: state, child: const WelcomePage()),
+          ),
+          GoRoute(
+            path: TvDetailsPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context,
+                state: state,
+                child: TvDetailsPage(seriesId: state.pathParameters['id']!)),
+          ),
+          GoRoute(
+            path: SearchPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context,
+                state: state,
+                child: SearchPage(query: state.uri.queryParameters["query"])),
+          ),
+        ]),
+        StatefulShellBranch(initialLocation: WelcomePage.routeMoivie, routes: [
+          GoRoute(
+            path: WelcomePage.routeMoivie,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context, state: state, child: const WelcomePage()),
+          ),
+          GoRoute(
+            path: MovieDetailsPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context,
+                state: state,
+                child: MovieDetailsPage(id: state.pathParameters['id']!)),
+          ),
+        ]),
+        StatefulShellBranch(routes: [
+          GoRoute(
+            path: ActivityPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context, state: state, child: const ActivityPage()),
+          ),
+        ]),
+        StatefulShellBranch(routes: [
+          GoRoute(
+            path: SystemSettingsPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context,
+                state: state,
+                child: const SystemSettingsPage()),
+          ),
+        ]),
+        StatefulShellBranch(routes: [
+          GoRoute(
+            path: SystemPage.route,
+            pageBuilder: (context, state) => buildPageWithDefaultTransition(
+                context: context, state: state, child: const SystemPage()),
+          )
+        ]),
       ],
     );
 
     final router = GoRouter(
       navigatorKey: APIs.navigatorKey,
+      initialLocation: WelcomePage.routeTv,
       routes: [
         shellRoute,
+        GoRoute(
+          path: "/",
+          redirect: (context, state) => WelcomePage.routeTv,
+        ),
         GoRoute(
           path: LoginScreen.route,
           builder: (context, state) => const LoginScreen(),
+        ),
+        GoRoute(
+          path: InitWizard.route,
+          builder: (context, state) => const InitWizard(),
         )
       ],
     );
 
-    return ProviderScope(
-      child: MaterialApp.router(
-        title: 'Polaris 影视追踪下载',
-        theme: ThemeData(
-            fontFamily: "NotoSansSC",
-            colorScheme: ColorScheme.fromSeed(
-                seedColor: Colors.blueAccent,
-                brightness: Brightness.dark,
-                surface: Colors.black87),
-            useMaterial3: true,
-            //scaffoldBackgroundColor: Color.fromARGB(255, 26, 24, 24)
-            tooltipTheme: TooltipThemeData(
-              textStyle: const TextStyle(
-                color: Colors.grey,
-              ),
-              decoration: BoxDecoration(
-                color: Colors.black54,
-                borderRadius: BorderRadius.circular(20),
-              ),
-            )),
-        routerConfig: router,
-      ),
+    return MaterialApp.router(
+      title: 'Polaris 影视追踪下载',
+      theme: ThemeData(
+          fontFamily: "NotoSansSC",
+          colorScheme: ColorScheme.fromSeed(
+              seedColor: Colors.blueAccent,
+              brightness: Brightness.dark,
+              surface: Colors.black87),
+          useMaterial3: true,
+          //scaffoldBackgroundColor: Color.fromARGB(255, 26, 24, 24)
+          tooltipTheme: TooltipThemeData(
+            textStyle: const TextStyle(
+              color: Colors.grey,
+            ),
+            decoration: BoxDecoration(
+              color: Colors.black54,
+              borderRadius: BorderRadius.circular(20),
+            ),
+          )),
+      routerConfig: router,
     );
   }
 }
 
 class MainSkeleton extends StatefulWidget {
-  final Widget body;
+  final StatefulNavigationShell body;
   const MainSkeleton({super.key, required this.body});
 
   @override
@@ -162,111 +174,17 @@ class MainSkeleton extends StatefulWidget {
 }
 
 class _MainSkeletonState extends State<MainSkeleton> {
-  var _selectedTab;
-
   @override
   Widget build(BuildContext context) {
-    var uri = GoRouterState.of(context).uri.toString();
-    if (uri.contains(WelcomePage.routeTv)) {
-      _selectedTab = 0;
-    } else if (uri.contains(WelcomePage.routeMoivie)) {
-      _selectedTab = 1;
-    } else if (uri.contains(ActivityPage.route)) {
-      _selectedTab = 2;
-    } else if (uri.contains(SystemSettingsPage.route)) {
-      _selectedTab = 3;
-    } else if (uri.contains(SystemPage.route)) {
-      _selectedTab = 4;
-    }
-
+    var padding = isSmallScreen(context) ? 5.0 : 20.0;
     return AdaptiveScaffold(
       appBarBreakpoint: Breakpoints.standard,
-      appBar: AppBar(
-        // TRY THIS: Try changing the color here to a specific color (to
-        // Colors.amber, perhaps?) and trigger a hot reload to see the AppBar
-        // change color while the other colors stay the same.
-        backgroundColor: Theme.of(context).colorScheme.inversePrimary,
-        // Here we take the value from the MyHomePage object that was created by
-        // the App.build method, and use it to set our appbar title.
-        title: TextButton(
-          onPressed: () => context.go(WelcomePage.routeTv),
-          child: Text(
-            "Polaris",
-            overflow: TextOverflow.clip,
-            style: TextStyle(fontSize: 28),
-          ),
-        ),
-
-        actions: [
-          SearchAnchor(
-              builder: (BuildContext context, SearchController controller) {
-            return Container(
-              constraints: const BoxConstraints(maxWidth: 250, maxHeight: 40),
-              child: Opacity(
-                opacity: 0.8,
-                child: SearchBar(
-                  hintText: "搜索...",
-                  leading: const Icon(Icons.search),
-                  controller: controller,
-                  shadowColor: WidgetStateColor.transparent,
-                  backgroundColor: WidgetStatePropertyAll(
-                      Theme.of(context).colorScheme.primaryContainer),
-                  onSubmitted: (value) => context.go(Uri(
-                      path: SearchPage.route,
-                      queryParameters: {'query': value}).toString()),
-                ),
-              ),
-            );
-          }, suggestionsBuilder:
-                  (BuildContext context, SearchController controller) {
-            return [Text("dadada")];
-          }),
-          IconButton(
-              onPressed: () => showCalendar(context),
-              icon: Icon(Icons.calendar_month)),
-          MenuAnchor(
-            menuChildren: [
-              MenuItemButton(
-                leadingIcon: const Icon(Icons.exit_to_app),
-                child: const Text("登出"),
-                onPressed: () async {
-                  await APIs.logout();
-                },
-              ),
-            ],
-            builder: (context, controller, child) {
-              return TextButton(
-                onPressed: () {
-                  if (controller.isOpen) {
-                    controller.close();
-                  } else {
-                    controller.open();
-                  }
-                },
-                child: const Icon(Icons.account_circle),
-              );
-            },
-          )
-        ],
-      ),
+      appBar: appBar(),
       useDrawer: false,
-      selectedIndex: _selectedTab,
-      onSelectedIndexChange: (int index) {
-        setState(() {
-          _selectedTab = index;
-        });
-        if (index == 0) {
-          context.go(WelcomePage.routeTv);
-        } else if (index == 1) {
-          context.go(WelcomePage.routeMoivie);
-        } else if (index == 2) {
-          context.go(ActivityPage.route);
-        } else if (index == 3) {
-          context.go(SystemSettingsPage.route);
-        } else if (index == 4) {
-          context.go(SystemPage.route);
-        }
-      },
+      selectedIndex: widget.body.currentIndex,
+      onSelectedIndexChange: (p0) => widget.body
+          .goBranch(p0, initialLocation: p0 == widget.body.currentIndex),
+
       destinations: const <NavigationDestination>[
         NavigationDestination(
           icon: Icon(Icons.live_tv_outlined),
@@ -294,11 +212,118 @@ class _MainSkeletonState extends State<MainSkeleton> {
           label: '系统',
         ),
       ],
-      body: (context) => SafeArea(child: widget.body),
+      body: (context) => SafeArea(
+          child: Padding(
+              padding: EdgeInsets.only(
+                  left: padding, right: padding, top: 5, bottom: 5),
+              child: widget.body)),
       // Define a default secondaryBody.
       // Override the default secondaryBody during the smallBreakpoint to be
       // empty. Must use AdaptiveScaffold.emptyBuilder to ensure it is properly
       // overridden.
     );
   }
+
+  showDonate(BuildContext context) {
+    return showDialog<void>(
+      context: context,
+      barrierDismissible: true,
+      builder: (BuildContext context) {
+        return AlertDialog(
+          title: Text("项目开发不易，给开发者加个鸡腿："),
+          content: SizedBox(
+              width: 350,
+              height: 400,
+              child: Ink.image(
+                  fit: BoxFit.fitWidth,
+                  image: AssetImage("assets/wechat.jpg"))),
+        );
+      },
+    );
+  }
+
+  AppBar appBar() {
+    return AppBar(
+      // TRY THIS: Try changing the color here to a specific color (to
+      // Colors.amber, perhaps?) and trigger a hot reload to see the AppBar
+      // change color while the other colors stay the same.
+      backgroundColor: Theme.of(context).colorScheme.inversePrimary,
+      // Here we take the value from the MyHomePage object that was created by
+      // the App.build method, and use it to set our appbar title.
+      leading: Container(
+        alignment: Alignment.centerLeft,
+        child: TextButton(
+          onPressed: () => context.go(WelcomePage.routeTv),
+          child: const Text(
+            "Polaris",
+            overflow: TextOverflow.clip,
+            style: TextStyle(fontSize: 28),
+          ),
+        ),
+      ),
+      leadingWidth: isSmallScreen(context) ? 0 : 190,
+      title: Container(
+        alignment: Alignment.bottomLeft,
+        child: SearchAnchor(
+            builder: (BuildContext context, SearchController controller) {
+          return Container(
+            constraints: const BoxConstraints(maxWidth: 250, maxHeight: 40),
+            child: Opacity(
+              opacity: 0.8,
+              child: SearchBar(
+                hintText: "在此搜索...",
+                leading: const Icon(Icons.search),
+                controller: controller,
+                shadowColor: WidgetStateColor.transparent,
+                backgroundColor: WidgetStatePropertyAll(
+                    Theme.of(context).colorScheme.primaryContainer),
+                onSubmitted: (value) => context.go(Uri(
+                    path: SearchPage.route,
+                    queryParameters: {'query': value}).toString()),
+              ),
+            ),
+          );
+        }, suggestionsBuilder:
+                (BuildContext context, SearchController controller) {
+          return [Text("dadada")];
+        }),
+      ),
+
+      actions: [
+        // IconButton(
+        //     onPressed: () => showCalendar(context),
+        //     icon: Icon(Icons.calendar_month)),
+        IconButton(
+            onPressed: () => showDonate(context),
+            icon: Icon(
+              Icons.favorite_rounded,
+              color: Colors.red,
+            )),
+
+        MenuAnchor(
+          menuChildren: [
+            MenuItemButton(
+              leadingIcon: const Icon(Icons.exit_to_app),
+              child: const Text("登出"),
+              onPressed: () async {
+                await APIs.logout();
+              },
+            ),
+          ],
+          builder: (context, controller, child) {
+            return TextButton(
+              onPressed: () {
+                if (controller.isOpen) {
+                  controller.close();
+                } else {
+                  controller.open();
+                }
+              },
+              child: const Icon(Icons.account_circle),
+            );
+          },
+        ),
+      ],
+    );
+  }
 }

ui/lib/movie_watchlist.dart

@@ -28,21 +28,22 @@ class _MovieDetailsPageState extends ConsumerState<MovieDetailsPage> {
   Widget build(BuildContext context) {
     var seriesDetails = ref.watch(mediaDetailsProvider(widget.id));
 
-    return seriesDetails.when(
-        data: (details) {
-          return ListView(
-            children: [
-              DetailCard(details: details),
-              NestedTabBar(
-                id: widget.id,
-              )
-            ],
-          );
-        },
-        error: (err, trace) {
-          return Text("$err");
-        },
-        loading: () => const MyProgressIndicator());
+    return SelectionArea(
+        child: seriesDetails.when(
+            data: (details) {
+              return ListView(
+                children: [
+                  DetailCard(details: details),
+                  NestedTabBar(
+                    id: widget.id,
+                  )
+                ],
+              );
+            },
+            error: (err, trace) {
+              return Text("$err");
+            },
+            loading: () => const MyProgressIndicator()));
   }
 }
 

ui/lib/providers/APIs.dart

@@ -1,3 +1,5 @@
+import 'dart:convert';
+
 import 'package:dio/dio.dart';
 import 'package:flutter/foundation.dart';
 import 'package:flutter/material.dart';
@@ -8,9 +10,12 @@ class APIs {
   static final _baseUrl = baseUrl();
   static final searchUrl = "$_baseUrl/api/v1/media/search";
   static final editMediaUrl = "$_baseUrl/api/v1/media/edit";
-  static final downloadAllUrl = "$_baseUrl/api/v1/media/downloadall/";
+  static final downloadAllEpisodesUrl = "$_baseUrl/api/v1/media/downloadall/";
+  static final downloadAllTvUrl = "$_baseUrl/api/v1/media/download/tv";
+  static final downloadAllMovieUrl = "$_baseUrl/api/v1/media/download/movie";
   static final settingsUrl = "$_baseUrl/api/v1/setting/do";
   static final settingsGeneralUrl = "$_baseUrl/api/v1/setting/general";
+  //static final singleSettingUrl = "$_baseUrl/api/v1/setting/";
   static final watchlistTvUrl = "$_baseUrl/api/v1/media/tv/watchlist";
   static final watchlistMovieUrl = "$_baseUrl/api/v1/media/movie/watchlist";
   static final availableTorrentsUrl = "$_baseUrl/api/v1/media/torrents/";
@@ -30,6 +35,7 @@ class APIs {
   static final logoutUrl = "$_baseUrl/api/v1/setting/logout";
   static final loginSettingUrl = "$_baseUrl/api/v1/setting/auth";
   static final activityUrl = "$_baseUrl/api/v1/activity/";
+  static final activityDeleteUrl = "$_baseUrl/api/v1/activity/delete";
   static final activityMediaUrl = "$_baseUrl/api/v1/activity/media/";
   static final imagesUrl = "$_baseUrl/api/v1/img";
   static final logsBaseUrl = "$_baseUrl/api/v1/logs/";
@@ -39,6 +45,8 @@ class APIs {
   static final addImportlistUrl = "$_baseUrl/api/v1/importlist/add";
   static final deleteImportlistUrl = "$_baseUrl/api/v1/importlist/delete";
   static final getAllImportlists = "$_baseUrl/api/v1/importlist/";
+  static final prowlarrUrl = "$_baseUrl/api/v1/setting/prowlarr";
+  static final aiConfigUrl = "$_baseUrl/api/v1/setting/ai";
 
   static final notifierAllUrl = "$_baseUrl/api/v1/notifier/all";
   static final notifierDeleteUrl = "$_baseUrl/api/v1/notifier/id/";
@@ -48,6 +56,11 @@ class APIs {
 
   static final cronJobUrl = "$_baseUrl/api/v1/setting/cron/trigger";
 
+  static final tvParseUrl = "$_baseUrl/api/v1/setting/parse/tv";
+  static final movieParseUrl = "$_baseUrl/api/v1/setting/parse/movie";
+
+  static final mediaSizeLimiterUrl = "$_baseUrl/api/v1/setting/limiter";
+
   static const tmdbApiKey = "tmdb_api_key";
   static const downloadDirKey = "download_dir";
 
@@ -112,4 +125,50 @@ class APIs {
       throw sp.message;
     }
   }
+
+  static Future<List<String>> downloadAllTv() async {
+    var resp = await getDio().get(APIs.downloadAllTvUrl);
+
+    var sp = ServerResponse.fromJson(resp.data);
+
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    return sp.data == null ? [] : sp.data as List<String>;
+  }
+
+  static Future<List<String>> downloadAllMovies() async {
+    var resp = await getDio().get(APIs.downloadAllMovieUrl);
+
+    var sp = ServerResponse.fromJson(resp.data);
+
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    return sp.data == null ? [] : sp.data as List<String>;
+  }
+
+  static Future<String> parseTvName(String s) async {
+    var resp = await getDio().post(APIs.tvParseUrl, data: {"s": s});
+
+    var sp = ServerResponse.fromJson(resp.data);
+
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    JsonEncoder encoder = new JsonEncoder.withIndent('  ');
+    return encoder.convert(sp.data);
+  }
+
+  static Future<String> parseMovieName(String s) async {
+    var resp = await getDio().post(APIs.movieParseUrl, data: {"s": s});
+
+    var sp = ServerResponse.fromJson(resp.data);
+
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    JsonEncoder encoder = new JsonEncoder.withIndent('  ');
+    return encoder.convert(sp.data);
+  }
 }

ui/lib/providers/activity.dart

@@ -4,9 +4,8 @@ import 'package:flutter_riverpod/flutter_riverpod.dart';
 import 'package:ui/providers/APIs.dart';
 import 'package:ui/providers/server_response.dart';
 
-var activitiesDataProvider =
-    AsyncNotifierProvider.autoDispose.family<ActivityData, List<Activity>, String>(
-        ActivityData.new);
+var activitiesDataProvider = AsyncNotifierProvider.autoDispose
+    .family<ActivityData, List<Activity>, String>(ActivityData.new);
 
 var mediaHistoryDataProvider = FutureProvider.autoDispose.family(
   (ref, arg) async {
@@ -24,16 +23,17 @@ var mediaHistoryDataProvider = FutureProvider.autoDispose.family(
   },
 );
 
-class ActivityData extends AutoDisposeFamilyAsyncNotifier<List<Activity>, String> {
+class ActivityData
+    extends AutoDisposeFamilyAsyncNotifier<List<Activity>, String> {
+  Timer? _timer;
+
   @override
   FutureOr<List<Activity>> build(String arg) async {
-    if (arg == "active") {
-      //refresh active downloads
-      Timer(const Duration(seconds: 5),
-          ref.invalidateSelf); //Periodically Refresh
+    if (_timer != null) {
+      _timer!.cancel();
     }
 
-    final dio = await APIs.getDio();
+    final dio = APIs.getDio();
     var resp =
         await dio.get(APIs.activityUrl, queryParameters: {"status": arg});
     final sp = ServerResponse.fromJson(resp.data);
@@ -44,12 +44,21 @@ class ActivityData extends AutoDisposeFamilyAsyncNotifier<List<Activity>, String
     for (final a in sp.data as List) {
       activities.add(Activity.fromJson(a));
     }
+
+    if (arg == "active") {
+      //refresh active downloads
+      _timer = Timer(const Duration(seconds: 5),
+          () => ref.invalidateSelf()); //Periodically Refresh
+    }
     return activities;
   }
 
-  Future<void> deleteActivity(String id) async {
-    final dio = await APIs.getDio();
-    var resp = await dio.delete("${APIs.activityUrl}$id");
+  Future<void> deleteActivity(int id) async {
+    final dio = APIs.getDio();
+    var resp = await dio.post(APIs.activityDeleteUrl, data: {
+      "id": id,
+      "add_2_blacklist": false,
+    });
     final sp = ServerResponse.fromJson(resp.data);
     if (sp.code != 0) {
       throw sp.message;
@@ -70,7 +79,8 @@ class Activity {
       required this.saved,
       required this.progress,
       required this.size,
-      required this.seedRatio});
+      required this.seedRatio,
+      required this.uploadProgress});
 
   final int? id;
   final int? mediaId;
@@ -83,6 +93,7 @@ class Activity {
   final int? progress;
   final int? size;
   final double seedRatio;
+  final double uploadProgress;
 
   factory Activity.fromJson(Map<String, dynamic> json) {
     return Activity(
@@ -96,6 +107,7 @@ class Activity {
         saved: json["saved"],
         progress: json["progress"],
         seedRatio: json["seed_ratio"],
-        size: json["size"]);
+        size: json["size"],
+        uploadProgress: json["upload_progress"]);
   }
 }

ui/lib/providers/series_details.dart

@@ -24,9 +24,9 @@ class SeriesDetailData
     return SeriesDetails.fromJson(rsp.data);
   }
 
-  Future<void> delete() async {
-    final dio = await APIs.getDio();
-    var resp = await dio.delete("${APIs.seriesDetailUrl}$id");
+  Future<void> delete(bool removeFiles ) async {
+    final dio = APIs.getDio();
+    var resp = await dio.delete("${APIs.seriesDetailUrl}$id", queryParameters: {"delete_files": removeFiles});
     var rsp = ServerResponse.fromJson(resp.data);
     if (rsp.code != 0) {
       throw rsp.message;
@@ -84,7 +84,7 @@ class SeriesDetailData
 
   Future<dynamic> downloadall() async {
     final dio = APIs.getDio();
-    var resp = await dio.get(APIs.downloadAllUrl + id!);
+    var resp = await dio.get(APIs.downloadAllEpisodesUrl + id!);
     var sp = ServerResponse.fromJson(resp.data);
     if (sp.code != 0) {
       throw sp.message;

ui/lib/providers/settings.dart

@@ -25,10 +25,17 @@ var importlistProvider =
     AsyncNotifierProvider.autoDispose<ImportListData, List<ImportList>>(
         ImportListData.new);
 
+var prowlarrSettingDataProvider =
+    AsyncNotifierProvider.autoDispose<ProwlarrSettingData, ProwlarrSetting>(
+        ProwlarrSettingData.new);
+
+var aiConfigDataProvider =
+    AsyncNotifierProvider.autoDispose<AIConfigData, AIConfig>(AIConfigData.new);
+
 class EditSettingData extends AutoDisposeAsyncNotifier<GeneralSetting> {
   @override
   FutureOr<GeneralSetting> build() async {
-    final dio = await APIs.getDio();
+    final dio = APIs.getDio();
 
     var resp = await dio.get(APIs.settingsGeneralUrl);
     var rrr = ServerResponse.fromJson(resp.data);
@@ -503,3 +510,91 @@ class ImportListData extends AutoDisposeAsyncNotifier<List<ImportList>> {
     ref.invalidateSelf();
   }
 }
+
+class ProwlarrSetting {
+  final bool disabled;
+  final String apiKey;
+  final String url;
+  ProwlarrSetting(
+      {required this.apiKey, required this.url, required this.disabled});
+  factory ProwlarrSetting.fromJson(Map<String, dynamic> json) {
+    return ProwlarrSetting(
+        apiKey: json["api_key"],
+        url: json["url"],
+        disabled: json["disabled"] ?? false);
+  }
+
+  Map<String, dynamic> tojson() =>
+      {"api_key": apiKey, "url": url, "disabled": disabled};
+}
+
+class ProwlarrSettingData extends AutoDisposeAsyncNotifier<ProwlarrSetting> {
+  @override
+  FutureOr<ProwlarrSetting> build() async {
+    final dio = APIs.getDio();
+    var resp = await dio.get(APIs.prowlarrUrl);
+    var sp = ServerResponse.fromJson(resp.data);
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    var se = ProwlarrSetting.fromJson(sp.data);
+    return se;
+  }
+
+  Future<void> save(ProwlarrSetting ps) async {
+    final dio = APIs.getDio();
+    var resp = await dio.post(APIs.prowlarrUrl, data: ps.tojson());
+    var sp = ServerResponse.fromJson(resp.data);
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    ref.invalidateSelf();
+  }
+}
+
+class AIConfigData extends AutoDisposeAsyncNotifier<AIConfig> {
+  @override
+  FutureOr<AIConfig> build() async {
+    final dio = APIs.getDio();
+    var resp = await dio.get(APIs.aiConfigUrl);
+    var sp = ServerResponse.fromJson(resp.data);
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    return AIConfig.fromJson(sp.data);
+  }
+
+  Future<void> save(AIConfig ai) async {
+    final dio = APIs.getDio();
+    var resp = await dio.post(APIs.aiConfigUrl, data: ai.toJson());
+    var sp = ServerResponse.fromJson(resp.data);
+    if (sp.code != 0) {
+      throw sp.message;
+    }
+    ref.invalidateSelf();
+  }
+}
+
+class AIConfig {
+  final bool enabled;
+  final String geminiApiKey;
+  final String geminiModelName;
+
+  AIConfig(
+      {required this.enabled,
+      required this.geminiApiKey,
+      required this.geminiModelName});
+
+  factory AIConfig.fromJson(Map<String, dynamic> json) {
+    return AIConfig(
+        enabled: json["enabled"],
+        geminiApiKey: json["gemini_api_key"],
+        geminiModelName: json["gemini_model_name"]);
+  }
+
+  Map<String, dynamic> toJson() => {
+        "enabled": enabled,
+        "gemini_api_key": geminiApiKey,
+        "gemini_model_name": geminiModelName
+      };
+}