Generate legit ResourceIds for SDK compatibility

internal/repositories/collections.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/google/uuid"
 	repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+	"github.com/pikami/cosmium/internal/resourceid"
 	structhidrators "github.com/pikami/cosmium/internal/struct_hidrators"
 	"golang.org/x/exp/maps"
 )
@@ -45,19 +46,22 @@ func DeleteCollection(databaseId string, collectionId string) repositorymodels.R
 }
 
 func CreateCollection(databaseId string, newCollection repositorymodels.Collection) (repositorymodels.Collection, repositorymodels.RepositoryStatus) {
-	if _, ok := storeState.Databases[databaseId]; !ok {
+	var ok bool
+	var database repositorymodels.Database
+	if database, ok = storeState.Databases[databaseId]; !ok {
 		return repositorymodels.Collection{}, repositorymodels.StatusNotFound
 	}
 
-	if _, ok := storeState.Collections[databaseId][newCollection.ID]; ok {
+	if _, ok = storeState.Collections[databaseId][newCollection.ID]; ok {
 		return repositorymodels.Collection{}, repositorymodels.Conflict
 	}
 
 	newCollection = structhidrators.Hidrate(newCollection).(repositorymodels.Collection)
 
 	newCollection.TimeStamp = time.Now().Unix()
-	newCollection.UniqueID = uuid.New().String()
-	newCollection.ETag = fmt.Sprintf("\"%s\"", newCollection.UniqueID)
+	newCollection.ResourceID = resourceid.NewCombined(database.ResourceID, resourceid.New())
+	newCollection.ETag = fmt.Sprintf("\"%s\"", uuid.New())
+	newCollection.Self = fmt.Sprintf("dbs/%s/colls/%s/", database.ResourceID, newCollection.ResourceID)
 
 	storeState.Collections[databaseId][newCollection.ID] = newCollection
 	storeState.Documents[databaseId][newCollection.ID] = make(map[string]repositorymodels.Document)

internal/repositories/databases.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/google/uuid"
 	repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+	"github.com/pikami/cosmium/internal/resourceid"
 	"golang.org/x/exp/maps"
 )
 
@@ -37,8 +38,10 @@ func CreateDatabase(newDatabase repositorymodels.Database) (repositorymodels.Dat
 	}
 
 	newDatabase.TimeStamp = time.Now().Unix()
-	newDatabase.UniqueID = uuid.New().String()
-	newDatabase.ETag = fmt.Sprintf("\"%s\"", newDatabase.UniqueID)
+	newDatabase.ResourceID = resourceid.New()
+	newDatabase.ETag = fmt.Sprintf("\"%s\"", uuid.New())
+	newDatabase.Self = fmt.Sprintf("dbs/%s/", newDatabase.ResourceID)
+
 	storeState.Databases[newDatabase.ID] = newDatabase
 	storeState.Collections[newDatabase.ID] = make(map[string]repositorymodels.Collection)
 	storeState.Documents[newDatabase.ID] = make(map[string]map[string]repositorymodels.Document)

internal/repositories/documents.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/google/uuid"
 	repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+	"github.com/pikami/cosmium/internal/resourceid"
 	"github.com/pikami/cosmium/parsers"
 	"github.com/pikami/cosmium/parsers/nosql"
 	memoryexecutor "github.com/pikami/cosmium/query_executors/memory_executor"
@@ -60,17 +61,19 @@ func DeleteDocument(databaseId string, collectionId string, documentId string) r
 }
 
 func CreateDocument(databaseId string, collectionId string, document map[string]interface{}) (repositorymodels.Document, repositorymodels.RepositoryStatus) {
-	var documentId string
 	var ok bool
+	var documentId string
+	var database repositorymodels.Database
+	var collection repositorymodels.Collection
 	if documentId, ok = document["id"].(string); !ok || documentId == "" {
 		return repositorymodels.Document{}, repositorymodels.BadRequest
 	}
 
-	if _, ok := storeState.Databases[databaseId]; !ok {
+	if database, ok = storeState.Databases[databaseId]; !ok {
 		return repositorymodels.Document{}, repositorymodels.StatusNotFound
 	}
 
-	if _, ok = storeState.Collections[databaseId][collectionId]; !ok {
+	if collection, ok = storeState.Collections[databaseId][collectionId]; !ok {
 		return repositorymodels.Document{}, repositorymodels.StatusNotFound
 	}
 
@@ -79,8 +82,9 @@ func CreateDocument(databaseId string, collectionId string, document map[string]
 	}
 
 	document["_ts"] = time.Now().Unix()
-	document["_rid"] = uuid.New().String()
-	document["_etag"] = fmt.Sprintf("\"%s\"", document["_rid"])
+	document["_rid"] = resourceid.NewCombined(database.ResourceID, collection.ResourceID, resourceid.New())
+	document["_etag"] = fmt.Sprintf("\"%s\"", uuid.New())
+	document["_self"] = fmt.Sprintf("dbs/%s/colls/%s/docs/%s/", database.ResourceID, collection.ResourceID, document["_rid"])
 
 	storeState.Documents[databaseId][collectionId][documentId] = document
 

internal/repositories/partition_key_ranges.go

@@ -1,22 +1,43 @@
 package repositories
 
-import repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+import (
+	"fmt"
 
+	"github.com/google/uuid"
+	repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+	"github.com/pikami/cosmium/internal/resourceid"
+)
+
+// I have no idea what this is tbh
 func GetPartitionKeyRanges(databaseId string, collectionId string) ([]repositorymodels.PartitionKeyRange, repositorymodels.RepositoryStatus) {
-	// I have no idea what this is tbh
+	var ok bool
+	var database repositorymodels.Database
+	var collection repositorymodels.Collection
+	if database, ok = storeState.Databases[databaseId]; !ok {
+		return make([]repositorymodels.PartitionKeyRange, 0), repositorymodels.StatusNotFound
+	}
+
+	if collection, ok = storeState.Collections[databaseId][collectionId]; !ok {
+		return make([]repositorymodels.PartitionKeyRange, 0), repositorymodels.StatusNotFound
+	}
+
+	pkrResourceId := resourceid.NewCombined(database.ResourceID, collection.ResourceID, resourceid.New())
+	pkrSelf := fmt.Sprintf("dbs/%s/colls/%s/pkranges/%s/", database.ResourceID, collection.ResourceID, pkrResourceId)
+	etag := fmt.Sprintf("\"%s\"", uuid.New())
+
 	return []repositorymodels.PartitionKeyRange{
 		{
-			Rid:                "ZxlyAP7rKwACAAAAAAAAUA==",
+			ResourceID:         pkrResourceId,
 			ID:                 "0",
-			Etag:               "\"00005504-0000-0100-0000-65c555490000\"",
+			Etag:               etag,
 			MinInclusive:       "",
 			MaxExclusive:       "FF",
 			RidPrefix:          0,
-			Self:               "dbs/ZxlyAA==/colls/ZxlyAP7rKwA=/pkranges/ZxlyAP7rKwACAAAAAAAAUA==/",
+			Self:               pkrSelf,
 			ThroughputFraction: 1,
 			Status:             "online",
 			Parents:            []interface{}{},
-			Ts:                 1707431241,
+			TimeStamp:          collection.TimeStamp,
 			Lsn:                17,
 		},
 	}, repositorymodels.StatusOk