surwiki/build.rs

78 lines
2.4 KiB
Rust
Raw Normal View History

2022-04-03 14:47:43 +03:00
#[macro_use]
extern crate diesel;
2017-08-19 23:48:51 +03:00
use diesel::prelude::*;
2022-04-03 14:47:43 +03:00
use diesel::Connection;
2022-04-03 23:16:45 +03:00
use quote::quote;
2017-08-19 23:48:51 +03:00
use std::env;
use std::fs::File;
use std::io::prelude::*;
use std::path::Path;
use walkdir::WalkDir;
#[allow(dead_code)]
mod sqlfunc {
use diesel::sql_types::Text;
sql_function!(fn markdown_to_fts(text: Text) -> Text);
sql_function!(fn theme_from_str_hash(text: Text) -> Text);
}
2017-08-19 23:48:51 +03:00
fn main() {
let out_dir = env::var("OUT_DIR").expect("cargo must set OUT_DIR");
let db_path = Path::new(&out_dir).join("build.db");
2022-04-03 14:47:43 +03:00
let db_path = db_path
.to_str()
.expect("Will only work for Unicode-representable paths");
2017-08-19 23:48:51 +03:00
let _ignore_failure = std::fs::remove_file(db_path);
2022-04-03 15:29:35 +03:00
let connection = SqliteConnection::establish(db_path)
.unwrap_or_else(|_| panic!("Error esablishing a database connection to {}", db_path));
2017-08-19 23:48:51 +03:00
// Integer is a dummy placeholder. Compiling fails when passing ().
diesel::expression::sql_literal::sql::<diesel::sql_types::Integer>("PRAGMA foreign_keys = ON")
.execute(&connection)
.expect("Should be able to enable foreign keys");
2017-08-19 23:48:51 +03:00
2022-04-03 14:47:43 +03:00
sqlfunc::markdown_to_fts::register_impl(&connection, |_: String| -> String { unreachable!() })
.unwrap();
sqlfunc::theme_from_str_hash::register_impl(&connection, |_: String| -> String {
unreachable!()
})
.unwrap();
2017-12-15 18:19:44 +03:00
diesel_migrations::run_pending_migrations(&connection).unwrap();
2017-08-19 23:48:51 +03:00
let infer_schema_path = Path::new(&out_dir).join("infer_schema.rs");
let mut file = File::create(infer_schema_path).expect("Unable to open file for writing");
2022-04-03 14:47:43 +03:00
file.write_all(
quote! {
mod __diesel_infer_schema_articles {
infer_table_from_schema!(#db_path, "articles");
}
pub use self::__diesel_infer_schema_articles::*;
2022-04-03 14:47:43 +03:00
mod __diesel_infer_schema_article_revisions {
infer_table_from_schema!(#db_path, "article_revisions");
}
pub use self::__diesel_infer_schema_article_revisions::*;
}
2022-04-03 23:16:45 +03:00
.to_string()
2022-04-03 14:47:43 +03:00
.as_bytes(),
)
.expect("Unable to write to file");
2017-08-19 23:48:51 +03:00
2022-04-03 14:47:43 +03:00
for entry in WalkDir::new("migrations")
.into_iter()
.filter_map(|e| e.ok())
{
2017-08-19 23:48:51 +03:00
println!("cargo:rerun-if-changed={}", entry.path().display());
}
// For build_config.rs
2022-04-03 14:47:43 +03:00
for env_var in &["CONTINUOUS_INTEGRATION", "TRAVIS_BRANCH", "TRAVIS_COMMIT"] {
println!("cargo:rerun-if-env-changed={}", env_var);
}
2017-08-19 23:48:51 +03:00
}