feat(server): start db query abstraction

libarchive/src/archive.rs

@@ -39,7 +39,7 @@ pub enum ReadFormat {
     Zip,
 }
 
-#[derive(Debug, PartialEq, Eq)]
+#[derive(Debug, PartialEq, Eq, Clone)]
 pub enum ReadFilter {
     All,
     Bzip2,

libarchive/src/write/file.rs

@@ -54,7 +54,7 @@ impl FileWriter {
                                 self.handle_mut(),
                                 &buf[written] as *const u8 as *const _,
                                 buf_len - written,
-                                )
+                            )
                         } as isize;
 
                         // Negative values signal errors

server/src/api/mod.rs

@@ -4,41 +4,41 @@ use axum::extract::{Path, Query, State};
 use axum::routing::get;
 use axum::Json;
 use axum::Router;
-use sea_orm::entity::EntityTrait;
+use sea_orm::ModelTrait;
-use sea_orm::query::QueryOrder;
-use sea_orm::PaginatorTrait;
 
 use pagination::PaginatedResponse;
 
-use crate::db::entities::package;
+use crate::db;
-use crate::db::entities::repo;
 
 pub fn router() -> Router<crate::Global> {
     Router::new()
         .route("/repos", get(get_repos))
         .route("/repos/:id", get(get_single_repo))
         .route("/packages", get(get_packages))
+        .route("/packages/:id", get(get_single_package))
 }
 
 async fn get_repos(
     State(global): State<crate::Global>,
     Query(pagination): Query<pagination::Query>,
-) -> crate::Result<Json<PaginatedResponse<repo::Model>>> {
+) -> crate::Result<Json<PaginatedResponse<db::repo::Model>>> {
-    let repos = repo::Entity::find()
+    let (total_pages, repos) = global
-        .order_by_asc(repo::Column::Id)
+        .db
-        .paginate(&global.db, pagination.per_page.unwrap_or(25))
+        .repos(
-        .fetch_page(pagination.page.unwrap_or(1) - 1)
+            pagination.per_page.unwrap_or(25),
+            pagination.page.unwrap_or(1) - 1,
+        )
         .await?;
-
+    Ok(Json(pagination.res(total_pages, repos)))
-    Ok(Json(pagination.res(repos)))
 }
 
 async fn get_single_repo(
     State(global): State<crate::Global>,
     Path(id): Path<i32>,
-) -> crate::Result<Json<repo::Model>> {
+) -> crate::Result<Json<db::repo::Model>> {
-    let repo = repo::Entity::find_by_id(id)
+    let repo = global
-        .one(&global.db)
+        .db
+        .repo(id)
         .await?
         .ok_or(axum::http::StatusCode::NOT_FOUND)?;
 
@@ -48,12 +48,43 @@ async fn get_single_repo(
 async fn get_packages(
     State(global): State<crate::Global>,
     Query(pagination): Query<pagination::Query>,
-) -> crate::Result<Json<PaginatedResponse<package::Model>>> {
+) -> crate::Result<Json<PaginatedResponse<db::package::Model>>> {
-    let pkgs = package::Entity::find()
+    let (total_pages, pkgs) = global
-        .order_by_asc(package::Column::Id)
+        .db
-        .paginate(&global.db, pagination.per_page.unwrap_or(25))
+        .packages(
-        .fetch_page(pagination.page.unwrap_or(1) - 1)
+            pagination.per_page.unwrap_or(25),
+            pagination.page.unwrap_or(1) - 1,
+        )
         .await?;
 
-    Ok(Json(pagination.res(pkgs)))
+    Ok(Json(pagination.res(total_pages, pkgs)))
+}
+
+#[derive(serde::Serialize)]
+pub struct PackageRes {
+    #[serde(flatten)]
+    entry: db::package::Model,
+    licenses: Vec<String>,
+}
+
+async fn get_single_package(
+    State(global): State<crate::Global>,
+    Path(id): Path<i32>,
+) -> crate::Result<Json<PackageRes>> {
+    let entry = global
+        .db
+        .package(id)
+        .await?
+        .ok_or(axum::http::StatusCode::NOT_FOUND)?;
+    let licenses = entry
+        .find_related(db::PackageLicense)
+        .all(&global.db)
+        .await?
+        .iter()
+        .map(|e| e.value.clone())
+        .collect();
+
+    let res = PackageRes { entry, licenses };
+
+    Ok(Json(res))
 }

server/src/api/pagination.rs

@@ -16,15 +16,21 @@ where
 {
     pub page: u64,
     pub per_page: u64,
+    pub total_pages: u64,
     pub count: usize,
     pub items: Vec<T>,
 }
 
 impl Query {
-    pub fn res<T: for<'de> Serialize>(self, items: Vec<T>) -> PaginatedResponse<T> {
+    pub fn res<T: for<'de> Serialize>(
+        self,
+        total_pages: u64,
+        items: Vec<T>,
+    ) -> PaginatedResponse<T> {
         PaginatedResponse {
             page: self.page.unwrap_or(DEFAULT_PAGE),
             per_page: self.per_page.unwrap_or(DEFAULT_PER_PAGE),
+            total_pages,
             count: items.len(),
             items,
         }

server/src/cli.rs

@@ -65,7 +65,7 @@ impl Cli {
 
         debug!("Connecting to database with URL {}", db_url);
 
-        let db = crate::db::init(db_url).await?;
+        let db = crate::db::RieterDb::connect(db_url).await?;
         // let db = crate::db::init("postgres://rieter:rieter@localhost:5432/rieter")
         // .await
         // .unwrap();

server/src/db/conn.rs

@@ -0,0 +1,61 @@
+use super::RieterDb;
+use sea_orm::{DbBackend, DbErr, ExecResult, QueryResult, Statement};
+use std::{future::Future, pin::Pin};
+
+// Allows RieterDb objects to be passed to ORM functions
+impl sea_orm::ConnectionTrait for RieterDb {
+    fn get_database_backend(&self) -> DbBackend {
+        self.conn.get_database_backend()
+    }
+    fn execute<'life0, 'async_trait>(
+        &'life0 self,
+        stmt: Statement,
+    ) -> Pin<Box<dyn Future<Output = std::result::Result<ExecResult, DbErr>> + Send + 'async_trait>>
+    where
+        Self: 'async_trait,
+        'life0: 'async_trait,
+    {
+        self.conn.execute(stmt)
+    }
+    fn execute_unprepared<'life0, 'life1, 'async_trait>(
+        &'life0 self,
+        sql: &'life1 str,
+    ) -> Pin<Box<dyn Future<Output = std::result::Result<ExecResult, DbErr>> + Send + 'async_trait>>
+    where
+        Self: 'async_trait,
+        'life0: 'async_trait,
+        'life1: 'async_trait,
+    {
+        self.conn.execute_unprepared(sql)
+    }
+    fn query_one<'life0, 'async_trait>(
+        &'life0 self,
+        stmt: Statement,
+    ) -> Pin<
+        Box<
+            dyn Future<Output = std::result::Result<Option<QueryResult>, DbErr>>
+                + Send
+                + 'async_trait,
+        >,
+    >
+    where
+        Self: 'async_trait,
+        'life0: 'async_trait,
+    {
+        self.conn.query_one(stmt)
+    }
+    fn query_all<'life0, 'async_trait>(
+        &'life0 self,
+        stmt: Statement,
+    ) -> Pin<
+        Box<
+            dyn Future<Output = std::result::Result<Vec<QueryResult>, DbErr>> + Send + 'async_trait,
+        >,
+    >
+    where
+        Self: 'async_trait,
+        'life0: 'async_trait,
+    {
+        self.conn.query_all(stmt)
+    }
+}

server/src/db/entities/mod.rs

@@ -3,5 +3,10 @@
 pub mod prelude;
 
 pub mod package;
+pub mod package_conflicts;
+pub mod package_depends;
+pub mod package_group;
 pub mod package_license;
+pub mod package_provides;
+pub mod package_replaces;
 pub mod repo;

server/src/db/entities/package.rs

@@ -26,8 +26,18 @@ pub struct Model {
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {
+    #[sea_orm(has_many = "super::package_conflicts::Entity")]
+    PackageConflicts,
+    #[sea_orm(has_many = "super::package_depends::Entity")]
+    PackageDepends,
+    #[sea_orm(has_many = "super::package_group::Entity")]
+    PackageGroup,
     #[sea_orm(has_many = "super::package_license::Entity")]
     PackageLicense,
+    #[sea_orm(has_many = "super::package_provides::Entity")]
+    PackageProvides,
+    #[sea_orm(has_many = "super::package_replaces::Entity")]
+    PackageReplaces,
     #[sea_orm(
         belongs_to = "super::repo::Entity",
         from = "Column::RepoId",
@@ -38,12 +48,42 @@ pub enum Relation {
     Repo,
 }
 
+impl Related<super::package_conflicts::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::PackageConflicts.def()
+    }
+}
+
+impl Related<super::package_depends::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::PackageDepends.def()
+    }
+}
+
+impl Related<super::package_group::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::PackageGroup.def()
+    }
+}
+
 impl Related<super::package_license::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::PackageLicense.def()
     }
 }
 
+impl Related<super::package_provides::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::PackageProvides.def()
+    }
+}
+
+impl Related<super::package_replaces::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::PackageReplaces.def()
+    }
+}
+
 impl Related<super::repo::Entity> for Entity {
     fn to() -> RelationDef {
         Relation::Repo.def()

server/src/db/entities/package_conflicts.rs

@@ -0,0 +1,33 @@
+//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "package_conflicts")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub package_id: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub value: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::package::Entity",
+        from = "Column::PackageId",
+        to = "super::package::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Package,
+}
+
+impl Related<super::package::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Package.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

server/src/db/entities/package_depends.rs

@@ -0,0 +1,35 @@
+//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "package_depends")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub package_id: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub r#type: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub value: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::package::Entity",
+        from = "Column::PackageId",
+        to = "super::package::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Package,
+}
+
+impl Related<super::package::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Package.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

server/src/db/entities/package_group.rs

@@ -0,0 +1,33 @@
+//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "package_group")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub package_id: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub value: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::package::Entity",
+        from = "Column::PackageId",
+        to = "super::package::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Package,
+}
+
+impl Related<super::package::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Package.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

server/src/db/entities/package_provides.rs

@@ -0,0 +1,33 @@
+//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "package_provides")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub package_id: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub value: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::package::Entity",
+        from = "Column::PackageId",
+        to = "super::package::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Package,
+}
+
+impl Related<super::package::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Package.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

server/src/db/entities/package_replaces.rs

@@ -0,0 +1,33 @@
+//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "package_replaces")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub package_id: i32,
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub value: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::package::Entity",
+        from = "Column::PackageId",
+        to = "super::package::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Package,
+}
+
+impl Related<super::package::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Package.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

server/src/db/entities/prelude.rs

@@ -1,5 +1,10 @@
 //! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.1
 
 pub use super::package::Entity as Package;
+pub use super::package_conflicts::Entity as PackageConflicts;
+pub use super::package_depends::Entity as PackageDepends;
+pub use super::package_group::Entity as PackageGroup;
 pub use super::package_license::Entity as PackageLicense;
+pub use super::package_provides::Entity as PackageProvides;
+pub use super::package_replaces::Entity as PackageReplaces;
 pub use super::repo::Entity as Repo;

server/src/db/migrator/m20230730_000001_create_repo_tables.rs

@@ -90,7 +90,152 @@ impl MigrationTrait for Migration {
                     )
                     .to_owned(),
             )
-            .await
+            .await?;
+        manager
+            .create_table(
+                Table::create()
+                    .table(PackageGroup::Table)
+                    .col(ColumnDef::new(PackageGroup::PackageId).integer().not_null())
+                    .col(
+                        ColumnDef::new(PackageGroup::Value)
+                            .string_len(255)
+                            .not_null(),
+                    )
+                    .primary_key(
+                        Index::create()
+                            .col(PackageGroup::PackageId)
+                            .col(PackageGroup::Value),
+                    )
+                    .foreign_key(
+                        ForeignKey::create()
+                            .name("fk-package_group-package_id")
+                            .from(PackageGroup::Table, PackageGroup::PackageId)
+                            .to(Package::Table, Package::Id)
+                            .on_delete(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+        manager
+            .create_table(
+                Table::create()
+                    .table(PackageReplaces::Table)
+                    .col(
+                        ColumnDef::new(PackageReplaces::PackageId)
+                            .integer()
+                            .not_null(),
+                    )
+                    .col(
+                        ColumnDef::new(PackageReplaces::Value)
+                            .string_len(255)
+                            .not_null(),
+                    )
+                    .primary_key(
+                        Index::create()
+                            .col(PackageReplaces::PackageId)
+                            .col(PackageReplaces::Value),
+                    )
+                    .foreign_key(
+                        ForeignKey::create()
+                            .name("fk-package_replaces-package_id")
+                            .from(PackageReplaces::Table, PackageReplaces::PackageId)
+                            .to(Package::Table, Package::Id)
+                            .on_delete(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+        manager
+            .create_table(
+                Table::create()
+                    .table(PackageConflicts::Table)
+                    .col(
+                        ColumnDef::new(PackageConflicts::PackageId)
+                            .integer()
+                            .not_null(),
+                    )
+                    .col(
+                        ColumnDef::new(PackageConflicts::Value)
+                            .string_len(255)
+                            .not_null(),
+                    )
+                    .primary_key(
+                        Index::create()
+                            .col(PackageConflicts::PackageId)
+                            .col(PackageConflicts::Value),
+                    )
+                    .foreign_key(
+                        ForeignKey::create()
+                            .name("fk-package_conflicts-package_id")
+                            .from(PackageConflicts::Table, PackageConflicts::PackageId)
+                            .to(Package::Table, Package::Id)
+                            .on_delete(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+        manager
+            .create_table(
+                Table::create()
+                    .table(PackageProvides::Table)
+                    .col(
+                        ColumnDef::new(PackageProvides::PackageId)
+                            .integer()
+                            .not_null(),
+                    )
+                    .col(
+                        ColumnDef::new(PackageProvides::Value)
+                            .string_len(255)
+                            .not_null(),
+                    )
+                    .primary_key(
+                        Index::create()
+                            .col(PackageProvides::PackageId)
+                            .col(PackageProvides::Value),
+                    )
+                    .foreign_key(
+                        ForeignKey::create()
+                            .name("fk-package_provides-package_id")
+                            .from(PackageProvides::Table, PackageProvides::PackageId)
+                            .to(Package::Table, Package::Id)
+                            .on_delete(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+        manager
+            .create_table(
+                Table::create()
+                    .table(PackageDepends::Table)
+                    .col(
+                        ColumnDef::new(PackageDepends::PackageId)
+                            .integer()
+                            .not_null(),
+                    )
+                    .col(ColumnDef::new(PackageDepends::Type).integer().not_null())
+                    .col(
+                        ColumnDef::new(PackageDepends::Value)
+                            .string_len(255)
+                            .not_null(),
+                    )
+                    .primary_key(
+                        Index::create()
+                            .col(PackageDepends::PackageId)
+                            .col(PackageDepends::Type)
+                            .col(PackageDepends::Value),
+                    )
+                    .foreign_key(
+                        ForeignKey::create()
+                            .name("fk-package_depends-package_id")
+                            .from(PackageDepends::Table, PackageDepends::PackageId)
+                            .to(Package::Table, Package::Id)
+                            .on_delete(ForeignKeyAction::Cascade),
+                    )
+                    .to_owned(),
+            )
+            .await?;
+
+        Ok(())
     }
 
     // Define how to rollback this migration: Drop the Bakery table.
@@ -98,6 +243,21 @@ impl MigrationTrait for Migration {
         manager
             .drop_table(Table::drop().table(PackageLicense::Table).to_owned())
             .await?;
+        manager
+            .drop_table(Table::drop().table(PackageGroup::Table).to_owned())
+            .await?;
+        manager
+            .drop_table(Table::drop().table(PackageReplaces::Table).to_owned())
+            .await?;
+        manager
+            .drop_table(Table::drop().table(PackageConflicts::Table).to_owned())
+            .await?;
+        manager
+            .drop_table(Table::drop().table(PackageProvides::Table).to_owned())
+            .await?;
+        manager
+            .drop_table(Table::drop().table(PackageDepends::Table).to_owned())
+            .await?;
         manager
             .drop_table(Table::drop().table(Package::Table).to_owned())
             .await?;
@@ -141,3 +301,39 @@ pub enum PackageLicense {
     PackageId,
     Value,
 }
+
+#[derive(Iden)]
+pub enum PackageGroup {
+    Table,
+    PackageId,
+    Value,
+}
+
+#[derive(Iden)]
+pub enum PackageReplaces {
+    Table,
+    PackageId,
+    Value,
+}
+
+#[derive(Iden)]
+pub enum PackageConflicts {
+    Table,
+    PackageId,
+    Value,
+}
+
+#[derive(Iden)]
+pub enum PackageProvides {
+    Table,
+    PackageId,
+    Value,
+}
+
+#[derive(Iden)]
+pub enum PackageDepends {
+    Table,
+    PackageId,
+    Type,
+    Value,
+}

server/src/db/mod.rs

@@ -1,17 +1,69 @@
+mod conn;
 pub mod entities;
 mod migrator;
 
 use migrator::Migrator;
+use sea_orm::ColumnTrait;
 use sea_orm::ConnectOptions;
 use sea_orm::Database;
+use sea_orm::DatabaseConnection;
+use sea_orm::EntityTrait;
+use sea_orm::PaginatorTrait;
+use sea_orm::QueryFilter;
+use sea_orm::QueryOrder;
 use sea_orm_migration::MigratorTrait;
 
-pub async fn init<C: Into<ConnectOptions>>(
+pub use entities::prelude::*;
-    opt: C,
+pub use entities::*;
-) -> Result<sea_orm::DatabaseConnection, sea_orm::DbErr> {
-    let db = Database::connect(opt).await?;
 
-    Migrator::up(&db, None).await?;
+type Result<T> = std::result::Result<T, sea_orm::DbErr>;
 
-    Ok(db)
+#[derive(Clone, Debug)]
+pub struct RieterDb {
+    pub conn: DatabaseConnection,
+}
+
+impl RieterDb {
+    pub async fn connect<C: Into<ConnectOptions>>(opt: C) -> Result<Self> {
+        let db = Database::connect(opt).await?;
+
+        Migrator::up(&db, None).await?;
+
+        Ok(Self { conn: db })
+    }
+
+    pub async fn repos(&self, per_page: u64, page: u64) -> Result<(u64, Vec<repo::Model>)> {
+        let paginator = Repo::find()
+            .order_by_asc(repo::Column::Id)
+            .paginate(&self.conn, per_page);
+        let repos = paginator.fetch_page(page).await?;
+        let total_pages = paginator.num_pages().await?;
+
+        Ok((total_pages, repos))
+    }
+
+    pub async fn repo(&self, id: i32) -> Result<Option<repo::Model>> {
+        repo::Entity::find_by_id(id).one(&self.conn).await
+    }
+
+    pub async fn repo_by_name(&self, name: &str) -> Result<Option<repo::Model>> {
+        Repo::find()
+            .filter(repo::Column::Name.eq(name))
+            .one(&self.conn)
+            .await
+    }
+
+    pub async fn packages(&self, per_page: u64, page: u64) -> Result<(u64, Vec<package::Model>)> {
+        let paginator = Package::find()
+            .order_by_asc(package::Column::Id)
+            .paginate(&self.conn, per_page);
+        let packages = paginator.fetch_page(page).await?;
+        let total_pages = paginator.num_pages().await?;
+
+        Ok((total_pages, packages))
+    }
+
+    pub async fn package(&self, id: i32) -> Result<Option<package::Model>> {
+        package::Entity::find_by_id(id).one(&self.conn).await
+    }
 }

server/src/main.rs

@@ -7,7 +7,6 @@ mod repo;
 use clap::Parser;
 pub use error::{Result, ServerError};
 use repo::RepoGroupManager;
-use sea_orm::DatabaseConnection;
 use std::path::PathBuf;
 use std::sync::{Arc, RwLock};
 
@@ -23,7 +22,7 @@ pub struct Config {
 pub struct Global {
     config: Config,
     repo_manager: Arc<RwLock<RepoGroupManager>>,
-    db: DatabaseConnection,
+    db: db::RieterDb,
 }
 
 #[tokio::main]

server/src/repo/manager.rs

@@ -136,6 +136,10 @@ impl RepoGroupManager {
 
     /// Add a package to the given repo, returning to what architectures the package was added.
     pub fn add_pkg(&mut self, repo: &str, pkg: &Package) -> io::Result<()> {
+        // TODO
+        // * if arch is "any", check if package doesn't already exist for other architecture
+        // * if arch isn't "any", check if package doesn't already exist for "any" architecture
+
         // We first remove any existing version of the package
         self.remove_pkg(repo, &pkg.info.arch, &pkg.info.name, false)?;
 

server/src/repo/mod.rs

@@ -5,7 +5,7 @@ pub use manager::RepoGroupManager;
 
 use std::path::PathBuf;
 
-use crate::db::entities::{package as db_package, repo as db_repo};
+use crate::db;
 use axum::body::Body;
 use axum::extract::{BodyStream, Path, State};
 use axum::http::Request;
@@ -127,30 +127,27 @@ async fn post_package_archive(
             tracing::info!("Added '{}' to repository '{}'", pkg.file_name(), repo);
 
             // Query the repo for its ID, or create it if it does not already exist
-            let res = db_repo::Entity::find()
+            let res = global.db.repo_by_name(&repo).await?;
-                .filter(db_repo::Column::Name.eq(&repo))
-                .one(&global.db)
-                .await?;
 
             let repo_id = if let Some(repo_entity) = res {
                 repo_entity.id
             } else {
-                let model = db_repo::ActiveModel {
+                let model = db::repo::ActiveModel {
                     name: sea_orm::Set(repo.clone()),
                     ..Default::default()
                 };
 
-                db_repo::Entity::insert(model)
+                db::Repo::insert(model)
                     .exec(&global.db)
                     .await?
                     .last_insert_id
             };
 
             // If the package already exists in the database, we remove it first
-            let res = db_package::Entity::find()
+            let res = db::Package::find()
-                .filter(db_package::Column::RepoId.eq(repo_id))
+                .filter(db::package::Column::RepoId.eq(repo_id))
-                .filter(db_package::Column::Name.eq(&pkg.info.name))
+                .filter(db::package::Column::Name.eq(&pkg.info.name))
-                .filter(db_package::Column::Arch.eq(&pkg.info.arch))
+                .filter(db::package::Column::Arch.eq(&pkg.info.arch))
                 .one(&global.db)
                 .await?;
 
@@ -159,10 +156,18 @@ async fn post_package_archive(
             }
 
             // Insert the package's data into the database
-            let mut model: db_package::ActiveModel = pkg.into();
+            let mut model: db::package::ActiveModel = pkg.clone().into();
             model.repo_id = sea_orm::Set(repo_id);
 
-            model.insert(&global.db).await?;
+            let pkg_entry = model.insert(&global.db).await?;
+            db::PackageLicense::insert_many(pkg.info.licenses.iter().map(|s| {
+                db::package_license::ActiveModel {
+                    package_id: sea_orm::Set(pkg_entry.id),
+                    value: sea_orm::Set(s.to_string()),
+                }
+            }))
+            .exec(&global.db)
+            .await?;
 
             Ok(())
         }
@@ -187,10 +192,7 @@ async fn delete_repo(
             .await??;
 
     if repo_removed {
-        let res = db_repo::Entity::find()
+        let res = global.db.repo_by_name(&repo).await?;
-            .filter(db_repo::Column::Name.eq(&repo))
-            .one(&global.db)
-            .await?;
 
         if let Some(repo_entry) = res {
             repo_entry.delete(&global.db).await?;
@@ -221,16 +223,13 @@ async fn delete_arch_repo(
     .await??;
 
     if repo_removed {
-        let res = db_repo::Entity::find()
+        let res = global.db.repo_by_name(&repo).await?;
-            .filter(db_repo::Column::Name.eq(&repo))
-            .one(&global.db)
-            .await?;
 
         if let Some(repo_entry) = res {
             // Also remove all packages for that architecture from database
-            db_package::Entity::delete_many()
+            db::Package::delete_many()
-                .filter(db_package::Column::RepoId.eq(repo_entry.id))
+                .filter(db::package::Column::RepoId.eq(repo_entry.id))
-                .filter(db_package::Column::Arch.eq(&arch))
+                .filter(db::package::Column::Arch.eq(&arch))
                 .exec(&global.db)
                 .await?;
         }
@@ -255,18 +254,15 @@ async fn delete_package(
     .await??;
 
     if let Some((name, version, release, arch)) = res {
-        let res = db_repo::Entity::find()
+        let res = global.db.repo_by_name(&repo).await?;
-            .filter(db_repo::Column::Name.eq(&repo))
-            .one(&global.db)
-            .await?;
 
         if let Some(repo_entry) = res {
             // Also remove entry from database
-            let res = db_package::Entity::find()
+            let res = db::Package::find()
-                .filter(db_package::Column::RepoId.eq(repo_entry.id))
+                .filter(db::package::Column::RepoId.eq(repo_entry.id))
-                .filter(db_package::Column::Name.eq(name))
+                .filter(db::package::Column::Name.eq(name))
-                .filter(db_package::Column::Version.eq(format!("{}-{}", version, release)))
+                .filter(db::package::Column::Version.eq(format!("{}-{}", version, release)))
-                .filter(db_package::Column::Arch.eq(arch))
+                .filter(db::package::Column::Arch.eq(arch))
                 .one(&global.db)
                 .await?;
 

server/src/repo/package.rs

@@ -11,7 +11,7 @@ use crate::db::entities::package;
 
 const IGNORED_FILES: [&str; 5] = [".BUILDINFO", ".INSTALL", ".MTREE", ".PKGINFO", ".CHANGELOG"];
 
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct Package {
     pub path: PathBuf,
     pub info: PkgInfo,
@@ -19,7 +19,7 @@ pub struct Package {
     pub compression: ReadFilter,
 }
 
-#[derive(Debug, Default)]
+#[derive(Debug, Default, Clone)]
 pub struct PkgInfo {
     pub base: String,
     pub name: String,