58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
|
// Copyright 2017 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 middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/miniflux/miniflux2/storage"
|
||
|
|
)
|
||
|
|
|
||
|
|
// FeverMiddleware is the middleware that handles Fever API.
|
||
|
|
type FeverMiddleware struct {
|
||
|
|
store *storage.Storage
|
||
|
|
}
|
||
|
|
|
||
|
|
// Handler executes the middleware.
|
||
|
|
func (f *FeverMiddleware) Handler(next http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
log.Println("[Middleware:Fever]")
|
||
|
|
|
||
|
|
apiKey := r.FormValue("api_key")
|
||
|
|
user, err := f.store.UserByFeverToken(apiKey)
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
w.Header().Set("Content-Type", "application/json")
|
||
|
|
w.Write([]byte(`{"api_version": 3, "auth": 0}`))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if user == nil {
|
||
|
|
log.Println("[Middleware:Fever] Fever authentication failure")
|
||
|
|
w.Header().Set("Content-Type", "application/json")
|
||
|
|
w.Write([]byte(`{"api_version": 3, "auth": 0}`))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Printf("[Middleware:Fever] User #%d is authenticated\n", user.ID)
|
||
|
|
f.store.SetLastLogin(user.ID)
|
||
|
|
|
||
|
|
ctx := r.Context()
|
||
|
|
ctx = context.WithValue(ctx, UserIDContextKey, user.ID)
|
||
|
|
ctx = context.WithValue(ctx, UserTimezoneContextKey, user.Timezone)
|
||
|
|
ctx = context.WithValue(ctx, IsAdminUserContextKey, user.IsAdmin)
|
||
|
|
ctx = context.WithValue(ctx, IsAuthenticatedContextKey, true)
|
||
|
|
|
||
|
|
next.ServeHTTP(w, r.WithContext(ctx))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewFeverMiddleware returns a new FeverMiddleware.
|
||
|
|
func NewFeverMiddleware(s *storage.Storage) *FeverMiddleware {
|
||
|
|
return &FeverMiddleware{store: s}
|
||
|
|
}
|