refactor: flatten store/postgres into store package

Only Postgres is used as a store backend, so the extra nesting is
unnecessary. Moves all files from store/postgres/ to store/ and
renames the package from postgres to store.

Author: ravisuhag

cli/embed.go

@@ -14,7 +14,7 @@ import (
 	"github.com/raystack/compass/core/namespace"
 	"github.com/raystack/compass/internal/config"
 	compassserver "github.com/raystack/compass/internal/server"
-	"github.com/raystack/compass/store/postgres"
+	"github.com/raystack/compass/store"
 	"github.com/spf13/cobra"
 )
 
@@ -57,21 +57,21 @@ func runEmbed(ctx context.Context, cfg *config.Config, embedType string, batchSi
 	slog.Info("embedding provider initialized", "provider", provider.Name())
 
 	// Init postgres
-	pgClient, err := postgres.NewClient(cfg.DB)
+	pgClient, err := store.NewClient(cfg.DB)
 	if err != nil {
 		return fmt.Errorf("connect to postgres: %w", err)
 	}
 	defer pgClient.Close()
 
-	embeddingRepo, err := postgres.NewEmbeddingRepository(pgClient)
+	embeddingRepo, err := store.NewEmbeddingRepository(pgClient)
 	if err != nil {
 		return err
 	}
 
 	ns := namespace.DefaultNamespace
 
 	if embedType == "all" || embedType == "entity" {
-		entityRepo, err := postgres.NewEntityRepository(pgClient)
+		entityRepo, err := store.NewEntityRepository(pgClient)
 		if err != nil {
 			return err
 		}
@@ -81,7 +81,7 @@ func runEmbed(ctx context.Context, cfg *config.Config, embedType string, batchSi
 	}
 
 	if embedType == "all" || embedType == "document" {
-		docRepo, err := postgres.NewDocumentRepository(pgClient)
+		docRepo, err := store.NewDocumentRepository(pgClient)
 		if err != nil {
 			return err
 		}

internal/config/config.go

@@ -6,14 +6,14 @@ import (
 	"github.com/raystack/compass/core/embedding"
 	"github.com/raystack/compass/internal/client"
 	"github.com/raystack/compass/internal/telemetry"
-	"github.com/raystack/compass/store/postgres"
+	"github.com/raystack/compass/store"
 )
 
 // Config is the root configuration for the compass application.
 type Config struct {
 	LogLevel  string           `yaml:"log_level" mapstructure:"log_level" default:"info"`
 	Telemetry telemetry.Config `mapstructure:"telemetry"`
-	DB        postgres.Config  `mapstructure:"db"`
+	DB        store.Config     `mapstructure:"db"`
 	Service   ServerConfig     `mapstructure:"service"`
 	Client    client.Config    `mapstructure:"client"`
 	Embedding EmbeddingConfig  `mapstructure:"embedding"`

internal/server/bootstrap.go

@@ -19,7 +19,7 @@ import (
 	"github.com/raystack/compass/internal/config"
 	compassmcp "github.com/raystack/compass/internal/mcp"
 	"github.com/raystack/compass/internal/telemetry"
-	"github.com/raystack/compass/store/postgres"
+	"github.com/raystack/compass/store"
 )
 
 // InitLogger sets up the global slog logger with a JSON handler.
@@ -66,39 +66,39 @@ func Start(ctx context.Context, cfg *config.Config, version string) error {
 	}()
 
 	// init user
-	userRepository, err := postgres.NewUserRepository(pgClient)
+	userRepository, err := store.NewUserRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create user repository: %w", err)
 	}
 	userService := user.NewService(userRepository)
 
 	// init star
-	starRepository, err := postgres.NewStarRepository(pgClient)
+	starRepository, err := store.NewStarRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create star repository: %w", err)
 	}
 	starService := star.NewService(starRepository)
 
 	// init namespace
-	namespaceService := namespace.NewService(postgres.NewNamespaceRepository(pgClient), nil)
+	namespaceService := namespace.NewService(store.NewNamespaceRepository(pgClient), nil)
 
 	// init entity system (Postgres-native: tsvector + pg_trgm + pgvector)
-	entityRepo, err := postgres.NewEntityRepository(pgClient)
+	entityRepo, err := store.NewEntityRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create entity repository: %w", err)
 	}
-	edgeRepo, err := postgres.NewEdgeRepository(pgClient)
+	edgeRepo, err := store.NewEdgeRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create edge repository: %w", err)
 	}
-	entitySearchRepo, err := postgres.NewEntitySearchRepository(pgClient)
+	entitySearchRepo, err := store.NewEntitySearchRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create entity search repository: %w", err)
 	}
 	entityService := entity.NewService(entityRepo, edgeRepo, entitySearchRepo)
 
 	// init document system
-	docRepo, err := postgres.NewDocumentRepository(pgClient)
+	docRepo, err := store.NewDocumentRepository(pgClient)
 	if err != nil {
 		return fmt.Errorf("failed to create document repository: %w", err)
 	}
@@ -112,7 +112,7 @@ func Start(ctx context.Context, cfg *config.Config, version string) error {
 		}
 		slog.Info("embedding pipeline enabled", "provider", provider.Name())
 
-		embeddingRepo, err := postgres.NewEmbeddingRepository(pgClient)
+		embeddingRepo, err := store.NewEmbeddingRepository(pgClient)
 		if err != nil {
 			return fmt.Errorf("failed to create embedding repository: %w", err)
 		}
@@ -173,7 +173,7 @@ func Migrate(ctx context.Context, cfg *config.Config, version string) error {
 	}
 
 	// create default namespace
-	nsService := namespace.NewService(postgres.NewNamespaceRepository(pgClient), nil)
+	nsService := namespace.NewService(store.NewNamespaceRepository(pgClient), nil)
 	if _, err = nsService.GetByID(ctx, namespace.DefaultNamespace.ID); errors.Is(err, namespace.ErrNotFound) {
 		if _, err := nsService.MigrateDefault(ctx); err != nil {
 			return fmt.Errorf("problem with migration %w", err)
@@ -204,8 +204,8 @@ func MigrateDown(ctx context.Context, cfg *config.Config, version string) error
 	return nil
 }
 
-func initPostgres(cfg postgres.Config) (*postgres.Client, error) {
-	pgClient, err := postgres.NewClient(cfg)
+func initPostgres(cfg store.Config) (*store.Client, error) {
+	pgClient, err := store.NewClient(cfg)
 	if err != nil {
 		return nil, fmt.Errorf("error creating postgres client: %w", err)
 	}

store/postgres/config.go store/config.gostore/postgres/config.go renamed to store/config.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"net"

store/postgres/document_repository.go store/document_repository.gostore/postgres/document_repository.go renamed to store/document_repository.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"context"

store/postgres/edge_repository.go store/edge_repository.gostore/postgres/edge_repository.go renamed to store/edge_repository.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"context"

store/postgres/embedding_repository.go store/embedding_repository.gostore/postgres/embedding_repository.go renamed to store/embedding_repository.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"context"

store/postgres/entity_repository.go store/entity_repository.gostore/postgres/entity_repository.go renamed to store/entity_repository.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"context"

…ore/postgres/entity_search_repository.go store/entity_search_repository.gostore/postgres/entity_search_repository.go renamed to store/entity_search_repository.go store/postgres/entity_search_repository.go renamed to store/entity_search_repository.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import (
 	"context"

store/postgres/errors.go store/errors.gostore/postgres/errors.go renamed to store/errors.go

@@ -1,4 +1,4 @@
-package postgres
+package store
 
 import "errors"