jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-02-06 23:21:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Simon Ding
2024-07-08 09:39:22 +08:00
parent 38aa250c60
commit a76adfdd29
4 changed files with 73 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
db/db.go
View File

@@ -135,9 +135,15 @@ func (c *Client) SaveTorznabInfo(name string, setting TorznabSetting) error {
return nil
}
func (c *Client) GetAllTorznabInfo() map[string]TorznabSetting {
type TorznabInfo struct {
Name string `json:"name"`
TorznabSetting
}
func (c *Client) GetAllTorznabInfo() []*TorznabInfo {
res := c.ent.Indexers.Query().Where(indexers.Implementation(IndexerTorznabImpl)).AllX(context.TODO())
var m = make(map[string]TorznabSetting, len(res))
var l = make([]*TorznabInfo, 0, len(res))
for _, r := range res {
var ss TorznabSetting
err := json.Unmarshal([]byte(r.Settings), &ss)
@@ -145,9 +151,12 @@ func (c *Client) GetAllTorznabInfo() map[string]TorznabSetting {
log.Errorf("unmarshal torznab %s error: %v", r.Name, err)
continue
}
m[r.Name] = ss
l = append(l, &TorznabInfo{
Name: r.Name,
TorznabSetting: ss,
})
}
return m
return l
}
func (c *Client) SaveTransmission(name, url, user, password string) error {
@@ -165,3 +174,12 @@ func (c *Client) GetTransmission() *ent.DownloadClients {
}
return dc
}
func (c *Client) GetAllDonloadClients() []*ent.DownloadClients {
cc, err := c.ent.DownloadClients.Query().All(context.TODO())
if err != nil {
log.Errorf("no download client")
return nil
}
return cc
}

22
server/resources.go
View File

@@ -16,10 +16,10 @@ func (s *Server) searchTvWithTorznab(name string, season, episode int) []torznab
var res []torznab.Result
allTorznab := s.db.GetAllTorznabInfo()
for name, setting := range allTorznab {
resp, err := torznab.Search(setting.URL, setting.ApiKey, q)
for _, tor := range allTorznab {
resp, err := torznab.Search(tor.URL, tor.ApiKey, q)
if err != nil {
log.Errorf("search %s error: %v", name, err)
log.Errorf("search %s error: %v", tor.Name, err)
continue
}
res = append(res, resp...)
@@ -48,6 +48,14 @@ func (s *Server) AddTorznabInfo(c *gin.Context) (interface{}, error) {
return nil, nil
}
func (s *Server) GetAllIndexers(c *gin.Context) (interface{}, error) {
indexers := s.db.GetAllTorznabInfo()
if len(indexers) == 0 {
return nil, fmt.Errorf("no indexer found")
}
return indexers, nil
}
type searchAndDownloadIn struct {
ID int `json:"id"`
Season int `json:"season"`
@@ -109,3 +117,11 @@ func (s *Server) AddDownloadClient(c *gin.Context) (interface{}, error) {
}
return nil, nil
}
func (s *Server) GetAllDonloadClients(c *gin.Context) (interface{}, error) {
res := s.db.GetAllDonloadClients()
if len(res) == 0 {
return nil, fmt.Errorf("no download client")
}
return res, nil
}

2
server/server.go
View File

@@ -64,12 +64,14 @@ func (s *Server) Serve() error {
}
indexer := api.Group("/indexer")
{
indexer.GET("/", HttpHandler(s.GetAllIndexers))
indexer.POST("/add", HttpHandler(s.AddTorznabInfo))
indexer.POST("/download", HttpHandler(s.SearchAndDownload))
}
downloader := api.Group("/downloader")
{
downloader.GET("/", HttpHandler(s.GetAllDonloadClients))
downloader.POST("/add", HttpHandler(s.AddDownloadClient))
}

62
ui/lib/weclome.dart
View File

@@ -29,42 +29,40 @@ class _WeclomePageState extends State<WelcomePage> {
return GridView.builder(
itemCount: favList.length,
gridDelegate:
const SliverGridDelegateWithFixedCrossAxisCount(crossAxisCount: 4),
const SliverGridDelegateWithFixedCrossAxisCount(crossAxisCount: 6),
itemBuilder: (context, i) {
var item = TvSeries.fromJson(favList[i]);
return Container(
child: Card(
margin: const EdgeInsets.all(4),
clipBehavior: Clip.hardEdge,
child: InkWell(
//splashColor: Colors.blue.withAlpha(30),
onTap: () {
context.go(TvDetailsPage.toRoute(item.id!));
//showDialog(context: context, builder: builder)
},
child: Column(
children: <Widget>[
Flexible(
child: SizedBox(
width: 300,
height: 600,
child: Image.network(
APIs.tmdbImgBaseUrl + item.posterPath!,
fit: BoxFit.contain,
),
return Card(
margin: const EdgeInsets.all(4),
clipBehavior: Clip.hardEdge,
child: InkWell(
//splashColor: Colors.blue.withAlpha(30),
onTap: () {
context.go(TvDetailsPage.toRoute(item.id!));
//showDialog(context: context, builder: builder)
},
child: Column(
children: <Widget>[
Flexible(
child: SizedBox(
width: 300,
height: 600,
child: Image.network(
APIs.tmdbImgBaseUrl + item.posterPath!,
fit: BoxFit.contain,
),
),
Flexible(
child: Text(
item.name!,
style: const TextStyle(
fontSize: 14, fontWeight: FontWeight.bold),
),
)
],
),
)),
);
),
Flexible(
child: Text(
item.name!,
style: const TextStyle(
fontSize: 14, fontWeight: FontWeight.bold),
),
)
],
),
));
});
}