survey/src/tests-migrate.rs

/*
 * Copyright (C) 2021  Aravinth Manivannan <realaravinth@batsense.net>
 *
 * 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::env;

use sqlx::postgres::PgPoolOptions;

mod settings;

pub use settings::Settings;

#[cfg(not(tarpaulin_include))]
#[actix_rt::main]
async fn main() {
    let settings = Settings::new().unwrap();

    let db = PgPoolOptions::new()
        .max_connections(settings.database.pool)
        .connect(&settings.database.url)
        .await
        .expect("Unable to form database pool");

    for arg in env::args() {
        if arg == "--build" {
            println!("Building cache buster config");
            build();
        }
    }

    sqlx::migrate!("./migrations/").run(&db).await.unwrap();
}

fn build() {
    use std::process::Command;

    // note: add error checking yourself.
    let output = Command::new("git")
        .args(["rev-parse", "HEAD"])
        .output()
        .unwrap();
    let git_hash = String::from_utf8(output.stdout).unwrap();
    println!("cargo:rustc-env=GIT_HASH={}", git_hash);

    //    let yml = include_str!("../openapi.yaml");
    //    let api_json: serde_json::Value = serde_yaml::from_str(yml).unwrap();
    //    println!(
    //        "cargo:rustc-env=OPEN_API_DOCS={}",
    //        serde_json::to_string(&api_json).unwrap()
    //    );
    cache_bust();
}

fn cache_bust() {
    use cache_buster::BusterBuilder;
    let types = vec![
        mime::IMAGE_PNG,
        mime::IMAGE_SVG,
        mime::IMAGE_JPEG,
        mime::IMAGE_GIF,
        mime::APPLICATION_JAVASCRIPT,
        mime::TEXT_CSS,
    ];

    let config = BusterBuilder::default()
        .source("./static/cache")
        .result("./assets")
        .mime_types(types)
        .follow_links(true)
        .build()
        .unwrap();

    config.process().unwrap();
}