Julio Biason
3 years ago
1 changed files with 35 additions and 2 deletions
@ -1,16 +1,49 @@
|
||||
/* |
||||
TIN - Time Tracking Application |
||||
Copyright (C) 2021 Julio Biason |
||||
|
||||
This program is free software: you can redistribute it and/or modify |
||||
it under the terms of the GNU Affero General Public License as published by |
||||
the Free Software Foundation, either version 3 of the License, or |
||||
(at your option) any later version. |
||||
|
||||
This program is distributed in the hope that it will be useful, |
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
GNU Affero General Public License for more details. |
||||
|
||||
You should have received a copy of the GNU Affero General Public License |
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/ |
||||
|
||||
use std::path::Path; |
||||
|
||||
use directories::ProjectDirs; |
||||
use sqlx::sqlite::Sqlite; |
||||
use sqlx::sqlite::SqliteConnectOptions; |
||||
use sqlx::sqlite::SqlitePoolOptions; |
||||
use sqlx::Pool; |
||||
|
||||
async fn connect() -> Result<Pool<Sqlite>, sqlx::Error> { |
||||
connect_with(&default_filename()).await |
||||
} |
||||
|
||||
async fn connect_with(filename: &Path) -> Result<Pool<Sqlite>, sqlx::Error> { |
||||
let pool = SqlitePoolOptions::new() |
||||
.connect_with( |
||||
SqliteConnectOptions::new() |
||||
.filename("testing.sqlite") |
||||
.filename(filename) |
||||
.create_if_missing(true), |
||||
) |
||||
.await?; |
||||
sqlx::migrate("migrations").run(&pool).await?; |
||||
sqlx::migrate!("./migrations").run(&pool).await?; |
||||
Ok(pool) |
||||
} |
||||
|
||||
/// Return the default filename for the database.
|
||||
fn default_filename() -> Path { |
||||
match ProjectDirs::from("me", "JulioBiason", "tin.sqlite3") { |
||||
Some(project_dir) => Path(project_dir.config_dir()), |
||||
None => Path("tin.sqlite3"), |
||||
} |
||||
} |
||||
|
Loading…
Reference in new issue