Added ability to load initial storage data

api/config/config.go

@@ -13,6 +13,7 @@ func ParseFlags() {
 	explorerPath := flag.String("ExplorerDir", "/home/pk/pro/cosmos-explorer/dist", "Path to cosmos-explorer files")
 	tlsCertificatePath := flag.String("Cert", "../example.crt", "Hostname")
 	tlsCertificateKey := flag.String("CertKey", "../example.key", "Hostname")
+	initialDataPath := flag.String("InitialData", "", "Path to JSON containing initial state")
 
 	flag.Parse()
 
@@ -21,6 +22,7 @@ func ParseFlags() {
 	Config.ExplorerPath = *explorerPath
 	Config.TLS_CertificatePath = *tlsCertificatePath
 	Config.TLS_CertificateKey = *tlsCertificateKey
+	Config.DataFilePath = *initialDataPath
 
 	Config.DatabaseAccount = Config.Host
 	Config.DatabaseDomain = Config.Host

api/config/models.go

@@ -10,4 +10,5 @@ type ServerConfig struct {
 	Host                string
 	TLS_CertificatePath string
 	TLS_CertificateKey  string
+	DataFilePath        string
 }

internal/repositories/state.go

@@ -0,0 +1,31 @@
+package repositories
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+
+	repositorymodels "github.com/pikami/cosmium/internal/repository_models"
+)
+
+func LoadStateFS(filePath string) {
+	data, err := os.ReadFile(filePath)
+	if err != nil {
+		log.Fatalf("Error reading state JSON file: %v", err)
+	}
+
+	var state repositorymodels.State
+	if err := json.Unmarshal(data, &state); err != nil {
+		log.Fatalf("Error unmarshalling state JSON: %v", err)
+	}
+
+	fmt.Println("Loaded state:")
+	fmt.Printf("Databases: %d\n", len(state.Databases))
+	fmt.Printf("Collections: %d\n", len(state.Collections))
+	fmt.Printf("Documents: %d\n", len(state.Documents))
+
+	databases = state.Databases
+	collections = state.Collections
+	documents = state.Documents
+}

internal/repository_models/models.go

@@ -110,3 +110,9 @@ type PartitionKeyRange struct {
 	Ts                 int    `json:"_ts"`
 	Lsn                int    `json:"lsn"`
 }
+
+type State struct {
+	Databases   []Database   `json:"databases"`
+	Collections []Collection `json:"collections"`
+	Documents   []Document   `json:"documents"`
+}

main.go

@@ -5,11 +5,16 @@ import (
 
 	"github.com/pikami/cosmium/api"
 	"github.com/pikami/cosmium/api/config"
+	"github.com/pikami/cosmium/internal/repositories"
 )
 
 func main() {
 	config.ParseFlags()
 
+	if config.Config.DataFilePath != "" {
+		repositories.LoadStateFS(config.Config.DataFilePath)
+	}
+
 	router := api.CreateRouter()
 	router.RunTLS(
 		fmt.Sprintf(":%d", config.Config.Port),