fix: moved the read csv into teh lib as its shared between two methods
This commit is contained in:
parent
e21af089d1
commit
88abff575f
3 changed files with 37 additions and 64 deletions
|
@ -7,9 +7,8 @@ use lettre::{
|
||||||
use maud::html;
|
use maud::html;
|
||||||
use rand::distributions::Alphanumeric;
|
use rand::distributions::Alphanumeric;
|
||||||
use rand::{thread_rng, Rng};
|
use rand::{thread_rng, Rng};
|
||||||
use skynet_ldap_backend::{db_init, get_config, Accounts, AccountsNew, Config};
|
use skynet_ldap_backend::{db_init, get_config, Accounts, AccountsNew, Config, read_csv, Record};
|
||||||
use sqlx::{Pool, Sqlite};
|
use sqlx::{Pool, Sqlite};
|
||||||
use std::error::Error;
|
|
||||||
|
|
||||||
#[async_std::main]
|
#[async_std::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
|
@ -42,40 +41,6 @@ async fn main() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, serde::Deserialize)]
|
|
||||||
struct Record {
|
|
||||||
#[serde(rename = "MemID")]
|
|
||||||
mem_id: String,
|
|
||||||
#[serde(rename = "Student Num")]
|
|
||||||
id_student: String,
|
|
||||||
#[serde(rename = "Contact Email")]
|
|
||||||
email: String,
|
|
||||||
#[serde(rename = "Expiry")]
|
|
||||||
expiry: String,
|
|
||||||
#[serde(rename = "First Name")]
|
|
||||||
name_first: String,
|
|
||||||
#[serde(rename = "Last Name")]
|
|
||||||
name_second: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_csv(config: &Config) -> Result<Vec<Record>, Box<dyn Error>> {
|
|
||||||
let mut records: Vec<Record> = vec![];
|
|
||||||
|
|
||||||
if let Ok(mut rdr) = csv::Reader::from_path(format!("{}/{}", &config.home, &config.csv)) {
|
|
||||||
for result in rdr.deserialize() {
|
|
||||||
// Notice that we need to provide a type hint for automatic
|
|
||||||
// deserialization.
|
|
||||||
let record: Record = result?;
|
|
||||||
if record.mem_id.is_empty() {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
records.push(record);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(records)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn check(db: &Pool<Sqlite>, mail: &str) -> bool {
|
async fn check(db: &Pool<Sqlite>, mail: &str) -> bool {
|
||||||
check_pending(db, mail).await && check_users(db, mail).await
|
check_pending(db, mail).await && check_users(db, mail).await
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
use chrono::{Datelike, Utc};
|
use chrono::{Datelike, Utc};
|
||||||
use dotenvy::dotenv;
|
use dotenvy::dotenv;
|
||||||
use ldap3::{LdapConn, Mod, Scope, SearchEntry};
|
use ldap3::{LdapConn, Mod, Scope, SearchEntry};
|
||||||
use skynet_ldap_backend::{get_config, Config};
|
use skynet_ldap_backend::{get_config, Config, read_csv};
|
||||||
use std::collections::{HashMap, HashSet};
|
use std::collections::{HashMap, HashSet};
|
||||||
use std::env;
|
use std::env;
|
||||||
use std::error::Error;
|
use std::error::Error;
|
||||||
|
@ -202,30 +202,3 @@ async fn from_csv(config: &Config) -> Result<HashSet<String>, Box<dyn Error>> {
|
||||||
|
|
||||||
Ok(uids)
|
Ok(uids)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, serde::Deserialize)]
|
|
||||||
struct Record {
|
|
||||||
// #[serde(rename = "MemID")]
|
|
||||||
// id_wolves: String,
|
|
||||||
#[serde(rename = "Student Num")]
|
|
||||||
id_student: String,
|
|
||||||
#[serde(rename = "Contact Email")]
|
|
||||||
email: String,
|
|
||||||
#[serde(rename = "Expiry")]
|
|
||||||
expiry: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_csv(config: &Config) -> Result<Vec<Record>, Box<dyn Error>> {
|
|
||||||
let mut records: Vec<Record> = vec![];
|
|
||||||
|
|
||||||
if let Ok(mut rdr) = csv::Reader::from_path(format!("{}/{}", &config.home, &config.csv)) {
|
|
||||||
for result in rdr.deserialize() {
|
|
||||||
// Notice that we need to provide a type hint for automatic
|
|
||||||
// deserialization.
|
|
||||||
let record: Record = result?;
|
|
||||||
records.push(record);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(records)
|
|
||||||
}
|
|
||||||
|
|
35
src/lib.rs
35
src/lib.rs
|
@ -251,3 +251,38 @@ async fn update_accounts(pool: &Pool<Sqlite>, config: &Config) {
|
||||||
// done with ldap
|
// done with ldap
|
||||||
ldap.unbind().unwrap();
|
ldap.unbind().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
#[derive(Debug, serde::Deserialize)]
|
||||||
|
pub struct Record {
|
||||||
|
#[serde(rename = "MemID")]
|
||||||
|
pub mem_id: String,
|
||||||
|
#[serde(rename = "Student Num")]
|
||||||
|
pub id_student: String,
|
||||||
|
#[serde(rename = "Contact Email")]
|
||||||
|
pub email: String,
|
||||||
|
#[serde(rename = "Expiry")]
|
||||||
|
pub expiry: String,
|
||||||
|
#[serde(rename = "First Name")]
|
||||||
|
pub name_first: String,
|
||||||
|
#[serde(rename = "Last Name")]
|
||||||
|
pub name_second: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn read_csv(config: &Config) -> Result<Vec<Record>, Box<dyn std::error::Error>> {
|
||||||
|
let mut records: Vec<Record> = vec![];
|
||||||
|
|
||||||
|
if let Ok(mut rdr) = csv::Reader::from_path(format!("{}/{}", &config.home, &config.csv)) {
|
||||||
|
for result in rdr.deserialize() {
|
||||||
|
// Notice that we need to provide a type hint for automatic
|
||||||
|
// deserialization.
|
||||||
|
let record: Record = result?;
|
||||||
|
if record.mem_id.is_empty() {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
records.push(record);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(records)
|
||||||
|
}
|
Loading…
Reference in a new issue