1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586 |
- package api
- import (
- "net/http"
- "github.com/drakkan/sftpgo/logger"
- "github.com/drakkan/sftpgo/sftpd"
- "github.com/drakkan/sftpgo/utils"
- "github.com/go-chi/chi"
- "github.com/go-chi/chi/middleware"
- "github.com/go-chi/render"
- )
- // GetHTTPRouter returns the configured HTTP handler
- func GetHTTPRouter() http.Handler {
- return router
- }
- func initializeRouter() {
- router = chi.NewRouter()
- router.Use(middleware.RequestID)
- router.Use(middleware.RealIP)
- router.Use(logger.NewStructuredLogger(logger.GetLogger()))
- router.Use(middleware.Recoverer)
- router.NotFound(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sendAPIResponse(w, r, nil, "Not Found", http.StatusNotFound)
- }))
- router.MethodNotAllowed(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sendAPIResponse(w, r, nil, "Method not allowed", http.StatusMethodNotAllowed)
- }))
- router.Get(versionPath, func(w http.ResponseWriter, r *http.Request) {
- render.JSON(w, r, utils.GetAppVersion())
- })
- router.Get(activeConnectionsPath, func(w http.ResponseWriter, r *http.Request) {
- render.JSON(w, r, sftpd.GetConnectionsStats())
- })
- router.Delete(activeConnectionsPath+"/{connectionID}", func(w http.ResponseWriter, r *http.Request) {
- handleCloseSFTPConnection(w, r)
- })
- router.Get(quotaScanPath, func(w http.ResponseWriter, r *http.Request) {
- getQuotaScans(w, r)
- })
- router.Post(quotaScanPath, func(w http.ResponseWriter, r *http.Request) {
- startQuotaScan(w, r)
- })
- router.Get(userPath, func(w http.ResponseWriter, r *http.Request) {
- getUsers(w, r)
- })
- router.Post(userPath, func(w http.ResponseWriter, r *http.Request) {
- addUser(w, r)
- })
- router.Get(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
- getUserByID(w, r)
- })
- router.Put(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
- updateUser(w, r)
- })
- router.Delete(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
- deleteUser(w, r)
- })
- }
- func handleCloseSFTPConnection(w http.ResponseWriter, r *http.Request) {
- connectionID := chi.URLParam(r, "connectionID")
- if connectionID == "" {
- sendAPIResponse(w, r, nil, "connectionID is mandatory", http.StatusBadRequest)
- return
- }
- if sftpd.CloseActiveConnection(connectionID) {
- sendAPIResponse(w, r, nil, "Connection closed", http.StatusOK)
- } else {
- sendAPIResponse(w, r, nil, "Not Found", http.StatusNotFound)
- }
- }
|