uqbar-dao/nectar
1
1
Fork 0
mirror of https://github.com/uqbar-dao/nectar.git synced 2024-12-21 07:31:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Format Rust code using rustfmt

Browse Source
This commit is contained in:
github-actions[bot] 2023-12-12 19:57:25 +00:00 committed by GitHub
parent 035b394a6c
commit 4990e83dee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 75 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main.rs
View File

@ -550,7 +550,6 @@ async fn main() {
} }
}; };
// gracefully abort all running processes in kernel // gracefully abort all running processes in kernel
let _ = kernel_message_sender let _ = kernel_message_sender
.send(KernelMessage { .send(KernelMessage {

2
src/types.rs
View File

@ -1060,7 +1060,7 @@ pub enum VfsAction {
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize, Deserialize)]
pub enum AddEntryType { pub enum AddEntryType {
Dir, Dir,
NewFile, // add a new file to fs and add name in vfs NewFile, // add a new file to fs and add name in vfs
ZipArchive, ZipArchive,
} }

94
src/vfs.rs
View File

@ -1,12 +1,12 @@
use dashmap::DashMap;
use std::collections::{HashMap, VecDeque}; use std::collections::{HashMap, VecDeque};
use std::io::prelude::*; use std::io::prelude::*;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::sync::Arc; use std::sync::Arc;
use tokio::sync::{Mutex, MutexGuard};
use tokio::fs; use tokio::fs;
use tokio::fs::OpenOptions; use tokio::fs::OpenOptions;
use tokio::io::{AsyncReadExt, AsyncSeekExt, AsyncWriteExt, SeekFrom}; use tokio::io::{AsyncReadExt, AsyncSeekExt, AsyncWriteExt, SeekFrom};
use dashmap::DashMap; use tokio::sync::{Mutex, MutexGuard};
use crate::types::*; use crate::types::*;
@ -126,8 +126,13 @@ async fn handle_request(
} }
}; };
check_caps(our_node.clone(), source.clone(), send_to_caps_oracle.clone(), &request) check_caps(
.await?; our_node.clone(),
source.clone(),
send_to_caps_oracle.clone(),
&request,
)
.await?;
let (ipc, bytes) = match request.action { let (ipc, bytes) = match request.action {
VfsAction::New => { VfsAction::New => {
@ -140,13 +145,17 @@ async fn handle_request(
} => { } => {
match entry_type { match entry_type {
AddEntryType::Dir => { AddEntryType::Dir => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone())
.await?;
// create dir. // create dir.
fs::create_dir_all(path).await.unwrap(); fs::create_dir_all(path).await.unwrap();
} }
AddEntryType::NewFile => { AddEntryType::NewFile => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone())
.await?;
// open and create file // open and create file
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
@ -187,7 +196,8 @@ async fn handle_request(
if is_file { if is_file {
// create file // create file
println!("writing a file!, orig filename {:?}", path); println!("writing a file!, orig filename {:?}", path);
let path = validate_path(vfs_path.clone(), request.drive.clone(), path).await?; let path = validate_path(vfs_path.clone(), request.drive.clone(), path)
.await?;
println!("with the path: {:?}", path); println!("with the path: {:?}", path);
println!("and original path: {:?}", full_path); println!("and original path: {:?}", full_path);
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
@ -195,9 +205,13 @@ async fn handle_request(
let mut file = file.lock().await; let mut file = file.lock().await;
file.write_all(&file_contents).await.unwrap(); file.write_all(&file_contents).await.unwrap();
println!("actually wrote file!"); println!("actually wrote file!");
} else if is_dir { } else if is_dir {
let path = validate_path(vfs_path.clone(), request.drive.clone(), path.clone()).await?; let path = validate_path(
vfs_path.clone(),
request.drive.clone(),
path.clone(),
)
.await?;
// If it's a directory, create it // If it's a directory, create it
fs::create_dir_all(path).await.unwrap(); fs::create_dir_all(path).await.unwrap();
@ -211,7 +225,8 @@ async fn handle_request(
(serde_json::to_vec(&VfsResponse::Ok).unwrap(), None) (serde_json::to_vec(&VfsResponse::Ok).unwrap(), None)
} }
VfsAction::Delete(mut full_path) => { VfsAction::Delete(mut full_path) => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
(serde_json::to_vec(&VfsResponse::Ok).unwrap(), None) (serde_json::to_vec(&VfsResponse::Ok).unwrap(), None)
} }
@ -219,7 +234,8 @@ async fn handle_request(
mut full_path, mut full_path,
offset, offset,
} => { } => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
file.seek(SeekFrom::Start(offset)).await.unwrap(); file.seek(SeekFrom::Start(offset)).await.unwrap();
@ -227,7 +243,8 @@ async fn handle_request(
(serde_json::to_vec(&VfsResponse::Ok).unwrap(), None) (serde_json::to_vec(&VfsResponse::Ok).unwrap(), None)
} }
VfsAction::Append(mut full_path) => { VfsAction::Append(mut full_path) => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
file.seek(SeekFrom::End(0)).await.unwrap(); file.seek(SeekFrom::End(0)).await.unwrap();
@ -239,7 +256,8 @@ async fn handle_request(
mut full_path, mut full_path,
size, size,
} => { } => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
file.set_len(size).await.unwrap(); file.set_len(size).await.unwrap();
@ -248,7 +266,8 @@ async fn handle_request(
} }
VfsAction::GetEntry(mut full_path) => { VfsAction::GetEntry(mut full_path) => {
println!("getting entry for path: {:?}", full_path); println!("getting entry for path: {:?}", full_path);
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
println!("getentry path resolved to: {:?}", path); println!("getentry path resolved to: {:?}", path);
let metadata = fs::metadata(&path).await.unwrap(); let metadata = fs::metadata(&path).await.unwrap();
if metadata.is_dir() { if metadata.is_dir() {
@ -259,7 +278,14 @@ async fn handle_request(
children.push(entry.path().display().to_string()); children.push(entry.path().display().to_string());
} }
(serde_json::to_vec(&VfsResponse::GetEntry { is_file: false, children }).unwrap(), None) (
serde_json::to_vec(&VfsResponse::GetEntry {
is_file: false,
children,
})
.unwrap(),
None,
)
} else if metadata.is_file() { } else if metadata.is_file() {
println!("is file!"); println!("is file!");
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
@ -268,10 +294,16 @@ async fn handle_request(
let mut contents = Vec::new(); let mut contents = Vec::new();
file.read_to_end(&mut contents).await.unwrap(); file.read_to_end(&mut contents).await.unwrap();
println!("read contents to last part"); println!("read contents to last part");
(serde_json::to_vec(&VfsResponse::GetEntry { is_file: true, children: Vec::new() }).unwrap(), Some(contents)) (
serde_json::to_vec(&VfsResponse::GetEntry {
is_file: true,
children: Vec::new(),
})
.unwrap(),
Some(contents),
)
} else { } else {
return Err(VfsError::InternalError) return Err(VfsError::InternalError);
} }
} }
VfsAction::GetFileChunk { VfsAction::GetFileChunk {
@ -279,31 +311,33 @@ async fn handle_request(
offset, offset,
length, length,
} => { } => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
let mut contents = vec![0; length as usize]; let mut contents = vec![0; length as usize];
file.seek(SeekFrom::Start(offset)).await.unwrap(); file.seek(SeekFrom::Start(offset)).await.unwrap();
file.read_exact(&mut contents).await.unwrap(); file.read_exact(&mut contents).await.unwrap();
( (
serde_json::to_vec(&VfsResponse::GetFileChunk).unwrap(), serde_json::to_vec(&VfsResponse::GetFileChunk).unwrap(),
Some(contents), Some(contents),
) )
} }
VfsAction::GetEntryLength(mut full_path) => { VfsAction::GetEntryLength(mut full_path) => {
let path = validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?; let path =
validate_path(vfs_path.clone(), request.drive.clone(), full_path.clone()).await?;
let file = open_file(open_files.clone(), path).await?; let file = open_file(open_files.clone(), path).await?;
let mut file = file.lock().await; let mut file = file.lock().await;
let length = file.metadata().await.unwrap().len(); let length = file.metadata().await.unwrap().len();
( (
serde_json::to_vec(&VfsResponse::GetEntryLength(Some(length))).unwrap(), serde_json::to_vec(&VfsResponse::GetEntryLength(Some(length))).unwrap(),
None, None,
) )
} }
}; };
if let Some(target) = km.rsvp.or_else(|| { if let Some(target) = km.rsvp.or_else(|| {
expects_response.map(|_| Address { expects_response.map(|_| Address {
@ -352,7 +386,11 @@ async fn handle_request(
Ok(()) Ok(())
} }
async fn validate_path(vfs_path: String, drive: String, request_path: String) -> Result<PathBuf, VfsError> { async fn validate_path(
vfs_path: String,
drive: String,
request_path: String,
) -> Result<PathBuf, VfsError> {
let drive_base = Path::new(&vfs_path).join(&drive); let drive_base = Path::new(&vfs_path).join(&drive);
if let Err(e) = fs::create_dir_all(&drive_base).await { if let Err(e) = fs::create_dir_all(&drive_base).await {
println!("failed creating drive dir! {:?}", e); println!("failed creating drive dir! {:?}", e);