Merge pull request #6 from codingpot/refactor-2-paper-repository-list

feat: implement PaperRepositoryList()

Author: deep-diver

client.go

@@ -85,39 +85,6 @@ func (c *Client) GetMethodList(ctx context.Context, id string) (*MethodList, err
 	return &res, nil
 }
 
-// Repository List implementing by Paper's ID
-type RepositoryList struct {
-	Count   int          `json:"count"`
-	Results []Repository `json:"results"`
-}
-
-type Repository struct {
-	URL         string `json:"url"`
-	IsOfficial  bool   `json:"is_official"`
-	Description string `json:"description"`
-	Stars       int    `json:"stars"`
-	Framework   string `json:"framework"`
-}
-
-func (c *Client) GetRepositoryList(ctx context.Context, id string) (*RepositoryList, error) {
-	fmt.Println(id)
-	url := fmt.Sprintf("%s/papers/%s/repositories", c.BaseURL, url.QueryEscape(id))
-	req, err := http.NewRequest("GET", url, nil)
-
-	if err != nil {
-		return nil, err
-	}
-
-	req = req.WithContext(ctx)
-
-	res := RepositoryList{}
-	if err := c.sendRequest(req, &res); err != nil {
-		return nil, err
-	}
-
-	return &res, nil
-}
-
 func (c *Client) sendRequest(req *http.Request, v interface{}) error {
 	req.Header.Set("Content-Type", "application/json; charset=utf-8")
 	req.Header.Set("Accept", "application/json; charset=utf-8")

cmd/client/main.go

@@ -31,7 +31,7 @@ func main() {
 	fmt.Println()
 	fmt.Println()
 
-	repositoryList, _ := c.GetRepositoryList(ctx, paper)
+	repositoryList, _ := c.PaperRepositoryList(paper)
 	fmt.Println(repositoryList)
 	fmt.Println()
 	fmt.Println()

models/repository.go

@@ -0,0 +1,17 @@
+package models
+
+// RepositoryList contains code repositories implemented by a certain paper.
+type RepositoryList struct {
+	Count    int64        `json:"count"`
+	Next     *string      `json:"next"`
+	Previous *string      `json:"previous"`
+	Results  []Repository `json:"results"`
+}
+
+type Repository struct {
+	URL         string `json:"url"`
+	IsOfficial  bool   `json:"is_official"`
+	Description string `json:"description"`
+	Stars       int    `json:"stars"`
+	Framework   string `json:"framework"`
+}

paper_repository_list.go

@@ -0,0 +1,24 @@
+package paperswithcode_go
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/codingpot/paperswithcode-go/models"
+	"net/url"
+)
+
+func (c *Client) PaperRepositoryList(paperID string) (*models.RepositoryList, error) {
+	paperURL := fmt.Sprintf("%s/papers/%s/repositories", c.BaseURL, url.QueryEscape(paperID))
+	response, err := c.HTTPClient.Get(paperURL)
+	if err != nil {
+		return nil, err
+	}
+
+	var repoList models.RepositoryList
+	err = json.NewDecoder(response.Body).Decode(&repoList)
+	if err != nil {
+		return nil, err
+	}
+
+	return &repoList, nil
+}

paper_repository_list_test.go

@@ -0,0 +1,13 @@
+package paperswithcode_go
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestClient_PaperRepositoryList(t *testing.T) {
+	c := NewClient()
+	list, err := c.PaperRepositoryList("generative-adversarial-networks")
+	assert.NoError(t, err)
+	assert.NotEmpty(t, list.Results[0].URL)
+}