diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 221 |
1 files changed, 221 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..eb7f0e8 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,221 @@ +#[macro_use] +mod actix_ructe; + +use std::time::{Duration, SystemTime}; + +use actix_web::body::{BoxBody, EitherBody, MessageBody}; +use actix_web::dev::ServiceResponse; +use actix_web::http::{header, StatusCode}; +use actix_web::middleware::{ErrorHandlerResponse, ErrorHandlers}; +use actix_web::{get, post, web, App, HttpResponse, HttpServer, Responder, ResponseError}; +use bcrypt::{hash, verify, DEFAULT_COST}; +use serde::Deserialize; +use sqlx::{postgres::PgPoolOptions, Pool, Postgres}; +use templates::statics::StaticFile; + +static FAR: Duration = Duration::from_secs(180 * 24 * 60 * 60); + +type Result<T> = std::result::Result<T, PinussyError>; + +#[derive(Clone)] +struct Pinussy { + db: Pool<Postgres>, +} + +#[actix_web::main] +async fn main() -> std::io::Result<()> { + let pool = PgPoolOptions::new() + .max_connections(5) + .connect("postgres://pinussy:pinussy@localhost/pinussy") + .await + .unwrap(); + + sqlx::migrate!("./migrations").run(&pool).await.unwrap(); + + let pinussy = Pinussy { db: pool }; + + HttpServer::new(move || { + App::new() + .wrap( + ErrorHandlers::new() + .handler(StatusCode::NOT_FOUND, render_404) + .handler(StatusCode::INTERNAL_SERVER_ERROR, render_500), + ) + .app_data(web::Data::new(pinussy.clone())) + .service(web::resource("/static/{filename}").to(static_file)) + .service(home) + .service(get_login) + .service(post_login) + .service(get_signup) + .service(post_signup) + .service(get_users) + // .service(get_pins) + // .service(post_pin) + // .service(get_pin) + // .service(post_board) + // .service(get_board) + }) + .bind(("127.0.0.1", 8080))? + .run() + .await +} + +#[get("/")] +async fn home() -> HttpResponse { + HttpResponse::Ok().body("Hello world!") +} + +#[get("/signup")] +async fn get_signup() -> HttpResponse { + HttpResponse::Ok().body(render!(templates::signup_html).unwrap()) +} + +#[derive(Deserialize)] +struct SignupForm { + username: String, + password: String, +} + +#[post("/signup")] +async fn post_signup( + state: web::Data<Pinussy>, + form: web::Form<SignupForm>, +) -> Result<HttpResponse> { + let password_hash = hash(&form.password, DEFAULT_COST)?; + sqlx::query!( + "insert into users(username, password) values ($1, $2)", + &form.username, + password_hash + ) + .execute(&state.db) + .await?; + Ok(HttpResponse::Ok().body(render!(templates::signup_html).unwrap())) +} + +#[get("/login")] +async fn get_login() -> HttpResponse { + HttpResponse::Ok().body(render!(templates::login_html).unwrap()) +} + +#[derive(Deserialize)] +struct LoginForm { + username: String, + password: String, + rememberme: bool, +} + +#[post("/login")] +async fn post_login(form: web::Form<LoginForm>) -> Result<HttpResponse> { + Ok(HttpResponse::Ok().body(render!(templates::login_html).unwrap())) +} + +#[derive(sqlx::Type)] +#[sqlx(type_name = "privacy", rename_all = "lowercase")] +enum Privacy { + Private, + Unlisted, + Public, +} + +#[derive(sqlx::FromRow)] +pub struct User { + id: i32, + username: String, + password: String, + email: Option<String>, + bio: Option<String>, + site: Option<String>, + privacy: Privacy, + admin: bool, +} + +#[get("/users")] +async fn get_users(state: web::Data<Pinussy>) -> Result<HttpResponse> { + let users: Vec<User> = sqlx::query_as("select * from users") + .fetch_all(&state.db) + .await + .unwrap(); + println!("lol"); + Ok(HttpResponse::Ok().body(render!(templates::users_html, users).unwrap())) +} + +async fn static_file(path: web::Path<String>) -> HttpResponse { + let name = &path.into_inner(); + if let Some(data) = StaticFile::get(name) { + let far_expires = SystemTime::now() + FAR; + HttpResponse::Ok() + .insert_header(header::Expires(far_expires.into())) + .insert_header(header::ContentType(data.mime.clone())) + .body(data.content) + } else { + HttpResponse::NotFound() + .reason("No such static file.") + .finish() + } +} + +fn render_404(res: ServiceResponse) -> actix_web::Result<ErrorHandlerResponse<BoxBody>> { + Ok(error_response( + res, + StatusCode::NOT_FOUND, + "The resource you requested can't be found.", + )) +} + +fn render_500(res: ServiceResponse) -> actix_web::Result<ErrorHandlerResponse<BoxBody>> { + Ok(error_response( + res, + StatusCode::INTERNAL_SERVER_ERROR, + "Sorry, Something went wrong. This is probably not your fault.", + )) +} + +fn error_response( + mut res: ServiceResponse, + status_code: StatusCode, + message: &str, +) -> ErrorHandlerResponse<BoxBody> { + res.headers_mut().insert( + header::CONTENT_TYPE, + header::HeaderValue::from_static(mime::TEXT_HTML_UTF_8.as_ref()), + ); + ErrorHandlerResponse::Response(res.map_body(|_head, _body| { + EitherBody::right(MessageBody::boxed( + render!(templates::error_html, status_code, message).unwrap(), + )) + })) +} + +#[derive(Debug)] +enum PinussyError { + Database(sqlx::Error), + Bcrypt, +} + +impl From<sqlx::Error> for PinussyError { + fn from(e: sqlx::Error) -> Self { + Self::Database(e) + } +} + +impl From<bcrypt::BcryptError> for PinussyError { + fn from(e: bcrypt::BcryptError) -> Self { + Self::Bcrypt + } +} + +impl std::fmt::Display for PinussyError { + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { + write!(f, "{:?}", self) + } +} + +impl std::error::Error for PinussyError {} + +impl ResponseError for PinussyError { + fn error_response(&self) -> HttpResponse<BoxBody> { + HttpResponse::new(self.status_code()) + } +} + +include!(concat!(env!("OUT_DIR"), "/templates.rs")); |