Add retention mode and legal hold mode on list objects api (#312)

Co-authored-by: Daniel Valdivia <hola@danielvaldivia.com>

Author: cesnietor

models/bucket_object.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/minio/minio-go/v7/pkg/replication"
 
@@ -54,6 +55,8 @@ type MinioClient interface {
 	getBucketNotification(ctx context.Context, bucketName string) (config notification.Configuration, err error)
 	getBucketPolicy(ctx context.Context, bucketName string) (string, error)
 	listObjects(ctx context.Context, bucket string, opts minio.ListObjectsOptions) <-chan minio.ObjectInfo
+	getObjectRetention(ctx context.Context, bucketName, objectName, versionID string) (mode *minio.RetentionMode, retainUntilDate *time.Time, err error)
+	getObjectLegalHold(ctx context.Context, bucketName, objectName string, opts minio.GetObjectLegalHoldOptions) (status *minio.LegalHoldStatus, err error)
 }
 
 // Interface implementation
@@ -116,6 +119,14 @@ func (c minioClient) listObjects(ctx context.Context, bucket string, opts minio.
 	return c.client.ListObjects(ctx, bucket, opts)
 }
 
+func (c minioClient) getObjectRetention(ctx context.Context, bucketName, objectName, versionID string) (mode *minio.RetentionMode, retainUntilDate *time.Time, err error) {
+	return c.client.GetObjectRetention(ctx, bucketName, objectName, versionID)
+}
+
+func (c minioClient) getObjectLegalHold(ctx context.Context, bucketName, objectName string, opts minio.GetObjectLegalHoldOptions) (status *minio.LegalHoldStatus, err error) {
+	return c.client.GetObjectLegalHold(ctx, bucketName, objectName, opts)
+}
+
 // MCClient interface with all functions to be implemented
 // by mock when testing, it should include all mc/S3Client respective api calls
 // that are used within this project.

restapi/client.go

@@ -19,6 +19,7 @@ package restapi
 import (
 	"context"
 	"fmt"
+	"log"
 	"path/filepath"
 	"regexp"
 	"strings"
@@ -29,6 +30,7 @@ import (
 	"github.com/minio/console/restapi/operations"
 	"github.com/minio/console/restapi/operations/user_api"
 	mc "github.com/minio/mc/cmd"
+	"github.com/minio/mc/pkg/probe"
 	"github.com/minio/minio-go/v7"
 )
 
@@ -62,12 +64,16 @@ func getListObjectsResponse(session *models.Principal, params user_api.ListObjec
 	defer cancel()
 	var prefix string
 	var recursive bool
+	var withVersions bool
 	if params.Prefix != nil {
 		prefix = *params.Prefix
 	}
 	if params.Recursive != nil {
 		recursive = *params.Recursive
 	}
+	if params.WithVersions != nil {
+		withVersions = *params.WithVersions
+	}
 	// bucket request needed to proceed
 	if params.BucketName == "" {
 		return nil, prepareError(errBucketNameNotInRequest)
@@ -80,7 +86,7 @@ func getListObjectsResponse(session *models.Principal, params user_api.ListObjec
 	// defining the client to be used
 	minioClient := minioClient{client: mClient}
 
-	objs, err := listBucketObjects(ctx, minioClient, params.BucketName, prefix, recursive)
+	objs, err := listBucketObjects(ctx, minioClient, params.BucketName, prefix, recursive, withVersions)
 	if err != nil {
 		return nil, prepareError(err)
 	}
@@ -93,17 +99,50 @@ func getListObjectsResponse(session *models.Principal, params user_api.ListObjec
 }
 
 // listBucketObjects gets an array of objects in a bucket
-func listBucketObjects(ctx context.Context, client MinioClient, bucketName string, prefix string, recursive bool) ([]*models.BucketObject, error) {
+func listBucketObjects(ctx context.Context, client MinioClient, bucketName string, prefix string, recursive, withVersions bool) ([]*models.BucketObject, error) {
 	var objects []*models.BucketObject
-	for lsObj := range client.listObjects(ctx, bucketName, minio.ListObjectsOptions{Prefix: prefix, Recursive: recursive}) {
+	for lsObj := range client.listObjects(ctx, bucketName, minio.ListObjectsOptions{Prefix: prefix, Recursive: recursive, WithVersions: withVersions}) {
 		if lsObj.Err != nil {
 			return nil, lsObj.Err
 		}
 		obj := &models.BucketObject{
-			Name:         lsObj.Key,
-			Size:         lsObj.Size,
-			LastModified: lsObj.LastModified.String(),
-			ContentType:  lsObj.ContentType,
+			Name:           lsObj.Key,
+			Size:           lsObj.Size,
+			LastModified:   lsObj.LastModified.String(),
+			ContentType:    lsObj.ContentType,
+			VersionID:      lsObj.VersionID,
+			IsLatest:       lsObj.IsLatest,
+			IsDeleteMarker: lsObj.IsDeleteMarker,
+			UserTags:       lsObj.UserTags,
+		}
+		if !lsObj.IsDeleteMarker {
+			// Add Legal Hold Status if available
+			legalHoldStatus, err := client.getObjectLegalHold(ctx, bucketName, lsObj.Key, minio.GetObjectLegalHoldOptions{VersionID: lsObj.VersionID})
+			if err != nil {
+				errResp := minio.ToErrorResponse(probe.NewError(err).ToGoError())
+				if errResp.Code != "NoSuchObjectLockConfiguration" {
+					log.Printf("error getting legal hold status for %s : %s", lsObj.VersionID, err)
+				}
+
+			} else {
+				if legalHoldStatus != nil {
+					obj.LegalHoldStatus = string(*legalHoldStatus)
+				}
+			}
+			// Add Retention Status if available
+			retention, retUntilDate, err := client.getObjectRetention(ctx, bucketName, lsObj.Key, lsObj.VersionID)
+			if err != nil {
+				errResp := minio.ToErrorResponse(probe.NewError(err).ToGoError())
+				if errResp.Code != "NoSuchObjectLockConfiguration" {
+					log.Printf("error getting retention status for %s : %s", lsObj.VersionID, err)
+				}
+			} else {
+				if retention != nil && retUntilDate != nil {
+					date := *retUntilDate
+					obj.RetentionMode = string(*retention)
+					obj.RetentionUntilDate = date.String()
+				}
+			}
 		}
 		objects = append(objects, obj)
 	}