Browse Source

Add get files endpoint

feature/add-get-files-endpoints
Roman Zipp 3 years ago
committed by Fabian Vowie
parent
commit
728f53c38f
No known key found for this signature in database GPG Key ID: C27317C33B27C410
  1. 50
      controllers/file.go
  2. 6
      main.go
  3. 35
      main_test.go

50
controllers/file.go

@ -0,0 +1,50 @@
package controllers
import (
"encoding/json"
"io/ioutil"
"net/http"
"time"
"github.com/geplauder/lithium/pipelines"
"github.com/geplauder/lithium/settings"
"github.com/geplauder/lithium/storage"
)
type File struct {
Bucket string `json:"bucket"`
FileName string `json:"file_name"`
Size int `json:"size"`
ModifiedAt time.Time `json:"modified_at"`
}
func FileHandler(w http.ResponseWriter, r *http.Request, pipes []pipelines.IPipeline, storageProvider storage.IStorageProvider, appSettings settings.Settings) {
buckets, err := ioutil.ReadDir("./files/" + appSettings.StorageProvider.BasePath)
if err != nil {
json.NewEncoder(w).Encode(struct {
Files []File `json:"files"`
}{})
return
}
var files []File
for _, b := range buckets {
bucketFiles, err := ioutil.ReadDir("./files/" + appSettings.StorageProvider.BasePath + "/" + b.Name())
if err != nil {
writeError(w, 500, "Base path not found")
return
}
for _, f := range bucketFiles {
files = append(files, File{b.Name(), f.Name(), int(f.Size()), f.ModTime()})
}
}
err = json.NewEncoder(w).Encode(struct {
Files []File `json:"files"`
}{files})
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
}
}

6
main.go

@ -29,6 +29,11 @@ func RegisterRoutes(r *mux.Router, appSettings settings.Settings, pipelines []pi
controllers.UploadHandler(w, r, pipelines, storageProvider)
})
files := r.Methods(http.MethodGet).Subrouter()
files.HandleFunc("/files", func(w http.ResponseWriter, r *http.Request) {
controllers.FileHandler(w, r, pipelines, storageProvider, appSettings)
})
pipeline := r.Methods(http.MethodGet).Subrouter()
pipeline.HandleFunc("/pipelines/{pipeline}", func(w http.ResponseWriter, r *http.Request) {
for _, pipeline := range pipelines {
@ -45,6 +50,7 @@ func RegisterRoutes(r *mux.Router, appSettings settings.Settings, pipelines []pi
authMiddleware := middlewares.CreateAuthenticationMiddleware(appSettings.Authentication.Token)
upload.Use(authMiddleware.Middleware)
files.Use(authMiddleware.Middleware)
pipeline.Use(authMiddleware.Middleware)
}
}

35
main_test.go

@ -6,6 +6,7 @@ import (
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/geplauder/lithium/controllers"
@ -141,3 +142,37 @@ func TestUploadRoute(t *testing.T) {
assert.JSONEq(t, string(str), responseRecorder.Body.String())
})
}
func TestFilesRoute(t *testing.T) {
data := pipelines.Pipeline{}
err := faker.FakeData(&data)
assert.Nil(t, err)
t.Run("Files endpoint returns files", func(t *testing.T) {
router := mux.NewRouter()
fs := storage.GetMemoryStorageProvider()
appSettings, _ := settings.LoadSettings(afero.NewMemMapFs())
RegisterRoutes(router, appSettings, []pipelines.IPipeline{data}, fs)
request, _ := http.NewRequest("GET", "/files", nil)
responseRecorder := httptest.NewRecorder()
router.ServeHTTP(responseRecorder, request)
type responseBody struct {
Files []struct {
Bucket string `json:"bucket"`
FileName string `json:"file_name"`
Size int `json:"size"`
ModifiedAt time.Time `json:"modified_at"`
} `json:"files"`
}
var res responseBody
assert.Equal(t, 200, responseRecorder.Code)
err := json.Unmarshal(responseRecorder.Body.Bytes(), &res)
assert.Nil(t, err)
})
}
Loading…
Cancel
Save