2020-05-06 13:25:54 +00:00
|
|
|
#[macro_use]
|
|
|
|
extern crate diesel;
|
2020-05-09 06:37:08 +00:00
|
|
|
#[macro_use]
|
2020-05-11 19:33:11 +00:00
|
|
|
extern crate derive_new;
|
|
|
|
// #[macro_use]
|
|
|
|
// extern crate comp;
|
|
|
|
// #[macro_use]
|
|
|
|
// extern crate validator_derive;
|
2020-05-09 06:37:08 +00:00
|
|
|
extern crate bcrypt;
|
|
|
|
extern crate custom_error;
|
|
|
|
extern crate regex;
|
|
|
|
extern crate validator;
|
2020-05-06 13:25:54 +00:00
|
|
|
|
2020-05-12 12:07:53 +00:00
|
|
|
use actix_web::{
|
|
|
|
error, get, middleware, post, web, App, Error, HttpResponse, HttpServer,
|
|
|
|
};
|
2020-05-06 13:25:54 +00:00
|
|
|
|
|
|
|
use yarte::Template;
|
|
|
|
|
2020-05-09 06:37:08 +00:00
|
|
|
use actix_web_httpauth::middleware::HttpAuthentication;
|
2020-05-06 13:25:54 +00:00
|
|
|
|
|
|
|
use actix_http::cookie::SameSite;
|
2020-05-09 06:37:08 +00:00
|
|
|
use actix_identity::{CookieIdentityPolicy, IdentityService};
|
2020-05-06 13:25:54 +00:00
|
|
|
use rand::Rng;
|
|
|
|
|
|
|
|
use actix_files as fs;
|
|
|
|
|
|
|
|
use diesel::prelude::*;
|
|
|
|
use diesel::r2d2::{self, ConnectionManager};
|
2020-05-09 06:37:08 +00:00
|
|
|
// use middlewares::csrf;
|
|
|
|
// use routes;
|
|
|
|
// use routes::users;
|
|
|
|
// use utils;
|
2020-05-06 13:25:54 +00:00
|
|
|
|
|
|
|
mod actions;
|
2020-05-09 06:37:08 +00:00
|
|
|
mod errors;
|
|
|
|
mod middlewares;
|
2020-05-06 13:25:54 +00:00
|
|
|
mod models;
|
|
|
|
mod routes;
|
|
|
|
mod schema;
|
|
|
|
mod types;
|
2020-05-09 06:37:08 +00:00
|
|
|
mod utils;
|
2020-05-06 13:25:54 +00:00
|
|
|
|
|
|
|
#[macro_use]
|
|
|
|
extern crate log;
|
|
|
|
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
|
|
struct MyObj {
|
|
|
|
name: String,
|
|
|
|
// number: i32,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[get("/{id}/{name}")]
|
|
|
|
async fn index(info: web::Path<(u32, String)>) -> Result<HttpResponse, Error> {
|
|
|
|
let (id, name) = (info.0, info.1.clone());
|
|
|
|
let template = models::CardTemplate {
|
|
|
|
title: "My Title",
|
|
|
|
body: name,
|
|
|
|
num: id,
|
|
|
|
};
|
|
|
|
template
|
|
|
|
.call()
|
|
|
|
.map(|body| HttpResponse::Ok().content_type("text/html").body(body))
|
2020-05-12 12:07:53 +00:00
|
|
|
.map_err(|_| {
|
|
|
|
error::ErrorInternalServerError("Error while parsing template")
|
|
|
|
})
|
2020-05-06 13:25:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// This handler uses json extractor
|
|
|
|
#[post("/extractor")]
|
|
|
|
async fn extract_my_obj(item: web::Json<MyObj>) -> HttpResponse {
|
|
|
|
debug!("model: {:?}", item);
|
|
|
|
HttpResponse::Ok().json(item.0) // <- send response
|
|
|
|
}
|
|
|
|
|
|
|
|
pub struct UserServiceImpl;
|
|
|
|
|
|
|
|
impl UserServiceImpl {
|
|
|
|
pub fn new() -> Self {
|
|
|
|
UserServiceImpl {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[actix_rt::main]
|
|
|
|
async fn main() -> std::io::Result<()> {
|
|
|
|
std::env::set_var("RUST_LOG", "debug");
|
|
|
|
env_logger::init();
|
|
|
|
dotenv::dotenv().ok();
|
|
|
|
|
2020-05-12 12:07:53 +00:00
|
|
|
let basic_auth_middleware =
|
|
|
|
HttpAuthentication::basic(utils::auth::validator);
|
2020-05-06 13:25:54 +00:00
|
|
|
|
|
|
|
// set up database connection pool
|
2020-05-12 12:07:53 +00:00
|
|
|
let connspec =
|
|
|
|
std::env::var("DATABASE_URL").expect("DATABASE_URL NOT FOUND");
|
2020-05-06 13:25:54 +00:00
|
|
|
let manager = ConnectionManager::<SqliteConnection>::new(connspec);
|
|
|
|
let pool = r2d2::Pool::builder()
|
|
|
|
.build(manager)
|
|
|
|
.expect("Failed to create pool.");
|
|
|
|
|
2020-05-11 19:33:11 +00:00
|
|
|
diesel_migrations::run_pending_migrations(&pool.get().unwrap())
|
|
|
|
.expect("Error running migrations");
|
|
|
|
|
2020-05-06 13:25:54 +00:00
|
|
|
let addr = std::env::var("BIND_ADDRESS").expect("BIND ADDRESS NOT FOUND");
|
|
|
|
info!("Starting server {}", addr);
|
|
|
|
let private_key = rand::thread_rng().gen::<[u8; 32]>();
|
|
|
|
let app = move || {
|
|
|
|
App::new()
|
|
|
|
.data(pool.clone())
|
|
|
|
.wrap(IdentityService::new(
|
|
|
|
CookieIdentityPolicy::new(&private_key)
|
|
|
|
.name("my-app-auth")
|
|
|
|
.secure(false)
|
2020-05-09 06:37:08 +00:00
|
|
|
.same_site(SameSite::Lax),
|
2020-05-06 13:25:54 +00:00
|
|
|
))
|
|
|
|
.wrap(middleware::Logger::default())
|
|
|
|
.service(web::scope("/chat").wrap(basic_auth_middleware.clone()))
|
|
|
|
// .service(extract_my_obj)
|
|
|
|
// .service(index)
|
2020-05-09 06:37:08 +00:00
|
|
|
.service(routes::users::get_user)
|
|
|
|
.service(routes::users::add_user)
|
|
|
|
.service(routes::users::get_all_users)
|
|
|
|
.service(routes::auth::login)
|
|
|
|
.service(routes::auth::logout)
|
|
|
|
.service(routes::auth::index)
|
2020-05-06 13:25:54 +00:00
|
|
|
.service(fs::Files::new("/", "./static"))
|
|
|
|
};
|
|
|
|
HttpServer::new(app).bind(addr)?.run().await
|
|
|
|
}
|