130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package consoleapi
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/common/storj"
|
|
"storj.io/storj/storagenode/console"
|
|
)
|
|
|
|
// ErrStorageNodeAPI - console storageNode api error type.
|
|
var ErrStorageNodeAPI = errs.Class("storageNode console web error")
|
|
|
|
// StorageNode is an api controller that exposes all dashboard related api.
|
|
type StorageNode struct {
|
|
service *console.Service
|
|
|
|
log *zap.Logger
|
|
}
|
|
|
|
// NewStorageNode is a constructor for sno controller.
|
|
func NewStorageNode(log *zap.Logger, service *console.Service) *StorageNode {
|
|
return &StorageNode{
|
|
log: log,
|
|
service: service,
|
|
}
|
|
}
|
|
|
|
// StorageNode handles StorageNode API requests.
|
|
func (dashboard *StorageNode) StorageNode(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
var err error
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
w.Header().Set(contentType, applicationJSON)
|
|
|
|
data, err := dashboard.service.GetDashboardData(ctx)
|
|
if err != nil {
|
|
dashboard.serveJSONError(w, http.StatusInternalServerError, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
if err := json.NewEncoder(w).Encode(data); err != nil {
|
|
dashboard.log.Error("failed to encode json response", zap.Error(ErrStorageNodeAPI.Wrap(err)))
|
|
return
|
|
}
|
|
}
|
|
|
|
// Satellites handles satellites API request.
|
|
func (dashboard *StorageNode) Satellites(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
var err error
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
w.Header().Set(contentType, applicationJSON)
|
|
|
|
data, err := dashboard.service.GetAllSatellitesData(ctx)
|
|
if err != nil {
|
|
dashboard.serveJSONError(w, http.StatusInternalServerError, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
if err := json.NewEncoder(w).Encode(data); err != nil {
|
|
dashboard.log.Error("failed to encode json response", zap.Error(ErrStorageNodeAPI.Wrap(err)))
|
|
return
|
|
}
|
|
}
|
|
|
|
// Satellite handles satellite API requests.
|
|
func (dashboard *StorageNode) Satellite(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
var err error
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
w.Header().Set(contentType, applicationJSON)
|
|
|
|
params := mux.Vars(r)
|
|
id, ok := params["id"]
|
|
if !ok {
|
|
dashboard.serveJSONError(w, http.StatusBadRequest, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
satelliteID, err := storj.NodeIDFromString(id)
|
|
if err != nil {
|
|
dashboard.serveJSONError(w, http.StatusBadRequest, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
if err = dashboard.service.VerifySatelliteID(ctx, satelliteID); err != nil {
|
|
dashboard.serveJSONError(w, http.StatusNotFound, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
data, err := dashboard.service.GetSatelliteData(ctx, satelliteID)
|
|
if err != nil {
|
|
dashboard.serveJSONError(w, http.StatusInternalServerError, ErrStorageNodeAPI.Wrap(err))
|
|
return
|
|
}
|
|
|
|
if err := json.NewEncoder(w).Encode(data); err != nil {
|
|
dashboard.log.Error("failed to encode json response", zap.Error(ErrStorageNodeAPI.Wrap(err)))
|
|
return
|
|
}
|
|
}
|
|
|
|
// serveJSONError writes JSON error to response output stream.
|
|
func (dashboard *StorageNode) serveJSONError(w http.ResponseWriter, status int, err error) {
|
|
w.WriteHeader(status)
|
|
|
|
var response struct {
|
|
Error string `json:"error"`
|
|
}
|
|
|
|
response.Error = err.Error()
|
|
|
|
err = json.NewEncoder(w).Encode(response)
|
|
if err != nil {
|
|
dashboard.log.Error("failed to write json error response", zap.Error(ErrStorageNodeAPI.Wrap(err)))
|
|
return
|
|
}
|
|
}
|