nblm-core 0.2.0

Core library for NotebookLM Enterprise API client
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use async_trait::async_trait;
use reqwest::Method;

use crate::client::api::backends::{AudioBackend, BackendContext};
use crate::error::Result;
use crate::models::enterprise::audio::{AudioOverviewRequest, AudioOverviewResponse};

use super::models::{
    requests::audio as wire_audio_req, responses::audio::AudioOverviewApiResponse,
};

pub(crate) struct EnterpriseAudioBackend {
    ctx: BackendContext,
}

impl EnterpriseAudioBackend {
    pub fn new(ctx: BackendContext) -> Self {
        Self { ctx }
    }
}

#[async_trait]
impl AudioBackend for EnterpriseAudioBackend {
    async fn create_audio_overview(
        &self,
        notebook_id: &str,
        request: AudioOverviewRequest,
    ) -> Result<AudioOverviewResponse> {
        let path = format!(
            "{}/audioOverviews",
            self.ctx.url_builder.notebook_path(notebook_id)
        );
        let url = self.ctx.url_builder.build_url(&path)?;

        let wire_request: wire_audio_req::AudioOverviewRequest = request.into();
        let api_response: AudioOverviewApiResponse = self
            .ctx
            .http
            .request_json(Method::POST, url, Some(&wire_request))
            .await?;

        Ok(api_response.audio_overview.into())
    }

    async fn delete_audio_overview(&self, notebook_id: &str) -> Result<()> {
        let path = format!(
            "{}/audioOverviews/default",
            self.ctx.url_builder.notebook_path(notebook_id)
        );
        let url = self.ctx.url_builder.build_url(&path)?;
        let _response: serde_json::Value = self
            .ctx
            .http
            .request_json(Method::DELETE, url, None::<&()>)
            .await?;
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::auth::StaticTokenProvider;
    use crate::client::http::HttpClient;
    use crate::client::url::new_url_builder;
    use crate::client::{RetryConfig, Retryer};
    use crate::env::EnvironmentConfig;
    use std::sync::Arc;
    use std::time::Duration;

    fn create_test_backend() -> EnterpriseAudioBackend {
        let env = EnvironmentConfig::enterprise("123", "global", "us").unwrap();
        let client = reqwest::Client::builder()
            .timeout(Duration::from_millis(10))
            .build()
            .unwrap();
        let token = Arc::new(StaticTokenProvider::new("token"));
        let retryer = Retryer::new(RetryConfig::default());
        let http = Arc::new(HttpClient::new(client, token, retryer, None));
        let url_builder = new_url_builder(
            env.profile(),
            env.base_url().to_string(),
            env.parent_path().to_string(),
        );
        let ctx = BackendContext::new(http, url_builder);
        EnterpriseAudioBackend::new(ctx)
    }

    #[test]
    fn create_audio_overview_url_construction() {
        let backend = create_test_backend();
        let path = format!(
            "{}/audioOverviews",
            backend.ctx.url_builder.notebook_path("test-notebook")
        );
        let url = backend.ctx.url_builder.build_url(&path).unwrap();
        assert!(url.as_str().contains("test-notebook"));
        assert!(url.as_str().contains("audioOverviews"));
        assert!(!url.as_str().contains("default"));
    }

    #[test]
    fn delete_audio_overview_url_construction() {
        let backend = create_test_backend();
        let path = format!(
            "{}/audioOverviews/default",
            backend.ctx.url_builder.notebook_path("test-notebook")
        );
        let url = backend.ctx.url_builder.build_url(&path).unwrap();
        assert!(url.as_str().contains("test-notebook"));
        assert!(url.as_str().contains("audioOverviews/default"));
    }

    #[test]
    fn backend_construction() {
        let backend = create_test_backend();
        assert!(Arc::strong_count(&backend.ctx.http) >= 1);
        assert!(Arc::strong_count(&backend.ctx.url_builder) >= 1);
    }
}