From 34cf6682fe307efb1652eaffc9703fc0f01fb6b1 Mon Sep 17 00:00:00 2001
From: "Frederick F. Kautz IV" <fkautz@alumni.cmu.edu>
Date: Mon, 16 Mar 2015 20:41:00 -0700
Subject: [PATCH] list objects in buckets

---
 pkg/storage/encoded/encoded.go   | 74 +++++++++++++++++++++++++++++++-
 pkg/storage/storage_api_suite.go |  3 +-
 2 files changed, 74 insertions(+), 3 deletions(-)

diff --git a/pkg/storage/encoded/encoded.go b/pkg/storage/encoded/encoded.go
index edfeee815..dd926b340 100644
--- a/pkg/storage/encoded/encoded.go
+++ b/pkg/storage/encoded/encoded.go
@@ -18,9 +18,13 @@ package encoded
 
 import (
 	"bytes"
+	"crypto/md5"
+	"encoding/hex"
 	"errors"
 	"io"
+	"sort"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/minio-io/minio/pkg/donutbox"
@@ -164,7 +168,71 @@ func (diskStorage StorageDriver) GetObjectMetadata(bucket, key string, prefix st
 
 // ListObjects lists objects
 func (diskStorage StorageDriver) ListObjects(bucket string, resources storage.BucketResourcesMetadata) ([]storage.ObjectMetadata, storage.BucketResourcesMetadata, error) {
-	return nil, storage.BucketResourcesMetadata{}, errors.New("Not Implemented")
+	objects, err := diskStorage.donutBox.ListObjectsInBucket(bucket, resources.Prefix)
+	if err != nil {
+		return nil, storage.BucketResourcesMetadata{}, err
+	}
+	var results []storage.ObjectMetadata
+	sort.Strings(objects)
+	for _, object := range withoutDelimiter(objects, resources.Prefix, resources.Delimiter) {
+		if len(results) < resources.Maxkeys {
+			objectMetadata, err := diskStorage.GetObjectMetadata(bucket, object, "")
+			if err != nil {
+				return nil, storage.BucketResourcesMetadata{}, err
+			}
+			results = append(results, objectMetadata)
+		} else {
+			resources.IsTruncated = true
+		}
+	}
+	if resources.Delimiter != "" {
+		objects = trimPrefixWithDelimiter(objects, resources.Prefix, resources.Delimiter)
+		objects = beforeDelimiter(objects, resources.Delimiter)
+		objects = removeDuplicates(objects)
+		resources.CommonPrefixes = objects
+	}
+	return results, resources, nil
+}
+
+func withoutDelimiter(inputs []string, prefix, delim string) (results []string) {
+	if delim == "" {
+		return inputs
+	}
+	for _, input := range inputs {
+		input = strings.TrimPrefix(input, prefix)
+		if !strings.Contains(input, delim) {
+			results = append(results, prefix+input)
+		}
+	}
+	return results
+}
+
+func trimPrefixWithDelimiter(inputs []string, prefix, delim string) (results []string) {
+	for _, input := range inputs {
+		input = strings.TrimPrefix(input, prefix)
+		if strings.Contains(input, delim) {
+			results = append(results, input)
+		}
+	}
+	return results
+}
+
+func beforeDelimiter(inputs []string, delim string) (results []string) {
+	for _, input := range inputs {
+		results = append(results, strings.Split(input, delim)[0]+delim)
+	}
+	return results
+}
+
+func removeDuplicates(inputs []string) (results []string) {
+	keys := make(map[string]string)
+	for _, input := range inputs {
+		keys[input] = input
+	}
+	for result := range keys {
+		results = append(results, result)
+	}
+	return results
 }
 
 // CreateObject creates a new object
@@ -182,11 +250,13 @@ func (diskStorage StorageDriver) CreateObject(bucketKey string, objectKey string
 	}
 	totalLength := uint64(0)
 	chunkCount := 0
+	hasher := md5.New()
 	for chunk := range splitStream {
 		params, err := erasure.ParseEncoderParams(8, 8, erasure.Cauchy)
 		if err != nil {
 			return err
 		}
+		hasher.Write(chunk.Data)
 		totalLength = totalLength + uint64(len(chunk.Data))
 		chunkCount = chunkCount + 1
 		encoder := erasure.NewEncoder(params)
@@ -213,7 +283,7 @@ func (diskStorage StorageDriver) CreateObject(bucketKey string, objectKey string
 
 		ContentType: contentType,
 		Created:     time.Now(),
-		Md5:         "md5",
+		Md5:         hex.EncodeToString(hasher.Sum(nil)),
 		Size:        int64(totalLength),
 	}
 
diff --git a/pkg/storage/storage_api_suite.go b/pkg/storage/storage_api_suite.go
index 7939b27ba..6d1b8e14f 100644
--- a/pkg/storage/storage_api_suite.go
+++ b/pkg/storage/storage_api_suite.go
@@ -85,9 +85,9 @@ func testPaging(c *check.C, create func() Storage) {
 	storage.CreateBucket("bucket")
 	resources := BucketResourcesMetadata{}
 	objects, resources, err := storage.ListObjects("bucket", resources)
+	c.Assert(err, check.IsNil)
 	c.Assert(len(objects), check.Equals, 0)
 	c.Assert(resources.IsTruncated, check.Equals, false)
-	c.Assert(err, check.IsNil)
 	// check before paging occurs
 	for i := 0; i < 5; i++ {
 		key := "obj" + strconv.Itoa(i)
@@ -140,6 +140,7 @@ func testPaging(c *check.C, create func() Storage) {
 		resources.Prefix = "this/is/"
 		resources.Maxkeys = 10
 		objects, resources, err = storage.ListObjects("bucket", resources)
+		c.Assert(err, check.IsNil)
 		c.Assert(len(objects), check.Equals, 1)
 		c.Assert(resources.CommonPrefixes[0], check.Equals, "also/")
 	}