Formatting
This commit is contained in:
		
							parent
							
								
									c0c56627c1
								
							
						
					
					
						commit
						8426bf7c8c
					
				| 
						 | 
				
			
			@ -106,7 +106,11 @@ impl CommitProcessor for PostIndexer {
 | 
			
		|||
                self.config.feed_generator_did, commit.seq, commit.time
 | 
			
		||||
            );
 | 
			
		||||
            self.database
 | 
			
		||||
                .update_subscription_cursor(Bluesky::FIREHOSE_HOST, &self.config.feed_generator_did, commit.seq)
 | 
			
		||||
                .update_subscription_cursor(
 | 
			
		||||
                    Bluesky::FIREHOSE_HOST,
 | 
			
		||||
                    &self.config.feed_generator_did,
 | 
			
		||||
                    commit.seq,
 | 
			
		||||
                )
 | 
			
		||||
                .await?;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -56,16 +56,18 @@ impl ProfileClassifier {
 | 
			
		|||
    }
 | 
			
		||||
 | 
			
		||||
    async fn fill_in_profile_details(&self, did: &str) -> Result<()> {
 | 
			
		||||
        let details = self.bluesky.fetch_profile_details(did).await.context("Could not fetch profile details")?;
 | 
			
		||||
        let details = self
 | 
			
		||||
            .bluesky
 | 
			
		||||
            .fetch_profile_details(did)
 | 
			
		||||
            .await
 | 
			
		||||
            .context("Could not fetch profile details")?;
 | 
			
		||||
 | 
			
		||||
        let country = match details {
 | 
			
		||||
            Some(details) => {
 | 
			
		||||
                self.ai
 | 
			
		||||
                    .infer_country_of_living(&details.display_name, &details.description)
 | 
			
		||||
                    .await
 | 
			
		||||
                    .context("Could not infer country of living")
 | 
			
		||||
                    ?
 | 
			
		||||
            }
 | 
			
		||||
            Some(details) => self
 | 
			
		||||
                .ai
 | 
			
		||||
                .infer_country_of_living(&details.display_name, &details.description)
 | 
			
		||||
                .await
 | 
			
		||||
                .context("Could not infer country of living")?,
 | 
			
		||||
            None => "xx".to_owned(),
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,9 +1,9 @@
 | 
			
		|||
use std::matches;
 | 
			
		||||
 | 
			
		||||
use anyhow::{anyhow, Result};
 | 
			
		||||
use atrium_api::agent::{AtpAgent, Session};
 | 
			
		||||
use atrium_api::blob::BlobRef;
 | 
			
		||||
use atrium_api::records::Record;
 | 
			
		||||
use atrium_api::agent::{AtpAgent, Session};
 | 
			
		||||
use atrium_xrpc::client::reqwest::ReqwestClient;
 | 
			
		||||
use axum::http::StatusCode;
 | 
			
		||||
use chrono::Utc;
 | 
			
		||||
| 
						 | 
				
			
			@ -11,7 +11,7 @@ use futures::StreamExt;
 | 
			
		|||
use log::error;
 | 
			
		||||
use tokio_tungstenite::{connect_async, tungstenite};
 | 
			
		||||
 | 
			
		||||
use super::entities::{ProfileDetails};
 | 
			
		||||
use super::entities::ProfileDetails;
 | 
			
		||||
use super::streaming::{handle_message, CommitProcessor};
 | 
			
		||||
 | 
			
		||||
pub struct Bluesky {
 | 
			
		||||
| 
						 | 
				
			
			@ -24,7 +24,7 @@ impl Bluesky {
 | 
			
		|||
 | 
			
		||||
    pub fn unauthenticated() -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            agent: AtpAgent::new(ReqwestClient::new(Self::XRPC_HOST.to_owned()))
 | 
			
		||||
            agent: AtpAgent::new(ReqwestClient::new(Self::XRPC_HOST.to_owned())),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -40,14 +40,7 @@ impl Bluesky {
 | 
			
		|||
    }
 | 
			
		||||
 | 
			
		||||
    pub async fn upload_blob(&self, blob: Vec<u8>) -> Result<BlobRef> {
 | 
			
		||||
        let result = self
 | 
			
		||||
            .agent
 | 
			
		||||
            .api
 | 
			
		||||
            .com
 | 
			
		||||
            .atproto
 | 
			
		||||
            .repo
 | 
			
		||||
            .upload_blob(blob)
 | 
			
		||||
            .await?;
 | 
			
		||||
        let result = self.agent.api.com.atproto.repo.upload_blob(blob).await?;
 | 
			
		||||
 | 
			
		||||
        Ok(result.blob)
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -247,7 +247,12 @@ impl Database {
 | 
			
		|||
        .map(|result| result.rows_affected() > 0)?)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub async fn update_subscription_cursor(&self, host: &str, did: &str, cursor: i32) -> Result<bool> {
 | 
			
		||||
    pub async fn update_subscription_cursor(
 | 
			
		||||
        &self,
 | 
			
		||||
        host: &str,
 | 
			
		||||
        did: &str,
 | 
			
		||||
        cursor: i32,
 | 
			
		||||
    ) -> Result<bool> {
 | 
			
		||||
        let mut params = Parameters::new();
 | 
			
		||||
 | 
			
		||||
        Ok(query(
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue