feat: add download method for sanqiu books.

bookstairs · Nov 15, 2022 · 2eacb6e · 2eacb6e
1 parent 6d4407c
commit 2eacb6e
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 5 deletions.
diff --git a/internal/driver/aliyun.go b/internal/driver/aliyun.go
@@ -47,6 +47,6 @@ func (a *aliyunDriver) Resolve(shareLink string, passcode string) ([]Share, erro
 	panic("TODO implement me")
 }
 
-func (a *aliyunDriver) Download(share Share) (io.ReadCloser, error) {
+func (a *aliyunDriver) Download(share Share) (io.ReadCloser, int64, error) {
 	panic("TODO implement me")
 }
diff --git a/internal/driver/common.go b/internal/driver/common.go
@@ -30,7 +30,7 @@ type (
 		Resolve(shareLink string, passcode string) ([]Share, error)
 
 		// Download the given link.
-		Download(share Share) (io.ReadCloser, error)
+		Download(share Share) (io.ReadCloser, int64, error)
 	}
 )
 

diff --git a/internal/driver/lanzou.go b/internal/driver/lanzou.go
@@ -37,6 +37,6 @@ func (l *lanzouDriver) Resolve(shareLink string, passcode string) ([]Share, erro
 	panic("TODO implement me")
 }
 
-func (l *lanzouDriver) Download(share Share) (io.ReadCloser, error) {
+func (l *lanzouDriver) Download(share Share) (io.ReadCloser, int64, error) {
 	panic("TODO implement me")
 }
diff --git a/internal/driver/telecom.go b/internal/driver/telecom.go
@@ -33,6 +33,6 @@ func (t *telecomDriver) Resolve(shareLink string, passcode string) ([]Share, err
 	panic("TODO implement me")
 }
 
-func (t *telecomDriver) Download(share Share) (io.ReadCloser, error) {
+func (t *telecomDriver) Download(share Share) (io.ReadCloser, int64, error) {
 	panic("TODO implement me")
 }
diff --git a/internal/fetcher/sanqiu.go b/internal/fetcher/sanqiu.go
@@ -120,5 +120,13 @@ func (s *sanqiuService) formats(id int64) (map[Format]driver.Share, error) {
 }
 
 func (s *sanqiuService) fetch(id int64, format Format, share driver.Share) (*fetch, error) {
-	panic("TODO implement me")
+	content, size, err := s.driver.Download(share)
+	if err != nil {
+		return nil, err
+	}
+
+	return &fetch{
+		content: content,
+		size:    size,
+	}, nil
 }