summary refs log tree commit diff stats
path: root/fetch
diff options
context:
space:
mode:
Diffstat (limited to 'fetch')
-rw-r--r--fetch/request.go41
-rw-r--r--fetch/wikipedia.go39
2 files changed, 80 insertions, 0 deletions
diff --git a/fetch/request.go b/fetch/request.go
new file mode 100644
index 0000000..a3c6f23
--- /dev/null
+++ b/fetch/request.go
@@ -0,0 +1,41 @@
+package fetch
+
+import (
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"time"
+)
+
+func getRes(api string) ([]byte, error) {
+	var body []byte
+
+	c := http.Client{
+		// TODO: timeout should be configurable by the user
+		Timeout: time.Second * 64,
+	}
+
+	req, err := http.NewRequest(http.MethodGet, api, nil)
+	if err != nil {
+		return body, err
+	}
+
+	req.Header.Set("User-Agent", "Andinus / Indus - framagit.org/andinus/indus")
+
+	res, err := c.Do(req)
+	if err != nil {
+		return body, err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode != 200 {
+		return body, fmt.Errorf("Unexpected response status code received: %d %s",
+			res.StatusCode, http.StatusText(res.StatusCode))
+	}
+
+	body, err = ioutil.ReadAll(res.Body)
+	if err != nil {
+		return body, err
+	}
+	return body, err
+}
diff --git a/fetch/wikipedia.go b/fetch/wikipedia.go
new file mode 100644
index 0000000..ee057d4
--- /dev/null
+++ b/fetch/wikipedia.go
@@ -0,0 +1,39 @@
+package fetch
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+// Wiki struct holds information retrieved from wikipedia api.
+type Wiki struct {
+	Type         string `json:"type"`
+	Title        string `json:"title"`
+	DisplayTitle string `json:"displaytitle"`
+	PageID       int    `json:"pageid"`
+	Description  string `json:"description"`
+	Language     string `json:"lang"`
+	Extract      string `json:"extract"`
+}
+
+// Wikipedia gets the summary from wikipedia api & returns it with an error (if
+// exists). It takes a string as input which will be the search query
+// sent to wikipedia.
+func Wikipedia(q string) (Wiki, error) {
+	w := Wiki{}
+	api := fmt.Sprintf("https://en.wikipedia.org/api/rest_v1/page/summary/%s", q)
+
+	body, err := getRes(api)
+	if err != nil {
+		return w, err
+	}
+
+	// Unmarshal json to w
+	err = json.Unmarshal([]byte(body), &w)
+	if err != nil {
+		err = fmt.Errorf("Unmarshalling Json failed\n%s", err.Error())
+		return w, err
+	}
+
+	return w, err
+}