Added schema validation. Fleshed out options for DisconnectAndClear() and Close().

Author: Chriztiaan

PowerSync/PowerSync.Common/Client/PowerSyncDatabase.cs

@@ -265,7 +265,7 @@ public async Task UpdateSchema(Schema schema)
 
         try
         {
-            // schema.Validate();
+            schema.Validate();
         }
         catch (Exception ex)
         {
@@ -356,12 +356,19 @@ public async Task Disconnect()
         syncStreamStatusCts?.Cancel();
     }
 
-    public async Task DisconnectAndClear()
+    /// <summary>
+    /// Disconnect and clear the database.
+    /// Use this when logging out.
+    /// The database can still be queried after this is called, but the tables
+    /// would be empty.
+    /// 
+    /// To preserve data in local-only tables, set clearLocal to false.
+    /// </summary>
+    public async Task DisconnectAndClear(bool clearLocal = true)
     {
         await Disconnect();
         await WaitForReady();
 
-        bool clearLocal = true;
 
         await Database.WriteTransaction(async tx =>
         {
@@ -373,11 +380,26 @@ await Database.WriteTransaction(async tx =>
         Emit(new PowerSyncDBEvent { StatusChanged = CurrentStatus });
     }
 
-    public new async Task Close()
+    /// <summary>
+    /// Close the database, releasing resources.
+    ///
+    /// Also disconnects any active connection.
+    /// 
+    /// Once close is called, this connection cannot be used again - a new one
+    /// must be constructed.
+    /// </summary>
+    public async Task Close(bool disconnect = true)
     {
-        base.Close();
         await WaitForReady();
 
+        if (Closed) return;
+
+        if (disconnect)
+        {
+            await Disconnect();
+        }
+
+        base.Close();
         syncStreamImplementation?.Close();
         BucketStorageAdapter?.Close();
 

PowerSync/PowerSync.Common/DB/Schema/Schema.cs

@@ -7,6 +7,14 @@ public class Schema(Dictionary<string, Table> tables)
 {
     private readonly Dictionary<string, Table> Tables = tables;
 
+    public void Validate()
+    {
+        foreach (var table in Tables.Values)
+        {
+            table.Validate();
+        }
+    }
+
     public string ToJSON()
     {
         var jsonObject = new

PowerSync/PowerSync.Common/DB/Schema/Table.cs

@@ -1,5 +1,6 @@
 namespace PowerSync.Common.DB.Schema;
 
+using System.Text.RegularExpressions;
 using Newtonsoft.Json;
 
 public class TableOptions(
@@ -19,7 +20,10 @@ public class TableOptions(
 
 public class Table
 {
-    protected TableOptions Options { get; set; }
+    private static readonly Regex InvalidSQLCharacters = new Regex(@"[""'%,.#\s\[\]]", RegexOptions.Compiled);
+
+
+    protected TableOptions Options { get; init; } = null!;
 
     public Dictionary<string, ColumnType> Columns;
     public Dictionary<string, List<string>> Indexes;
@@ -48,6 +52,53 @@ [.. kv.Value.Select(name =>
         Indexes = Options?.Indexes ?? [];
     }
 
+    public void Validate()
+    {
+        if (!string.IsNullOrWhiteSpace(Options.ViewName) && InvalidSQLCharacters.IsMatch(Options.ViewName!))
+        {
+            throw new Exception($"Invalid characters in view name: {Options.ViewName}");
+        }
+
+        if (Columns.Count > Column.MAX_AMOUNT_OF_COLUMNS)
+        {
+            throw new Exception($"Table has too many columns. The maximum number of columns is {Column.MAX_AMOUNT_OF_COLUMNS}.");
+        }
+
+        var columnNames = new HashSet<string> { "id" };
+
+        foreach (var columnName in Columns.Keys)
+        {
+            if (columnName == "id")
+            {
+                throw new Exception("An id column is automatically added, custom id columns are not supported");
+            }
+
+            if (InvalidSQLCharacters.IsMatch(columnName))
+            {
+                throw new Exception($"Invalid characters in column name: {columnName}");
+            }
+
+            columnNames.Add(columnName);
+        }
+
+        foreach (var (indexName, indexColumns) in Indexes)
+        {
+
+            if (InvalidSQLCharacters.IsMatch(indexName))
+            {
+                throw new Exception($"Invalid characters in index name: {indexName}");
+            }
+
+            foreach (var indexColumn in indexColumns)
+            {
+                if (!columnNames.Contains(indexColumn))
+                {
+                    throw new Exception($"Column {indexColumn} not found for index {indexName}");
+                }
+            }
+        }
+    }
+
     public string ToJSON(string Name = "")
     {
         var jsonObject = new

Tests/PowerSync/PowerSync.Common.Tests/Client/Sync/BucketStorageTests.cs

@@ -70,7 +70,7 @@ public async Task InitializeAsync()
         db = new PowerSyncDatabase(new PowerSyncDatabaseOptions
         {
             Database = new SQLOpenOptions { DbFilename = "powersync.db" },
-            Schema = TestSchema.appSchema,
+            Schema = TestSchema.AppSchema,
         });
         await db.Init();
         bucketStorage = new SqliteBucketStorage(db.Database, createLogger());
@@ -496,7 +496,7 @@ await Assert.ThrowsAsync<SqliteException>(async () =>
         powersync = new PowerSyncDatabase(new PowerSyncDatabaseOptions
         {
             Database = new SQLOpenOptions { DbFilename = dbName },
-            Schema = TestSchema.appSchema,
+            Schema = TestSchema.AppSchema,
         });
         await powersync.Init();
 
@@ -515,7 +515,7 @@ public async Task ShouldRemoveTypes()
         var powersync = new PowerSyncDatabase(new PowerSyncDatabaseOptions
         {
             Database = new SQLOpenOptions { DbFilename = dbName },
-            Schema = TestSchema.appSchema,
+            Schema = TestSchema.AppSchema,
         });
 
         await powersync.Init();
@@ -557,7 +557,7 @@ await Assert.ThrowsAsync<SqliteException>(async () =>
         powersync = new PowerSyncDatabase(new PowerSyncDatabaseOptions
         {
             Database = new SQLOpenOptions { DbFilename = dbName },
-            Schema = TestSchema.appSchema,
+            Schema = TestSchema.AppSchema,
         });
         await powersync.Init();
 

Tests/PowerSync/PowerSync.Common.Tests/Client/Sync/CRUDTests.cs

@@ -20,7 +20,7 @@ public async Task InitializeAsync()
         db = new PowerSyncDatabase(new PowerSyncDatabaseOptions
         {
             Database = new SQLOpenOptions { DbFilename = dbName },
-            Schema = TestSchema.appSchema,
+            Schema = TestSchema.AppSchema,
         });
         await db.Init();
     }

Tests/PowerSync/PowerSync.Common.Tests/TestSchema.cs

@@ -4,7 +4,7 @@ namespace PowerSync.Common.Tests;
 
 public class TestSchema
 {
-    public static Table assets = new Table(new Dictionary<string, ColumnType>
+    public static readonly Table Assets = new Table(new Dictionary<string, ColumnType>
         {
             { "created_at", ColumnType.TEXT },
             { "make", ColumnType.TEXT },
@@ -19,15 +19,15 @@ public class TestSchema
             Indexes = new Dictionary<string, List<string>> { { "makemodel", new List<string> { "make", "model" } } }
         });
 
-    public static Table customers = new Table(new Dictionary<string, ColumnType>
+    public static readonly Table Customers = new Table(new Dictionary<string, ColumnType>
         {
             { "name", ColumnType.TEXT },
             { "email", ColumnType.TEXT }
         });
 
-    public static Schema appSchema = new Schema(new Dictionary<string, Table>
+    public static readonly Schema AppSchema = new Schema(new Dictionary<string, Table>
         {
-            { "assets", assets },
-            { "customers", customers }
+            { "assets", Assets },
+            { "customers", Customers }
         });
 }