Compare commits

...

14 Commits

Author SHA1 Message Date
125ce9c955 fix get value db 2022-02-07 15:59:52 +01:00
131f19deed db interface 2022-02-07 15:59:43 +01:00
2ac552e840 added gql tests 2022-02-07 15:56:09 +01:00
48c50a5b7e typesInit flag 2022-02-07 15:53:14 +01:00
48f770f703 added loader tests 2021-11-27 04:07:51 +01:00
2ae14cdfd4 fixed mock s3 list dirs 2021-11-27 04:07:41 +01:00
a10593a318 loader cache config 2021-11-27 04:07:27 +01:00
0971301562 added tests 2021-11-25 01:57:38 +01:00
47befe6db1 fixed id parent on root on root 2021-11-25 01:57:24 +01:00
979ebee677 fixed mock s3 list recursive 2021-11-25 01:56:57 +01:00
8d85d645d6 use s3Service interface 2021-11-23 20:12:24 +01:00
60817c2249 added mock s3 implementation 2021-11-23 01:54:02 +01:00
bead881af2 added s3 interface with minio implementation 2021-11-23 01:53:51 +01:00
80302b62f4 added Name func to ID 2021-11-23 01:53:04 +01:00
20 changed files with 993 additions and 281 deletions

1
go.mod
View File

@@ -14,5 +14,6 @@ require (
github.com/opentracing/opentracing-go v1.2.0 // indirect
github.com/patrickmn/go-cache v2.1.0+incompatible
github.com/sirupsen/logrus v1.8.1
github.com/stretchr/testify v1.7.0
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f
)

4
go.sum
View File

@@ -67,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.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
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-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/crypto v0.0.0-20201216223049-8b5274cf687f h1:aZp0e2vLN4MToVqnjNEYEtrEA8RH8U8FN1CU7JgqsPU=
@@ -94,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.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
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=

View File

@@ -1,94 +1,9 @@
package db
import (
"context"
"database/sql"
_ "embed"
"time"
import "context"
_ "github.com/go-sql-driver/mysql"
"golang.org/x/crypto/bcrypt"
)
//go:embed setup.sql
var setupSql string
const DB_NAME = "s3Browser"
type DB 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 &DB{
dbConn: db,
}, nil
}
func (d *DB) 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 *DB) 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 *DB) 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
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
View 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
}

75
internal/gql/gql_test.go Normal file
View 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)
}

View File

@@ -13,6 +13,7 @@ import (
types "git.kapelle.org/niklas/s3browser/internal/types"
)
var typesInit bool = false
var graphqlDirType *graphql.Object
var graphqlFileType *graphql.Object
var graphqlLoginResultType *graphql.Object
@@ -246,6 +247,8 @@ func GraphqlTypes() {
},
})
typesInit = true
}
//loadFile helper func for using the dataloader to get a file

View File

@@ -5,17 +5,16 @@ import (
"fmt"
"strings"
"github.com/minio/minio-go/v7"
"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").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return fmt.Errorf("Failed to get s3Client from context")
@@ -24,7 +23,7 @@ func deleteMutation(ctx context.Context, id types.ID) error {
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.Bucket, id.Key, minio.RemoveObjectOptions{})
err := s3Client.RemoveObject(ctx, id)
if err != nil {
return err
@@ -36,7 +35,7 @@ func deleteMutation(ctx context.Context, id types.ID) error {
}
func copyMutation(ctx context.Context, src, dest types.ID) (*types.File, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
@@ -50,35 +49,22 @@ func copyMutation(ctx context.Context, src, dest types.ID) (*types.File, error)
}
log.Debug("S3 'CopyObject': ", src, "-->", dest)
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
Bucket: dest.Bucket,
Object: dest.Key,
}, minio.CopySrcOptions{
Bucket: src.Bucket,
Object: src.Key,
})
err := s3Client.CopyObject(ctx, src, dest)
if err != nil {
return nil, err
}
newID := types.ID{
Bucket: info.Bucket,
Key: info.Key,
}
newID.Normalize()
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, newID)
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, dest)
return &types.File{
ID: newID,
ID: dest,
}, nil
}
func moveDirMutation(ctx context.Context, src, dest types.ID) ([]*types.File, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
@@ -107,13 +93,7 @@ func moveDirMutation(ctx context.Context, src, dest types.ID) ([]*types.File, er
newID.Normalize()
log.Debug("S3 'CopyObject': ", src, "-->", dest)
_, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
Bucket: dest.Bucket,
Object: newID.Key,
}, minio.CopySrcOptions{
Bucket: file.ID.Bucket,
Object: file.ID.Key,
})
err := s3Client.CopyObject(ctx, file.ID, dest)
if err != nil {
// TODO: handle error
@@ -135,7 +115,7 @@ func moveDirMutation(ctx context.Context, src, dest types.ID) ([]*types.File, er
}
func moveFileMutation(ctx context.Context, src, dest types.ID) (*types.File, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
@@ -150,13 +130,7 @@ func moveFileMutation(ctx context.Context, src, dest types.ID) (*types.File, err
log.Debug("S3 'CopyObject': ", src, "-->", dest)
// There is no (spoon) move. Only copy and delete
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
Bucket: dest.Bucket,
Object: dest.Key,
}, minio.CopySrcOptions{
Bucket: src.Bucket,
Object: src.Key,
})
err := s3Client.CopyObject(ctx, src, dest)
if err != nil {
return nil, err
@@ -168,54 +142,37 @@ func moveFileMutation(ctx context.Context, src, dest types.ID) (*types.File, err
return nil, err
}
newId := types.ID{
Bucket: info.Bucket,
Key: info.Key,
}
newId.Normalize()
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, newId)
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, dest)
return &types.File{
ID: newId,
ID: dest,
}, nil
}
func createDirectory(ctx context.Context, id types.ID) (*types.Directory, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
}
log.Debug("S3 'PutObject': ", id)
info, err := s3Client.PutObject(ctx, id.Bucket, id.Key, strings.NewReader(""), 0, minio.PutObjectOptions{
ContentType: "application/x-directory",
})
err := s3Client.PutObject(ctx, id, strings.NewReader(""), 0) // TODO: s3client interface needs content type parameter
if err != nil {
return nil, err
}
newID := types.ID{
Bucket: info.Bucket,
Key: info.Key,
}
newID.Normalize()
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, newID)
ctx.Value("loader").(*loader.Loader).InvalidedCacheForId(ctx, id)
return &types.Directory{
ID: newID,
ID: id,
}, nil
}
func deleteDirectory(ctx context.Context, id types.ID) error {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return fmt.Errorf("Failed to get s3Client from context")
@@ -236,11 +193,9 @@ func deleteDirectory(ctx context.Context, id types.ID) error {
}
// Delete all child files
var keysToDel []string
for _, file := range files {
keysToDel = append(keysToDel, file.ID.Key)
s3Client.RemoveObject(ctx, file.ID)
}
err = helper.DeleteMultiple(ctx, *s3Client, id.Bucket, keysToDel)
if err != nil {
return err
@@ -253,7 +208,7 @@ func deleteDirectory(ctx context.Context, id types.ID) error {
// 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.Bucket, id.Key, minio.RemoveObjectOptions{})
err := s3Client.RemoveObject(ctx, id)
if err != nil {
return err
}
@@ -267,7 +222,7 @@ func deleteDirectory(ctx context.Context, id types.ID) error {
//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)
dbStore := ctx.Value("dbStore").(db.DB)
succes, err := dbStore.CheckLogin(ctx, username, password)

View File

@@ -15,6 +15,10 @@ import (
//GraphqlSchema generate the schema with its root query and mutation
func GraphqlSchema() (graphql.Schema, error) {
if !typesInit {
GraphqlTypes()
}
queryFields := graphql.Fields{
"files": &graphql.Field{
Type: graphql.NewNonNull(graphql.NewList(graphql.NewNonNull(graphqlFileType))),

View File

@@ -6,37 +6,15 @@ import (
"strings"
"time"
"git.kapelle.org/niklas/s3browser/internal/s3"
types "git.kapelle.org/niklas/s3browser/internal/types"
"github.com/golang-jwt/jwt"
"github.com/minio/minio-go/v7"
log "github.com/sirupsen/logrus"
)
func GetFilenameFromKey(id string) string {
return filepath.Base(id)
}
func DeleteMultiple(ctx context.Context, s3Client minio.Client, bucket string, keys []string) error {
objectsCh := make(chan minio.ObjectInfo, 1)
go func() {
defer close(objectsCh)
for _, id := range keys {
objectsCh <- minio.ObjectInfo{
Key: id,
}
}
}()
log.Debug("S3 'RemoveObject': ", keys)
for err := range s3Client.RemoveObjects(ctx, bucket, objectsCh, minio.RemoveObjectsOptions{}) {
log.Error("Failed to delete object ", err.ObjectName, " because: ", err.Err.Error())
// TODO: error handel
}
return nil
}
func GetParentDir(id types.ID) types.ID {
dirs := strings.Split(id.Key, "/")
@@ -57,17 +35,10 @@ func GetParentDir(id types.ID) types.ID {
return parent
}
func ObjInfoToFile(objInfo minio.ObjectInfo, bucket string) *types.File {
objID := types.ID{
Bucket: bucket,
Key: objInfo.Key,
}
objID.Normalize()
func ObjInfoToFile(objInfo s3.Object, bucket string) *types.File {
return &types.File{
ID: objID,
Name: GetFilenameFromKey(objID.Key),
ID: objInfo.ID,
Name: objInfo.ID.Name(),
Size: objInfo.Size,
ContentType: objInfo.ContentType,
ETag: objInfo.ETag,

View File

@@ -4,9 +4,7 @@ import (
"context"
"fmt"
"io"
"mime"
"net/http"
"path/filepath"
"time"
"github.com/golang-jwt/jwt"
@@ -15,12 +13,12 @@ import (
"github.com/graphql-go/graphql"
"github.com/graphql-go/graphql/gqlerrors"
"github.com/graphql-go/handler"
"github.com/minio/minio-go/v7"
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"
)
@@ -112,7 +110,7 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
return
}
s3Client := ctx.Value("s3Client").(*minio.Client)
s3Client := ctx.Value("s3Client").(s3.S3Service)
idString := r.URL.Query().Get("id")
id := types.ParseID(idString)
@@ -124,7 +122,7 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
}
log.Debug("S3 'StatObject': ", id)
objInfo, err := s3Client.StatObject(context.Background(), id.Bucket, id.Key, minio.GetObjectOptions{})
objInfo, err := s3Client.StatObject(context.Background(), *id)
if err != nil {
log.Error("Failed to get object info: ", err)
@@ -139,7 +137,7 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
}
log.Debug("S3 'GetObject': ", id)
obj, err := s3Client.GetObject(context.Background(), id.Bucket, id.Key, minio.GetObjectOptions{})
obj, err := s3Client.GetObject(context.Background(), *id)
if err != nil {
log.Error("Failed to get object: ", err)
@@ -148,7 +146,7 @@ func httpGetFile(ctx context.Context, rw http.ResponseWriter, r *http.Request) {
}
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("ETag", objInfo.ETag)
@@ -166,7 +164,7 @@ func httpPostFile(ctx context.Context, rw http.ResponseWriter, r *http.Request)
return
}
s3Client := ctx.Value("s3Client").(*minio.Client)
s3Client := ctx.Value("s3Client").(s3.S3Service)
idString := r.URL.Query().Get("id")
@@ -180,13 +178,11 @@ func httpPostFile(ctx context.Context, rw http.ResponseWriter, r *http.Request)
id.Normalize()
contentType := r.Header.Get("Content-Type")
mimeType, _, _ := mime.ParseMediaType(contentType)
// contentType := r.Header.Get("Content-Type")
// mimeType, _, _ := mime.ParseMediaType(contentType)
log.Debug("S3 'PutObject': ", id)
_, err := s3Client.PutObject(context.Background(), id.Bucket, id.Key, r.Body, r.ContentLength, minio.PutObjectOptions{
ContentType: mimeType,
})
err := s3Client.PutObject(context.Background(), *id, r.Body, r.ContentLength) // TODO: put content type
if err != nil {
rw.WriteHeader(http.StatusInternalServerError)

View File

@@ -4,9 +4,9 @@ import (
"context"
"fmt"
types "git.kapelle.org/niklas/s3browser/internal/types"
"git.kapelle.org/niklas/s3browser/internal/s3"
"git.kapelle.org/niklas/s3browser/internal/types"
"github.com/graph-gophers/dataloader"
"github.com/minio/minio-go/v7"
log "github.com/sirupsen/logrus"
)
@@ -14,7 +14,7 @@ import (
func listObjectsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
var results []*dataloader.Result
s3Client, ok := c.Value("s3Client").(*minio.Client)
s3Client, ok := c.Value("s3Client").(s3.S3Service)
if !ok {
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
@@ -22,10 +22,18 @@ func listObjectsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result
for _, v := range k {
id := v.Raw().(types.ID)
results = append(results, &dataloader.Result{
Data: listObjects(s3Client, id, false),
Error: nil,
})
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
@@ -35,7 +43,7 @@ func listObjectsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result
func listObjectsRecursiveBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
var results []*dataloader.Result
s3Client, ok := c.Value("s3Client").(*minio.Client)
s3Client, ok := c.Value("s3Client").(s3.S3Service)
if !ok {
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
@@ -43,35 +51,27 @@ func listObjectsRecursiveBatch(c context.Context, k dataloader.Keys) []*dataload
for _, v := range k {
id := v.Raw().(types.ID)
results = append(results, &dataloader.Result{
Data: listObjects(s3Client, id, true),
Error: nil,
})
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
}
// listObjects helper func for listObjectsBatch
func listObjects(s3Client *minio.Client, id types.ID, recursive bool) []minio.ObjectInfo {
log.Debug("S3 'ListObjects': ", id)
objectCh := s3Client.ListObjects(context.Background(), id.Bucket, minio.ListObjectsOptions{
Prefix: id.Key,
Recursive: recursive,
})
result := make([]minio.ObjectInfo, 0)
for obj := range objectCh {
result = append(result, obj)
}
return result
}
func listBucketsBatch(c context.Context, k dataloader.Keys) []*dataloader.Result {
var results []*dataloader.Result
s3Client, ok := c.Value("s3Client").(*minio.Client)
s3Client, ok := c.Value("s3Client").(s3.S3Service)
if !ok {
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
@@ -100,7 +100,7 @@ func statObjectBatch(ctx context.Context, k dataloader.Keys) []*dataloader.Resul
log.Debug("statObjectBatch")
var results []*dataloader.Result
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
s3Client, ok := ctx.Value("s3Client").(s3.S3Service)
if !ok {
return handleLoaderError(k, fmt.Errorf("Failed to get s3Client from context"))
@@ -109,7 +109,7 @@ func statObjectBatch(ctx context.Context, k dataloader.Keys) []*dataloader.Resul
for _, v := range k {
id := v.Raw().(types.ID)
log.Debug("S3 'StatObject': ", id)
stat, err := s3Client.StatObject(ctx, id.Bucket, id.Key, minio.GetObjectOptions{})
stat, err := s3Client.StatObject(ctx, id)
results = append(results, &dataloader.Result{
Data: stat,
Error: err,

View File

@@ -3,14 +3,12 @@ package loader
import (
"context"
"fmt"
"path/filepath"
"strings"
"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"
"github.com/minio/minio-go/v7"
)
type Loader struct {
@@ -25,11 +23,18 @@ type Loader struct {
listBucketsLoaderCache cache.S3Cache
}
func NewLoader(config types.AppConfig) *Loader {
listObjectsLoaderCache := &dataloader.NoCache{}
listObjectsRecursiveLoaderCache := &dataloader.NoCache{}
statObjectLoaderCache := cache.NewTTLCache(config.CacheTTL, config.CacheCleanup)
listBucketsLoaderCache := cache.NewTTLCache(config.CacheTTL, config.CacheCleanup)
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(
@@ -68,10 +73,8 @@ func (l *Loader) GetFiles(ctx context.Context, path types.ID) ([]types.File, err
var files []types.File
for _, obj := range objects.([]minio.ObjectInfo) {
if obj.Err != nil {
return nil, obj.Err
} else if !strings.HasSuffix(obj.Key, "/") {
for _, obj := range objects.([]s3.Object) {
if !obj.ID.IsDirectory() {
files = append(files, *helper.ObjInfoToFile(obj, path.Bucket))
}
}
@@ -86,13 +89,13 @@ func (l *Loader) GetFile(ctx context.Context, id types.ID) (*types.File, error)
if err != nil {
return nil, err
}
objInfo, ok := result.(minio.ObjectInfo)
objInfo, ok := result.(*s3.Object)
if !ok {
return nil, fmt.Errorf("Failed to stats object")
}
return helper.ObjInfoToFile(objInfo, id.Bucket), nil
return helper.ObjInfoToFile(*objInfo, id.Bucket), nil
}
func (l *Loader) GetDirs(ctx context.Context, path types.ID) ([]types.Directory, error) {
@@ -104,20 +107,10 @@ func (l *Loader) GetDirs(ctx context.Context, path types.ID) ([]types.Directory,
}
var dirs []types.Directory
for _, obj := range result.([]minio.ObjectInfo) {
if obj.Err != nil {
return nil, obj.Err
} else if strings.HasSuffix(obj.Key, "/") {
resultID := types.ID{
Bucket: path.Bucket,
Key: obj.Key,
}
resultID.Normalize()
for _, obj := range result.([]s3.Object) {
if obj.ID.IsDirectory() {
dirs = append(dirs, types.Directory{
ID: resultID,
Name: filepath.Base(obj.Key),
ID: obj.ID,
})
}
}
@@ -134,13 +127,7 @@ func (l *Loader) GetBuckets(ctx context.Context) ([]string, error) {
return nil, err
}
bucketsInfo := result.([]minio.BucketInfo)
var buckets []string
for _, i := range bucketsInfo {
buckets = append(buckets, i.Name)
}
return buckets, nil
return result.([]string), nil
}
func (l *Loader) GetFilesRecursive(ctx context.Context, path types.ID) ([]types.File, error) {
@@ -151,7 +138,7 @@ func (l *Loader) GetFilesRecursive(ctx context.Context, path types.ID) ([]types.
return nil, err
}
objects := result.([]minio.ObjectInfo)
objects := result.([]s3.Object)
var files []types.File
for _, obj := range objects {

View 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)
}

131
internal/s3/minio.go Normal file
View 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
View 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
View 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
View 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
}

View File

@@ -3,25 +3,18 @@ package s3browser
import (
"context"
"github.com/minio/minio-go/v7"
"github.com/minio/minio-go/v7/pkg/credentials"
"github.com/graph-gophers/dataloader"
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"
)
// setupS3Client connect the s3Client
func setupS3Client(config types.AppConfig) (*minio.Client, error) {
return minio.New(config.S3Endoint, &minio.Options{
Creds: credentials.NewStaticV4(config.S3AccessKey, config.S3SecretKey, ""),
Secure: config.S3SSL,
})
}
// Start starts the app
func Start(config types.AppConfig) {
@@ -30,7 +23,7 @@ func Start(config types.AppConfig) {
}
log.Info("Starting")
s3Client, err := setupS3Client(config)
s3Client, err := s3.NewMinio(config)
if err != nil {
log.Error("Failed to setup s3 client: ", err.Error())
@@ -42,7 +35,12 @@ func Start(config types.AppConfig) {
log.Error("Failed to connect DB: ", err.Error())
}
loader := loader.NewLoader(config)
loader := loader.NewLoader(loader.CacheConfig{
ListObjectsLoaderCache: &dataloader.NoCache{},
ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
StatObjectLoaderCache: cache.NewTTLCache(config.CacheTTL, config.CacheCleanup),
ListBucketsLoaderCache: cache.NewTTLCache(config.CacheTTL, config.CacheCleanup),
})
gql.GraphqlTypes()
schema, err := gql.GraphqlSchema()

View File

@@ -69,9 +69,15 @@ func (i ID) Parent() *ID {
Key: strings.Join(parts[:len(parts)-2], "/") + "/",
}
} else {
dir := filepath.Dir(i.Key)
if dir != "/" {
dir += "/"
}
parent = &ID{
Bucket: i.Bucket,
Key: filepath.Dir(i.Key) + "/",
Key: dir,
}
}
@@ -80,6 +86,11 @@ func (i ID) Parent() *ID {
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)

109
internal/types/id_test.go Normal file
View 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)
}