107 lines
2.5 KiB
Rust
107 lines
2.5 KiB
Rust
mod content;
|
|
|
|
use anyhow::{anyhow, Result};
|
|
use axum::{
|
|
extract::State,
|
|
http::StatusCode,
|
|
response::{IntoResponse, Response},
|
|
routing::get,
|
|
Router, Server,
|
|
};
|
|
use figment::{
|
|
providers::{Env, Format, Serialized, Toml},
|
|
Figment,
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
use sqlx::{postgres::PgPoolOptions, query_as, Pool, Postgres};
|
|
use std::{net::SocketAddr, sync::Arc};
|
|
|
|
use crate::content::{Page, PageBlock};
|
|
|
|
#[derive(Deserialize, Serialize)]
|
|
struct Config {
|
|
bind: String,
|
|
database_url: String,
|
|
}
|
|
|
|
impl Default for Config {
|
|
fn default() -> Self {
|
|
Config {
|
|
bind: "127.0.0.1:3000".to_owned(),
|
|
database_url: "postgres://artificiale:changeme@localhost/artificiale".to_owned(),
|
|
}
|
|
}
|
|
}
|
|
|
|
struct GenericError(anyhow::Error);
|
|
|
|
impl IntoResponse for GenericError {
|
|
fn into_response(self) -> Response {
|
|
(
|
|
StatusCode::INTERNAL_SERVER_ERROR,
|
|
format!("Something went wrong: {}", self.0),
|
|
)
|
|
.into_response()
|
|
}
|
|
}
|
|
|
|
impl From<sqlx::Error> for GenericError {
|
|
fn from(value: sqlx::Error) -> Self {
|
|
GenericError(anyhow!("Database error: {}", value))
|
|
}
|
|
}
|
|
|
|
struct AppState {
|
|
database: Pool<Postgres>,
|
|
}
|
|
|
|
async fn root(State(state): State<Arc<AppState>>) -> Result<String, GenericError> {
|
|
let page = query_as!(
|
|
Page,
|
|
r#"select
|
|
id,
|
|
author,
|
|
title,
|
|
description,
|
|
tags,
|
|
slug,
|
|
created_at,
|
|
modified_at,
|
|
deleted_at,
|
|
blocks as "blocks!: sqlx::types::Json<Vec<PageBlock>>"
|
|
from pages limit 1"#
|
|
)
|
|
.fetch_one(&state.database)
|
|
.await?;
|
|
Ok(page.title)
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
tracing_subscriber::fmt::init();
|
|
|
|
let config: Config = Figment::from(Serialized::defaults(Config::default()))
|
|
.merge(Toml::file("mabel.toml"))
|
|
.merge(Env::prefixed("MABEL_"))
|
|
.extract()?;
|
|
|
|
let pool = PgPoolOptions::new()
|
|
.max_connections(5)
|
|
.connect(config.database_url.as_str())
|
|
.await?;
|
|
|
|
sqlx::migrate!().run(&pool).await?;
|
|
|
|
let shared_state = Arc::new(AppState { database: pool });
|
|
|
|
let app = Router::new().route("/", get(root)).with_state(shared_state);
|
|
|
|
let addr: SocketAddr = config.bind.parse()?;
|
|
tracing::debug!("listening on {}", addr);
|
|
Server::bind(&addr)
|
|
.serve(app.into_make_service())
|
|
.await
|
|
.unwrap();
|
|
Ok(())
|
|
}
|