3214682499

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/wilburt/wilburx9.dev/commit/<a class=hub.com/wilburt/wilburx9.dev/commit/<a class="double-link" href="https://git"><a class=hub.com/wilburt/wilburx9.dev/commit/<a class="double-link" href="https://git"><a class=hub.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66ac3">b88d996dd<a href="https://github.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66ac3">&quot;&gt;&amp;lt;a href=&amp;quot;https://github.com/wilburt/wilburx9.dev/commit/&lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/wilburt/wilburx9.dev/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66ac3&quot;&gt;b88d996dd&lt;/a&gt;&lt;a href=&quot;https://github.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66ac3&quot;&gt;&amp;lt;a href=&amp;quot;https://github.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14a&lt;/a&gt;ffe66ac3">&quot;&gt;Merge &lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/wilburt/wilburx9.dev/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/wilburt/wilburx9.dev/commit/<a class="double-link" href="https://github.com/wilburt/wilburx9.dev/commit/474eaec2f9839c2979b7bdcc459385c924d4ef7e">474eaec2f</a><a href="https://github.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66ac3">&amp;quot;&amp;gt;474eaec2f&amp;lt;/a&amp;gt;&amp;quot;&amp;gt;474eaec2f&amp;lt;/a&amp;gt;&amp;lt;a href=&amp;quot;https://github.com/wilburt/wilburx9.dev/commit/b88d996dd93d57a5f00096ad82b8e14affe66&lt;/a&gt;ac3&quot;&gt; into &lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/wilburt/wilburx9.dev/commit/<a class="double-link" href="https://github.com/wilburt/wilburx9.dev/commit/953800e6c">953800e6c">953800e6c</a>

Pull Request Pull Request #26: Tests and Revamped UI

Run Details

24 of 24 new or added lines in 6 files covered. (100.0%)

82 of 180 relevant lines covered (45.56%)

0.51 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/backend/api/gallery/instagram.go

package gallery

import (
        "context"
        "encoding/json"
        "fmt"
        log "github.com/sirupsen/logrus"
        "github.com/wilburt/wilburx9.dev/backend/api/gallery/internal/models"
        "github.com/wilburt/wilburx9.dev/backend/api/internal"
        "github.com/wilburt/wilburx9.dev/backend/api/internal/database"
        "math"
        "net/http"
        "net/url"
        "time"
)

const (
        instagramKey          = "instagram"
        minTokenRemainingLife = 24 * time.Hour * 5 // 5 Days
        instagramLimit        = "20"
)

// Instagram encapsulates the fetching of Instagram images and access token management
type Instagram struct {
        AccessToken string
        Db          database.ReadWrite
        HttpClient  internal.HttpClient
}

// Cache fetches and caches Instagram images to db
func (i Instagram) Cache(ctx context.Context) (int, error) {
        result, err := i.fetchImages(ctx)
        if err != nil {
                return 0, err
        }

        return len(result), i.Db.Write(ctx, internal.DbGalleryKey, result...)
}

// Recursively fetch all the images
func (i Instagram) fetchImages(ctx context.Context) ([]database.Model, error) {
        token, err := i.getToken(ctx)
        if err != nil {
                log.WithFields(log.Fields{"error": err}).Warning("Couldn't get token")
                return nil, err
        }

        u, _ := url.Parse("https://graph.instagram.com/me/media")
        q := u.Query()
        q.Set("fields", "caption,id,media_url,timestamp,permalink,thumbnail_url,media_type")
        q.Set("access_token", token)
        q.Set("limit", instagramLimit)
        u.RawQuery = q.Encode()

        req, _ := http.NewRequest(http.MethodGet, u.String(), nil)
        res, err := i.HttpClient.Do(req)
        if err != nil {
                log.WithFields(log.Fields{"error": err}).Warning("Couldn't send request")
                return nil, err
        }
        defer res.Body.Close()

        var data models.InstaImg
        err = json.NewDecoder(res.Body).Decode(&data)
        if err != nil {
                log.WithFields(log.Fields{"error": err}).Warning("Couldn't Unmarshall data")
                return nil, err
        }

        return data.Data.ToImages(instagramKey), nil
}

func (i Instagram) getToken(ctx context.Context) (string, error) {

        // Attempt to get token from Db
        keys, _, err := i.Db.Read(ctx, internal.DbKeys, "", math.MaxInt)
        // If we haven't saved the token before, log an error and refresh the token we have now
        if err != nil || len(keys) == 0 {
                log.Warningf("Couldn't get Keys from the db. Possible error: %v", err)
                return i.refreshToken(ctx, i.AccessToken)
        }

        var tk models.InstaToken
        // Get token map from keys
        for _, m := range keys {
                if _, ok := m["id"]; ok && m["id"] == instagramKey {
                        if bytes, err := json.Marshal(m); err == nil {
                                json.Unmarshal(bytes, &tk)
                        }
                        break
                }
        }

        if tk.ID == "" || tk.Value == "" {
                return i.refreshToken(ctx, i.AccessToken)
        }

        // Check for expired token.
        if tk.Expired() {
                // Token has expired and it can't be refreshed. This should never happen.
                return "", fmt.Errorf("instagram access token has expired")
        }

        // Refresh the token if need be
        if tk.IsAboutToExpire(minTokenRemainingLife) {
                return i.refreshToken(ctx, tk.Value)
        }
        return tk.Value, nil
}

func (i Instagram) refreshToken(ctx context.Context, oldToken string) (string, error) {
        u, _ := url.Parse("https://graph.instagram.com/refresh_access_token")
        q := u.Query()
        q.Set("grant_type", "ig_refresh_token")
        q.Set("access_token", oldToken)
        u.RawQuery = q.Encode()

        req, _ := http.NewRequest(http.MethodGet, u.String(), nil)

        res, err := i.HttpClient.Do(req)
        if err != nil {
                return "", err
        }
        defer res.Body.Close()

        var newT = models.NewInstaToken(instagramKey)
        err = json.NewDecoder(res.Body).Decode(&newT)
        if err != nil {
                return "", err
        }

        newT.RefreshedAt = time.Now()
        err = i.Db.Write(ctx, internal.DbKeys, newT)
        if err != nil {
                return "", err
        }

        return newT.Value, nil
}

1	package gallery
2
3	import (
4	"context"
5	"encoding/json"
6	"fmt"
7	log "github.com/sirupsen/logrus"
8	"github.com/wilburt/wilburx9.dev/backend/api/gallery/internal/models"
9	"github.com/wilburt/wilburx9.dev/backend/api/internal"
10	"github.com/wilburt/wilburx9.dev/backend/api/internal/database"
11	"math"
12	"net/http"
13	"net/url"
14	"time"
15	)
16
17	const (
18	instagramKey = "instagram"
19	minTokenRemainingLife = 24 * time.Hour * 5 // 5 Days
20	instagramLimit = "20"
21	)
22
23	// Instagram encapsulates the fetching of Instagram images and access token management
24	type Instagram struct {
25	AccessToken string
26	Db database.ReadWrite
27	HttpClient internal.HttpClient
28	}
29
30	// Cache fetches and caches Instagram images to db
31	func (i Instagram) Cache(ctx context.Context) (int, error) {	×
32	result, err := i.fetchImages(ctx)	×
33	if err != nil {	×
34	return 0, err	×
35	}	×
36
37	return len(result), i.Db.Write(ctx, internal.DbGalleryKey, result...)	×
38	}
39
40	// Recursively fetch all the images
41	func (i Instagram) fetchImages(ctx context.Context) ([]database.Model, error) {	×
42	token, err := i.getToken(ctx)	×
43	if err != nil {	×
44	log.WithFields(log.Fields{"error": err}).Warning("Couldn't get token")	×
45	return nil, err	×
46	}	×
47
48	u, _ := url.Parse("https://graph.instagram.com/me/media")	×
49	q := u.Query()	×
50	q.Set("fields", "caption,id,media_url,timestamp,permalink,thumbnail_url,media_type")	×
51	q.Set("access_token", token)	×
52	q.Set("limit", instagramLimit)	×
53	u.RawQuery = q.Encode()	×
54		×
55	req, _ := http.NewRequest(http.MethodGet, u.String(), nil)	×
56	res, err := i.HttpClient.Do(req)	×
57	if err != nil {	×
58	log.WithFields(log.Fields{"error": err}).Warning("Couldn't send request")	×
59	return nil, err	×
60	}	×
61	defer res.Body.Close()	×
62		×
63	var data models.InstaImg	×
64	err = json.NewDecoder(res.Body).Decode(&data)	×
65	if err != nil {	×
66	log.WithFields(log.Fields{"error": err}).Warning("Couldn't Unmarshall data")	×
67	return nil, err	×
68	}	×
69
70	return data.Data.ToImages(instagramKey), nil	×
71	}
72
73	func (i Instagram) getToken(ctx context.Context) (string, error) {	×
74		×
75	// Attempt to get token from Db	×
76	keys, _, err := i.Db.Read(ctx, internal.DbKeys, "", math.MaxInt)	×
77	// If we haven't saved the token before, log an error and refresh the token we have now	×
78	if err != nil \|\| len(keys) == 0 {	×
79	log.Warningf("Couldn't get Keys from the db. Possible error: %v", err)	×
80	return i.refreshToken(ctx, i.AccessToken)	×
81	}	×
82
83	var tk models.InstaToken	×
84	// Get token map from keys	×
85	for _, m := range keys {	×
86	if _, ok := m["id"]; ok && m["id"] == instagramKey {	×
87	if bytes, err := json.Marshal(m); err == nil {	×
88	json.Unmarshal(bytes, &tk)	×
89	}	×
90	break	×
91	}
92	}
93
94	if tk.ID == "" \|\| tk.Value == "" {	×
95	return i.refreshToken(ctx, i.AccessToken)	×
96	}	×
97
98	// Check for expired token.
99	if tk.Expired() {	×
100	// Token has expired and it can't be refreshed. This should never happen.	×
101	return "", fmt.Errorf("instagram access token has expired")	×
102	}	×
103
104	// Refresh the token if need be
105	if tk.IsAboutToExpire(minTokenRemainingLife) {	×
106	return i.refreshToken(ctx, tk.Value)	×
107	}	×
108	return tk.Value, nil	×
109	}
110
111	func (i Instagram) refreshToken(ctx context.Context, oldToken string) (string, error) {	×
112	u, _ := url.Parse("https://graph.instagram.com/refresh_access_token")	×
113	q := u.Query()	×
114	q.Set("grant_type", "ig_refresh_token")	×
115	q.Set("access_token", oldToken)	×
116	u.RawQuery = q.Encode()	×
117		×
118	req, _ := http.NewRequest(http.MethodGet, u.String(), nil)	×
119		×
120	res, err := i.HttpClient.Do(req)	×
121	if err != nil {	×
122	return "", err	×
123	}	×
124	defer res.Body.Close()	×
125		×
126	var newT = models.NewInstaToken(instagramKey)	×
127	err = json.NewDecoder(res.Body).Decode(&newT)	×
128	if err != nil {	×
129	return "", err	×
130	}	×
131
132	newT.RefreshedAt = time.Now()	×
133	err = i.Db.Write(ctx, internal.DbKeys, newT)	×
134	if err != nil {	×
135	return "", err	×
136	}	×
137
138	return newT.Value, nil	×
139	}

wilburt / wilburx9.dev / 3214682499

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous