dnutiu/social-media-news-bot
dnutiu/social-media-news-bot
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

implement upload_image method

Browse source
This commit is contained in:
Denis-Cosmin NUTIU 2024-12-30 22:34:50 +02:00
parent 67c6336b04
commit 0997ab8c20
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
bot/src/bluesky.rs
View file

@ -1,8 +1,10 @@
pub(crate) mod atproto; pub(crate) mod atproto;
mod token; mod token;
use crate::bluesky::atproto::ATProtoServerCreateSession; use crate::bluesky::atproto::{ATProtoServerCreateSession, BlobResponse};
use anyhow::anyhow; use anyhow::anyhow;
use base64::prelude::BASE64_STANDARD;
use base64::Engine;
use reqwest::Body; use reqwest::Body;
use token::Token; use token::Token;
@ -13,6 +15,7 @@ pub struct BlueSkyClient {
} }
impl BlueSkyClient { impl BlueSkyClient {
/// Creates a new BlueSky client instance for the given account credentials.
pub async fn new(user_handle: &str, user_password: &str) -> Result<Self, anyhow::Error> { pub async fn new(user_handle: &str, user_password: &str) -> Result<Self, anyhow::Error> {
let client = reqwest::Client::new(); let client = reqwest::Client::new();
let server_create_session = ATProtoServerCreateSession { let server_create_session = ATProtoServerCreateSession {
@ -35,6 +38,7 @@ impl BlueSkyClient {
}) })
} }
/// Makes a new tweet.
pub async fn post<T>(&mut self, body: T) -> Result<(), anyhow::Error> pub async fn post<T>(&mut self, body: T) -> Result<(), anyhow::Error>
where where
T: Into<Body>, T: Into<Body>,
@ -62,6 +66,7 @@ impl BlueSkyClient {
Ok(()) Ok(())
} }
/// Renews the Authentication JWT bearer token using the refresh token.
async fn renew_token(&mut self) -> Result<(), anyhow::Error> { async fn renew_token(&mut self) -> Result<(), anyhow::Error> {
let result: Token = self let result: Token = self
.client .client
@ -78,4 +83,22 @@ impl BlueSkyClient {
self.auth_token = result; self.auth_token = result;
Ok(()) Ok(())
} }
/// Uploads an image.
async fn upload_image(&mut self, payload: Vec<u8>) -> Result<BlobResponse, anyhow::Error> {
let base64_data = BASE64_STANDARD.encode(payload);
Ok(self
.client
.post("https://bsky.social/xrpc/com.atproto.repo.uploadBlob")
.header("Content-Type", "image/png")
.header(
"Authorization",
format!("Bearer {}", self.auth_token.access_jwt),
)
.body(base64_data)
.send()
.await?
.json()
.await?)
}
} }