2024-04-06 02:41:10 +00:00
|
|
|
use std::net::SocketAddr;
|
|
|
|
|
|
|
|
use axum::routing::{get, post};
|
|
|
|
use axum::Router;
|
|
|
|
|
|
|
|
use sqlx::postgres::PgPoolOptions;
|
|
|
|
use sqlx::{Pool, Postgres};
|
|
|
|
|
|
|
|
use info_utils::prelude::*;
|
|
|
|
|
|
|
|
pub mod get;
|
|
|
|
pub mod post;
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct ServerState {
|
|
|
|
pub db_pool: Pool<Postgres>,
|
|
|
|
pub host: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
pub struct UrlRow {
|
|
|
|
pub index: i32,
|
|
|
|
pub id: String,
|
|
|
|
pub url: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
async fn main() -> eyre::Result<()> {
|
|
|
|
color_eyre::install()?;
|
|
|
|
|
|
|
|
let db_pool = init_db().await?;
|
|
|
|
|
2024-04-06 02:48:54 +00:00
|
|
|
let host = std::env::var("CHELA_HOST").unwrap_or("localhost".to_string());
|
|
|
|
let server_state = ServerState { db_pool, host };
|
|
|
|
|
|
|
|
let address = std::env::var("LISTEN_ADDRESS").unwrap_or("0.0.0.0".to_string());
|
|
|
|
let port = std::env::var("LISTEN_PORT").unwrap_or("3000".to_string());
|
2024-04-06 02:41:10 +00:00
|
|
|
|
|
|
|
let router = init_routes(server_state)?;
|
2024-04-06 02:48:54 +00:00
|
|
|
let listener = tokio::net::TcpListener::bind(format!("{}:{}", address, port)).await?;
|
|
|
|
log!("Listening at {}:{}", address, port);
|
2024-04-06 02:41:10 +00:00
|
|
|
axum::serve(
|
|
|
|
listener,
|
|
|
|
router.into_make_service_with_connect_info::<SocketAddr>(),
|
|
|
|
)
|
|
|
|
.await?;
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn init_db() -> eyre::Result<Pool<Postgres>> {
|
|
|
|
let db_pool = PgPoolOptions::new()
|
|
|
|
.max_connections(15)
|
|
|
|
.connect(std::env::var("DATABASE_URL")?.as_str())
|
|
|
|
.await?;
|
|
|
|
log!("Successfully connected to database");
|
|
|
|
|
|
|
|
sqlx::query!("CREATE SCHEMA IF NOT EXISTS chela")
|
|
|
|
.execute(&db_pool)
|
|
|
|
.await?;
|
|
|
|
log!("Created schema chela");
|
|
|
|
|
|
|
|
sqlx::query!(
|
|
|
|
"
|
|
|
|
CREATE TABLE IF NOT EXISTS chela.urls (
|
|
|
|
index SERIAL PRIMARY KEY,
|
|
|
|
id TEXT NOT NULL UNIQUE,
|
|
|
|
url TEXT NOT NULL
|
|
|
|
)
|
|
|
|
",
|
|
|
|
)
|
|
|
|
.execute(&db_pool)
|
|
|
|
.await?;
|
|
|
|
log!("Created table chela.urls");
|
|
|
|
|
|
|
|
sqlx::query!(
|
|
|
|
"
|
|
|
|
CREATE TABLE IF NOT EXISTS chela.tracking (
|
|
|
|
timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|
|
|
id TEXT NOT NULL,
|
|
|
|
ip TEXT NOT NULL,
|
|
|
|
referrer TEXT,
|
|
|
|
user_agent TEXT
|
|
|
|
)
|
|
|
|
",
|
|
|
|
)
|
|
|
|
.execute(&db_pool)
|
|
|
|
.await?;
|
|
|
|
log!("Created table chela.tracking");
|
|
|
|
|
|
|
|
Ok(db_pool)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn init_routes(state: ServerState) -> eyre::Result<Router> {
|
|
|
|
let router = Router::new()
|
|
|
|
.route("/", get(get::get_index))
|
|
|
|
.route("/:id", get(get::get_id))
|
|
|
|
.route("/", post(post::create_link))
|
|
|
|
.layer(axum::Extension(state));
|
|
|
|
|
|
|
|
Ok(router)
|
|
|
|
}
|