router.go 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. package api
  2. import (
  3. "net/http"
  4. "github.com/drakkan/sftpgo/logger"
  5. "github.com/drakkan/sftpgo/sftpd"
  6. "github.com/drakkan/sftpgo/utils"
  7. "github.com/go-chi/chi"
  8. "github.com/go-chi/chi/middleware"
  9. "github.com/go-chi/render"
  10. )
  11. // GetHTTPRouter returns the configured HTTP handler
  12. func GetHTTPRouter() http.Handler {
  13. return router
  14. }
  15. func initializeRouter() {
  16. router = chi.NewRouter()
  17. router.Use(middleware.RequestID)
  18. router.Use(middleware.RealIP)
  19. router.Use(logger.NewStructuredLogger(logger.GetLogger()))
  20. router.Use(middleware.Recoverer)
  21. router.NotFound(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  22. sendAPIResponse(w, r, nil, "Not Found", http.StatusNotFound)
  23. }))
  24. router.MethodNotAllowed(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  25. sendAPIResponse(w, r, nil, "Method not allowed", http.StatusMethodNotAllowed)
  26. }))
  27. router.Get(versionPath, func(w http.ResponseWriter, r *http.Request) {
  28. render.JSON(w, r, utils.GetAppVersion())
  29. })
  30. router.Get(activeConnectionsPath, func(w http.ResponseWriter, r *http.Request) {
  31. render.JSON(w, r, sftpd.GetConnectionsStats())
  32. })
  33. router.Delete(activeConnectionsPath+"/{connectionID}", func(w http.ResponseWriter, r *http.Request) {
  34. handleCloseSFTPConnection(w, r)
  35. })
  36. router.Get(quotaScanPath, func(w http.ResponseWriter, r *http.Request) {
  37. getQuotaScans(w, r)
  38. })
  39. router.Post(quotaScanPath, func(w http.ResponseWriter, r *http.Request) {
  40. startQuotaScan(w, r)
  41. })
  42. router.Get(userPath, func(w http.ResponseWriter, r *http.Request) {
  43. getUsers(w, r)
  44. })
  45. router.Post(userPath, func(w http.ResponseWriter, r *http.Request) {
  46. addUser(w, r)
  47. })
  48. router.Get(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
  49. getUserByID(w, r)
  50. })
  51. router.Put(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
  52. updateUser(w, r)
  53. })
  54. router.Delete(userPath+"/{userID}", func(w http.ResponseWriter, r *http.Request) {
  55. deleteUser(w, r)
  56. })
  57. }
  58. func handleCloseSFTPConnection(w http.ResponseWriter, r *http.Request) {
  59. connectionID := chi.URLParam(r, "connectionID")
  60. if connectionID == "" {
  61. sendAPIResponse(w, r, nil, "connectionID is mandatory", http.StatusBadRequest)
  62. return
  63. }
  64. if sftpd.CloseActiveConnection(connectionID) {
  65. sendAPIResponse(w, r, nil, "Connection closed", http.StatusOK)
  66. } else {
  67. sendAPIResponse(w, r, nil, "Not Found", http.StatusNotFound)
  68. }
  69. }