mrfluffy/bitBeam
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update added download and user authentecation

Browse Source
This commit is contained in:
zastian@mrthoddata.com
2025-05-09 16:48:29 +01:00
parent 9b208edfde
commit 1c123f5a45
10 changed files with 456 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

111
src/main.rs
View File

@@ -1,11 +1,11 @@
use axum::{
extract::DefaultBodyLimit,
//response::IntoResponse,
routing::{get, post},
response::IntoResponse,
Extension, Router,
};
use log::{debug, error, info, warn};
use sqlx::{any::AnyPoolOptions, migrate::MigrateDatabase, AnyPool, Sqlite};
use log::{info, warn, error, debug};
use std::path::Path;
use tokio::fs;
@@ -31,28 +31,59 @@ async fn main() {
sqlx::any::install_default_drivers();
// Load the configuration from environment variables
let config = data::Config {
db_type: std::env::var("BITBEAM_DB_TYPE").unwrap_or_else(|_| "postgres".to_string()),
database_url: match std::env::var("BITBEAM_DB_TYPE").unwrap().as_str() {
"postgres" => std::env::var("BITBEAM_DATABASE_URL")
.expect("BITBEAM_DATABASE_URL must be set for Postgres"),
"sqlite" => std::env::var("BITBEAM_DATABASE_URL")
.expect("BITBEAM_DATABASE_URL must be set for SQLite"),
other => panic!("Unsupported BITBEAM_DB_TYPE: {}", other),
db_type: std::env::var("BITBEAM_DB_TYPE").unwrap_or_else(|_| "sqlite".to_string()),
// Determine the correct database URL
database_url: match std::env::var("BITBEAM_DB_TYPE")
.unwrap_or_else(|_| "sqlite".to_string())
.as_str()
{
"postgres" => {
// For Postgres, BITBEAM_DATABASE_URL must be set
std::env::var("BITBEAM_DATABASE_URL")
.expect("BITBEAM_DATABASE_URL must be set for Postgres")
}
"sqlite" => {
// For SQLite, use BITBEAM_DATABASE_URL if set, otherwise default
std::env::var("BITBEAM_DATABASE_URL")
.unwrap_or_else(|_| "sqlite://./bitbeam.sqlite".to_string())
}
other => {
panic!("Unsupported BITBEAM_DB_TYPE: {}", other);
}
},
data_path: std::env::var("BITBEAM_DATA_PATH").unwrap_or_else(|_| "./media_store".to_string()),
data_path: std::env::var("BITBEAM_DATA_PATH")
.unwrap_or_else(|_| "./media_store".to_string()),
port: std::env::var("BITBEAM_PORT").unwrap_or_else(|_| "3000".to_string()),
listener_addr: std::env::var("BITBEAM_ADDR").unwrap_or_else(|_| "127.0.0.1".to_string()),
log_level: std::env::var("BITBEAM_LOG_LEVEL").unwrap_or_else(|_| "info".to_string()),
log_location: std::env::var("BITBEAM_LOG_LOCATION").unwrap_or_else(|_| "./bitbeam.log".to_string()),
log_location: std::env::var("BITBEAM_LOG_LOCATION")
.unwrap_or_else(|_| "./bitbeam.log".to_string()),
use_tls: std::env::var("BITBEAM_USE_TLS")
.unwrap_or_else(|_| "false".to_string())
.parse()
.unwrap_or(false),
base_url: std::env::var("BITBEAM_BASE_URL").unwrap_or_else(|_| {
format!(
"localhost:{}",
std::env::var("BITBEAM_PORT").unwrap_or_else(|_| "3000".to_string())
)
.to_string()
}),
allow_register: std::env::var("BITBEAM_ALLOW_REGISTER")
.unwrap_or_else(|_| "true".to_string())
.parse()
.unwrap_or(true),
};
// Setting up the logging system
// The log level is set based on the environment variable BITBEAM_LOG_LEVEL
let level = match config.log_level.as_str() {
"debug" => log::LevelFilter::Debug,
"info" => log::LevelFilter::Info,
"warn" => log::LevelFilter::Warn,
"info" => log::LevelFilter::Info,
"warn" => log::LevelFilter::Warn,
"error" => log::LevelFilter::Error,
_ => log::LevelFilter::Info,
_ => log::LevelFilter::Info,
};
// Initialize the logging system
let log_path = &config.log_location;
@@ -73,7 +104,7 @@ async fn main() {
Err(error) => {
error!("Error creating database: {}", error);
panic!("error: {}", error)
},
}
}
} else {
info!("Database already exists");
@@ -94,11 +125,14 @@ async fn main() {
r#"
CREATE TABLE IF NOT EXISTS files (
id TEXT PRIMARY KEY,
file_name TEXT NOT NULL,
content_type TEXT NOT NULL,
upload_time BIGINT NOT NULL,
download_limit INTEGER NOT NULL,
download_count INTEGER NOT NULL,
file_size BIGINT NOT NULL
file_size BIGINT NOT NULL,
download_url TEXT NOT NULL,
owner TEXT NOT NULL
);
"#,
)
@@ -107,11 +141,25 @@ async fn main() {
{
info!("DB created");
};
// create the user table
if let Err(_e) = sqlx::query(
r#"
CREATE TABLE IF NOT EXISTS users (
key TEXT PRIMARY KEY,
username TEXT NOT NULL,
password TEXT NOT NULL
);
"#,
)
.execute(&pool)
.await
{
info!("DB created");
};
//create the directory if it doesn't exist
let dir = Path::new(&config.data_path);
if let Err(e) = fs::create_dir_all(dir).await {
warn!("could not make dir at {} error: {}", &config.data_path ,e);
warn!("could not make dir at {} error: {}", &config.data_path, e);
}
//let file_path = dir.join(&id);
@@ -122,6 +170,8 @@ async fn main() {
.route("/", get(|| async { "Hello, World!" }))
.route("/upload", post(api::upload))
.route("/all_files", get(api::all_files))
.route("/download/{uuid}", get(api::download_file))
.route("/user/register", post(api::register_user))
.layer(DefaultBodyLimit::max(100 * 1024 * 1024))
.layer(Extension(pool))
.layer(Extension(config.clone()))
@@ -130,10 +180,15 @@ async fn main() {
// The web server is started using the Axum framework
// The server listens on the address and port specified in the configuration
axum::serve(
match tokio::net::TcpListener::bind(format!("{}:{}",&config.listener_addr,&config.port)).await {
match tokio::net::TcpListener::bind(format!("{}:{}", &config.listener_addr, &config.port))
.await
{
Ok(listener) => listener,
Err(e) => {
error!("Error binding to address {}:{} : {}",&config.listener_addr,&config.port, e);
error!(
"Error binding to address {}:{} : {}",
&config.listener_addr, &config.port, e
);
return;
}
},
@@ -149,16 +204,19 @@ async fn main() {
/// It formats the log messages to include the date, time, log level, target, and message.
/// It also sets the log level based on the provided level filter.
/// It takes the log file path and log level as parameters.
fn init_logging(log_file_path: &str, level: log::LevelFilter) -> Result<(), Box<dyn std::error::Error>> {
fn init_logging(
log_file_path: &str,
level: log::LevelFilter,
) -> Result<(), Box<dyn std::error::Error>> {
// Build a Dispatch for stdout
let stdout_dispatch = fern::Dispatch::new()
.format(|out, message, record| {
out.finish(format_args!(
"[{date}][{lvl}][{target}] {msg}",
date = chrono::Local::now().format("%Y-%m-%d %H:%M:%S"),
lvl = record.level(),
lvl = record.level(),
target = record.target(),
msg = message,
msg = message,
))
})
.level(level)
@@ -170,9 +228,9 @@ fn init_logging(log_file_path: &str, level: log::LevelFilter) -> Result<(), Box<
out.finish(format_args!(
"[{date}][{lvl}][{target}] {msg}",
date = chrono::Local::now().format("%Y-%m-%d %H:%M:%S"),
lvl = record.level(),
lvl = record.level(),
target = record.target(),
msg = message,
msg = message,
))
})
.level(level)
@@ -188,6 +246,3 @@ fn init_logging(log_file_path: &str, level: log::LevelFilter) -> Result<(), Box<
Ok(())
}