Compare commits
37 Commits
a3e66cd351
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| 125ce9c955 | |||
| 131f19deed | |||
| 2ac552e840 | |||
| 48c50a5b7e | |||
| 48f770f703 | |||
| 2ae14cdfd4 | |||
| a10593a318 | |||
| 0971301562 | |||
| 47befe6db1 | |||
| 979ebee677 | |||
| 8d85d645d6 | |||
| 60817c2249 | |||
| bead881af2 | |||
| 80302b62f4 | |||
| 2353a0bf53 | |||
| c5ab0156fd | |||
| 686630b2df | |||
| 8725def3a1 | |||
| 9be7b6c18f | |||
| 13f3217a38 | |||
| aa82cd938c | |||
| aac1ca8891 | |||
| 2268b518b1 | |||
| 8c6f59a6b4 | |||
| 439e5473b6 | |||
| d2b0364445 | |||
| 4344bf841c | |||
| f2a8e0197d | |||
| 74037dfab5 | |||
| 91e217e472 | |||
| f42d12ffd9 | |||
| 87dc249371 | |||
| a060ed0fe4 | |||
| a35ce7c24b | |||
| 9cfc4da5f3 | |||
| f59f3183f2 | |||
| ed932e3c92 |
2
.env
2
.env
@@ -2,7 +2,7 @@
|
|||||||
S3_ENDPOINT=localhost:9000
|
S3_ENDPOINT=localhost:9000
|
||||||
S3_ACCESS_KEY=testo
|
S3_ACCESS_KEY=testo
|
||||||
S3_SECRET_KEY=testotesto
|
S3_SECRET_KEY=testotesto
|
||||||
S3_BUCKET=dev
|
|
||||||
S3_DISABLE_SSL=true
|
S3_DISABLE_SSL=true
|
||||||
ADDRESS=:8080
|
ADDRESS=:8080
|
||||||
VERBOSE=true
|
VERBOSE=true
|
||||||
|
DB_CONNECTION=s3Browser:hunter2@/s3Browser
|
||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
s3browser "git.kapelle.org/niklas/s3browser/internal"
|
s3browser "git.kapelle.org/niklas/s3browser/internal"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
"github.com/alexflint/go-arg"
|
"github.com/alexflint/go-arg"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -11,12 +12,12 @@ type args struct {
|
|||||||
S3Endpoint string `arg:"--s3-endpoint,required,env:S3_ENDPOINT" help:"host[:port]" placeholder:"ENDPOINT"`
|
S3Endpoint string `arg:"--s3-endpoint,required,env:S3_ENDPOINT" help:"host[:port]" placeholder:"ENDPOINT"`
|
||||||
S3AccessKey string `arg:"--s3-access-key,required,env:S3_ACCESS_KEY" placeholder:"ACCESS_KEY"`
|
S3AccessKey string `arg:"--s3-access-key,required,env:S3_ACCESS_KEY" placeholder:"ACCESS_KEY"`
|
||||||
S3SecretKey string `arg:"--s3-secret-key,required,env:S3_SECRET_KEY" placeholder:"SECRET_KEY"`
|
S3SecretKey string `arg:"--s3-secret-key,required,env:S3_SECRET_KEY" placeholder:"SECRET_KEY"`
|
||||||
S3Bucket string `arg:"--s3-bucket,required,env:S3_BUCKET" placeholder:"BUCKET"`
|
|
||||||
S3DisableSSL bool `arg:"--s3-disable-ssl,env:S3_DISABLE_SSL" default:"false"`
|
S3DisableSSL bool `arg:"--s3-disable-ssl,env:S3_DISABLE_SSL" default:"false"`
|
||||||
Address string `arg:"--address,env:ADDRESS" default:":3000" help:"what address to listen on" placeholder:"ADDRESS"`
|
Address string `arg:"--address,env:ADDRESS" default:":3000" help:"what address to listen on" placeholder:"ADDRESS"`
|
||||||
CacheTTL int64 `arg:"--cache-ttl,env:CACHE_TTL" help:"Time in seconds" default:"30" placeholder:"TTL"`
|
CacheTTL int64 `arg:"--cache-ttl,env:CACHE_TTL" help:"Time in seconds" default:"30" placeholder:"TTL"`
|
||||||
CacheCleanup int64 `arg:"--cache-cleanup,env:CACHE_CLEANUP" help:"Time in seconds" default:"60" placeholder:"CLEANUP"`
|
CacheCleanup int64 `arg:"--cache-cleanup,env:CACHE_CLEANUP" help:"Time in seconds" default:"60" placeholder:"CLEANUP"`
|
||||||
Verbose bool `arg:"-v,--verbose,env:VERBOSE" help:"verbosity level" default:"false"`
|
Verbose bool `arg:"-v,--verbose,env:VERBOSE" help:"verbosity level" default:"false"`
|
||||||
|
DBConnection string `arg:"--db,required,env:DB_CONNECTION" help:"DSN in format: https://github.com/go-sql-driver/mysql#dsn-data-source-name"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (args) Version() string {
|
func (args) Version() string {
|
||||||
@@ -28,12 +29,12 @@ func main() {
|
|||||||
var args args
|
var args args
|
||||||
arg.MustParse(&args)
|
arg.MustParse(&args)
|
||||||
|
|
||||||
s3browser.Start(s3browser.AppConfig{
|
s3browser.Start(types.AppConfig{
|
||||||
S3Endoint: args.S3Endpoint,
|
S3Endoint: args.S3Endpoint,
|
||||||
S3SSL: !args.S3DisableSSL,
|
S3SSL: !args.S3DisableSSL,
|
||||||
S3AccessKey: args.S3AccessKey,
|
S3AccessKey: args.S3AccessKey,
|
||||||
S3SecretKey: args.S3SecretKey,
|
S3SecretKey: args.S3SecretKey,
|
||||||
S3Bucket: args.S3Bucket,
|
DSN: args.DBConnection,
|
||||||
CacheTTL: time.Duration(args.CacheTTL) * time.Second,
|
CacheTTL: time.Duration(args.CacheTTL) * time.Second,
|
||||||
CacheCleanup: time.Duration(args.CacheCleanup) * time.Second,
|
CacheCleanup: time.Duration(args.CacheCleanup) * time.Second,
|
||||||
Address: args.Address,
|
Address: args.Address,
|
||||||
|
|||||||
@@ -13,6 +13,20 @@ services:
|
|||||||
command: server /data --console-address ":9001"
|
command: server /data --console-address ":9001"
|
||||||
volumes:
|
volumes:
|
||||||
- minio_dev:/data
|
- minio_dev:/data
|
||||||
|
db:
|
||||||
|
container_name: db
|
||||||
|
image: mariadb
|
||||||
|
environment:
|
||||||
|
- MARIADB_ROOT_PASSWORD=hunter2
|
||||||
|
- MARIADB_DATABASE=s3Browser
|
||||||
|
- MARIADB_USER=s3Browser
|
||||||
|
- MARIADB_PASSWORD=hunter2
|
||||||
|
ports:
|
||||||
|
- 3306:3306
|
||||||
|
volumes:
|
||||||
|
- mariadb_dev:/var/lib/mysql
|
||||||
volumes:
|
volumes:
|
||||||
minio_dev:
|
minio_dev:
|
||||||
name: minio_dev
|
name: minio_dev
|
||||||
|
mariadb_dev:
|
||||||
|
name: mariadb_dev
|
||||||
3
go.mod
3
go.mod
@@ -4,6 +4,7 @@ go 1.16
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/alexflint/go-arg v1.4.2
|
github.com/alexflint/go-arg v1.4.2
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0
|
||||||
github.com/golang-jwt/jwt v3.2.2+incompatible
|
github.com/golang-jwt/jwt v3.2.2+incompatible
|
||||||
github.com/gorilla/mux v1.8.0
|
github.com/gorilla/mux v1.8.0
|
||||||
github.com/graph-gophers/dataloader v5.0.0+incompatible
|
github.com/graph-gophers/dataloader v5.0.0+incompatible
|
||||||
@@ -13,4 +14,6 @@ require (
|
|||||||
github.com/opentracing/opentracing-go v1.2.0 // indirect
|
github.com/opentracing/opentracing-go v1.2.0 // indirect
|
||||||
github.com/patrickmn/go-cache v2.1.0+incompatible
|
github.com/patrickmn/go-cache v2.1.0+incompatible
|
||||||
github.com/sirupsen/logrus v1.8.1
|
github.com/sirupsen/logrus v1.8.1
|
||||||
|
github.com/stretchr/testify v1.7.0
|
||||||
|
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f
|
||||||
)
|
)
|
||||||
|
|||||||
6
go.sum
6
go.sum
@@ -7,6 +7,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
|
|||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
|
github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
|
||||||
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
|
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
|
||||||
|
github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
|
||||||
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
|
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
|
||||||
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
|
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
|
||||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
@@ -65,6 +67,8 @@ github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXf
|
|||||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
|
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
|
||||||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||||
|
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f h1:aZp0e2vLN4MToVqnjNEYEtrEA8RH8U8FN1CU7JgqsPU=
|
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f h1:aZp0e2vLN4MToVqnjNEYEtrEA8RH8U8FN1CU7JgqsPU=
|
||||||
@@ -92,3 +96,5 @@ gopkg.in/ini.v1 v1.57.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
|
|||||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
|
gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
|
||||||
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|||||||
9
internal/cache/cache.go
vendored
Normal file
9
internal/cache/cache.go
vendored
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package cache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/graph-gophers/dataloader"
|
||||||
|
)
|
||||||
|
|
||||||
|
type S3Cache interface {
|
||||||
|
dataloader.Cache
|
||||||
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
package s3browser
|
package cache
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
@@ -8,17 +8,18 @@ import (
|
|||||||
gocache "github.com/patrickmn/go-cache"
|
gocache "github.com/patrickmn/go-cache"
|
||||||
)
|
)
|
||||||
|
|
||||||
type cache struct {
|
type TTLCache struct {
|
||||||
c *gocache.Cache
|
c *gocache.Cache
|
||||||
}
|
}
|
||||||
|
|
||||||
func newCache(ttl, cleanupInterval time.Duration) *cache {
|
// Create new ttl cache
|
||||||
return &cache{
|
func NewTTLCache(ttl, cleanupInterval time.Duration) *TTLCache {
|
||||||
|
return &TTLCache{
|
||||||
c: gocache.New(ttl, cleanupInterval),
|
c: gocache.New(ttl, cleanupInterval),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *cache) Get(_ context.Context, key dataloader.Key) (dataloader.Thunk, bool) {
|
func (c *TTLCache) Get(_ context.Context, key dataloader.Key) (dataloader.Thunk, bool) {
|
||||||
v, ok := c.c.Get(key.String())
|
v, ok := c.c.Get(key.String())
|
||||||
if ok {
|
if ok {
|
||||||
return v.(dataloader.Thunk), ok
|
return v.(dataloader.Thunk), ok
|
||||||
@@ -26,11 +27,11 @@ func (c *cache) Get(_ context.Context, key dataloader.Key) (dataloader.Thunk, bo
|
|||||||
return nil, ok
|
return nil, ok
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *cache) Set(_ context.Context, key dataloader.Key, value dataloader.Thunk) {
|
func (c *TTLCache) Set(_ context.Context, key dataloader.Key, value dataloader.Thunk) {
|
||||||
c.c.Set(key.String(), value, 0)
|
c.c.Set(key.String(), value, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *cache) Delete(_ context.Context, key dataloader.Key) bool {
|
func (c *TTLCache) Delete(_ context.Context, key dataloader.Key) bool {
|
||||||
if _, found := c.c.Get(key.String()); found {
|
if _, found := c.c.Get(key.String()); found {
|
||||||
c.c.Delete(key.String())
|
c.c.Delete(key.String())
|
||||||
return true
|
return true
|
||||||
@@ -38,6 +39,6 @@ func (c *cache) Delete(_ context.Context, key dataloader.Key) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *cache) Clear() {
|
func (c *TTLCache) Clear() {
|
||||||
c.c.Flush()
|
c.c.Flush()
|
||||||
}
|
}
|
||||||
@@ -1,245 +0,0 @@
|
|||||||
package s3browser
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/graph-gophers/dataloader"
|
|
||||||
"github.com/minio/minio-go/v7"
|
|
||||||
log "github.com/sirupsen/logrus"
|
|
||||||
)
|
|
||||||
|
|
||||||
// listObjectsBatch batch func for calling s3.ListObjects()
|
|
||||||
func listObjectsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
|
||||||
log.Debug("listObjectsBatch: ", k.Keys())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
|
|
||||||
s3Client, ok := c.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, v := range k {
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: listObjects(s3Client, bucketName, v.String(), false),
|
|
||||||
Error: nil,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// listObjectsRecursiveBatch just like listObjectsBatch but with recursive set to true
|
|
||||||
func listObjectsRecursiveBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
|
||||||
log.Debug("listObjectsRecursiveBatch: ", k.Keys())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
|
|
||||||
s3Client, ok := c.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, v := range k {
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: listObjects(s3Client, bucketName, v.String(), true),
|
|
||||||
Error: nil,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// listObjects helper func for listObjectsBatch
|
|
||||||
func listObjects(s3Client *minio.Client, bukitName, path string, recursive bool) []minio.ObjectInfo {
|
|
||||||
log.Debug("S3 call 'ListObjects': ", path)
|
|
||||||
objectCh := s3Client.ListObjects(context.Background(), bukitName, minio.ListObjectsOptions{
|
|
||||||
Prefix: path,
|
|
||||||
Recursive: recursive,
|
|
||||||
})
|
|
||||||
|
|
||||||
result := make([]minio.ObjectInfo, 0)
|
|
||||||
for obj := range objectCh {
|
|
||||||
result = append(result, obj)
|
|
||||||
}
|
|
||||||
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
// getFilesBatch batch func for getting all files in path. Uses "listObjects" dataloader
|
|
||||||
func getFilesBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
|
||||||
log.Debug("getFilesBatch: ", k.Keys())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
|
|
||||||
loader, ok := c.Value("loader").(map[string]*dataloader.Loader)
|
|
||||||
if !ok {
|
|
||||||
return handleLoaderError(k, fmt.Errorf("Failed to get loader from context"))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, v := range k {
|
|
||||||
path := v.String()
|
|
||||||
files := make([]File, 0)
|
|
||||||
|
|
||||||
if !strings.HasSuffix(path, "/") {
|
|
||||||
path += "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
thunk := loader["listObjects"].Load(c, dataloader.StringKey(path))
|
|
||||||
|
|
||||||
objects, _ := thunk()
|
|
||||||
|
|
||||||
// TODO: handle thunk error
|
|
||||||
|
|
||||||
for _, obj := range objects.([]minio.ObjectInfo) {
|
|
||||||
if obj.Err != nil {
|
|
||||||
// TODO: how to handle?
|
|
||||||
} else if !strings.HasSuffix(obj.Key, "/") {
|
|
||||||
files = append(files, File{
|
|
||||||
ID: obj.Key,
|
|
||||||
Name: filepath.Base(obj.Key),
|
|
||||||
Size: obj.Size,
|
|
||||||
ContentType: obj.ContentType,
|
|
||||||
ETag: obj.ETag,
|
|
||||||
LastModified: obj.LastModified,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: files,
|
|
||||||
Error: nil,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// getFileBatch batch func for getting object info
|
|
||||||
func getFileBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
|
||||||
log.Debug("getFileBatch: ", k.Keys())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
|
|
||||||
s3Client, ok := c.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, v := range k {
|
|
||||||
log.Debug("S3 call 'StatObject': ", v.String())
|
|
||||||
obj, err := s3Client.StatObject(context.Background(), bucketName, v.String(), minio.StatObjectOptions{})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: nil,
|
|
||||||
Error: err,
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: &File{
|
|
||||||
ID: obj.Key,
|
|
||||||
Size: obj.Size,
|
|
||||||
ContentType: obj.ContentType,
|
|
||||||
ETag: obj.ETag,
|
|
||||||
LastModified: obj.LastModified,
|
|
||||||
},
|
|
||||||
Error: nil,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// getDirsBatch batch func for getting dirs in a path
|
|
||||||
func getDirsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
|
||||||
log.Debug("getDirsBatch: ", k.Keys())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
|
|
||||||
loader, ok := c.Value("loader").(map[string]*dataloader.Loader)
|
|
||||||
if !ok {
|
|
||||||
return handleLoaderError(k, fmt.Errorf("Failed to get loader from context"))
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, v := range k {
|
|
||||||
path := v.String()
|
|
||||||
dirs := make([]Directory, 0)
|
|
||||||
|
|
||||||
if !strings.HasSuffix(path, "/") {
|
|
||||||
path += "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
thunk := loader["listObjects"].Load(c, dataloader.StringKey(path))
|
|
||||||
|
|
||||||
objects, _ := thunk()
|
|
||||||
|
|
||||||
// TODO: handle thunk error
|
|
||||||
|
|
||||||
for _, obj := range objects.([]minio.ObjectInfo) {
|
|
||||||
if obj.Err != nil {
|
|
||||||
// TODO: how to handle?
|
|
||||||
} else if strings.HasSuffix(obj.Key, "/") {
|
|
||||||
dirs = append(dirs, Directory{
|
|
||||||
ID: obj.Key,
|
|
||||||
Name: filepath.Base(obj.Key),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: dirs,
|
|
||||||
Error: nil,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// handleLoaderError helper func when the whole batch failed
|
|
||||||
func handleLoaderError(k dataloader.Keys, err error) []*dataloader.Result {
|
|
||||||
log.Error(err.Error())
|
|
||||||
var results []*dataloader.Result
|
|
||||||
for range k {
|
|
||||||
results = append(results, &dataloader.Result{
|
|
||||||
Data: nil,
|
|
||||||
Error: err,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return results
|
|
||||||
}
|
|
||||||
|
|
||||||
// createDataloader create all dataloaders and return a map of them plus a cache for objects
|
|
||||||
func createDataloader(config AppConfig) map[string]*dataloader.Loader {
|
|
||||||
loaderMap := make(map[string]*dataloader.Loader, 0)
|
|
||||||
|
|
||||||
loaderMap["getFiles"] = dataloader.NewBatchedLoader(
|
|
||||||
getFilesBatch,
|
|
||||||
dataloader.WithCache(newCache(config.CacheTTL, config.CacheCleanup)),
|
|
||||||
)
|
|
||||||
|
|
||||||
loaderMap["getFile"] = dataloader.NewBatchedLoader(
|
|
||||||
getFileBatch,
|
|
||||||
dataloader.WithCache(newCache(config.CacheTTL, config.CacheCleanup)),
|
|
||||||
)
|
|
||||||
|
|
||||||
loaderMap["listObjects"] = dataloader.NewBatchedLoader(
|
|
||||||
listObjectsBatch,
|
|
||||||
dataloader.WithCache(newCache(config.CacheTTL, config.CacheCleanup)),
|
|
||||||
)
|
|
||||||
|
|
||||||
loaderMap["listObjectsRecursive"] = dataloader.NewBatchedLoader(
|
|
||||||
listObjectsRecursiveBatch,
|
|
||||||
dataloader.WithCache(newCache(config.CacheTTL, config.CacheCleanup)),
|
|
||||||
)
|
|
||||||
|
|
||||||
loaderMap["getDirs"] = dataloader.NewBatchedLoader(
|
|
||||||
getDirsBatch,
|
|
||||||
dataloader.WithCache(newCache(config.CacheTTL, config.CacheCleanup)),
|
|
||||||
)
|
|
||||||
|
|
||||||
return loaderMap
|
|
||||||
}
|
|
||||||
9
internal/db/db.go
Normal file
9
internal/db/db.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import "context"
|
||||||
|
|
||||||
|
type DB interface {
|
||||||
|
Setup() error
|
||||||
|
CheckLogin(ctx context.Context, username, password string) (bool, error)
|
||||||
|
AddUser(ctx context.Context, username, password string) error
|
||||||
|
}
|
||||||
94
internal/db/mysql.go
Normal file
94
internal/db/mysql.go
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
package db
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
_ "embed"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
_ "github.com/go-sql-driver/mysql"
|
||||||
|
"golang.org/x/crypto/bcrypt"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed setup.sql
|
||||||
|
var setupSql string
|
||||||
|
|
||||||
|
const DB_NAME = "s3Browser"
|
||||||
|
|
||||||
|
type mysqlDB struct {
|
||||||
|
dbConn *sql.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewDB(dataSourceName string) (DB, error) {
|
||||||
|
db, err := sql.Open("mysql", dataSourceName)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
db.SetConnMaxLifetime(time.Minute * 3)
|
||||||
|
db.SetMaxOpenConns(10)
|
||||||
|
db.SetMaxIdleConns(10)
|
||||||
|
|
||||||
|
return &mysqlDB{
|
||||||
|
dbConn: db,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *mysqlDB) Setup() error {
|
||||||
|
tx, err := d.dbConn.Begin()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = tx.Exec(setupSql)
|
||||||
|
if err != nil {
|
||||||
|
tx.Rollback()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = tx.Commit()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *mysqlDB) CheckLogin(ctx context.Context, username, password string) (bool, error) {
|
||||||
|
rows, err := d.dbConn.QueryContext(ctx, "SELECT password FROM user WHERE username = ?", username)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !rows.Next() {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var passwordHash []byte
|
||||||
|
err = rows.Scan(&passwordHash)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if bcrypt.CompareHashAndPassword(passwordHash, []byte(password)) != nil {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *mysqlDB) AddUser(ctx context.Context, username, password string) error {
|
||||||
|
hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = d.dbConn.ExecContext(ctx, "INSERT INTO user (username,password) VALUES (?,?)", username, hash)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
11
internal/db/setup.sql
Normal file
11
internal/db/setup.sql
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
CREATE TABLE s3Browser.`user` (
|
||||||
|
id INT auto_increment NOT NULL,
|
||||||
|
username varchar(100) NOT NULL,
|
||||||
|
password varchar(60) NOT NULL,
|
||||||
|
CONSTRAINT user_PK PRIMARY KEY (id),
|
||||||
|
CONSTRAINT user_UN UNIQUE KEY (username)
|
||||||
|
)
|
||||||
|
|
||||||
|
ENGINE=InnoDB
|
||||||
|
DEFAULT CHARSET=utf8mb4
|
||||||
|
COLLATE=utf8mb4_general_ci;
|
||||||
@@ -1,25 +0,0 @@
|
|||||||
package s3browser
|
|
||||||
|
|
||||||
import "fmt"
|
|
||||||
|
|
||||||
type extendedError struct {
|
|
||||||
Message string
|
|
||||||
Code string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (err *extendedError) Error() string {
|
|
||||||
return err.Message
|
|
||||||
}
|
|
||||||
|
|
||||||
func (err *extendedError) Extensions() map[string]interface{} {
|
|
||||||
return map[string]interface{}{
|
|
||||||
"code": err.Code,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func extendError(code, format string, a ...interface{}) *extendedError {
|
|
||||||
return &extendedError{
|
|
||||||
Message: fmt.Sprintf(format, a...),
|
|
||||||
Code: code,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
29
internal/errors/errors.go
Normal file
29
internal/errors/errors.go
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
package errors
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrNotAuthenticated = ExtendError("UNAUTHENTICATED", "No valid authentication provided")
|
||||||
|
)
|
||||||
|
|
||||||
|
type ExtendedError struct {
|
||||||
|
Message string
|
||||||
|
Code string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (err *ExtendedError) Error() string {
|
||||||
|
return err.Message
|
||||||
|
}
|
||||||
|
|
||||||
|
func (err *ExtendedError) Extensions() map[string]interface{} {
|
||||||
|
return map[string]interface{}{
|
||||||
|
"code": err.Code,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ExtendError(code, format string, a ...interface{}) *ExtendedError {
|
||||||
|
return &ExtendedError{
|
||||||
|
Message: fmt.Sprintf(format, a...),
|
||||||
|
Code: code,
|
||||||
|
}
|
||||||
|
}
|
||||||
75
internal/gql/gql_test.go
Normal file
75
internal/gql/gql_test.go
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
package gql_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/gql"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
"github.com/graph-gophers/dataloader"
|
||||||
|
"github.com/graphql-go/graphql"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func setup(t *testing.T) (*assert.Assertions, context.Context, graphql.Schema) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
schema, _ := gql.GraphqlSchema()
|
||||||
|
|
||||||
|
s3, err := s3.NewMockS3([]string{"bucket1"})
|
||||||
|
assert.NoError(err)
|
||||||
|
ctx = context.WithValue(ctx, "s3Client", s3)
|
||||||
|
|
||||||
|
loader := loader.NewLoader(loader.CacheConfig{
|
||||||
|
ListObjectsLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
|
||||||
|
StatObjectLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListBucketsLoaderCache: &dataloader.NoCache{},
|
||||||
|
})
|
||||||
|
assert.NotNil(loader)
|
||||||
|
ctx = context.WithValue(ctx, "loader", loader)
|
||||||
|
|
||||||
|
return assert, ctx, schema
|
||||||
|
}
|
||||||
|
|
||||||
|
func do(ctx context.Context, schema graphql.Schema, query string) *graphql.Result {
|
||||||
|
params := graphql.Params{
|
||||||
|
Schema: schema,
|
||||||
|
RequestString: query,
|
||||||
|
Context: ctx,
|
||||||
|
}
|
||||||
|
r := graphql.Do(params)
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateSchema(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
assert.NotPanics(func() {
|
||||||
|
gql.GraphqlTypes()
|
||||||
|
})
|
||||||
|
|
||||||
|
var schema graphql.Schema
|
||||||
|
var err error
|
||||||
|
|
||||||
|
assert.NotPanics(func() {
|
||||||
|
schema, err = gql.GraphqlSchema()
|
||||||
|
})
|
||||||
|
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.NotNil(schema)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAuth(t *testing.T) {
|
||||||
|
assert, ctx, schema := setup(t)
|
||||||
|
|
||||||
|
r := do(ctx, schema, `
|
||||||
|
{
|
||||||
|
authorized
|
||||||
|
}
|
||||||
|
`)
|
||||||
|
t.Logf("Data: %v", r.Data)
|
||||||
|
assert.Len(r.Errors, 0)
|
||||||
|
}
|
||||||
@@ -1,26 +1,26 @@
|
|||||||
package s3browser
|
package gql
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/graph-gophers/dataloader"
|
|
||||||
"github.com/graphql-go/graphql"
|
"github.com/graphql-go/graphql"
|
||||||
"github.com/graphql-go/graphql/language/ast"
|
"github.com/graphql-go/graphql/language/ast"
|
||||||
|
|
||||||
|
helper "git.kapelle.org/niklas/s3browser/internal/helper"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
type LoginResult struct {
|
var typesInit bool = false
|
||||||
Token string `json:"token"`
|
|
||||||
Successful bool `json:"successful"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var graphqlDirType *graphql.Object
|
var graphqlDirType *graphql.Object
|
||||||
var graphqlFileType *graphql.Object
|
var graphqlFileType *graphql.Object
|
||||||
var graphqlLoginResultType *graphql.Object
|
var graphqlLoginResultType *graphql.Object
|
||||||
|
var objIDType *graphql.Scalar
|
||||||
|
|
||||||
// graphqlTypes create all graphql types and stores the in the global variables
|
//GraphqlTypes create all graphql types and stores the in the global variables
|
||||||
func graphqlTypes() {
|
func GraphqlTypes() {
|
||||||
|
|
||||||
var dateTimeType = graphql.NewScalar(graphql.ScalarConfig{
|
var dateTimeType = graphql.NewScalar(graphql.ScalarConfig{
|
||||||
Name: "DateTime",
|
Name: "DateTime",
|
||||||
@@ -47,7 +47,39 @@ func graphqlTypes() {
|
|||||||
ParseLiteral: func(valueAST ast.Value) interface{} {
|
ParseLiteral: func(valueAST ast.Value) interface{} {
|
||||||
switch valueAST := valueAST.(type) {
|
switch valueAST := valueAST.(type) {
|
||||||
case *ast.StringValue:
|
case *ast.StringValue:
|
||||||
return valueAST.Value
|
if tval, err := time.Parse(time.RFC3339, valueAST.Value); err != nil {
|
||||||
|
return nil
|
||||||
|
} else {
|
||||||
|
return tval
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
objIDType = graphql.NewScalar(graphql.ScalarConfig{
|
||||||
|
Name: "objID",
|
||||||
|
Description: `String representing a bucket, key and version combination.
|
||||||
|
Looks like this: "bucketName:/name/of/key" or "bucketName@version:/name/of/key"`,
|
||||||
|
Serialize: func(value interface{}) interface{} {
|
||||||
|
switch value := value.(type) {
|
||||||
|
case types.ID:
|
||||||
|
return value.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
return "INVALID"
|
||||||
|
},
|
||||||
|
ParseValue: func(value interface{}) interface{} {
|
||||||
|
switch tvalue := value.(type) {
|
||||||
|
case string:
|
||||||
|
return types.ParseID(tvalue)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
ParseLiteral: func(valueAST ast.Value) interface{} {
|
||||||
|
switch valueAST := valueAST.(type) {
|
||||||
|
case *ast.StringValue:
|
||||||
|
return types.ParseID(valueAST.Value)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
@@ -58,17 +90,17 @@ func graphqlTypes() {
|
|||||||
Description: "Represents a directory",
|
Description: "Represents a directory",
|
||||||
Fields: graphql.Fields{
|
Fields: graphql.Fields{
|
||||||
"id": &graphql.Field{
|
"id": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
"name": &graphql.Field{
|
"name": &graphql.Field{
|
||||||
Type: graphql.String,
|
Type: graphql.String,
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(Directory)
|
source, ok := p.Source.(types.Directory)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse source for resolve")
|
return nil, fmt.Errorf("Failed to parse source for resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
return filepath.Base(source.ID), nil
|
return filepath.Base(source.ID.Key), nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -79,18 +111,18 @@ func graphqlTypes() {
|
|||||||
Description: "Represents a file, not a directory",
|
Description: "Represents a file, not a directory",
|
||||||
Fields: graphql.Fields{
|
Fields: graphql.Fields{
|
||||||
"id": &graphql.Field{
|
"id": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
Description: "The uniqe ID of the file. Represents the path and the s3 key.",
|
Description: "The uniqe ID of the file. Represents the path and the s3 key.",
|
||||||
},
|
},
|
||||||
"name": &graphql.Field{
|
"name": &graphql.Field{
|
||||||
Type: graphql.String,
|
Type: graphql.String,
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(File)
|
source, ok := p.Source.(types.File)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse source for resolve")
|
return nil, fmt.Errorf("Failed to parse source for resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
return filepath.Base(source.ID), nil
|
return filepath.Base(source.ID.Key), nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"size": &graphql.Field{
|
"size": &graphql.Field{
|
||||||
@@ -140,15 +172,19 @@ func graphqlTypes() {
|
|||||||
"parent": &graphql.Field{
|
"parent": &graphql.Field{
|
||||||
Type: graphqlDirType,
|
Type: graphqlDirType,
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(File)
|
source, ok := p.Source.(types.File)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse Source for parent resolve")
|
return nil, fmt.Errorf("Failed to parse Source for parent resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
basename := getPathFromId(source.ID)
|
parent := source.ID.Parent()
|
||||||
|
|
||||||
return Directory{
|
if parent == nil {
|
||||||
ID: basename,
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return types.Directory{
|
||||||
|
ID: *source.ID.Parent(),
|
||||||
}, nil
|
}, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -158,43 +194,40 @@ func graphqlTypes() {
|
|||||||
graphqlDirType.AddFieldConfig("files", &graphql.Field{
|
graphqlDirType.AddFieldConfig("files", &graphql.Field{
|
||||||
Type: graphql.NewList(graphqlFileType),
|
Type: graphql.NewList(graphqlFileType),
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(Directory)
|
source, ok := p.Source.(types.Directory)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse Source for files resolve")
|
return nil, fmt.Errorf("Failed to parse Source for files resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
loader := p.Context.Value("loader").(map[string]*dataloader.Loader)
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
|
|
||||||
thunk := loader["getFiles"].Load(p.Context, dataloader.StringKey(nomalizeID(source.ID)))
|
return loader.GetFiles(p.Context, source.ID)
|
||||||
return thunk()
|
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
graphqlDirType.AddFieldConfig("directorys", &graphql.Field{
|
graphqlDirType.AddFieldConfig("directorys", &graphql.Field{
|
||||||
Type: graphql.NewList(graphqlDirType),
|
Type: graphql.NewList(graphqlDirType),
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(Directory)
|
source, ok := p.Source.(types.Directory)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse Source for directories resolve")
|
return nil, fmt.Errorf("Failed to parse Source for directories resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
loader := p.Context.Value("loader").(map[string]*dataloader.Loader)
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
thunk := loader["getDirs"].Load(p.Context, dataloader.StringKey(nomalizeID(source.ID)))
|
return loader.GetDirs(p.Context, source.ID)
|
||||||
|
|
||||||
return thunk()
|
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
graphqlDirType.AddFieldConfig("parent", &graphql.Field{
|
graphqlDirType.AddFieldConfig("parent", &graphql.Field{
|
||||||
Type: graphqlDirType,
|
Type: graphqlDirType,
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
source, ok := p.Source.(Directory)
|
source, ok := p.Source.(types.Directory)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse Source for directories resolve")
|
return nil, fmt.Errorf("Failed to parse Source for directories resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
return Directory{
|
return types.Directory{
|
||||||
ID: getParentDir(source.ID),
|
ID: helper.GetParentDir(source.ID),
|
||||||
}, nil
|
}, nil
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
@@ -214,26 +247,25 @@ func graphqlTypes() {
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
|
typesInit = true
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// graphqlTypes helper func for using the dataloader to get a file
|
//loadFile helper func for using the dataloader to get a file
|
||||||
func loadFile(p graphql.ResolveParams) (*File, error) {
|
func loadFile(p graphql.ResolveParams) (*types.File, error) {
|
||||||
source, ok := p.Source.(File)
|
source, ok := p.Source.(types.File)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse source for resolve")
|
return nil, fmt.Errorf("Failed to parse source for resolve")
|
||||||
}
|
}
|
||||||
|
|
||||||
loader := p.Context.Value("loader").(map[string]*dataloader.Loader)
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
|
|
||||||
thunk := loader["getFile"].Load(p.Context, dataloader.StringKey(nomalizeID(source.ID)))
|
file, err := loader.GetFile(p.Context, source.ID)
|
||||||
result, err := thunk()
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
file, ok := result.(*File)
|
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to load file")
|
return nil, fmt.Errorf("Failed to load file")
|
||||||
}
|
}
|
||||||
249
internal/gql/mutations.go
Normal file
249
internal/gql/mutations.go
Normal file
@@ -0,0 +1,249 @@
|
|||||||
|
package gql
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/db"
|
||||||
|
helper "git.kapelle.org/niklas/s3browser/internal/helper"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
func deleteMutation(ctx context.Context, id types.ID) error {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'RemoveObject': ", id)
|
||||||
|
// TODO: it is posible to remove multiple objects with a single call.
|
||||||
|
// Is it better to batch this?
|
||||||
|
err := s3Client.RemoveObject(ctx, id)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, id)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func copyMutation(ctx context.Context, src, dest types.ID) (*types.File, error) {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if dest is a file or a dir
|
||||||
|
if dest.IsDirectory() {
|
||||||
|
// create new dest id
|
||||||
|
// TODO: What if a file with this id already exists?
|
||||||
|
dest.Key += helper.GetFilenameFromKey(src.Key)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'CopyObject': ", src, "-->", dest)
|
||||||
|
err := s3Client.CopyObject(ctx, src, dest)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, dest)
|
||||||
|
|
||||||
|
return &types.File{
|
||||||
|
ID: dest,
|
||||||
|
}, nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func moveDirMutation(ctx context.Context, src, dest types.ID) ([]*types.File, error) {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !dest.IsDirectory() {
|
||||||
|
return nil, fmt.Errorf("Dest must be a directory")
|
||||||
|
}
|
||||||
|
|
||||||
|
loader, ok := ctx.Value("loader").(*loader.Loader)
|
||||||
|
|
||||||
|
// "move" all file inside dir
|
||||||
|
files, err := loader.GetFilesRecursive(ctx, src)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var result []*types.File
|
||||||
|
parent := src.Parent()
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
newID := types.ID{
|
||||||
|
Bucket: dest.Bucket,
|
||||||
|
Key: strings.Replace(file.ID.Key, parent.Key, dest.Key, 1),
|
||||||
|
}
|
||||||
|
newID.Normalize()
|
||||||
|
|
||||||
|
log.Debug("S3 'CopyObject': ", src, "-->", dest)
|
||||||
|
err := s3Client.CopyObject(ctx, file.ID, dest)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
// TODO: handle error
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteMutation(ctx, file.ID)
|
||||||
|
|
||||||
|
loader.InvalidedCacheForId(ctx, newID)
|
||||||
|
loader.InvalidedCacheForId(ctx, file.ID)
|
||||||
|
|
||||||
|
result = append(result, &types.File{
|
||||||
|
ID: newID,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
loader.InvalidedCacheForId(ctx, src)
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func moveFileMutation(ctx context.Context, src, dest types.ID) (*types.File, error) {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if dest is a file or a dir
|
||||||
|
if dest.IsDirectory() {
|
||||||
|
// create new dest id
|
||||||
|
// TODO: What if a file with this id already exists?
|
||||||
|
dest.Key += helper.GetFilenameFromKey(src.Key)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'CopyObject': ", src, "-->", dest)
|
||||||
|
// There is no (spoon) move. Only copy and delete
|
||||||
|
err := s3Client.CopyObject(ctx, src, dest)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = deleteMutation(ctx, src)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, dest)
|
||||||
|
|
||||||
|
return &types.File{
|
||||||
|
ID: dest,
|
||||||
|
}, nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func createDirectory(ctx context.Context, id types.ID) (*types.Directory, error) {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'PutObject': ", id)
|
||||||
|
err := s3Client.PutObject(ctx, id, strings.NewReader(""), 0) // TODO: s3client interface needs content type parameter
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, id)
|
||||||
|
|
||||||
|
return &types.Directory{
|
||||||
|
ID: id,
|
||||||
|
}, nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func deleteDirectory(ctx context.Context, id types.ID) error {
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("Failed to get s3Client from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
loader, ok := ctx.Value("loader").(*loader.Loader)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("Failed to get dataloader from context")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get all files inside the directory
|
||||||
|
|
||||||
|
files, err := loader.GetFilesRecursive(ctx, id)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete all child files
|
||||||
|
for _, file := range files {
|
||||||
|
s3Client.RemoveObject(ctx, file.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the dir had no children it exists as an object (object with "/" at the end).
|
||||||
|
// If it exists as an object and had children it will get delete once the last child has been deleted
|
||||||
|
// If it had no children we have to delete it manualy
|
||||||
|
// This is at least the behavior when working with minio as s3 backend
|
||||||
|
// TODO: check if this is normal behavior when working with s3
|
||||||
|
if len(files) == 0 {
|
||||||
|
log.Debug("S3 'RemoveObject': ", id)
|
||||||
|
err := s3Client.RemoveObject(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
loader.InvalidedCacheForId(ctx, id)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//login Checks for valid username password combination. Returns singed jwt string
|
||||||
|
func login(ctx context.Context, username, password string) (types.LoginResult, error) {
|
||||||
|
|
||||||
|
dbStore := ctx.Value("dbStore").(db.DB)
|
||||||
|
|
||||||
|
succes, err := dbStore.CheckLogin(ctx, username, password)
|
||||||
|
|
||||||
|
if !succes {
|
||||||
|
return types.LoginResult{
|
||||||
|
Successful: false,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
token := helper.CreateJWT(helper.CreateClaims(username))
|
||||||
|
|
||||||
|
tokenString, err := token.SignedString([]byte("TODO"))
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return types.LoginResult{
|
||||||
|
Successful: false,
|
||||||
|
}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return types.LoginResult{
|
||||||
|
Token: tokenString,
|
||||||
|
Successful: true,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
@@ -1,56 +1,62 @@
|
|||||||
package s3browser
|
package gql
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"github.com/graph-gophers/dataloader"
|
|
||||||
"github.com/graphql-go/graphql"
|
"github.com/graphql-go/graphql"
|
||||||
|
|
||||||
|
s3errors "git.kapelle.org/niklas/s3browser/internal/errors"
|
||||||
|
helper "git.kapelle.org/niklas/s3browser/internal/helper"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
// graphqlSchema generate the schema with its root query and mutation
|
//GraphqlSchema generate the schema with its root query and mutation
|
||||||
func graphqlSchema() (graphql.Schema, error) {
|
func GraphqlSchema() (graphql.Schema, error) {
|
||||||
|
|
||||||
|
if !typesInit {
|
||||||
|
GraphqlTypes()
|
||||||
|
}
|
||||||
|
|
||||||
queryFields := graphql.Fields{
|
queryFields := graphql.Fields{
|
||||||
"files": &graphql.Field{
|
"files": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphql.NewList(graphql.NewNonNull(graphqlFileType))),
|
Type: graphql.NewNonNull(graphql.NewList(graphql.NewNonNull(graphqlFileType))),
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"path": &graphql.ArgumentConfig{
|
"path": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.String),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
path, ok := p.Args["path"].(string)
|
path, ok := p.Args["path"].(*types.ID)
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, nil
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("querry 'files': ", path)
|
log.Debug("querry 'files': ", path)
|
||||||
|
|
||||||
loader := p.Context.Value("loader").(map[string]*dataloader.Loader)
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
thunk := loader["getFiles"].Load(p.Context, dataloader.StringKey(path))
|
return loader.GetFiles(p.Context, *path)
|
||||||
return thunk()
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"directorys": &graphql.Field{
|
"directories": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphql.NewList(graphql.NewNonNull(graphqlDirType))),
|
Type: graphql.NewNonNull(graphql.NewList(graphql.NewNonNull(graphqlDirType))),
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"path": &graphql.ArgumentConfig{
|
"path": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.String),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
path, ok := p.Args["path"].(string)
|
path, ok := p.Args["path"].(*types.ID)
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
@@ -58,32 +64,31 @@ func graphqlSchema() (graphql.Schema, error) {
|
|||||||
|
|
||||||
log.Debug("querry 'directorys': ", path)
|
log.Debug("querry 'directorys': ", path)
|
||||||
|
|
||||||
loader := p.Context.Value("loader").(map[string]*dataloader.Loader)
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
thunk := loader["getDirs"].Load(p.Context, dataloader.StringKey(path))
|
return loader.GetDirs(p.Context, *path)
|
||||||
return thunk()
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"file": &graphql.Field{
|
"file": &graphql.Field{
|
||||||
Type: graphqlFileType,
|
Type: graphqlFileType,
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"id": &graphql.ArgumentConfig{
|
"id": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
id, ok := p.Args["id"].(string)
|
id, ok := p.Args["id"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("querry 'file': ", id)
|
log.Debug("querry 'file': ", id)
|
||||||
|
|
||||||
return File{
|
return types.File{
|
||||||
ID: id,
|
ID: *id,
|
||||||
}, nil
|
}, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -92,11 +97,24 @@ func graphqlSchema() (graphql.Schema, error) {
|
|||||||
Type: graphql.NewNonNull(graphql.Boolean),
|
Type: graphql.NewNonNull(graphql.Boolean),
|
||||||
Description: "True if the user is authorized",
|
Description: "True if the user is authorized",
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
auth, _ := isAuth(p.Context)
|
auth := helper.IsAuthenticated(p.Context)
|
||||||
|
|
||||||
return auth, nil
|
return auth, nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
"buckets": &graphql.Field{
|
||||||
|
Name: "buckets",
|
||||||
|
Type: graphql.NewNonNull(graphql.NewList(graphql.String)),
|
||||||
|
Description: "List available buckets",
|
||||||
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
|
}
|
||||||
|
|
||||||
|
loader := p.Context.Value("loader").(*loader.Loader)
|
||||||
|
return loader.GetBuckets(p.Context)
|
||||||
|
},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
mutationFields := graphql.Fields{
|
mutationFields := graphql.Fields{
|
||||||
@@ -104,117 +122,146 @@ func graphqlSchema() (graphql.Schema, error) {
|
|||||||
Type: graphql.String,
|
Type: graphql.String,
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"id": &graphql.ArgumentConfig{
|
"id": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
id, ok := p.Args["id"].(string)
|
id, ok := p.Args["id"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("mutation 'delete': ", id)
|
log.Debug("mutation 'delete': ", id)
|
||||||
|
|
||||||
return id, deleteMutation(p.Context, id)
|
return id, deleteMutation(p.Context, *id)
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"copy": &graphql.Field{
|
"copy": &graphql.Field{
|
||||||
Type: graphqlFileType,
|
Type: graphqlFileType,
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"src": &graphql.ArgumentConfig{
|
"src": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
"dest": &graphql.ArgumentConfig{
|
"dest": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
src, ok := p.Args["src"].(string)
|
src, ok := p.Args["src"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
dest, ok := p.Args["dest"].(string)
|
dest, ok := p.Args["dest"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("mutation 'copy': ", src, "-->", dest)
|
log.Debug("mutation 'copy': ", src, "-->", dest)
|
||||||
|
|
||||||
return copyMutation(p.Context, src, dest)
|
return copyMutation(p.Context, *src, *dest)
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"move": &graphql.Field{
|
"move": &graphql.Field{
|
||||||
Type: graphqlFileType,
|
Type: graphql.NewNonNull(graphqlFileType),
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"src": &graphql.ArgumentConfig{
|
"src": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
"dest": &graphql.ArgumentConfig{
|
"dest": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
src, ok := p.Args["src"].(string)
|
src, ok := p.Args["src"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
dest, ok := p.Args["dest"].(string)
|
dest, ok := p.Args["dest"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("mutation 'move': ", src, "-->", dest)
|
log.Debug("mutation 'move': ", src, "-->", dest)
|
||||||
|
|
||||||
return moveMutation(p.Context, src, dest)
|
return moveFileMutation(p.Context, *src, *dest)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"moveDir": &graphql.Field{
|
||||||
|
Type: graphql.NewNonNull(graphql.NewList(graphqlFileType)),
|
||||||
|
Args: graphql.FieldConfigArgument{
|
||||||
|
"src": &graphql.ArgumentConfig{
|
||||||
|
Type: graphql.NewNonNull(objIDType),
|
||||||
|
},
|
||||||
|
"dest": &graphql.ArgumentConfig{
|
||||||
|
Type: graphql.NewNonNull(objIDType),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
|
}
|
||||||
|
|
||||||
|
src, ok := p.Args["src"].(*types.ID)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
|
}
|
||||||
|
dest, ok := p.Args["dest"].(*types.ID)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("mutation 'moveDir': ", src, "-->", dest)
|
||||||
|
|
||||||
|
return moveDirMutation(p.Context, *src, *dest)
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"createDir": &graphql.Field{
|
"createDir": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphqlDirType),
|
Type: graphql.NewNonNull(graphqlDirType),
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"path": &graphql.ArgumentConfig{
|
"path": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
path, ok := p.Args["path"].(string)
|
path, ok := p.Args["path"].(*types.ID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("mutation 'createDir': ", path)
|
log.Debug("mutation 'createDir': ", path)
|
||||||
|
|
||||||
return createDirectory(p.Context, path)
|
return createDirectory(p.Context, *path)
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"deleteDir": &graphql.Field{
|
"deleteDir": &graphql.Field{
|
||||||
Type: graphql.NewNonNull(graphql.String),
|
Type: graphql.NewNonNull(graphql.String),
|
||||||
Args: graphql.FieldConfigArgument{
|
Args: graphql.FieldConfigArgument{
|
||||||
"path": &graphql.ArgumentConfig{
|
"path": &graphql.ArgumentConfig{
|
||||||
Type: graphql.NewNonNull(graphql.ID),
|
Type: graphql.NewNonNull(objIDType),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
|
||||||
if is, err := isAuth(p.Context); !is {
|
if !helper.IsAuthenticated(p.Context) {
|
||||||
return nil, err
|
return nil, s3errors.ErrNotAuthenticated
|
||||||
}
|
}
|
||||||
|
|
||||||
path, ok := p.Args["path"].(string)
|
path, ok := p.Args["path"].(*types.ID)
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("Failed to parse args")
|
return nil, fmt.Errorf("Failed to parse args")
|
||||||
@@ -222,7 +269,7 @@ func graphqlSchema() (graphql.Schema, error) {
|
|||||||
|
|
||||||
log.Debug("mutation 'deleteDir': ", path)
|
log.Debug("mutation 'deleteDir': ", path)
|
||||||
|
|
||||||
return path, deleteDirectory(p.Context, path)
|
return path, deleteDirectory(p.Context, *path)
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"login": &graphql.Field{
|
"login": &graphql.Field{
|
||||||
@@ -1,141 +0,0 @@
|
|||||||
package s3browser
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/golang-jwt/jwt"
|
|
||||||
"github.com/graph-gophers/dataloader"
|
|
||||||
"github.com/minio/minio-go/v7"
|
|
||||||
log "github.com/sirupsen/logrus"
|
|
||||||
)
|
|
||||||
|
|
||||||
func invalidateCache(ctx context.Context, id string) error {
|
|
||||||
loader, ok := ctx.Value("loader").(map[string]*dataloader.Loader)
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get loader from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Debug("Invalidate cache for id: ", id)
|
|
||||||
|
|
||||||
path := getPathFromId(id)
|
|
||||||
|
|
||||||
loader["getFile"].Clear(ctx, dataloader.StringKey(id))
|
|
||||||
loader["getFiles"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["listObjects"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["listObjectsRecursive"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func getPathFromId(id string) string {
|
|
||||||
dir := filepath.Dir(id)
|
|
||||||
|
|
||||||
if dir == "." {
|
|
||||||
return "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
return nomalizeID(dir + "/")
|
|
||||||
}
|
|
||||||
|
|
||||||
func getFilenameFromID(id string) string {
|
|
||||||
return filepath.Base(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
func invalidateCacheForDir(ctx context.Context, path string) error {
|
|
||||||
loader, ok := ctx.Value("loader").(map[string]*dataloader.Loader)
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get loader from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Debug("Invalidate cache for dir: ", path)
|
|
||||||
|
|
||||||
parent := getParentDir(path)
|
|
||||||
|
|
||||||
log.Debug("Cache clear dir: ", path, " parent: ", parent)
|
|
||||||
|
|
||||||
loader["getFile"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["listObjects"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["listObjectsRecursive"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["getFiles"].Clear(ctx, dataloader.StringKey(path))
|
|
||||||
loader["getDirs"].Clear(ctx, dataloader.StringKey(parent))
|
|
||||||
loader["listObjects"].Clear(ctx, dataloader.StringKey(parent))
|
|
||||||
loader["listObjectsRecursive"].Clear(ctx, dataloader.StringKey(parent))
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func deleteMultiple(ctx context.Context, s3Client minio.Client, ids []minio.ObjectInfo) error {
|
|
||||||
log.Debug("Delte multiple")
|
|
||||||
objectsCh := make(chan minio.ObjectInfo, 1)
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
defer close(objectsCh)
|
|
||||||
for _, id := range ids {
|
|
||||||
objectsCh <- id
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
for err := range s3Client.RemoveObjects(ctx, bucketName, objectsCh, minio.RemoveObjectsOptions{}) {
|
|
||||||
log.Error("Failed to delete object ", err.ObjectName, " because: ", err.Err.Error())
|
|
||||||
// TODO: error handel
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// nomalizeID makes sure there is a leading "/" in the id
|
|
||||||
func nomalizeID(id string) string {
|
|
||||||
if !strings.HasPrefix(id, "/") {
|
|
||||||
if id == "." {
|
|
||||||
return "/"
|
|
||||||
}
|
|
||||||
id = "/" + id
|
|
||||||
}
|
|
||||||
|
|
||||||
return id
|
|
||||||
}
|
|
||||||
|
|
||||||
func getParentDir(id string) string {
|
|
||||||
dirs := strings.Split(id, "/")
|
|
||||||
|
|
||||||
cut := 1
|
|
||||||
if strings.HasSuffix(id, "/") {
|
|
||||||
cut = 2
|
|
||||||
}
|
|
||||||
|
|
||||||
parent := strings.Join(dirs[:len(dirs)-cut], "/") + "/"
|
|
||||||
|
|
||||||
return nomalizeID(parent)
|
|
||||||
}
|
|
||||||
|
|
||||||
func isAuth(ctx context.Context) (bool, error) {
|
|
||||||
token, ok := ctx.Value("jwt").(*jwt.Token)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return false, extendError("UNAUTHORIZED", "Unauthorized")
|
|
||||||
}
|
|
||||||
|
|
||||||
if token.Valid {
|
|
||||||
return true, nil
|
|
||||||
} else {
|
|
||||||
return false, extendError("UNAUTHORIZED", "Unauthorized")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func createJWT(claims *JWTClaims) *jwt.Token {
|
|
||||||
|
|
||||||
claims.ExpiresAt = time.Now().Add(time.Hour * 24).Unix()
|
|
||||||
|
|
||||||
return jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
|
||||||
}
|
|
||||||
|
|
||||||
func createClaims(username string) *JWTClaims {
|
|
||||||
return &JWTClaims{
|
|
||||||
StandardClaims: jwt.StandardClaims{
|
|
||||||
Subject: username,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
67
internal/helper/helper.go
Normal file
67
internal/helper/helper.go
Normal file
@@ -0,0 +1,67 @@
|
|||||||
|
package helper
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/golang-jwt/jwt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GetFilenameFromKey(id string) string {
|
||||||
|
return filepath.Base(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetParentDir(id types.ID) types.ID {
|
||||||
|
dirs := strings.Split(id.Key, "/")
|
||||||
|
|
||||||
|
cut := 1
|
||||||
|
if strings.HasSuffix(id.Key, "/") {
|
||||||
|
cut = 2
|
||||||
|
}
|
||||||
|
|
||||||
|
parentKey := strings.Join(dirs[:len(dirs)-cut], "/") + "/"
|
||||||
|
|
||||||
|
parent := types.ID{
|
||||||
|
Bucket: id.Bucket,
|
||||||
|
Key: parentKey,
|
||||||
|
}
|
||||||
|
|
||||||
|
parent.Normalize()
|
||||||
|
|
||||||
|
return parent
|
||||||
|
}
|
||||||
|
|
||||||
|
func ObjInfoToFile(objInfo s3.Object, bucket string) *types.File {
|
||||||
|
return &types.File{
|
||||||
|
ID: objInfo.ID,
|
||||||
|
Name: objInfo.ID.Name(),
|
||||||
|
Size: objInfo.Size,
|
||||||
|
ContentType: objInfo.ContentType,
|
||||||
|
ETag: objInfo.ETag,
|
||||||
|
LastModified: objInfo.LastModified,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsAuthenticated(ctx context.Context) bool {
|
||||||
|
token, ok := ctx.Value("jwt").(*jwt.Token)
|
||||||
|
return (ok && token.Valid)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateJWT(claims *types.JWTClaims) *jwt.Token {
|
||||||
|
|
||||||
|
claims.ExpiresAt = time.Now().Add(time.Hour * 24).Unix()
|
||||||
|
|
||||||
|
return jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateClaims(username string) *types.JWTClaims {
|
||||||
|
return &types.JWTClaims{
|
||||||
|
StandardClaims: jwt.StandardClaims{
|
||||||
|
Subject: username,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
//go:build !prod
|
//go:build !prod
|
||||||
// +build !prod
|
// +build !prod
|
||||||
|
|
||||||
package s3browser
|
package httpserver
|
||||||
|
|
||||||
import "github.com/gorilla/mux"
|
import "github.com/gorilla/mux"
|
||||||
|
|
||||||
@@ -1,12 +1,10 @@
|
|||||||
package s3browser
|
package httpserver
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"mime"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"path/filepath"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/golang-jwt/jwt"
|
"github.com/golang-jwt/jwt"
|
||||||
@@ -15,20 +13,24 @@ import (
|
|||||||
"github.com/graphql-go/graphql"
|
"github.com/graphql-go/graphql"
|
||||||
"github.com/graphql-go/graphql/gqlerrors"
|
"github.com/graphql-go/graphql/gqlerrors"
|
||||||
"github.com/graphql-go/handler"
|
"github.com/graphql-go/handler"
|
||||||
"github.com/minio/minio-go/v7"
|
|
||||||
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
|
helper "git.kapelle.org/niklas/s3browser/internal/helper"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
type JWTClaims struct {
|
var (
|
||||||
jwt.StandardClaims
|
tokenExp = int64((time.Hour * 23).Seconds())
|
||||||
}
|
)
|
||||||
|
|
||||||
type CookieExtractor struct {
|
type cookieExtractor struct {
|
||||||
Name string
|
Name string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *CookieExtractor) ExtractToken(req *http.Request) (string, error) {
|
func (c *cookieExtractor) ExtractToken(req *http.Request) (string, error) {
|
||||||
cookie, err := req.Cookie(c.Name)
|
cookie, err := req.Cookie(c.Name)
|
||||||
|
|
||||||
if err == nil && len(cookie.Value) != 0 {
|
if err == nil && len(cookie.Value) != 0 {
|
||||||
@@ -36,11 +38,10 @@ func (c *CookieExtractor) ExtractToken(req *http.Request) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return "", jwtRequest.ErrNoTokenInRequest
|
return "", jwtRequest.ErrNoTokenInRequest
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// initHttp setup and start the http server. Blocking
|
// InitHttp setup and start the http server. Blocking
|
||||||
func initHttp(resolveContext context.Context, schema graphql.Schema, address string) error {
|
func InitHttp(resolveContext context.Context, schema graphql.Schema, address string) error {
|
||||||
r := mux.NewRouter()
|
r := mux.NewRouter()
|
||||||
|
|
||||||
gqlHandler := handler.New(&handler.Config{
|
gqlHandler := handler.New(&handler.Config{
|
||||||
@@ -64,8 +65,8 @@ func initHttp(resolveContext context.Context, schema graphql.Schema, address str
|
|||||||
|
|
||||||
parsedToken, err := jwtRequest.ParseFromRequestWithClaims(r, jwtRequest.MultiExtractor{
|
parsedToken, err := jwtRequest.ParseFromRequestWithClaims(r, jwtRequest.MultiExtractor{
|
||||||
jwtRequest.AuthorizationHeaderExtractor,
|
jwtRequest.AuthorizationHeaderExtractor,
|
||||||
&CookieExtractor{Name: "jwt"},
|
&cookieExtractor{Name: "jwt"},
|
||||||
}, &JWTClaims{}, jwtKeyFunc)
|
}, &types.JWTClaims{}, jwtKeyFunc)
|
||||||
|
|
||||||
if err == nil && parsedToken.Valid {
|
if err == nil && parsedToken.Valid {
|
||||||
newRequest := r.WithContext(context.WithValue(r.Context(), "jwt", parsedToken))
|
newRequest := r.WithContext(context.WithValue(r.Context(), "jwt", parsedToken))
|
||||||
@@ -79,14 +80,17 @@ func initHttp(resolveContext context.Context, schema graphql.Schema, address str
|
|||||||
|
|
||||||
r.HandleFunc("/api/graphql", func(rw http.ResponseWriter, r *http.Request) {
|
r.HandleFunc("/api/graphql", func(rw http.ResponseWriter, r *http.Request) {
|
||||||
token := r.Context().Value("jwt")
|
token := r.Context().Value("jwt")
|
||||||
|
refreshTokenIfNeeded(rw, r)
|
||||||
gqlHandler.ContextHandler(context.WithValue(resolveContext, "jwt", token), rw, r)
|
gqlHandler.ContextHandler(context.WithValue(resolveContext, "jwt", token), rw, r)
|
||||||
})
|
})
|
||||||
|
|
||||||
r.HandleFunc("/api/file", func(rw http.ResponseWriter, r *http.Request) {
|
r.HandleFunc("/api/file", func(rw http.ResponseWriter, r *http.Request) {
|
||||||
|
refreshTokenIfNeeded(rw, r)
|
||||||
httpGetFile(resolveContext, rw, r)
|
httpGetFile(resolveContext, rw, r)
|
||||||
}).Methods("GET")
|
}).Methods("GET")
|
||||||
|
|
||||||
r.HandleFunc("/api/file", func(rw http.ResponseWriter, r *http.Request) {
|
r.HandleFunc("/api/file", func(rw http.ResponseWriter, r *http.Request) {
|
||||||
|
refreshTokenIfNeeded(rw, r)
|
||||||
httpPostFile(resolveContext, rw, r)
|
httpPostFile(resolveContext, rw, r)
|
||||||
}).Methods("POST")
|
}).Methods("POST")
|
||||||
|
|
||||||
@@ -94,8 +98,6 @@ func initHttp(resolveContext context.Context, schema graphql.Schema, address str
|
|||||||
|
|
||||||
r.HandleFunc("/api/logout", logout).Methods("POST")
|
r.HandleFunc("/api/logout", logout).Methods("POST")
|
||||||
|
|
||||||
r.HandleFunc("/api/refresh", refreshToken).Methods("POST")
|
|
||||||
|
|
||||||
// Init the embedded static files
|
// Init the embedded static files
|
||||||
initStatic(r)
|
initStatic(r)
|
||||||
|
|
||||||
@@ -103,18 +105,27 @@ func initHttp(resolveContext context.Context, schema graphql.Schema, address str
|
|||||||
}
|
}
|
||||||
|
|
||||||
func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
||||||
if is, _ := isAuth(r.Context()); !is {
|
if !helper.IsAuthenticated(r.Context()) {
|
||||||
rw.WriteHeader(http.StatusUnauthorized)
|
rw.WriteHeader(http.StatusUnauthorized)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
s3Client := ctx.Value("s3Client").(*minio.Client)
|
s3Client := ctx.Value("s3Client").(s3.S3Service)
|
||||||
id := r.URL.Query().Get("id")
|
idString := r.URL.Query().Get("id")
|
||||||
|
|
||||||
log.Debug("S3 call 'StatObject': ", id)
|
id := types.ParseID(idString)
|
||||||
objInfo, err := s3Client.StatObject(context.Background(), bucketName, id, minio.GetObjectOptions{})
|
|
||||||
|
if id == nil {
|
||||||
|
// Failed to parse ID
|
||||||
|
rw.WriteHeader(http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'StatObject': ", id)
|
||||||
|
objInfo, err := s3Client.StatObject(context.Background(), *id)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
log.Error("Failed to get object info: ", err)
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
rw.WriteHeader(http.StatusInternalServerError)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -125,16 +136,17 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Debug("S3 call 'GetObject': ", id)
|
log.Debug("S3 'GetObject': ", id)
|
||||||
obj, err := s3Client.GetObject(context.Background(), bucketName, id, minio.GetObjectOptions{})
|
obj, err := s3Client.GetObject(context.Background(), *id)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
log.Error("Failed to get object: ", err)
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
rw.WriteHeader(http.StatusInternalServerError)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
rw.Header().Set("Cache-Control", "must-revalidate")
|
rw.Header().Set("Cache-Control", "must-revalidate")
|
||||||
rw.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", filepath.Base((objInfo.Key))))
|
rw.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", id.Name()))
|
||||||
rw.Header().Set("Content-Type", objInfo.ContentType)
|
rw.Header().Set("Content-Type", objInfo.ContentType)
|
||||||
rw.Header().Set("ETag", objInfo.ETag)
|
rw.Header().Set("ETag", objInfo.ETag)
|
||||||
|
|
||||||
@@ -147,32 +159,38 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func httpPostFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
func httpPostFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
|
||||||
if is, _ := isAuth(r.Context()); !is {
|
if !helper.IsAuthenticated(r.Context()) {
|
||||||
rw.WriteHeader(http.StatusUnauthorized)
|
rw.WriteHeader(http.StatusUnauthorized)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
s3Client := ctx.Value("s3Client").(*minio.Client)
|
s3Client := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
id := r.URL.Query().Get("id")
|
idString := r.URL.Query().Get("id")
|
||||||
|
|
||||||
log.Debug("Upload file: ", id)
|
id := types.ParseID(idString)
|
||||||
|
|
||||||
contentType := r.Header.Get("Content-Type")
|
if id == nil {
|
||||||
mimeType, _, _ := mime.ParseMediaType(contentType)
|
// Failed to parse ID
|
||||||
|
rw.WriteHeader(http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
log.Debug("S3 call 'PutObject': ", id)
|
id.Normalize()
|
||||||
info, err := s3Client.PutObject(context.Background(), bucketName, id, r.Body, r.ContentLength, minio.PutObjectOptions{
|
|
||||||
ContentType: mimeType,
|
// contentType := r.Header.Get("Content-Type")
|
||||||
})
|
// mimeType, _, _ := mime.ParseMediaType(contentType)
|
||||||
|
|
||||||
|
log.Debug("S3 'PutObject': ", id)
|
||||||
|
err := s3Client.PutObject(context.Background(), *id, r.Body, r.ContentLength) // TODO: put content type
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
rw.WriteHeader(http.StatusInternalServerError)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Invalidate cache
|
loader := ctx.Value("loader").(*loader.Loader)
|
||||||
invalidateCache(ctx, info.Key)
|
loader.InvalidedCacheForId(ctx, *id)
|
||||||
|
|
||||||
rw.WriteHeader(http.StatusCreated)
|
rw.WriteHeader(http.StatusCreated)
|
||||||
}
|
}
|
||||||
@@ -196,7 +214,7 @@ func setLoginCookie(rw http.ResponseWriter, r *http.Request) {
|
|||||||
|
|
||||||
tokenString := string(body)
|
tokenString := string(body)
|
||||||
|
|
||||||
token, err := jwt.ParseWithClaims(tokenString, &JWTClaims{}, jwtKeyFunc)
|
token, err := jwt.ParseWithClaims(tokenString, &types.JWTClaims{}, jwtKeyFunc)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
rw.WriteHeader(http.StatusInternalServerError)
|
||||||
@@ -208,7 +226,7 @@ func setLoginCookie(rw http.ResponseWriter, r *http.Request) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
claims, ok := token.Claims.(*JWTClaims)
|
claims, ok := token.Claims.(*types.JWTClaims)
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
rw.WriteHeader(http.StatusInternalServerError)
|
||||||
@@ -245,32 +263,31 @@ func logout(rw http.ResponseWriter, r *http.Request) {
|
|||||||
rw.WriteHeader(http.StatusNoContent)
|
rw.WriteHeader(http.StatusNoContent)
|
||||||
}
|
}
|
||||||
|
|
||||||
func refreshToken(rw http.ResponseWriter, r *http.Request) {
|
func refreshTokenIfNeeded(rw http.ResponseWriter, r *http.Request) {
|
||||||
if is, _ := isAuth(r.Context()); !is {
|
currentToken, ok := r.Context().Value("jwt").(*jwt.Token)
|
||||||
rw.WriteHeader(http.StatusUnauthorized)
|
|
||||||
|
if !ok && currentToken == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
oldToken, ok := r.Context().Value("jwt").(*jwt.Token)
|
claims, ok := currentToken.Claims.(*types.JWTClaims)
|
||||||
|
|
||||||
if !ok {
|
if !ok {
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
log.Error("Failed to refresh JWT")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
claims, ok := oldToken.Claims.(*JWTClaims)
|
// Refresh only if token older than 1 hour
|
||||||
|
if (claims.ExpiresAt - time.Now().Unix()) > tokenExp {
|
||||||
if !ok {
|
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
token := createJWT(claims)
|
newToken := helper.CreateJWT(claims)
|
||||||
|
|
||||||
tokenString, err := token.SignedString([]byte("TODO"))
|
tokenString, err := newToken.SignedString([]byte("TODO"))
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
rw.WriteHeader(http.StatusInternalServerError)
|
log.Error("Failed to refresh JWT")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -284,4 +301,6 @@ func refreshToken(rw http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
http.SetCookie(rw, cookie)
|
http.SetCookie(rw, cookie)
|
||||||
|
|
||||||
|
log.Debug("Refreshed JWT")
|
||||||
}
|
}
|
||||||
@@ -1,10 +1,11 @@
|
|||||||
//go:build prod
|
//go:build prod
|
||||||
// +build prod
|
// +build prod
|
||||||
|
|
||||||
package s3browser
|
package httpserver
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"embed"
|
"embed"
|
||||||
|
"github.com/gorilla/mux"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
@@ -26,7 +27,7 @@ func (spa *spaFileSystem) Open(name string) (http.File, error) {
|
|||||||
return f, err
|
return f, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func initStatic(e *mux.Router) {
|
func initStatic(r *mux.Router) {
|
||||||
staticFS, _ := fs.Sub(staticFiles, "static")
|
staticFS, _ := fs.Sub(staticFiles, "static")
|
||||||
r.Handle("/", http.FileServer(&spaFileSystem{http.FS(staticFS)}))
|
r.PathPrefix("/").Handler(http.FileServer(&spaFileSystem{http.FS(staticFS)}))
|
||||||
}
|
}
|
||||||
134
internal/loader/batch.go
Normal file
134
internal/loader/batch.go
Normal file
@@ -0,0 +1,134 @@
|
|||||||
|
package loader
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/graph-gophers/dataloader"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
)
|
||||||
|
|
||||||
|
// listObjectsBatch batch func for calling s3.ListObjects()
|
||||||
|
func listObjectsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
||||||
|
var results []*dataloader.Result
|
||||||
|
|
||||||
|
s3Client, ok := c.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range k {
|
||||||
|
id := v.Raw().(types.ID)
|
||||||
|
objects, err := s3Client.ListObjects(c, id)
|
||||||
|
if err != nil {
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: nil,
|
||||||
|
Error: err,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: objects,
|
||||||
|
Error: nil,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
|
// listObjectsRecursiveBatch just like listObjectsBatch but with recursive set to true
|
||||||
|
func listObjectsRecursiveBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
||||||
|
var results []*dataloader.Result
|
||||||
|
|
||||||
|
s3Client, ok := c.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range k {
|
||||||
|
id := v.Raw().(types.ID)
|
||||||
|
objects, err := s3Client.ListObjectsRecursive(c, id)
|
||||||
|
if err != nil {
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: nil,
|
||||||
|
Error: err,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: objects,
|
||||||
|
Error: nil,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
|
func listBucketsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
|
||||||
|
var results []*dataloader.Result
|
||||||
|
|
||||||
|
s3Client, ok := c.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Debug("S3 'ListBuckets'")
|
||||||
|
buckets, err := s3Client.ListBuckets(c)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return handleLoaderError(k, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
result := &dataloader.Result{
|
||||||
|
Data: buckets,
|
||||||
|
Error: nil,
|
||||||
|
}
|
||||||
|
|
||||||
|
for range k {
|
||||||
|
results = append(results, result)
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
|
func statObjectBatch(ctx context.Context, k dataloader.Keys) []*dataloader.Result {
|
||||||
|
log.Debug("statObjectBatch")
|
||||||
|
|
||||||
|
var results []*dataloader.Result
|
||||||
|
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range k {
|
||||||
|
id := v.Raw().(types.ID)
|
||||||
|
log.Debug("S3 'StatObject': ", id)
|
||||||
|
stat, err := s3Client.StatObject(ctx, id)
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: stat,
|
||||||
|
Error: err,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
|
|
||||||
|
// handleLoaderError helper func when the whole batch failed
|
||||||
|
func handleLoaderError(k dataloader.Keys, err error) []*dataloader.Result {
|
||||||
|
log.Error(err.Error())
|
||||||
|
var results []*dataloader.Result
|
||||||
|
for range k {
|
||||||
|
results = append(results, &dataloader.Result{
|
||||||
|
Data: nil,
|
||||||
|
Error: err,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return results
|
||||||
|
}
|
||||||
164
internal/loader/loader.go
Normal file
164
internal/loader/loader.go
Normal file
@@ -0,0 +1,164 @@
|
|||||||
|
package loader
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/cache"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/helper"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/graph-gophers/dataloader"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Loader struct {
|
||||||
|
listObjectsLoader *dataloader.Loader
|
||||||
|
listObjectsRecursiveLoader *dataloader.Loader
|
||||||
|
statObjectLoader *dataloader.Loader
|
||||||
|
listBucketsLoader *dataloader.Loader
|
||||||
|
|
||||||
|
listObjectsLoaderCache cache.S3Cache
|
||||||
|
listObjectsRecursiveLoaderCache cache.S3Cache
|
||||||
|
statObjectLoaderCache cache.S3Cache
|
||||||
|
listBucketsLoaderCache cache.S3Cache
|
||||||
|
}
|
||||||
|
|
||||||
|
type CacheConfig struct {
|
||||||
|
ListObjectsLoaderCache cache.S3Cache
|
||||||
|
ListObjectsRecursiveLoaderCache cache.S3Cache
|
||||||
|
StatObjectLoaderCache cache.S3Cache
|
||||||
|
ListBucketsLoaderCache cache.S3Cache
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLoader(cacheConfig CacheConfig) *Loader {
|
||||||
|
listObjectsLoaderCache := cacheConfig.ListObjectsLoaderCache
|
||||||
|
listObjectsRecursiveLoaderCache := cacheConfig.ListObjectsRecursiveLoaderCache
|
||||||
|
statObjectLoaderCache := cacheConfig.StatObjectLoaderCache
|
||||||
|
listBucketsLoaderCache := cacheConfig.ListBucketsLoaderCache
|
||||||
|
|
||||||
|
return &Loader{
|
||||||
|
listObjectsLoader: dataloader.NewBatchedLoader(
|
||||||
|
listObjectsBatch,
|
||||||
|
dataloader.WithCache(listObjectsLoaderCache),
|
||||||
|
),
|
||||||
|
listObjectsLoaderCache: listObjectsLoaderCache,
|
||||||
|
|
||||||
|
listObjectsRecursiveLoader: dataloader.NewBatchedLoader(
|
||||||
|
listObjectsRecursiveBatch,
|
||||||
|
dataloader.WithCache(listObjectsRecursiveLoaderCache),
|
||||||
|
),
|
||||||
|
listObjectsRecursiveLoaderCache: listObjectsRecursiveLoaderCache,
|
||||||
|
|
||||||
|
statObjectLoader: dataloader.NewBatchedLoader(
|
||||||
|
statObjectBatch,
|
||||||
|
dataloader.WithCache(statObjectLoaderCache),
|
||||||
|
),
|
||||||
|
statObjectLoaderCache: statObjectLoaderCache,
|
||||||
|
|
||||||
|
listBucketsLoader: dataloader.NewBatchedLoader(
|
||||||
|
listBucketsBatch,
|
||||||
|
dataloader.WithCache(listBucketsLoaderCache),
|
||||||
|
),
|
||||||
|
listBucketsLoaderCache: listBucketsLoaderCache,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) GetFiles(ctx context.Context, path types.ID) ([]types.File, error) {
|
||||||
|
thunk := l.listObjectsLoader.Load(ctx, path)
|
||||||
|
objects, err := thunk()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var files []types.File
|
||||||
|
|
||||||
|
for _, obj := range objects.([]s3.Object) {
|
||||||
|
if !obj.ID.IsDirectory() {
|
||||||
|
files = append(files, *helper.ObjInfoToFile(obj, path.Bucket))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) GetFile(ctx context.Context, id types.ID) (*types.File, error) {
|
||||||
|
thunk := l.statObjectLoader.Load(ctx, id)
|
||||||
|
|
||||||
|
result, err := thunk()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
objInfo, ok := result.(*s3.Object)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("Failed to stats object")
|
||||||
|
}
|
||||||
|
|
||||||
|
return helper.ObjInfoToFile(*objInfo, id.Bucket), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) GetDirs(ctx context.Context, path types.ID) ([]types.Directory, error) {
|
||||||
|
thunk := l.listObjectsLoader.Load(ctx, path)
|
||||||
|
|
||||||
|
result, err := thunk()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var dirs []types.Directory
|
||||||
|
for _, obj := range result.([]s3.Object) {
|
||||||
|
if obj.ID.IsDirectory() {
|
||||||
|
dirs = append(dirs, types.Directory{
|
||||||
|
ID: obj.ID,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return dirs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) GetBuckets(ctx context.Context) ([]string, error) {
|
||||||
|
thunk := l.listBucketsLoader.Load(ctx, dataloader.StringKey(""))
|
||||||
|
|
||||||
|
result, err := thunk()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return result.([]string), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) GetFilesRecursive(ctx context.Context, path types.ID) ([]types.File, error) {
|
||||||
|
thunk := l.listObjectsRecursiveLoader.Load(ctx, path)
|
||||||
|
result, err := thunk()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
objects := result.([]s3.Object)
|
||||||
|
|
||||||
|
var files []types.File
|
||||||
|
for _, obj := range objects {
|
||||||
|
files = append(files, *helper.ObjInfoToFile(obj, path.Bucket))
|
||||||
|
}
|
||||||
|
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Loader) InvalidedCacheForId(ctx context.Context, id types.ID) {
|
||||||
|
parent := id.Parent()
|
||||||
|
|
||||||
|
l.statObjectLoader.Clear(ctx, id)
|
||||||
|
|
||||||
|
// Code below is useless for now until we use a propper cache for "listObjectsLoader" and "listObjectsRecursiveLoader"
|
||||||
|
// TODO: implement cache invalidation for "listObjectsLoader" and "listObjectsRecursiveLoader"
|
||||||
|
l.listObjectsLoader.Clear(ctx, id).Clear(ctx, parent)
|
||||||
|
|
||||||
|
// Remove up from recursive list
|
||||||
|
for rParent := parent; rParent != nil; rParent = rParent.Parent() {
|
||||||
|
l.listObjectsRecursiveLoader.Clear(ctx, rParent)
|
||||||
|
}
|
||||||
|
}
|
||||||
105
internal/loader/loader_test.go
Normal file
105
internal/loader/loader_test.go
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
package loader_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/graph-gophers/dataloader"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func setup(t *testing.T) (context.Context, *loader.Loader, *assert.Assertions) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
s3, _ := s3.NewMockS3([]string{"bucket1", "bucket2"})
|
||||||
|
loader := loader.NewLoader(loader.CacheConfig{
|
||||||
|
ListObjectsLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
|
||||||
|
StatObjectLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListBucketsLoaderCache: &dataloader.NoCache{},
|
||||||
|
})
|
||||||
|
|
||||||
|
fillS3(s3)
|
||||||
|
ctx := context.WithValue(context.Background(), "s3Client", s3)
|
||||||
|
|
||||||
|
return ctx, loader, assert
|
||||||
|
}
|
||||||
|
|
||||||
|
func fillS3(s3 s3.S3Service) {
|
||||||
|
ctx := context.Background()
|
||||||
|
length := int64(len("content"))
|
||||||
|
|
||||||
|
for _, v := range []string{
|
||||||
|
"bucket1:/file1", "bucket1:/file2", "bucket1:/dir1/file1",
|
||||||
|
"bucket1:/dir1/file2", "bucket1:/dir2/file1", "bucket1:/dir1/sub1/file1",
|
||||||
|
"bucket1:/dir1/sub1/file2",
|
||||||
|
} {
|
||||||
|
s3.PutObject(ctx, *types.ParseID(v), strings.NewReader("content"), length)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateLoader(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
loader := loader.NewLoader(loader.CacheConfig{
|
||||||
|
ListObjectsLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
|
||||||
|
StatObjectLoaderCache: &dataloader.NoCache{},
|
||||||
|
ListBucketsLoaderCache: &dataloader.NoCache{},
|
||||||
|
})
|
||||||
|
|
||||||
|
assert.NotNil(loader)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetBuckets(t *testing.T) {
|
||||||
|
ctx, loader, assert := setup(t)
|
||||||
|
|
||||||
|
buckets, err := loader.GetBuckets(ctx)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Len(buckets, 2)
|
||||||
|
assert.Contains(buckets, "bucket1")
|
||||||
|
assert.Contains(buckets, "bucket2")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetFile(t *testing.T) {
|
||||||
|
ctx, loader, assert := setup(t)
|
||||||
|
|
||||||
|
file, err := loader.GetFile(ctx, *types.ParseID("bucket1:/dir1/file1"))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
assert.Equal("bucket1:/dir1/file1", file.ID.String())
|
||||||
|
assert.Equal("file1", file.Name)
|
||||||
|
assert.Equal(int64(len("content")), file.Size)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetFiles(t *testing.T) {
|
||||||
|
ctx, loader, assert := setup(t)
|
||||||
|
|
||||||
|
id := types.ParseID("bucket1:/")
|
||||||
|
|
||||||
|
files, err := loader.GetFiles(ctx, *id)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Len(files, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetDir(t *testing.T) {
|
||||||
|
ctx, loader, assert := setup(t)
|
||||||
|
|
||||||
|
id := types.ParseID("bucket1:/")
|
||||||
|
|
||||||
|
dirs, err := loader.GetDirs(ctx, *id)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Len(dirs, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test(t *testing.T) {
|
||||||
|
ctx, loader, assert := setup(t)
|
||||||
|
|
||||||
|
id := types.ParseID("bucket1:/dir1/")
|
||||||
|
|
||||||
|
files, err := loader.GetFilesRecursive(ctx, *id)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Len(files, 4)
|
||||||
|
}
|
||||||
@@ -1,217 +0,0 @@
|
|||||||
package s3browser
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/graph-gophers/dataloader"
|
|
||||||
"github.com/minio/minio-go/v7"
|
|
||||||
)
|
|
||||||
|
|
||||||
func deleteMutation(ctx context.Context, id string) error {
|
|
||||||
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get s3Client from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: it is posible to remove multiple objects with a single call.
|
|
||||||
// Is it better to batch this?
|
|
||||||
err := s3Client.RemoveObject(ctx, bucketName, id, minio.RemoveObjectOptions{})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Invalidate cache
|
|
||||||
return invalidateCache(ctx, nomalizeID(id))
|
|
||||||
}
|
|
||||||
|
|
||||||
func copyMutation(ctx context.Context, src, dest string) (*File, error) {
|
|
||||||
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("Failed to get s3Client from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if dest is a file or a dir
|
|
||||||
if strings.HasSuffix(dest, "/") {
|
|
||||||
// create new dest id
|
|
||||||
// TODO: What if a file with this id already exists?
|
|
||||||
dest += getFilenameFromID(src)
|
|
||||||
}
|
|
||||||
|
|
||||||
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
|
|
||||||
Bucket: bucketName,
|
|
||||||
Object: dest,
|
|
||||||
}, minio.CopySrcOptions{
|
|
||||||
Bucket: bucketName,
|
|
||||||
Object: src,
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Invalidate cache
|
|
||||||
// TODO: check error
|
|
||||||
invalidateCache(ctx, nomalizeID(info.Key))
|
|
||||||
|
|
||||||
return &File{
|
|
||||||
ID: info.Key,
|
|
||||||
}, nil
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func moveMutation(ctx context.Context, src, dest string) (*File, error) {
|
|
||||||
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("Failed to get s3Client from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if dest is a file or a dir
|
|
||||||
if strings.HasSuffix(dest, "/") {
|
|
||||||
// create new dest id
|
|
||||||
// TODO: What if a file with this id already exists?
|
|
||||||
dest += getFilenameFromID(src)
|
|
||||||
}
|
|
||||||
|
|
||||||
// There is no (spoon) move. Only copy and delete
|
|
||||||
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
|
|
||||||
Bucket: bucketName,
|
|
||||||
Object: dest,
|
|
||||||
}, minio.CopySrcOptions{
|
|
||||||
Bucket: bucketName,
|
|
||||||
Object: src,
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
err = deleteMutation(ctx, src)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
invalidateCache(ctx, nomalizeID(info.Key))
|
|
||||||
|
|
||||||
return &File{
|
|
||||||
ID: info.Key,
|
|
||||||
}, nil
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func createDirectory(ctx context.Context, path string) (*Directory, error) {
|
|
||||||
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("Failed to get s3Client from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
if !strings.HasSuffix(path, "/") {
|
|
||||||
path += "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
info, err := s3Client.PutObject(ctx, bucketName, path, strings.NewReader(""), 0, minio.PutObjectOptions{
|
|
||||||
ContentType: "application/x-directory",
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Invalidate cache
|
|
||||||
// TODO: check error
|
|
||||||
invalidateCacheForDir(ctx, nomalizeID(info.Key))
|
|
||||||
|
|
||||||
return &Directory{
|
|
||||||
ID: info.Key,
|
|
||||||
}, nil
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func deleteDirectory(ctx context.Context, path string) error {
|
|
||||||
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get s3Client from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
loader, ok := ctx.Value("loader").(map[string]*dataloader.Loader)
|
|
||||||
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get dataloader from context")
|
|
||||||
}
|
|
||||||
|
|
||||||
if !strings.HasSuffix(path, "/") {
|
|
||||||
path += "/"
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get all files inside the directory
|
|
||||||
thunk := loader["listObjectsRecursive"].Load(ctx, dataloader.StringKey(nomalizeID(path)))
|
|
||||||
|
|
||||||
result, err := thunk()
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
files, ok := result.([]minio.ObjectInfo)
|
|
||||||
if !ok {
|
|
||||||
return fmt.Errorf("Failed to get parse result from listObjects")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Delete all child files
|
|
||||||
err = deleteMultiple(ctx, *s3Client, files)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the dir had no children it exists as an object (object with "/" at the end).
|
|
||||||
// If it exists as an object and had children it will get delete once the last child has been deleted
|
|
||||||
// If it had no children we have to delete it manualy
|
|
||||||
// This is at least the behavior when working with minio as s3 backend
|
|
||||||
// TODO: check if this is normal behavior when working with s3
|
|
||||||
if len(files) == 0 {
|
|
||||||
err := s3Client.RemoveObject(ctx, bucketName, path, minio.RemoveObjectOptions{})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//Invalidate cache
|
|
||||||
invalidateCacheForDir(ctx, nomalizeID(path))
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
//login Checks for valid username password combination. Returns singed jwt string
|
|
||||||
func login(ctx context.Context, username, password string) (LoginResult, error) {
|
|
||||||
|
|
||||||
// TODO: replace with propper user management
|
|
||||||
if username != "admin" && password != "hunter2" {
|
|
||||||
return LoginResult{
|
|
||||||
Successful: false,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
token := createJWT(createClaims(username))
|
|
||||||
|
|
||||||
tokenString, err := token.SignedString([]byte("TODO"))
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return LoginResult{
|
|
||||||
Successful: false,
|
|
||||||
}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return LoginResult{
|
|
||||||
Token: tokenString,
|
|
||||||
Successful: true,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
131
internal/s3/minio.go
Normal file
131
internal/s3/minio.go
Normal file
@@ -0,0 +1,131 @@
|
|||||||
|
package s3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/minio/minio-go/v7"
|
||||||
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
||||||
|
)
|
||||||
|
|
||||||
|
type minioS3 struct {
|
||||||
|
client *minio.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewMinio(config types.AppConfig) (S3Service, error) {
|
||||||
|
client, err := minio.New(config.S3Endoint, &minio.Options{
|
||||||
|
Creds: credentials.NewStaticV4(config.S3AccessKey, config.S3SecretKey, ""),
|
||||||
|
Secure: config.S3SSL,
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &minioS3{
|
||||||
|
client: client,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) ListBuckets(ctx context.Context) ([]string, error) {
|
||||||
|
buckets, err := m.client.ListBuckets(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var rtn []string
|
||||||
|
|
||||||
|
for _, v := range buckets {
|
||||||
|
rtn = append(rtn, v.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
return rtn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) ListObjects(ctx context.Context, id types.ID) ([]Object, error) {
|
||||||
|
var result []Object
|
||||||
|
|
||||||
|
for objInfo := range m.client.ListObjects(ctx, id.Bucket, minio.ListObjectsOptions{
|
||||||
|
Prefix: id.Key,
|
||||||
|
Recursive: false,
|
||||||
|
}) {
|
||||||
|
objId := types.ID{
|
||||||
|
Bucket: id.Bucket,
|
||||||
|
Key: objInfo.Key,
|
||||||
|
}
|
||||||
|
|
||||||
|
result = append(result, Object{
|
||||||
|
ID: objId,
|
||||||
|
Size: objInfo.Size,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) ListObjectsRecursive(ctx context.Context, id types.ID) ([]Object, error) {
|
||||||
|
var result []Object
|
||||||
|
|
||||||
|
for objInfo := range m.client.ListObjects(ctx, id.Bucket, minio.ListObjectsOptions{
|
||||||
|
Prefix: id.Key,
|
||||||
|
Recursive: true,
|
||||||
|
}) {
|
||||||
|
objId := types.ID{
|
||||||
|
Bucket: id.Bucket,
|
||||||
|
Key: objInfo.Key,
|
||||||
|
}
|
||||||
|
result = append(result, Object{
|
||||||
|
ID: objId,
|
||||||
|
Size: objInfo.Size,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) GetObject(ctx context.Context, id types.ID) (ObjectReader, error) {
|
||||||
|
object, err := m.client.GetObject(ctx, id.Bucket, id.Key, minio.GetObjectOptions{})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return object, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) PutObject(ctx context.Context, id types.ID, reader io.Reader, objectSize int64) error {
|
||||||
|
_, err := m.client.PutObject(ctx, id.Bucket, id.Key, reader, objectSize, minio.PutObjectOptions{})
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) CopyObject(ctx context.Context, src types.ID, dest types.ID) error {
|
||||||
|
_, err := m.client.CopyObject(ctx, minio.CopyDestOptions{
|
||||||
|
Bucket: dest.Bucket,
|
||||||
|
Object: dest.Key,
|
||||||
|
}, minio.CopySrcOptions{
|
||||||
|
Bucket: src.Bucket,
|
||||||
|
Object: src.Key,
|
||||||
|
})
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) StatObject(ctx context.Context, id types.ID) (*Object, error) {
|
||||||
|
info, err := m.client.StatObject(ctx, id.Bucket, id.Key, minio.GetObjectOptions{})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Object{
|
||||||
|
ID: id,
|
||||||
|
Size: info.Size,
|
||||||
|
LastModified: info.LastModified,
|
||||||
|
ContentType: info.ContentType,
|
||||||
|
ETag: info.ETag,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *minioS3) RemoveObject(ctx context.Context, id types.ID) error {
|
||||||
|
return m.client.RemoveObject(ctx, id.Bucket, id.Key, minio.RemoveObjectOptions{})
|
||||||
|
}
|
||||||
154
internal/s3/mock.go
Normal file
154
internal/s3/mock.go
Normal file
@@ -0,0 +1,154 @@
|
|||||||
|
package s3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"crypto/md5"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
)
|
||||||
|
|
||||||
|
type mockS3 struct {
|
||||||
|
buckets []string
|
||||||
|
objects map[types.ID]mockObject
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockObject struct {
|
||||||
|
content []byte
|
||||||
|
contentType string
|
||||||
|
lastMod time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
type mockObjectReader struct {
|
||||||
|
*bytes.Reader
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r mockObjectReader) Close() error {
|
||||||
|
// NOOP
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewMockS3(buckets []string) (S3Service, error) {
|
||||||
|
return &mockS3{
|
||||||
|
buckets: buckets,
|
||||||
|
objects: map[types.ID]mockObject{},
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) ListBuckets(ctx context.Context) ([]string, error) {
|
||||||
|
return m.buckets, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) ListObjects(ctx context.Context, id types.ID) ([]Object, error) {
|
||||||
|
var results []Object
|
||||||
|
|
||||||
|
dirs := make(map[string]bool)
|
||||||
|
|
||||||
|
depth := len(strings.Split(id.Key, "/"))
|
||||||
|
|
||||||
|
for k, v := range m.objects {
|
||||||
|
if k.Bucket == id.Bucket {
|
||||||
|
if k.Parent().Key == id.Key {
|
||||||
|
results = append(results, *mockObjToObject(v, k))
|
||||||
|
} else if strings.HasPrefix(k.Key, id.Key) {
|
||||||
|
s := strings.Join(strings.Split(k.Key, "/")[:depth], "/") + "/"
|
||||||
|
dirs[s] = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for k := range dirs {
|
||||||
|
results = append(results, Object{
|
||||||
|
ID: types.ID{
|
||||||
|
Bucket: id.Bucket,
|
||||||
|
Key: k,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return results, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) ListObjectsRecursive(ctx context.Context, id types.ID) ([]Object, error) {
|
||||||
|
var results []Object
|
||||||
|
|
||||||
|
for k, v := range m.objects {
|
||||||
|
if k.Bucket == id.Bucket {
|
||||||
|
if strings.HasPrefix(k.Key, id.Key) {
|
||||||
|
results = append(results, *mockObjToObject(v, k))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return results, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) GetObject(ctx context.Context, id types.ID) (ObjectReader, error) {
|
||||||
|
mockObj, exist := m.objects[id]
|
||||||
|
|
||||||
|
if !exist {
|
||||||
|
return nil, fmt.Errorf("Object not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
reader := bytes.NewReader(mockObj.content)
|
||||||
|
|
||||||
|
return mockObjectReader{reader}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) PutObject(ctx context.Context, id types.ID, reader io.Reader, objectSize int64) error {
|
||||||
|
content, err := ioutil.ReadAll(reader)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
m.objects[id] = mockObject{
|
||||||
|
content: content,
|
||||||
|
lastMod: time.Now(),
|
||||||
|
contentType: "application/octet-stream", // TODO: detect MIME type or dont its just a mock after all
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) CopyObject(ctx context.Context, src types.ID, dest types.ID) error {
|
||||||
|
srcObj, exist := m.objects[src]
|
||||||
|
|
||||||
|
if !exist {
|
||||||
|
return fmt.Errorf("Object not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
m.objects[dest] = srcObj
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) StatObject(ctx context.Context, id types.ID) (*Object, error) {
|
||||||
|
mockObj, exist := m.objects[id]
|
||||||
|
|
||||||
|
if !exist {
|
||||||
|
return nil, fmt.Errorf("Object not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
return mockObjToObject(mockObj, id), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockS3) RemoveObject(ctx context.Context, id types.ID) error {
|
||||||
|
delete(m.objects, id)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func mockObjToObject(mockObj mockObject, id types.ID) *Object {
|
||||||
|
return &Object{
|
||||||
|
ID: id,
|
||||||
|
Size: int64(len(mockObj.content)),
|
||||||
|
ContentType: mockObj.contentType,
|
||||||
|
LastModified: mockObj.lastMod,
|
||||||
|
ETag: fmt.Sprintf("%x", md5.Sum(mockObj.content)),
|
||||||
|
}
|
||||||
|
}
|
||||||
162
internal/s3/mock_test.go
Normal file
162
internal/s3/mock_test.go
Normal file
@@ -0,0 +1,162 @@
|
|||||||
|
package s3_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io/ioutil"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func setup(t *testing.T) (s3.S3Service, context.Context, *assert.Assertions) {
|
||||||
|
service, _ := s3.NewMockS3([]string{"bucket1", "bucket2"})
|
||||||
|
ctx := context.Background()
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
return service, ctx, assert
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuckets(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
buckets, err := s3.ListBuckets(ctx)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
assert.Len(buckets, 2)
|
||||||
|
|
||||||
|
assert.Contains(buckets, "bucket1")
|
||||||
|
assert.Contains(buckets, "bucket2")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPut(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
content := "FileContent"
|
||||||
|
|
||||||
|
err := s3.PutObject(ctx, *types.ParseID("bucket1:/file1"), strings.NewReader(content), int64(len(content)))
|
||||||
|
assert.NoError(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPutAndGet(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
content := "FileContent"
|
||||||
|
id := *types.ParseID("bucket1:/file1")
|
||||||
|
|
||||||
|
err := s3.PutObject(ctx, *types.ParseID("bucket1:/file1"), strings.NewReader(content), int64(len(content)))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
reader, err := s3.GetObject(ctx, id)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
readerContent, err := ioutil.ReadAll(reader)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
assert.Equal(content, string(readerContent))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStat(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
content := "FileContent"
|
||||||
|
id := *types.ParseID("bucket1:/file1")
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
|
||||||
|
err := s3.PutObject(ctx, id, strings.NewReader(content), int64(len(content)))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
obj, err := s3.StatObject(ctx, id)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
assert.Equal(id.String(), obj.ID.String())
|
||||||
|
assert.Equal(int64(len(content)), obj.Size)
|
||||||
|
assert.NotEmpty(obj.ETag)
|
||||||
|
assert.WithinDuration(now, obj.LastModified, time.Second*1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemove(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
content := "FileContent"
|
||||||
|
id := *types.ParseID("bucket1:/file1")
|
||||||
|
|
||||||
|
err := s3.PutObject(ctx, id, strings.NewReader(content), int64(len(content)))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
err = s3.RemoveObject(ctx, id)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
_, err = s3.StatObject(ctx, id)
|
||||||
|
assert.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestList(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
content1 := "FileContent1"
|
||||||
|
id1 := *types.ParseID("bucket1:/file1")
|
||||||
|
|
||||||
|
err := s3.PutObject(ctx, id1, strings.NewReader(content1), int64(len(content1)))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
content2 := "FileContent2"
|
||||||
|
id2 := *types.ParseID("bucket1:/file2")
|
||||||
|
|
||||||
|
err = s3.PutObject(ctx, id2, strings.NewReader(content2), int64(len(content2)))
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
listID := types.ParseID("bucket1:/")
|
||||||
|
|
||||||
|
objects, err := s3.ListObjects(ctx, *listID)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
assert.Len(objects, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestListRecursive(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
s3.PutObject(ctx, *types.ParseID("bucket1:/file1"), strings.NewReader("content"), int64(len("content")))
|
||||||
|
s3.PutObject(ctx, *types.ParseID("bucket1:/path1/file1"), strings.NewReader("content"), int64(len("content")))
|
||||||
|
s3.PutObject(ctx, *types.ParseID("bucket1:/path1/file2"), strings.NewReader("content"), int64(len("content")))
|
||||||
|
s3.PutObject(ctx, *types.ParseID("bucket1:/path1/path2/file1"), strings.NewReader("content"), int64(len("content")))
|
||||||
|
s3.PutObject(ctx, *types.ParseID("bucket1:/path3/path4/file1"), strings.NewReader("content"), int64(len("content")))
|
||||||
|
|
||||||
|
objects, err := s3.ListObjectsRecursive(ctx, *types.ParseID("bucket1:/path1/"))
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Len(objects, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCopy(t *testing.T) {
|
||||||
|
s3, ctx, assert := setup(t)
|
||||||
|
|
||||||
|
id1 := *types.ParseID("bucket1:/file1")
|
||||||
|
id2 := *types.ParseID("bucket1:/file2")
|
||||||
|
|
||||||
|
s3.PutObject(ctx, id1, strings.NewReader("content"), int64(len("content")))
|
||||||
|
|
||||||
|
err := s3.CopyObject(ctx, id1, id2)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
obj1, err := s3.StatObject(ctx, id1)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.NotNil(obj1)
|
||||||
|
|
||||||
|
obj2, err := s3.StatObject(ctx, id1)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.NotNil(obj2)
|
||||||
|
|
||||||
|
assert.Equal(obj1.ETag, obj2.ETag)
|
||||||
|
assert.Equal(obj1.Size, obj2.Size)
|
||||||
|
|
||||||
|
obj2Reader, err := s3.GetObject(ctx, id2)
|
||||||
|
assert.NoError(err)
|
||||||
|
|
||||||
|
obj2Content, err := ioutil.ReadAll(obj2Reader)
|
||||||
|
assert.NoError(err)
|
||||||
|
assert.Equal([]byte("content"), obj2Content)
|
||||||
|
}
|
||||||
37
internal/s3/s3.go
Normal file
37
internal/s3/s3.go
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
package s3
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ObjectReader interface {
|
||||||
|
io.Reader
|
||||||
|
io.Seeker
|
||||||
|
io.ReaderAt
|
||||||
|
io.Closer
|
||||||
|
}
|
||||||
|
|
||||||
|
type Object struct {
|
||||||
|
ID types.ID
|
||||||
|
Size int64
|
||||||
|
LastModified time.Time
|
||||||
|
ContentType string
|
||||||
|
ETag string
|
||||||
|
}
|
||||||
|
|
||||||
|
type S3Service interface {
|
||||||
|
ListBuckets(ctx context.Context) ([]string, error)
|
||||||
|
|
||||||
|
GetObject(ctx context.Context, id types.ID) (ObjectReader, error)
|
||||||
|
PutObject(ctx context.Context, id types.ID, reader io.Reader, objectSize int64) error
|
||||||
|
|
||||||
|
ListObjects(ctx context.Context, id types.ID) ([]Object, error)
|
||||||
|
ListObjectsRecursive(ctx context.Context, id types.ID) ([]Object, error)
|
||||||
|
CopyObject(ctx context.Context, src types.ID, dest types.ID) error
|
||||||
|
StatObject(ctx context.Context, id types.ID) (*Object, error)
|
||||||
|
RemoveObject(ctx context.Context, id types.ID) error
|
||||||
|
}
|
||||||
@@ -2,96 +2,48 @@ package s3browser
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/minio/minio-go/v7"
|
"github.com/graph-gophers/dataloader"
|
||||||
"github.com/minio/minio-go/v7/pkg/credentials"
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/cache"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/db"
|
||||||
|
gql "git.kapelle.org/niklas/s3browser/internal/gql"
|
||||||
|
httpserver "git.kapelle.org/niklas/s3browser/internal/httpserver"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/loader"
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/s3"
|
||||||
|
types "git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
// AppConfig general config
|
|
||||||
type AppConfig struct {
|
|
||||||
S3Endoint string
|
|
||||||
S3AccessKey string
|
|
||||||
S3SecretKey string
|
|
||||||
S3SSL bool
|
|
||||||
S3Bucket string
|
|
||||||
CacheTTL time.Duration
|
|
||||||
CacheCleanup time.Duration
|
|
||||||
Address string
|
|
||||||
LogDebug bool
|
|
||||||
}
|
|
||||||
|
|
||||||
// File represents a file with its metadata
|
|
||||||
type File struct {
|
|
||||||
ID string `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
Size int64 `json:"size"`
|
|
||||||
ContentType string `json:"contentType"`
|
|
||||||
ETag string `json:"etag"`
|
|
||||||
LastModified time.Time `json:"lastModified"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Directory represents a directory with its metadata
|
|
||||||
type Directory struct {
|
|
||||||
ID string `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
Files []File `json:"files"`
|
|
||||||
Directorys []Directory `json:"directorys"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var bucketName string
|
|
||||||
|
|
||||||
// setupS3Client connect the s3Client
|
|
||||||
func setupS3Client(config AppConfig) (*minio.Client, error) {
|
|
||||||
minioClient, err := minio.New(config.S3Endoint, &minio.Options{
|
|
||||||
Creds: credentials.NewStaticV4(config.S3AccessKey, config.S3SecretKey, ""),
|
|
||||||
Secure: config.S3SSL,
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
exists, err := minioClient.BucketExists(context.Background(), config.S3Bucket)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if !exists {
|
|
||||||
return nil, fmt.Errorf("Bucket '%s' does not exist", config.S3Bucket)
|
|
||||||
}
|
|
||||||
|
|
||||||
bucketName = config.S3Bucket
|
|
||||||
|
|
||||||
return minioClient, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start starts the app
|
// Start starts the app
|
||||||
func Start(config AppConfig) {
|
func Start(config types.AppConfig) {
|
||||||
|
|
||||||
if config.LogDebug {
|
if config.LogDebug {
|
||||||
log.SetLevel(log.DebugLevel)
|
log.SetLevel(log.DebugLevel)
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Info("Starting")
|
log.Info("Starting")
|
||||||
log.Debug("Setting up s3 client")
|
s3Client, err := s3.NewMinio(config)
|
||||||
s3Client, err := setupS3Client(config)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Failed to setup s3 client: ", err.Error())
|
log.Error("Failed to setup s3 client: ", err.Error())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Info("s3 client connected")
|
|
||||||
|
|
||||||
log.Debug("Creating dataloader")
|
dbStore, err := db.NewDB(config.DSN)
|
||||||
loaderMap := createDataloader(config)
|
if err != nil {
|
||||||
|
log.Error("Failed to connect DB: ", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
log.Debug("Generating graphq schema")
|
loader := loader.NewLoader(loader.CacheConfig{
|
||||||
graphqlTypes()
|
ListObjectsLoaderCache: &dataloader.NoCache{},
|
||||||
schema, err := graphqlSchema()
|
ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
|
||||||
|
StatObjectLoaderCache: cache.NewTTLCache(config.CacheTTL, config.CacheCleanup),
|
||||||
|
ListBucketsLoaderCache: cache.NewTTLCache(config.CacheTTL, config.CacheCleanup),
|
||||||
|
})
|
||||||
|
|
||||||
|
gql.GraphqlTypes()
|
||||||
|
schema, err := gql.GraphqlSchema()
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Failed to generate graphq schemas: ", err.Error())
|
log.Error("Failed to generate graphq schemas: ", err.Error())
|
||||||
@@ -99,10 +51,10 @@ func Start(config AppConfig) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
resolveContext := context.WithValue(context.Background(), "s3Client", s3Client)
|
resolveContext := context.WithValue(context.Background(), "s3Client", s3Client)
|
||||||
resolveContext = context.WithValue(resolveContext, "loader", loaderMap)
|
resolveContext = context.WithValue(resolveContext, "loader", loader)
|
||||||
|
resolveContext = context.WithValue(resolveContext, "dbStore", dbStore)
|
||||||
|
|
||||||
log.Debug("Starting HTTP server")
|
err = httpserver.InitHttp(resolveContext, schema, config.Address)
|
||||||
err = initHttp(resolveContext, schema, config.Address)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Failed to start webserver: ", err.Error())
|
log.Error("Failed to start webserver: ", err.Error())
|
||||||
|
|||||||
114
internal/types/id.go
Normal file
114
internal/types/id.go
Normal file
@@ -0,0 +1,114 @@
|
|||||||
|
package types
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
idRegex = regexp.MustCompile(`(.*?)(@(.*))?:(.*)`)
|
||||||
|
)
|
||||||
|
|
||||||
|
// ID an id of a file consists of at least a Bucket and a Key. Version is optional.
|
||||||
|
// Can also be used as an ID for a directory. When the key ends with "/" it is treated as dir.
|
||||||
|
type ID struct {
|
||||||
|
Bucket string `json:"bucket"` // Name of the bucket
|
||||||
|
Key string `json:"key"` // Key of the object
|
||||||
|
Version string `json:"version"` // Version of the object. For now we ignore it
|
||||||
|
}
|
||||||
|
|
||||||
|
// String Return String representation of an ID
|
||||||
|
// Looks like this: "bucketName@version:/id/of/obj" or "bucketName:/id/of/obj"
|
||||||
|
func (i ID) String() string {
|
||||||
|
if i.Version == "" {
|
||||||
|
return fmt.Sprintf("%s:%s", i.Bucket, i.Key)
|
||||||
|
} else {
|
||||||
|
return fmt.Sprintf("%s@%s:%s", i.Bucket, i.Version, i.Key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Normalize normalzes the key to have a "/" prefix
|
||||||
|
func (i *ID) Normalize() {
|
||||||
|
if i.Key == "." {
|
||||||
|
i.Key = "/"
|
||||||
|
} else if !strings.HasPrefix(i.Key, "/") {
|
||||||
|
i.Key = "/" + i.Key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Valid checks if bucket and key is not empty
|
||||||
|
func (i *ID) Valid() bool {
|
||||||
|
return i.Bucket != "" && i.Key != ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i *ID) IsDirectory() bool {
|
||||||
|
return strings.HasSuffix(i.Key, "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Raw for the Key interface for the dataloaders so ID can be used as a dataloader key
|
||||||
|
func (i ID) Raw() interface{} {
|
||||||
|
return i
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parent returns the parent dir ID. If its a file then return containing directory.
|
||||||
|
// If this is a directory then return the dir one up.
|
||||||
|
func (i ID) Parent() *ID {
|
||||||
|
if i.Key == "/" {
|
||||||
|
// Already at root. We dont have a parent
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var parent *ID
|
||||||
|
|
||||||
|
if i.IsDirectory() {
|
||||||
|
parts := strings.Split(i.Key, "/")
|
||||||
|
parent = &ID{
|
||||||
|
Bucket: i.Bucket,
|
||||||
|
Key: strings.Join(parts[:len(parts)-2], "/") + "/",
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dir := filepath.Dir(i.Key)
|
||||||
|
|
||||||
|
if dir != "/" {
|
||||||
|
dir += "/"
|
||||||
|
}
|
||||||
|
|
||||||
|
parent = &ID{
|
||||||
|
Bucket: i.Bucket,
|
||||||
|
Key: dir,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
parent.Normalize()
|
||||||
|
|
||||||
|
return parent
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns filename or directory name
|
||||||
|
func (i ID) Name() string {
|
||||||
|
return filepath.Base(i.Key)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseID parses a string to an ID. Null if invalid
|
||||||
|
func ParseID(id string) *ID {
|
||||||
|
match := idRegex.FindStringSubmatch(id)
|
||||||
|
if match == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
rtn := &ID{
|
||||||
|
Bucket: match[1],
|
||||||
|
Version: match[3],
|
||||||
|
Key: match[4],
|
||||||
|
}
|
||||||
|
|
||||||
|
if !rtn.Valid() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
rtn.Normalize()
|
||||||
|
|
||||||
|
return rtn
|
||||||
|
}
|
||||||
109
internal/types/id_test.go
Normal file
109
internal/types/id_test.go
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
package types_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.kapelle.org/niklas/s3browser/internal/types"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TODO: test version component (not yet used in code)
|
||||||
|
|
||||||
|
func TestIDParse(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
id := types.ParseID("test:/path/key")
|
||||||
|
|
||||||
|
assert.NotNil(id)
|
||||||
|
assert.True(id.Valid())
|
||||||
|
assert.Equal("test", id.Bucket)
|
||||||
|
assert.Equal("/path/key", id.Key)
|
||||||
|
assert.False(id.IsDirectory())
|
||||||
|
assert.Equal("key", id.Name())
|
||||||
|
assert.Equal("test:/path/key", id.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDParseInvalid(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
assert.Nil(types.ParseID("/asd/ad"))
|
||||||
|
assert.Nil(types.ParseID("test"))
|
||||||
|
assert.Nil(types.ParseID("test:"))
|
||||||
|
assert.Nil(types.ParseID(""))
|
||||||
|
assert.Nil(types.ParseID("/"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDIsDir(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
idFile := types.ParseID("test:/path/key")
|
||||||
|
assert.NotNil(idFile)
|
||||||
|
assert.False(idFile.IsDirectory())
|
||||||
|
|
||||||
|
idDir := types.ParseID("test:/path/key/")
|
||||||
|
assert.NotNil(idDir)
|
||||||
|
assert.True(idDir.IsDirectory())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDRoot(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
id := types.ParseID("test:/")
|
||||||
|
|
||||||
|
assert.NotNil(id)
|
||||||
|
assert.True(id.Valid())
|
||||||
|
assert.Equal("test", id.Bucket)
|
||||||
|
assert.Equal("/", id.Key)
|
||||||
|
assert.True(id.IsDirectory())
|
||||||
|
assert.Equal("/", id.Name())
|
||||||
|
assert.Equal("test:/", id.String())
|
||||||
|
|
||||||
|
assert.Nil(id.Parent())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDParentFromFile(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
id := types.ParseID("test:/path1/path2/key")
|
||||||
|
|
||||||
|
assert.NotNil(id)
|
||||||
|
|
||||||
|
parent := id.Parent()
|
||||||
|
|
||||||
|
assert.NotNil(parent)
|
||||||
|
assert.True(parent.Valid())
|
||||||
|
assert.Equal("test", parent.Bucket)
|
||||||
|
assert.Equal("/path1/path2/", parent.Key)
|
||||||
|
assert.True(parent.IsDirectory())
|
||||||
|
assert.Equal("path2", parent.Name())
|
||||||
|
assert.Equal("test:/path1/path2/", parent.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDParentFromDir(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
id := types.ParseID("test:/path1/path2/")
|
||||||
|
|
||||||
|
assert.NotNil(id)
|
||||||
|
|
||||||
|
parent := id.Parent()
|
||||||
|
|
||||||
|
assert.NotNil(parent)
|
||||||
|
assert.True(parent.Valid())
|
||||||
|
assert.Equal("test", parent.Bucket)
|
||||||
|
assert.Equal("/path1/", parent.Key)
|
||||||
|
assert.True(parent.IsDirectory())
|
||||||
|
assert.Equal("path1", parent.Name())
|
||||||
|
assert.Equal("test:/path1/", parent.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIDParentRoot(t *testing.T) {
|
||||||
|
assert := assert.New(t)
|
||||||
|
|
||||||
|
id := types.ParseID("test:/key1")
|
||||||
|
|
||||||
|
parent := id.Parent()
|
||||||
|
|
||||||
|
assert.NotNil(parent)
|
||||||
|
assert.Equal("/", parent.Key)
|
||||||
|
}
|
||||||
47
internal/types/types.go
Normal file
47
internal/types/types.go
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
package types
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/golang-jwt/jwt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// AppConfig general config
|
||||||
|
type AppConfig struct {
|
||||||
|
S3Endoint string
|
||||||
|
S3AccessKey string
|
||||||
|
S3SecretKey string
|
||||||
|
S3SSL bool
|
||||||
|
DSN string
|
||||||
|
CacheTTL time.Duration
|
||||||
|
CacheCleanup time.Duration
|
||||||
|
Address string
|
||||||
|
LogDebug bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// File represents a file with its metadata
|
||||||
|
type File struct {
|
||||||
|
ID ID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
ContentType string `json:"contentType"`
|
||||||
|
ETag string `json:"etag"`
|
||||||
|
LastModified time.Time `json:"lastModified"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Directory represents a directory with its metadata
|
||||||
|
type Directory struct {
|
||||||
|
ID ID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Files []File `json:"files"`
|
||||||
|
Directorys []Directory `json:"directorys"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type JWTClaims struct {
|
||||||
|
jwt.StandardClaims
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoginResult struct {
|
||||||
|
Token string `json:"token"`
|
||||||
|
Successful bool `json:"successful"`
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user