diff --git a/services/graph/pkg/service/v0/driveitems.go b/services/graph/pkg/service/v0/driveitems.go index 69832e4ee0..e23ceb3ad2 100644 --- a/services/graph/pkg/service/v0/driveitems.go +++ b/services/graph/pkg/service/v0/driveitems.go @@ -237,12 +237,9 @@ func (g Graph) getSpecialDriveItems(ctx context.Context, baseURL *url.URL, space return nil } metadata := space.Opaque.Map - names := map[string]string{ - SpaceImageSpecialFolderName: "/.space/logo.png", - ReadmeSpecialFolderName: "/.space/readme.md", - } + names := [2]string{SpaceImageSpecialFolderName, ReadmeSpecialFolderName} - for itemName, itemPath := range names { + for _, itemName := range names { // The default is a path relative to the space root var ref storageprovider.Reference if itemID, ok := metadata[itemName]; ok { @@ -252,16 +249,11 @@ func (g Graph) getSpecialDriveItems(ctx context.Context, baseURL *url.URL, space ref = storageprovider.Reference{ ResourceId: &rid, } - } else { - ref = storageprovider.Reference{ - ResourceId: space.GetRoot(), - Path: itemPath, + spaceItem := g.getSpecialDriveItem(ctx, ref, itemName, baseURL, space) + if spaceItem != nil { + spaceItems = append(spaceItems, *spaceItem) } } - spaceItem := g.getSpecialDriveItem(ctx, ref, itemName, baseURL, space) - if spaceItem != nil { - spaceItems = append(spaceItems, *spaceItem) - } } // cache properties