117 lines
3.5 KiB
Go
117 lines
3.5 KiB
Go
package httpd
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/go-chi/render"
|
|
|
|
"github.com/drakkan/sftpgo/v2/dataprovider"
|
|
"github.com/drakkan/sftpgo/v2/vfs"
|
|
)
|
|
|
|
func getFolders(w http.ResponseWriter, r *http.Request) {
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
|
|
limit, offset, order, err := getSearchFilters(w, r)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
folders, err := dataprovider.GetFolders(limit, offset, order)
|
|
if err == nil {
|
|
render.JSON(w, r, folders)
|
|
} else {
|
|
sendAPIResponse(w, r, err, "", http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
func addFolder(w http.ResponseWriter, r *http.Request) {
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
|
|
var folder vfs.BaseVirtualFolder
|
|
err := render.DecodeJSON(r.Body, &folder)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", http.StatusBadRequest)
|
|
return
|
|
}
|
|
err = dataprovider.AddFolder(&folder)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", getRespStatus(err))
|
|
return
|
|
}
|
|
renderFolder(w, r, folder.Name, http.StatusCreated)
|
|
}
|
|
|
|
func updateFolder(w http.ResponseWriter, r *http.Request) {
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
|
|
var err error
|
|
|
|
name := getURLParam(r, "name")
|
|
folder, err := dataprovider.GetFolderByName(name)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", getRespStatus(err))
|
|
return
|
|
}
|
|
users := folder.Users
|
|
folderID := folder.ID
|
|
currentS3AccessSecret := folder.FsConfig.S3Config.AccessSecret
|
|
currentAzAccountKey := folder.FsConfig.AzBlobConfig.AccountKey
|
|
currentAzSASUrl := folder.FsConfig.AzBlobConfig.SASURL
|
|
currentGCSCredentials := folder.FsConfig.GCSConfig.Credentials
|
|
currentCryptoPassphrase := folder.FsConfig.CryptConfig.Passphrase
|
|
currentSFTPPassword := folder.FsConfig.SFTPConfig.Password
|
|
currentSFTPKey := folder.FsConfig.SFTPConfig.PrivateKey
|
|
|
|
folder.FsConfig.S3Config = vfs.S3FsConfig{}
|
|
folder.FsConfig.AzBlobConfig = vfs.AzBlobFsConfig{}
|
|
folder.FsConfig.GCSConfig = vfs.GCSFsConfig{}
|
|
folder.FsConfig.CryptConfig = vfs.CryptFsConfig{}
|
|
folder.FsConfig.SFTPConfig = vfs.SFTPFsConfig{}
|
|
err = render.DecodeJSON(r.Body, &folder)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", http.StatusBadRequest)
|
|
return
|
|
}
|
|
folder.ID = folderID
|
|
folder.Name = name
|
|
folder.FsConfig.SetEmptySecretsIfNil()
|
|
updateEncryptedSecrets(&folder.FsConfig, currentS3AccessSecret, currentAzAccountKey, currentAzSASUrl, currentGCSCredentials,
|
|
currentCryptoPassphrase, currentSFTPPassword, currentSFTPKey)
|
|
err = dataprovider.UpdateFolder(&folder, users)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", getRespStatus(err))
|
|
return
|
|
}
|
|
sendAPIResponse(w, r, nil, "Folder updated", http.StatusOK)
|
|
}
|
|
|
|
func renderFolder(w http.ResponseWriter, r *http.Request, name string, status int) {
|
|
folder, err := dataprovider.GetFolderByName(name)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", getRespStatus(err))
|
|
return
|
|
}
|
|
folder.PrepareForRendering()
|
|
if status != http.StatusOK {
|
|
ctx := context.WithValue(r.Context(), render.StatusCtxKey, status)
|
|
render.JSON(w, r.WithContext(ctx), folder)
|
|
} else {
|
|
render.JSON(w, r, folder)
|
|
}
|
|
}
|
|
|
|
func getFolderByName(w http.ResponseWriter, r *http.Request) {
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
|
|
name := getURLParam(r, "name")
|
|
renderFolder(w, r, name, http.StatusOK)
|
|
}
|
|
|
|
func deleteFolder(w http.ResponseWriter, r *http.Request) {
|
|
r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
|
|
name := getURLParam(r, "name")
|
|
err := dataprovider.DeleteFolder(name)
|
|
if err != nil {
|
|
sendAPIResponse(w, r, err, "", getRespStatus(err))
|
|
return
|
|
}
|
|
sendAPIResponse(w, r, err, "Folder deleted", http.StatusOK)
|
|
}
|