110 lines
3.5 KiB
Go
110 lines
3.5 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/leandrofars/oktopus/internal/api/cors"
|
|
"github.com/leandrofars/oktopus/internal/api/middleware"
|
|
"github.com/leandrofars/oktopus/internal/bridge"
|
|
"github.com/leandrofars/oktopus/internal/config"
|
|
"github.com/leandrofars/oktopus/internal/db"
|
|
"github.com/nats-io/nats.go"
|
|
"github.com/nats-io/nats.go/jetstream"
|
|
)
|
|
|
|
type Api struct {
|
|
port string
|
|
js jetstream.JetStream
|
|
nc *nats.Conn
|
|
bridge bridge.Bridge
|
|
db db.Database
|
|
ctx context.Context
|
|
}
|
|
|
|
const REQUEST_TIMEOUT = time.Second * 30
|
|
|
|
const (
|
|
NormalUser = iota
|
|
AdminUser
|
|
)
|
|
|
|
func NewApi(c config.RestApi, js jetstream.JetStream, nc *nats.Conn, bridge bridge.Bridge, d db.Database) Api {
|
|
return Api{
|
|
port: c.Port,
|
|
js: js,
|
|
nc: nc,
|
|
ctx: c.Ctx,
|
|
bridge: bridge,
|
|
db: d,
|
|
}
|
|
}
|
|
|
|
func (a *Api) StartApi() {
|
|
r := mux.NewRouter()
|
|
authentication := r.PathPrefix("/api/auth").Subrouter()
|
|
authentication.HandleFunc("/login", a.generateToken).Methods("PUT")
|
|
authentication.HandleFunc("/register", a.registerUser).Methods("POST")
|
|
authentication.HandleFunc("/admin/register", a.registerAdminUser).Methods("POST")
|
|
authentication.HandleFunc("/admin/exists", a.adminUserExists).Methods("GET")
|
|
iot := r.PathPrefix("/api/device").Subrouter()
|
|
iot.HandleFunc("", a.retrieveDevices).Methods("GET")
|
|
iot.HandleFunc("/{id}", a.retrieveDevices).Methods("GET")
|
|
iot.HandleFunc("/{sn}/{mtp}/get", a.deviceGetMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/add", a.deviceCreateMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/del", a.deviceDeleteMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/set", a.deviceUpdateMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/parameters", a.deviceGetSupportedParametersMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/instances", a.deviceGetParameterInstances).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/operate", a.deviceOperateMsg).Methods("PUT")
|
|
iot.HandleFunc("/{sn}/{mtp}/fw_update", a.deviceFwUpdate).Methods("PUT")
|
|
// iot.HandleFunc("/{sn}/{mtp}/wifi", a.deviceWifi).Methods("PUT", "GET")
|
|
// mtp := r.PathPrefix("/api/mtp").Subrouter()
|
|
// mtp.HandleFunc("", a.mtpInfo).Methods("GET")
|
|
dash := r.PathPrefix("/api/info").Subrouter()
|
|
dash.HandleFunc("/vendors", a.vendorsInfo).Methods("GET")
|
|
dash.HandleFunc("/status", a.statusInfo).Methods("GET")
|
|
dash.HandleFunc("/device_class", a.productClassInfo).Methods("GET")
|
|
dash.HandleFunc("/general", a.generalInfo).Methods("GET")
|
|
users := r.PathPrefix("/api/users").Subrouter()
|
|
users.HandleFunc("", a.retrieveUsers).Methods("GET")
|
|
|
|
/* ----- Middleware for requests which requires user to be authenticated ---- */
|
|
iot.Use(func(handler http.Handler) http.Handler {
|
|
return middleware.Middleware(handler)
|
|
})
|
|
|
|
// mtp.Use(func(handler http.Handler) http.Handler {
|
|
// return middleware.Middleware(handler)
|
|
// })
|
|
|
|
dash.Use(func(handler http.Handler) http.Handler {
|
|
return middleware.Middleware(handler)
|
|
})
|
|
|
|
users.Use(func(handler http.Handler) http.Handler {
|
|
return middleware.Middleware(handler)
|
|
})
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
corsOpts := cors.GetCorsConfig()
|
|
|
|
srv := &http.Server{
|
|
Addr: "0.0.0.0:" + a.port,
|
|
WriteTimeout: time.Second * 60,
|
|
ReadTimeout: time.Second * 60,
|
|
IdleTimeout: time.Second * 60,
|
|
Handler: corsOpts.Handler(r),
|
|
}
|
|
|
|
go func() {
|
|
if err := srv.ListenAndServe(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
log.Println("Running REST API at port", a.port)
|
|
}
|