skynet_api 0.6.5

API for Skynet plugin.
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

use crate::{HyUuid, entity::webpush_subscriptions, hyuuid::uuids2strings, request::Condition};
use anyhow::Result;
use sea_orm::{
    ActiveModelTrait, ColumnTrait, ConnectionTrait, EntityTrait, PaginatorTrait, QueryFilter, Set,
};
use skynet_macro::default_viewer;

pub struct WebpushSubscriptionViewer;

#[default_viewer(webpush_subscriptions)]
impl WebpushSubscriptionViewer {
    pub async fn subscribe<C>(
        db: &C,
        uid: &HyUuid,
        topic: &HyUuid,
    ) -> Result<webpush_subscriptions::Model>
    where
        C: ConnectionTrait,
    {
        webpush_subscriptions::ActiveModel {
            uid: Set(*uid),
            topic: Set(*topic),
            ..Default::default()
        }
        .insert(db)
        .await
        .map_err(Into::into)
    }

    pub async fn unsubscribe<C>(db: &C, uid: &HyUuid, topic: &HyUuid) -> Result<u64>
    where
        C: ConnectionTrait,
    {
        webpush_subscriptions::Entity::delete_many()
            .filter(webpush_subscriptions::Column::Uid.eq(*uid))
            .filter(webpush_subscriptions::Column::Topic.eq(*topic))
            .exec(db)
            .await
            .map(|x| x.rows_affected)
            .map_err(Into::into)
    }
}