add: migrate to models module

src/models/dtos/keyframes/mod.rs

@@ -0,0 +1,38 @@
+use crate::entities::{keyframes::KeyframeEntity, videos::VideoEntity};
+
+#[derive(Debug, Clone, serde::Serialize, utoipa::ToSchema)]
+pub struct KeyframeDto {
+    id: i64,
+    path: String,
+    timestamp: f64,
+}
+
+impl From<(KeyframeEntity, VideoEntity)> for KeyframeDto {
+    fn from((keyframe, video): (KeyframeEntity, VideoEntity)) -> Self {
+        let id = keyframe.id();
+        let timestamp = keyframe.video_related_frame_timestamp();
+        let path = format!(
+            "/static/keyframes/L{}_V{:0>3}/{:0>3}.jpg",
+            video.l(),
+            video.v(),
+            keyframe.video_related_frame_id()
+        );
+        Self {
+            id,
+            path,
+            timestamp,
+        }
+    }
+}
+
+impl From<(VideoEntity, KeyframeEntity)> for KeyframeDto {
+    fn from((video, keyframe): (VideoEntity, KeyframeEntity)) -> Self {
+        Self::from((keyframe, video))
+    }
+}
+
+#[derive(Debug, Clone, Copy, serde::Deserialize)]
+pub struct VectoredKeyframeDto {
+    id: i64,
+    score: f64,
+}

src/models/dtos/mod.rs

@@ -0,0 +1,3 @@
+pub mod keyframes;
+pub mod vectors;
+pub mod videos;

src/models/dtos/videos/mod.rs

@@ -0,0 +1,17 @@
+use crate::entities::videos::VideoEntity;
+
+#[derive(Debug, Clone, serde::Serialize, utoipa::ToSchema)]
+pub struct VideoDto {
+    id: i64,
+    path: String,
+}
+
+impl From<VideoEntity> for VideoDto {
+    fn from(value: VideoEntity) -> Self {
+        let path = format!("/static/videos/L{}_V{:0>3}.mp4", value.l(), value.v());
+        Self {
+            id: value.id(),
+            path,
+        }
+    }
+}

src/models/entities/keyframes/mod.rs

@@ -0,0 +1,22 @@
+#[derive(Debug, Clone, diesel::Queryable, diesel::Selectable)]
+#[diesel(table_name = crate::schema::keyframes)]
+#[diesel(check_for_backend(diesel::pg::Pg))]
+pub struct KeyframeEntity {
+    id: i64,
+    video_related_frame_id: i16,
+    video_related_frame_timestamp: f64,
+}
+
+impl KeyframeEntity {
+    pub fn id(&self) -> i64 {
+        self.id
+    }
+
+    pub fn video_related_frame_id(&self) -> i16 {
+        self.video_related_frame_id
+    }
+
+    pub fn video_related_frame_timestamp(&self) -> f64 {
+        self.video_related_frame_timestamp
+    }
+}

src/models/entities/mod.rs

@@ -0,0 +1,2 @@
+pub mod keyframes;
+pub mod videos;

src/models/entities/videos/mod.rs

@@ -0,0 +1,22 @@
+#[derive(Debug, Clone, diesel::Queryable, diesel::Selectable)]
+#[diesel(table_name = crate::schema::videos)]
+#[diesel(check_for_backend(diesel::pg::Pg))]
+pub struct VideoEntity {
+    id: i64,
+    l: i16,
+    v: i16,
+}
+
+impl VideoEntity {
+    pub fn id(&self) -> i64 {
+        self.id
+    }
+
+    pub fn l(&self) -> i16 {
+        self.l
+    }
+
+    pub fn v(&self) -> i16 {
+        self.v
+    }
+}

src/models/mod.rs

@@ -0,0 +1,3 @@
+pub mod dtos;
+pub mod entities;
+pub mod states;

src/models/states/mod.rs

@@ -0,0 +1,48 @@
+use std::sync::Arc;
+
+use deadpool_diesel::{
+    Runtime,
+    postgres::{Manager, Pool},
+};
+use qdrant_client::Qdrant;
+
+#[derive(Clone)]
+pub struct AppState {
+    diesel_pool: Pool,
+    qdrant_client: Arc<Qdrant>,
+}
+
+impl AppState {
+    pub fn new() -> anyhow::Result<Self> {
+        Ok(Self {
+            diesel_pool: Self::diesel_pool_helper()?,
+            qdrant_client: Self::qdrant_client_helper()?,
+        })
+    }
+
+    fn diesel_pool_helper() -> anyhow::Result<Pool> {
+        Pool::builder(Manager::new(
+            std::env::var("DATABASE_URL")?,
+            Runtime::Tokio1,
+        ))
+        .build()
+        .map_err(Into::into)
+    }
+
+    fn qdrant_client_helper() -> anyhow::Result<Arc<Qdrant>> {
+        Qdrant::from_url(&std::env::var("QDRANT_URL")?)
+            .build()
+            .map(Arc::new)
+            .map_err(Into::into)
+    }
+}
+
+impl AppState {
+    pub fn pool(&self) -> &Pool {
+        &self.diesel_pool
+    }
+
+    pub fn qdrant_client(&self) -> &Qdrant {
+        &self.qdrant_client
+    }
+}