84 lines
2.4 KiB
Rust
84 lines
2.4 KiB
Rust
use chrono::NaiveDateTime;
|
|
use diesel::{self, QueryDsl, RunQueryDsl, ExpressionMethods, PgConnection};
|
|
use std::iter::Iterator;
|
|
|
|
use models::users::User;
|
|
use schema::{instances, users};
|
|
|
|
#[derive(Identifiable, Queryable, Serialize)]
|
|
pub struct Instance {
|
|
pub id: i32,
|
|
pub public_domain: String,
|
|
pub name: String,
|
|
pub local: bool,
|
|
pub blocked: bool,
|
|
pub creation_date: NaiveDateTime
|
|
}
|
|
|
|
#[derive(Insertable)]
|
|
#[table_name = "instances"]
|
|
pub struct NewInstance {
|
|
pub public_domain: String,
|
|
pub name: String,
|
|
pub local: bool
|
|
}
|
|
|
|
impl Instance {
|
|
pub fn get_local(conn: &PgConnection) -> Option<Instance> {
|
|
instances::table.filter(instances::local.eq(true))
|
|
.limit(1)
|
|
.load::<Instance>(conn)
|
|
.expect("Error loading local instance infos")
|
|
.into_iter().nth(0)
|
|
}
|
|
|
|
pub fn get_remotes(conn: &PgConnection) -> Vec<Instance> {
|
|
instances::table.filter(instances::local.eq(false))
|
|
.load::<Instance>(conn)
|
|
.expect("Error loading remote instances infos")
|
|
}
|
|
|
|
pub fn local_id(conn: &PgConnection) -> i32 {
|
|
Instance::get_local(conn).unwrap().id
|
|
}
|
|
|
|
pub fn insert<'a>(conn: &PgConnection, pub_dom: String, name: String, local: bool) -> Instance {
|
|
diesel::insert_into(instances::table)
|
|
.values(NewInstance {
|
|
public_domain: pub_dom,
|
|
name: name,
|
|
local: local
|
|
})
|
|
.get_result(conn)
|
|
.expect("Error saving new instance")
|
|
}
|
|
|
|
pub fn get(conn: &PgConnection, id: i32) -> Option<Instance> {
|
|
instances::table.filter(instances::id.eq(id))
|
|
.limit(1)
|
|
.load::<Instance>(conn)
|
|
.expect("Error loading local instance infos")
|
|
.into_iter().nth(0)
|
|
}
|
|
|
|
pub fn find_by_domain(conn: &PgConnection, domain: String) -> Option<Instance> {
|
|
instances::table.filter(instances::public_domain.eq(domain))
|
|
.limit(1)
|
|
.load::<Instance>(conn)
|
|
.expect("Couldn't load instance by domain")
|
|
.into_iter().nth(0)
|
|
}
|
|
|
|
pub fn block(&self) {
|
|
unimplemented!()
|
|
}
|
|
|
|
pub fn has_admin(&self, conn: &PgConnection) -> bool {
|
|
users::table.filter(users::instance_id.eq(self.id))
|
|
.filter(users::is_admin.eq(true))
|
|
.load::<User>(conn)
|
|
.expect("Couldn't load admins")
|
|
.len() > 0
|
|
}
|
|
}
|