feat (v2): v2

Author: zinic

cypher/models/cypher/model.go

@@ -770,14 +770,23 @@ type Literal struct {
 }
 
 func NewLiteral(value any, null bool) *Literal {
+	if !null {
+		if strValue, typeOK := value.(string); typeOK {
+			return NewStringLiteral(strValue)
+		}
+	}
+
 	return &Literal{
 		Value: value,
 		Null:  null,
 	}
 }
 
 func NewStringLiteral(value string) *Literal {
-	return NewLiteral("'"+value+"'", false)
+	return &Literal{
+		Value: "'" + value + "'",
+		Null:  false,
+	}
 }
 
 func (s *Literal) copy() *Literal {

cypher/models/pgsql/format/format.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/specterops/dawgs/cypher/models/pgsql"
 )
@@ -70,6 +71,9 @@ func formatSlice[T any, TS []T](builder *OutputBuilder, slice TS, dataType pgsql
 
 func formatValue(builder *OutputBuilder, value any) error {
 	switch typedValue := value.(type) {
+	case time.Time:
+		builder.Write("'", typedValue.Format(time.RFC3339Nano), "'::timestamp with time zone")
+
 	case uint:
 		builder.Write(strconv.FormatUint(uint64(typedValue), 10))
 
@@ -116,7 +120,11 @@ func formatValue(builder *OutputBuilder, value any) error {
 		return formatSlice(builder, typedValue, pgsql.Int8Array)
 
 	case string:
-		builder.Write("'", typedValue, "'")
+		// Double single quotes per SQL string literal rules
+		builder.Write("'", strings.ReplaceAll(typedValue, "'", "''"), "'")
+
+	case []string:
+		return formatSlice(builder, typedValue, pgsql.TextArray)
 
 	case bool:
 		builder.Write(strconv.FormatBool(typedValue))

cypher/models/pgsql/test/testcase.go

@@ -13,14 +13,13 @@ import (
 	"testing"
 	"time"
 
-	"github.com/specterops/dawgs/drivers/pg"
-
 	"cuelang.org/go/pkg/regexp"
 	"github.com/specterops/dawgs/cypher/frontend"
 	"github.com/specterops/dawgs/cypher/models/cypher"
 	"github.com/specterops/dawgs/cypher/models/pgsql"
 	"github.com/specterops/dawgs/cypher/models/pgsql/translate"
 	"github.com/specterops/dawgs/cypher/models/walk"
+	"github.com/specterops/dawgs/database"
 	"github.com/stretchr/testify/require"
 )
 
@@ -178,7 +177,7 @@ func (s *TranslationTestCase) Assert(t *testing.T, expectedSQL string, kindMappe
 	}
 }
 
-func (s *TranslationTestCase) AssertLive(ctx context.Context, t *testing.T, driver *pg.Driver) {
+func (s *TranslationTestCase) AssertLive(ctx context.Context, t *testing.T, db database.Instance) {
 	if regularQuery, err := frontend.ParseCypher(frontend.NewContext(), s.Cypher); err != nil {
 		t.Fatalf("Failed to compile cypher query: %s - %v", s.Cypher, err)
 	} else {
@@ -195,13 +194,15 @@ func (s *TranslationTestCase) AssertLive(ctx context.Context, t *testing.T, driv
 			}
 		}
 
-		if translation, err := translate.Translate(context.Background(), regularQuery, driver.KindMapper(), s.CypherParams); err != nil {
-			t.Fatalf("Failed to translate cypher query: %s - %v", s.Cypher, err)
-		} else if formattedQuery, err := translate.Translated(translation); err != nil {
-			t.Fatalf("Failed to format SQL translatedQuery: %v", err)
-		} else {
-			require.Nil(t, driver.Run(ctx, "explain "+formattedQuery, translation.Parameters))
-		}
+		require.NoError(t, db.Session(ctx, func(ctx context.Context, driver database.Driver) error {
+			result := driver.Explain(ctx, regularQuery, s.CypherParams)
+
+			if err := result.Close(ctx); err != nil {
+				return err
+			}
+
+			return result.Error()
+		}))
 	}
 }
 

cypher/models/pgsql/test/translation_test.go

@@ -7,7 +7,7 @@ import (
 	"strings"
 	"testing"
 
-	"github.com/specterops/dawgs/drivers/pg/pgutil"
+	"github.com/specterops/dawgs/database/pg/pgutil"
 
 	"github.com/specterops/dawgs/cypher/models/pgsql"
 	"github.com/specterops/dawgs/graph"

cypher/models/pgsql/test/validation_integration_test.go

@@ -9,10 +9,10 @@ import (
 	"runtime/debug"
 	"testing"
 
-	"github.com/jackc/pgx/v5/pgxpool"
+	"github.com/specterops/dawgs/database"
 
 	"github.com/specterops/dawgs"
-	"github.com/specterops/dawgs/drivers/pg"
+	"github.com/specterops/dawgs/database/pg"
 	"github.com/specterops/dawgs/graph"
 	"github.com/specterops/dawgs/util/size"
 	"github.com/stretchr/testify/require"
@@ -32,34 +32,25 @@ func TestTranslationTestCases(t *testing.T) {
 
 	require.NotEmpty(t, pgConnectionStr)
 
-	if pgxPool, err := pgxpool.New(testCtx, pgConnectionStr); err != nil {
-		t.Fatalf("Failed opening database connection: %v", err)
-	} else if connection, err := dawgs.Open(context.TODO(), pg.DriverName, dawgs.Config{
+	if connection, err := dawgs.Open(context.TODO(), pg.DriverName, dawgs.Config{
 		GraphQueryMemoryLimit: size.Gibibyte,
-		Pool:                  pgxPool,
+		ConnectionString:      pgConnectionStr,
 	}); err != nil {
 		t.Fatalf("Failed opening database connection: %v", err)
-	} else if pgConnection, typeOK := connection.(*pg.Driver); !typeOK {
-		t.Fatalf("Invalid connection type: %T", connection)
 	} else {
 		defer connection.Close(testCtx)
 
-		graphSchema := graph.Schema{
-			Graphs: []graph.Graph{{
-				Name: "test",
-				Nodes: graph.Kinds{
-					graph.StringKind("NodeKind1"),
-					graph.StringKind("NodeKind2"),
-				},
-				Edges: graph.Kinds{
-					graph.StringKind("EdgeKind1"),
-					graph.StringKind("EdgeKind2"),
-				},
-			}},
-			DefaultGraph: graph.Graph{
-				Name: "test",
+		graphSchema := database.NewSchema("test", database.Graph{
+			Name: "test",
+			Nodes: graph.Kinds{
+				graph.StringKind("NodeKind1"),
+				graph.StringKind("NodeKind2"),
 			},
-		}
+			Edges: graph.Kinds{
+				graph.StringKind("EdgeKind1"),
+				graph.StringKind("EdgeKind2"),
+			},
+		})
 
 		if err := connection.AssertSchema(testCtx, graphSchema); err != nil {
 			t.Fatalf("Failed asserting graph schema: %v", err)
@@ -79,7 +70,7 @@ func TestTranslationTestCases(t *testing.T) {
 						}
 					}()
 
-					testCase.AssertLive(testCtx, t, pgConnection)
+					testCase.AssertLive(testCtx, t, connection)
 				})
 
 				casesRun += 1

cypher/models/pgsql/visualization/visualizer_test.go

@@ -5,7 +5,7 @@ import (
 	"context"
 	"testing"
 
-	"github.com/specterops/dawgs/drivers/pg/pgutil"
+	"github.com/specterops/dawgs/database/pg/pgutil"
 
 	"github.com/specterops/dawgs/cypher/frontend"
 	"github.com/specterops/dawgs/cypher/models/pgsql/translate"

database/driver.go

@@ -0,0 +1,82 @@
+package database
+
+import (
+	"context"
+
+	"github.com/specterops/dawgs/graph"
+
+	"github.com/specterops/dawgs/cypher/models/cypher"
+)
+
+type Option int
+
+const (
+	OptionReadOnly  Option = 0
+	OptionReadWrite Option = 1
+)
+
+type Result interface {
+	HasNext(ctx context.Context) bool
+	Scan(scanTargets ...any) error
+	Error() error
+	Close(ctx context.Context) error
+
+	// Values returns the next values array from the result.
+	//
+	// Deprecated: This function will be removed in future version.
+	Values() []any
+}
+
+type Driver interface {
+	WithGraph(target Graph) Driver
+
+	CreateNode(ctx context.Context, node *graph.Node) (graph.ID, error)
+	CreateRelationship(ctx context.Context, relationship *graph.Relationship) (graph.ID, error)
+
+	Exec(ctx context.Context, query *cypher.RegularQuery, parameters map[string]any) Result
+	Explain(ctx context.Context, query *cypher.RegularQuery, parameters map[string]any) Result
+	Profile(ctx context.Context, query *cypher.RegularQuery, parameters map[string]any) Result
+	Mapper() graph.ValueMapper
+}
+
+type QueryLogic func(ctx context.Context, driver Driver) error
+
+type Instance interface {
+	AssertSchema(ctx context.Context, schema Schema) error
+	Session(ctx context.Context, driverLogic QueryLogic, options ...Option) error
+	Transaction(ctx context.Context, driverLogic QueryLogic, options ...Option) error
+	Close(ctx context.Context) error
+
+	// FetchKinds retrieves the complete list of kinds available to the database.
+	FetchKinds(ctx context.Context) (graph.Kinds, error)
+}
+
+type errorResult struct {
+	err error
+}
+
+func (s errorResult) HasNext(ctx context.Context) bool {
+	return false
+}
+
+func (s errorResult) Scan(scanTargets ...any) error {
+	return s.err
+}
+
+func (s errorResult) Error() error {
+	return s.err
+}
+
+func (s errorResult) Values() []any {
+	return nil
+}
+
+func (s errorResult) Close(ctx context.Context) error {
+	return nil
+}
+
+func NewErrorResult(err error) Result {
+	return errorResult{
+		err: err,
+	}
+}