Source file database.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
(*****************************************************************************)
(*                                                                           *)
(*  Copyright (C) 2026 Yves Ndiaye                                           *)
(*                                                                           *)
(* This Source Code Form is subject to the terms of the Mozilla Public       *)
(* License, v. 2.0. If a copy of the MPL was not distributed with this       *)
(* file, You can obtain one at https://mozilla.org/MPL/2.0/.                 *)
(*                                                                           *)
(*****************************************************************************)

type t = Sqlite3.db

let transaction database f =
  let _ = Sqlite3.exec database "BEGIN TRANSACTION;" in
  match f database with
  | Ok _ as r ->
      let _ = Sqlite3.exec database "COMMIT;" in
      r
  | Error _ as e ->
      let _ = Sqlite3.exec database "ROLLBACK;" in
      e

let open' ?mode ?uri ?memory ?mutex ?cache ?vfs path =
  let db = Sqlite3.db_open ?mode ?uri ?memory ?mutex ?cache ?vfs path in
  let statement = Sqlite3.prepare db "PRAGMA foreign_keys = ON;" in
  let _ = Sqlite3.step statement in
  let _ = Sqlite3.finalize statement in
  db

let create_tables db =
  let sql =
    Printf.sprintf "BEGIN TRANSACTION; %s %s %s %s %s %s %s %s COMMIT;"
      Album.sql_table Artist.sql_table Playlist.sql_table MusicFile.sql_table
      AlbumParticipant.sql_table MusicParticipant.sql_table
      MusicProperty.sql_table PlaylistMusics.sql_table
  in
  Sqlite3.exec db sql

let close db = Sqlite3.db_close db
let errmsg database = Sqlite3.errmsg database
let sqlite3 db = db