oko 0.1.14

Fully local home security system.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use futures_util::{Stream, StreamExt};
use tokio_util::sync::CancellationToken;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

pub use crate::web::{ApiChannelMessage, App, ImageContainer};

mod db;
mod users;
mod web;

pub use {
    db::Camera, db::CameraPermission, db::CameraPermissionUserView, db::CameraPermissionView,
    db::CameraSetting, db::CameraSettingNoMeta, db::Model, db::User, db::Video,
    db::VideoCameraView,
};

// Taken from https://github.com/hyperium/hyper/issues/2787#issuecomment-1073229886
/// Run a stream until it completes or we receive the shutdown signal.
///
/// Uses the `async-stream` to make things easier to write.
pub fn or_until_shutdown<S>(
    stream: S,
    shutdown_token: CancellationToken,
) -> impl Stream<Item = S::Item>
where
    S: Stream,
{
    async_stream::stream! {
        futures_util::pin_mut!(stream);

        loop {
            tokio::select! {
                Some(item) = stream.next() => {
                    yield item
                }
                () = shutdown_token.cancelled() => {
                    break;
                }
            }
        }
    }
}

pub async fn run() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    // TODO: Improve this, log to file, etc.
    // TODO: Use tracing wherever '?' is used.
    tracing_subscriber::registry()
        .with(
            tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| {
                format!(
                    "{}=debug,axum_login=debug,tower_sessions=debug,sqlx=warn,tower_http=debug",
                    env!("CARGO_CRATE_NAME")
                )
                .into()
            }),
        )
        .with(tracing_subscriber::fmt::layer().compact().without_time())
        .try_init()?;

    // TODO: Properly handle errors.
    App::new().await?.serve().await?;

    Ok(())
}