initial commit

Author: deep-diver

go.mod

@@ -0,0 +1,3 @@
+module paperswithcode
+
+go 1.16

internal/client.go

@@ -0,0 +1,189 @@
+package client
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"net/http"
+	"net/url"
+)
+
+type Client struct {
+	BaseURL    string
+	HTTPClient *http.Client
+}
+
+type errorResponse struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+}
+
+// Specific Paper's Information by the title
+type Paper struct {
+	ID        string   `json:"id"`
+	ArxivID   string   `json:"arxiv_id,omitempty"`
+	URLAbs    string   `json:"url_abs"`
+	URLPDF    string   `json:"url_pdf"`
+	Title     string   `json:"title"`
+	Abstract  string   `json:"abstract"`
+	Authors   []string `json:"authors"`
+	Published string   `json:"published"`
+}
+
+func (c *Client) GetPaper(ctx context.Context, id string) (*Paper, error) {
+	fmt.Println(id)
+	url := fmt.Sprintf("%s/papers/%s/", c.BaseURL, url.QueryEscape(id))
+	req, err := http.NewRequest("GET", url, nil)
+
+	if err != nil {
+		return nil, err
+	}
+
+	req = req.WithContext(ctx)
+
+	res := Paper{}
+	if err := c.sendRequest(req, &res); err != nil {
+		return nil, err
+	}
+
+	return &res, nil
+}
+
+// Method list used by Paper's ID
+type MethodList struct {
+	Count   int      `json:"count"`
+	Methods []Method `json:"results"`
+}
+
+type Method struct {
+	ID          string `json:"id"`
+	Name        string `json:"name"`
+	FullName    string `json:"full_name"`
+	Description string `json:"description"`
+	Paper       string `json:"paper"`
+}
+
+func (c *Client) GetMethodList(ctx context.Context, id string) (*MethodList, error) {
+	fmt.Println(id)
+	url := fmt.Sprintf("%s/papers/%s/methods", c.BaseURL, url.QueryEscape(id))
+	req, err := http.NewRequest("GET", url, nil)
+
+	if err != nil {
+		return nil, err
+	}
+
+	req = req.WithContext(ctx)
+
+	res := MethodList{}
+	if err := c.sendRequest(req, &res); err != nil {
+		return nil, 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")
+
+	res, err := c.HTTPClient.Do(req)
+	if err != nil {
+		return err
+	}
+
+	defer res.Body.Close()
+
+	if res.StatusCode < http.StatusOK || res.StatusCode >= http.StatusBadRequest {
+		var errRes errorResponse
+		if err = json.NewDecoder(res.Body).Decode(&errRes); err == nil {
+			return errors.New(errRes.Message)
+		}
+
+		return fmt.Errorf("unknown error, status code: %d", res.StatusCode)
+	}
+
+	if err = json.NewDecoder(res.Body).Decode(&v); err != nil {
+		return err
+	}
+
+	// fmt.Println(fullResponse)
+
+	return nil
+}
+
+/*
+	GetPapers API is currently broken. need to wait until it gets fixed
+*/
+
+// type PaperList struct {
+// 	Count    int    `json:"count"`
+// 	NextPage string `json:"next,omitempty"`
+// 	PrePage  string `json:"previous,omitempty"`
+// 	Results  []struct {
+// 		ID               string   `json:"id"`
+// 		ArxivID          []string `json:"arxiv_id,omitempty"`
+// 		NipsID           []string `json:"nips_id,omitempty"`
+// 		URLAbs           string   `json:"url_abs"`
+// 		URLPdf           string   `json:"url_pdf"`
+// 		Title            string   `json:"title"`
+// 		Abstract         string   `json:"abstract"`
+// 		Authors          []string `json:"authors"`
+// 		Published        string   `json:"published"`
+// 		Conference       []string `json:"conference,omitempty"`
+// 		ConferenceURLAbs []string `json:"conference_url_abs,omitempty"`
+// 		ConferenceURLPdf []string `json:"conference_url_pdf,omitempty"`
+// 		Proceeding       []string `json:"proceeding,omitempty"`
+// 	} `json:"results"`
+// }
+
+// func (c *Client) GetPapers(ctx context.Context, query string, page int, limit int) (*PaperList, error) {
+// 	url := fmt.Sprintf("%s/papers?q=%s&items_per_page=%d&page=%d", c.BaseURL, url.QueryEscape(query), limit, page)
+// 	req, err := http.NewRequest("GET", url, nil)
+
+// 	if err != nil {
+// 		return nil, err
+// 	}
+
+// 	req = req.WithContext(ctx)
+
+// 	res := PaperList{}
+// 	if err := c.sendRequest(req, &res); err != nil {
+// 		return nil, err
+// 	}
+
+// 	return &res, nil
+// }

main

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"strings"
+	"time"
+
+	client "paperswithcode/internal"
+)
+
+const (
+	BASE_URL = "https://paperswithcode.com/api/v1"
+)
+
+func NewClient() *client.Client {
+	return &client.Client{
+		BaseURL: BASE_URL,
+		HTTPClient: &http.Client{
+			Timeout: time.Minute,
+		},
+	}
+}
+
+func main() {
+	c := NewClient()
+	ctx := context.Background()
+	paper := "generative adversarial networks"
+	paper = strings.ReplaceAll(paper, " ", "-")
+
+	paperList, _ := c.GetPaper(ctx, paper)
+	fmt.Println(paperList)
+	fmt.Println()
+	fmt.Println()
+	fmt.Println()
+
+	methodList, _ := c.GetMethodList(ctx, paper)
+	fmt.Println(methodList)
+	fmt.Println()
+	fmt.Println()
+	fmt.Println()
+
+	repositoryList, _ := c.GetRepositoryList(ctx, paper)
+	fmt.Println(repositoryList)
+	fmt.Println()
+	fmt.Println()
+	fmt.Println()
+}