initial setup + tests

Author: davidabram

libsql.go

@@ -395,6 +395,36 @@ type conn struct {
 	nativePtr C.libsql_connection_t
 }
 
+type extension struct {
+	Library string
+	Entry string
+}
+
+func (c *conn) loadExtensions(exts []extension) error {
+    for _, ext := range exts {
+        if err := c.loadExtension(ext.Library, ext.Entry); err != nil {
+            return err
+        }
+    }
+    return nil
+}
+
+func (c *conn) LoadExtension(lib string, entry string) error {
+	if err := c.loadExtension(lib, entry); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *conn) loadExtension(lib string, entry string) error {
+	var errMsg *C.char
+	statusCode := C.libsql_load_extension(c.nativePtr, C.CString(lib), C.CString(entry), nil)
+	if statusCode != 0 {
+		return libsqlError(fmt.Sprintf("failed to load extension %s with entry point %s", lib, entry), statusCode, errMsg)
+	}
+	return nil
+}
+
 func (c *conn) Prepare(query string) (sqldriver.Stmt, error) {
 	return c.PrepareContext(context.Background(), query)
 }

libsql_test.go

@@ -1359,3 +1359,71 @@ func TestErrorRowsNext(t *testing.T) {
 		}
 	})
 }
+
+// To run this, set LIBSQL_TEST_EXTENSION to the full path of a valid SQLite extension
+// and (optionally) LIBSQL_TEST_EXTENSION_ENTRY to its init symbol (defaults to "sqlite3_extension_init").
+func TestLoadExtension_Existing(t *testing.T) {
+    extPath := os.Getenv("LIBSQL_TEST_EXTENSION")
+    if extPath == "" {
+        t.Skip("LIBSQL_TEST_EXTENSION not set; skipping existing‐extension load test")
+    }
+    entryPoint := os.Getenv("LIBSQL_TEST_EXTENSION_ENTRY")
+    if entryPoint == "" {
+        entryPoint = "sqlite3_extension_init"
+    }
+
+    db, err := sql.Open("libsql", ":memory:")
+    if err != nil {
+        t.Fatal(err)
+    }
+    defer db.Close()
+
+    ctx := context.Background()
+    sqlConn, err := db.Conn(ctx)
+    if err != nil {
+        t.Fatal(err)
+    }
+    defer sqlConn.Close()
+
+    err = sqlConn.Raw(func(driverConn interface{}) error {
+        cImpl, ok := driverConn.(*conn)
+        if !ok {
+            return fmt.Errorf("unexpected driverConn type %T", driverConn)
+        }
+        return cImpl.LoadExtension(extPath, entryPoint)
+    })
+
+    if err != nil {
+        t.Fatalf("failed to load existing extension %q (entry %q): %v", extPath, entryPoint, err)
+    }
+}
+
+func TestLoadExtension_Nonexistent(t *testing.T) {
+    db, err := sql.Open("libsql", ":memory:")
+    if err != nil {
+        t.Fatal(err)
+    }
+    defer db.Close()
+
+    ctx := context.Background()
+    sqlConn, err := db.Conn(ctx)
+    if err != nil {
+        t.Fatal(err)
+    }
+    defer sqlConn.Close()
+
+    err = sqlConn.Raw(func(driverConn interface{}) error {
+        cImpl, ok := driverConn.(*conn)
+        if !ok {
+            return fmt.Errorf("unexpected driverConn type %T", driverConn)
+        }
+        return cImpl.LoadExtension("nonexistent_extension.so", "entry_point")
+    })
+
+    if err == nil {
+        t.Fatal("expected error loading nonexistent extension, got nil")
+    }
+    if !strings.Contains(err.Error(), "failed to load extension") {
+        t.Fatalf("unexpected error loading extension: %v", err)
+    }
+}