AWG: Added migration

src/main.rs

@@ -4,12 +4,14 @@ mod auth;
 mod config;
 mod i18n;
 mod oidc;
+mod sqlite_migration;
 mod telegram;
 mod user;
 mod vpn;
 
 use std::sync::Arc;
 
+use async_trait::async_trait;
 use cot::auth::PasswordVerificationResult;
 use cot::cli::CliMetadata;
 use cot::common_types::Password;
@@ -21,7 +23,7 @@ use cot::db::Database;
 use cot::form::{Form, FormResult};
 use cot::html::Html;
 use cot::middleware::SessionMiddleware;
-use cot::project::RegisterAppsContext;
+use cot::project::{ProjectContext, RegisterAppsContext};
 use cot::request::extractors::{RequestForm, UrlQuery};
 use cot::response::IntoResponse;
 use cot::router::method::get;
@@ -181,11 +183,19 @@ struct AmneziaFellowApp {
     config: Arc<AppConfig>,
 }
 
+#[async_trait]
 impl App for AmneziaFellowApp {
     fn name(&self) -> &'static str {
         env!("CARGO_PKG_NAME")
     }
 
+    async fn init(&self, context: &mut ProjectContext) -> cot::Result<()> {
+        if let Some(db) = context.try_database() {
+            sqlite_migration::run_if_requested(&self.config, db).await?;
+        }
+        Ok(())
+    }
+
     fn router(&self) -> Router {
         Router::with_urls([
             Route::with_handler_and_name(
@@ -296,8 +306,11 @@ impl Project for AmneziaFellowProject {
                 "  Priority: env var > DB override > compiled default.\n",
                 "\n",
                 "  Database (required for most features):\n",
-                "    AMNEZIA_FELLOW_DATABASE_URL    SQLite connection URL\n",
+                "    AMNEZIA_FELLOW_DATABASE_URL    SQLite or PostgreSQL connection URL\n",
                 "                                   Example: sqlite:///data/amnezia-fellow.sqlite3?mode=rwc\n",
+                "                                   Example: postgresql://user:pass@postgres:5432/amnezia_fellow\n",
+                "    AMNEZIA_FELLOW_MIGRATE_SQLITE SQLite path/URL to import into PostgreSQL once\n",
+                "                                   Example: /data/amnezia-fellow.sqlite3\n",
                 "\n",
                 "  Server:\n",
                 "    AMNEZIA_FELLOW_LOG_LEVEL       Tracing filter (default: info)\n",