Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-05-24 08:44:48 -07:00
Code Issues Packages Projects Releases Wiki Activity

Change containter builder struct to func

Browse Source
This commit is contained in:
Širhoe Biazhkovič
2021-09-05 18:22:10 +03:00
parent 114e8c0f06
commit d0e13b3273
4 changed files with 11 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
middleware/torrentapproval/torrentapproval.go
View File

@@ -44,7 +44,7 @@ func (d driver) NewHook(optionBytes []byte) (middleware.Hook, error) {
return nil, err
}
if c, err := container.GetContainer(cfg.Name, confBytes); err == nil{
if c, err := container.GetContainer(cfg.Name, confBytes); err == nil {
return &hook{c}, nil
} else {
return nil, err
@@ -58,24 +58,23 @@ type hook struct {
hashContainer container.Container
}
func (h *hook) HandleAnnounce(ctx context.Context, req *bittorrent.AnnounceRequest, resp *bittorrent.AnnounceResponse) (context.Context, error) {
func (h *hook) HandleAnnounce(ctx context.Context, req *bittorrent.AnnounceRequest, _ *bittorrent.AnnounceResponse) (context.Context, error) {
var err error
if !h.hashContainer.Contains(req.InfoHash){
if !h.hashContainer.Contains(req.InfoHash) {
err = ErrTorrentUnapproved
}
return ctx, err
}
func (h *hook) HandleScrape(ctx context.Context, req *bittorrent.ScrapeRequest, resp *bittorrent.ScrapeResponse) (context.Context, error) {
func (h *hook) HandleScrape(ctx context.Context, req *bittorrent.ScrapeRequest, _ *bittorrent.ScrapeResponse) (context.Context, error) {
// Scrapes don't require any protection.
return ctx, nil
}
func (h *hook) Stop() stop.Result {
if st, isOk := h.hashContainer.(stop.Stopper); isOk{
if st, isOk := h.hashContainer.(stop.Stopper); isOk {
return st.Stop()
}
return stop.AlreadyStopped