diff options
Diffstat (limited to 'ui/integration_pocket.go')
-rw-r--r-- | ui/integration_pocket.go | 87 |
1 files changed, 87 insertions, 0 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")) +} |