85 lines
1.9 KiB
Rust
85 lines
1.9 KiB
Rust
use std::time::Duration;
|
|
|
|
use axum::body::Bytes;
|
|
use reqwest::{
|
|
multipart::{Form, Part},
|
|
Response,
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::{error::ServiceError, JMServiceInner};
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct IPFSFile {
|
|
#[serde(rename = "Hash")]
|
|
pub hash: String,
|
|
#[serde(rename = "Name")]
|
|
pub name: String,
|
|
#[serde(rename = "Size")]
|
|
pub size: String,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
pub struct CatQuery {
|
|
pub arg: String,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
pub struct AddQuery {
|
|
pub pin: bool,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
pub struct PinQuery {
|
|
pub arg: String,
|
|
}
|
|
|
|
impl JMServiceInner {
|
|
pub async fn cat(&self, cid: String) -> Result<Response, ServiceError> {
|
|
let request = self
|
|
.client
|
|
.post(self.ipfs_url.join("/api/v0/cat")?)
|
|
.query(&CatQuery::new(cid));
|
|
Ok(request.send().await?)
|
|
}
|
|
|
|
pub async fn add(&self, file: Bytes, filename: String) -> Result<IPFSFile, ServiceError> {
|
|
let request = self
|
|
.client
|
|
.post(self.ipfs_url.join("/api/v0/add")?)
|
|
.query(&AddQuery::new(false))
|
|
.multipart(Form::new().part("file", Part::stream(file).file_name(filename)));
|
|
let response = request.send().await?;
|
|
let res: IPFSFile = response.json().await?;
|
|
Ok(res)
|
|
}
|
|
|
|
pub async fn pin(&self, cid: String) -> Result<(), ServiceError> {
|
|
let request = self
|
|
.client
|
|
.post(self.ipfs_url.join("/api/v0/pin/add")?)
|
|
.query(&PinQuery::new(cid))
|
|
.timeout(Duration::from_secs(60));
|
|
request.send().await?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl CatQuery {
|
|
pub fn new(cid: String) -> Self {
|
|
Self { arg: cid }
|
|
}
|
|
}
|
|
|
|
impl AddQuery {
|
|
pub fn new(pin: bool) -> Self {
|
|
Self { pin }
|
|
}
|
|
}
|
|
|
|
impl PinQuery {
|
|
pub fn new(cid: String) -> Self {
|
|
Self { arg: cid }
|
|
}
|
|
}
|