5 files changed, 95 insertions, 3 deletions
diff --git a/ui/integration_pocket.go b/ui/integration_pocket.go
new file mode 100644
index 0000000..4f65d61
--- /dev/null
+++ b/ui/integration_pocket.go
@@ -0,0 +1,87 @@
+// Copyright 2018 Frédéric Guillot. All rights reserved.
+// Use of this source code is governed by the Apache 2.0
+// license that can be found in the LICENSE file.
+
+package ui
+
+import (
+	"net/http"
+
+	"github.com/miniflux/miniflux/http/context"
+	"github.com/miniflux/miniflux/http/response"
+	"github.com/miniflux/miniflux/http/response/html"
+	"github.com/miniflux/miniflux/http/route"
+	"github.com/miniflux/miniflux/integration/pocket"
+	"github.com/miniflux/miniflux/logger"
+	"github.com/miniflux/miniflux/ui/session"
+)
+
+// PocketAuthorize redirects the end-user to Pocket website to authorize the application.
+func (c *Controller) PocketAuthorize(w http.ResponseWriter, r *http.Request) {
+	ctx := context.New(r)
+
+	user, err := c.store.UserByID(ctx.UserID())
+	if err != nil {
+		html.ServerError(w, err)
+		return
+	}
+
+	integration, err := c.store.Integration(user.ID)
+	if err != nil {
+		html.ServerError(w, err)
+		return
+	}
+
+	sess := session.New(c.store, ctx)
+	connector := pocket.NewConnector(integration.PocketConsumerKey)
+	redirectURL := c.cfg.BaseURL() + route.Path(c.router, "pocketCallback")
+	requestToken, err := connector.RequestToken(redirectURL)
+	if err != nil {
+		logger.Error("[Pocket:Authorize] %v", err)
+		sess.NewFlashErrorMessage(c.translator.GetLanguage(ctx.UserLanguage()).Get("Unable to fetch request token from Pocket!"))
+		response.Redirect(w, r, route.Path(c.router, "integrations"))
+		return
+	}
+
+	sess.SetPocketRequestToken(requestToken)
+	response.Redirect(w, r, connector.AuthorizationURL(requestToken, redirectURL))
+}
+
+// PocketCallback saves the personal access token after the authorization step.
+func (c *Controller) PocketCallback(w http.ResponseWriter, r *http.Request) {
+	ctx := context.New(r)
+	sess := session.New(c.store, ctx)
+
+	user, err := c.store.UserByID(ctx.UserID())
+	if err != nil {
+		html.ServerError(w, err)
+		return
+	}
+
+	integration, err := c.store.Integration(user.ID)
+	if err != nil {
+		html.ServerError(w, err)
+		return
+	}
+
+	connector := pocket.NewConnector(integration.PocketConsumerKey)
+	accessToken, err := connector.AccessToken(ctx.PocketRequestToken())
+	if err != nil {
+		logger.Error("[Pocket:Callback] %v", err)
+		sess.NewFlashErrorMessage(c.translator.GetLanguage(ctx.UserLanguage()).Get("Unable to fetch access token from Pocket!"))
+		response.Redirect(w, r, route.Path(c.router, "integrations"))
+		return
+	}
+
+	sess.SetPocketRequestToken("")
+	integration.PocketAccessToken = accessToken
+
+	err = c.store.UpdateIntegration(integration)
+	if err != nil {
+		html.ServerError(w, err)
+		return
+	}
+
+	sess.NewFlashMessage(c.translator.GetLanguage(ctx.UserLanguage()).Get("Your Pocket account is now linked!"))
+	response.Redirect(w, r, route.Path(c.router, "integrations"))
+}
diff --git a/ui/session/session.go b/ui/session/session.go
index fc25406..3d630a8 100644
--- a/ui/session/session.go
+++ b/ui/session/session.go
@@ -56,6 +56,11 @@ func (s *Session) SetLanguage(language string) {
 	s.store.UpdateSessionField(s.ctx.SessionID(), "language", language)
 }
 
+// SetPocketRequestToken updates Pocket Request Token.
+func (s *Session) SetPocketRequestToken(requestToken string) {
+	s.store.UpdateSessionField(s.ctx.SessionID(), "pocket_request_token", requestToken)
+}
+
 // New returns a new session handler.
 func New(store *storage.Storage, ctx *context.Context) *Session {
 	return &Session{store, ctx}
diff --git a/ui/static/bin.go b/ui/static/bin.go
index a5c0ed0..6c74cd9 100644
--- a/ui/static/bin.go
+++ b/ui/static/bin.go
@@ -1,5 +1,5 @@
 // Code generated by go generate; DO NOT EDIT.
-// 2018-05-20 11:35:19.492969127 -0700 PDT m=+0.004803782
+// 2018-05-20 15:22:33.779971968 -0700 PDT m=+0.011442481
 
 package static
 
diff --git a/ui/static/css.go b/ui/static/css.go
index 8e28baf..6755dcf 100644
--- a/ui/static/css.go
+++ b/ui/static/css.go
@@ -1,5 +1,5 @@
 // Code generated by go generate; DO NOT EDIT.
-// 2018-05-20 11:35:19.493979584 -0700 PDT m=+0.005814260
+// 2018-05-20 15:22:33.782567368 -0700 PDT m=+0.014037881
 
 package static
 
diff --git a/ui/static/js.go b/ui/static/js.go
index b507236..d4558c2 100644
--- a/ui/static/js.go
+++ b/ui/static/js.go
@@ -1,5 +1,5 @@
 // Code generated by go generate; DO NOT EDIT.
-// 2018-05-20 11:35:19.494515654 -0700 PDT m=+0.006350329
+// 2018-05-20 15:22:33.784038886 -0700 PDT m=+0.015509399
 
 package static