use std::path::PathBuf; use actix::prelude::*; use actix_files::NamedFile; use actix_web::{error, get, post, web, Error, HttpResponse}; use log::debug; use serde::Deserialize; use crate::hash::{decode, Hash}; #[derive(Clone)] pub struct State { pub directory: PathBuf, pub db: Addr, pub hasher: Addr, } #[get("/raw/{hash}")] pub async fn get_raw(state: web::Data, hash: web::Path) -> Result { let response = state .db .send(crate::database::RetrieveByHash { hash: Hash(decode(hash.into_inner()).map_err(error::ErrorInternalServerError)?), }) .await?; debug!("{:?}", response); match response { Ok(result) => match result { Some(path) => Ok(NamedFile::open(path)?), None => Err(error::ErrorNotFound("NOT FOUND")), }, Err(e) => Err(error::ErrorInternalServerError(e)), } } #[derive(Deserialize)] pub struct LookupRequest { query: String, } #[get("/api/lookup")] pub async fn get_lookup( state: web::Data, web::Query(info): web::Query, ) -> Result { let response = state .db .send(crate::database::LookupByFilename { query: info.query }) .await?; Ok(HttpResponse::Ok().json(response.map_err(error::ErrorInternalServerError)?)) } #[post("/api/refresh")] pub async fn api_refresh(state: web::Data) -> Result { actix::spawn(crate::filesystem::reimport_directory( state.directory.clone(), state.db.clone(), state.hasher.clone(), )); Ok(HttpResponse::Ok().finish()) }