Add libsql+file:// support

Fixes ariga#2198 Signed-off-by: Piotr Jastrzebski <[email protected]>
haaawk · Oct 23, 2023 · 3ecbe34 · 3ecbe34
1 parent 3a137f9
commit 3ecbe34
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/sql/sqlite/driver.go b/sql/sqlite/driver.go
@@ -65,9 +65,13 @@ func init() {
 		sqlclient.DriverOpener(Open),
 		sqlclient.RegisterTxOpener(OpenTx),
 		sqlclient.RegisterCodec(MarshalHCL, EvalHCL),
-		sqlclient.RegisterFlavours("libsql+ws", "libsql+wss"),
+		sqlclient.RegisterFlavours("libsql+ws", "libsql+wss", "libsql+file"),
 		sqlclient.RegisterURLParser(sqlclient.URLParserFunc(func(u *url.URL) *sqlclient.URL {
-			return &sqlclient.URL{URL: u, DSN: strings.TrimPrefix(u.String(), "libsql+"), Schema: mainFile}
+			dns := strings.TrimPrefix(u.String(), "libsql+")
+			if strings.HasPrefix(dns, "file://") {
+				dns = strings.Replace(dns, "file://", "file:", 1)
+			}
+			return &sqlclient.URL{URL: u, DSN: dns, Schema: mainFile}
 		})),
 	)
 }