diff --git a/rust/rsc/Cargo.toml b/rust/rsc/Cargo.toml index 66b01eb9f..b66370f36 100644 --- a/rust/rsc/Cargo.toml +++ b/rust/rsc/Cargo.toml @@ -3,14 +3,6 @@ name = "rsc" version = "0.1.0" edition = "2021" -[[bin]] -name = "rsc" -path = "src/rsc/main.rs" - -[[bin]] -name = "rsc_tool" -path = "src/rsc_tool/main.rs" - [dependencies] entity = { path = "../entity" } migration = { path = "../migration" } diff --git a/rust/rsc/src/rsc/add_job.rs b/rust/rsc/src/bin/rsc/add_job.rs similarity index 98% rename from rust/rsc/src/rsc/add_job.rs rename to rust/rsc/src/bin/rsc/add_job.rs index e3eed73e2..088e393fc 100644 --- a/rust/rsc/src/rsc/add_job.rs +++ b/rust/rsc/src/bin/rsc/add_job.rs @@ -7,8 +7,7 @@ use sea_orm::{ActiveModelTrait, ActiveValue::*, DatabaseConnection, DbErr, Trans use std::sync::Arc; use tracing; -#[path = "../common/database.rs"] -mod database; +use rsc::database; #[tracing::instrument(skip_all)] pub async fn add_job( diff --git a/rust/rsc/src/rsc/api_key_check.rs b/rust/rsc/src/bin/rsc/api_key_check.rs similarity index 100% rename from rust/rsc/src/rsc/api_key_check.rs rename to rust/rsc/src/bin/rsc/api_key_check.rs diff --git a/rust/rsc/src/rsc/blob.rs b/rust/rsc/src/bin/rsc/blob.rs similarity index 97% rename from rust/rsc/src/rsc/blob.rs rename to rust/rsc/src/bin/rsc/blob.rs index eb36a830f..60faac9ce 100644 --- a/rust/rsc/src/rsc/blob.rs +++ b/rust/rsc/src/bin/rsc/blob.rs @@ -6,7 +6,7 @@ use entity::blob; use futures::stream::BoxStream; use futures::TryStreamExt; use sea_orm::prelude::Uuid; -use sea_orm::{ActiveModelTrait, ActiveValue::*, DatabaseConnection}; +use sea_orm::{ActiveValue::*, DatabaseConnection}; use std::sync::Arc; use tokio_util::bytes::Bytes; use tracing; diff --git a/rust/rsc/src/rsc/blob_store_impls.rs b/rust/rsc/src/bin/rsc/blob_store_impls.rs similarity index 100% rename from rust/rsc/src/rsc/blob_store_impls.rs rename to rust/rsc/src/bin/rsc/blob_store_impls.rs diff --git a/rust/rsc/src/rsc/main.rs b/rust/rsc/src/bin/rsc/main.rs similarity index 99% rename from rust/rsc/src/rsc/main.rs rename to rust/rsc/src/bin/rsc/main.rs index 05f577a20..76c51036f 100644 --- a/rust/rsc/src/rsc/main.rs +++ b/rust/rsc/src/bin/rsc/main.rs @@ -3,24 +3,23 @@ use axum::{ routing::{get, post}, Router, }; +use chrono::Utc; use clap::Parser; use data_encoding::HEXLOWER; use migration::{Migrator, MigratorTrait}; use rand_core::{OsRng, RngCore}; -use rlimit::{getrlimit, Resource}; +use rlimit::Resource; +use rsc::{config, database}; +use sea_orm::{ + prelude::Uuid, ActiveModelTrait, ActiveValue::*, ColumnTrait, ConnectOptions, ConnectionTrait, + Database, DatabaseConnection, EntityTrait, QueryFilter, +}; use std::collections::HashMap; use std::io::{Error, ErrorKind}; use std::sync::Arc; use std::time::Duration; use tracing; -use sea_orm::{ - prelude::Uuid, ActiveModelTrait, ActiveValue::*, ColumnTrait, ConnectOptions, ConnectionTrait, - Database, DatabaseConnection, EntityTrait, QueryFilter, -}; - -use chrono::Utc; - mod add_job; mod api_key_check; mod blob; @@ -28,11 +27,6 @@ mod blob_store_impls; mod read_job; mod types; -#[path = "../common/config.rs"] -mod config; -#[path = "../common/database.rs"] -mod database; - #[derive(Debug, Parser)] struct ServerOptions { #[arg(help = "Specify a config override file", value_name = "CONFIG", long)] diff --git a/rust/rsc/src/rsc/read_job.rs b/rust/rsc/src/bin/rsc/read_job.rs similarity index 100% rename from rust/rsc/src/rsc/read_job.rs rename to rust/rsc/src/bin/rsc/read_job.rs diff --git a/rust/rsc/src/rsc/types.rs b/rust/rsc/src/bin/rsc/types.rs similarity index 100% rename from rust/rsc/src/rsc/types.rs rename to rust/rsc/src/bin/rsc/types.rs diff --git a/rust/rsc/src/rsc_tool/main.rs b/rust/rsc/src/bin/rsc_tool/main.rs similarity index 99% rename from rust/rsc/src/rsc_tool/main.rs rename to rust/rsc/src/bin/rsc_tool/main.rs index 5d6f32922..59dc6e538 100644 --- a/rust/rsc/src/rsc_tool/main.rs +++ b/rust/rsc/src/bin/rsc_tool/main.rs @@ -2,17 +2,13 @@ use clap::{Parser, Subcommand}; use inquire::{Confirm, Text}; use is_terminal::IsTerminal; use migration::{DbErr, Migrator, MigratorTrait}; +use rsc::{config, database}; use sea_orm::{prelude::Uuid, DatabaseConnection}; use std::io::{Error, ErrorKind}; use tracing; mod table; -#[path = "../common/config.rs"] -mod config; -#[path = "../common/database.rs"] -mod database; - async fn add_api_key( key: Option, desc: String, diff --git a/rust/rsc/src/rsc_tool/table.rs b/rust/rsc/src/bin/rsc_tool/table.rs similarity index 100% rename from rust/rsc/src/rsc_tool/table.rs rename to rust/rsc/src/bin/rsc_tool/table.rs diff --git a/rust/rsc/src/common/config.rs b/rust/rsc/src/config.rs similarity index 100% rename from rust/rsc/src/common/config.rs rename to rust/rsc/src/config.rs diff --git a/rust/rsc/src/common/database.rs b/rust/rsc/src/database.rs similarity index 99% rename from rust/rsc/src/common/database.rs rename to rust/rsc/src/database.rs index 878b81e32..9e40f979b 100644 --- a/rust/rsc/src/common/database.rs +++ b/rust/rsc/src/database.rs @@ -1,7 +1,7 @@ use chrono::NaiveDateTime; use data_encoding::BASE64; use entity::prelude::{ - Blob, BlobStore, Job, LocalBlobStore, OutputDir, OutputFile, OutputSymlink, VisibleFile, + Blob, BlobStore, LocalBlobStore, OutputDir, OutputFile, OutputSymlink, VisibleFile, }; use entity::{ api_key, blob, blob_store, job, local_blob_store, output_dir, output_file, output_symlink, @@ -13,8 +13,7 @@ use rand_core::{OsRng, RngCore}; use sea_orm::ExecResult; use sea_orm::{ prelude::Uuid, ActiveModelTrait, ActiveValue::*, ColumnTrait, ConnectionTrait, DbBackend, - DbErr, DeleteResult, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder, QuerySelect, - Statement, + DbErr, DeleteResult, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder, Statement, }; use tracing; diff --git a/rust/rsc/src/lib.rs b/rust/rsc/src/lib.rs new file mode 100644 index 000000000..2ee385a05 --- /dev/null +++ b/rust/rsc/src/lib.rs @@ -0,0 +1,2 @@ +pub mod config; +pub mod database;