feat: construct request function
wow a helper function changed get_live_id to private
This commit is contained in:
16
src/grpc.rs
16
src/grpc.rs
@@ -1,12 +1,13 @@
|
|||||||
use std::process::Command;
|
use std::process::Command;
|
||||||
|
|
||||||
use tonic::{metadata::MetadataValue, Request, Status};
|
use stream_list::{v3_data_live_chat_message_service_client::V3DataLiveChatMessageServiceClient, LiveChatMessageListRequest};
|
||||||
|
use tonic::{metadata::MetadataValue, service::interceptor::InterceptedService, transport::Channel, Request, Status};
|
||||||
|
|
||||||
pub mod stream_list {
|
pub mod stream_list {
|
||||||
tonic::include_proto!("youtube.api.v3");
|
tonic::include_proto!("youtube.api.v3");
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_live_id(channel: String) -> String {
|
fn get_live_id(channel: String) -> String {
|
||||||
let video_id = Command::new("./get_url.sh")
|
let video_id = Command::new("./get_url.sh")
|
||||||
.arg(channel)
|
.arg(channel)
|
||||||
.output()
|
.output()
|
||||||
@@ -25,6 +26,17 @@ pub fn auth_header(mut req: Request<()>) -> Result<Request<()>, Status> {
|
|||||||
.expect("WHAT");
|
.expect("WHAT");
|
||||||
Ok(req)
|
Ok(req)
|
||||||
}
|
}
|
||||||
|
pub fn construct_request(channel: String) -> Request<LiveChatMessageListRequest>{
|
||||||
|
let request = Request::new(LiveChatMessageListRequest {
|
||||||
|
part: vec!["snippet".to_string()],
|
||||||
|
live_chat_id: Some(get_live_id(channel)),
|
||||||
|
max_results: Some(20),
|
||||||
|
page_token: None,
|
||||||
|
hl: None,
|
||||||
|
profile_image_size: None,
|
||||||
|
});
|
||||||
|
request
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_get_live_id() {
|
fn test_get_live_id() {
|
||||||
|
|||||||
Reference in New Issue
Block a user