tauri-workspace/tauri/src/endpoints/file_system.rs

use crate::ApplicationDispatcherExt;

use tauri_api::dir;
use tauri_api::file;
use tauri_api::path::resolve_path;

use std::fs;
use std::fs::File;
use std::io::Write;
use std::path::PathBuf;

use super::cmd::{DirOperationOptions, FileOperationOptions};

/// Reads a directory.
#[cfg(read_dir)]
pub async fn read_dir<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<DirOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let (recursive, dir) = if let Some(options_value) = options {
        (options_value.recursive, options_value.dir)
      } else {
        (false, None)
      };
      dir::read_dir(resolve_path(path, dir)?, recursive).map_err(crate::Error::FailedToExecuteApi)
    },
    callback,
    error,
  )
  .await;
}

/// Copies a file.
#[cfg(copy_file)]
pub async fn copy_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  source: PathBuf,
  destination: PathBuf,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let (src, dest) = match options.and_then(|o| o.dir) {
        Some(dir) => (
          resolve_path(source, Some(dir.clone()))?,
          resolve_path(destination, Some(dir))?,
        ),
        None => (source, destination),
      };
      fs::copy(src, dest)?;
      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Creates a directory.
#[cfg(create_dir)]
pub async fn create_dir<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<DirOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let (recursive, dir) = if let Some(options_value) = options {
        (options_value.recursive, options_value.dir)
      } else {
        (false, None)
      };
      let resolved_path = resolve_path(path, dir)?;
      if recursive {
        fs::create_dir_all(resolved_path)?;
      } else {
        fs::create_dir(resolved_path)?;
      }

      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Removes a directory.
#[cfg(remove_dir)]
pub async fn remove_dir<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<DirOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let (recursive, dir) = if let Some(options_value) = options {
        (options_value.recursive, options_value.dir)
      } else {
        (false, None)
      };
      let resolved_path = resolve_path(path, dir)?;
      if recursive {
        fs::remove_dir_all(resolved_path)?;
      } else {
        fs::remove_dir(resolved_path)?;
      }

      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Removes a file
#[cfg(remove_file)]
pub async fn remove_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let resolved_path = resolve_path(path, options.and_then(|o| o.dir))?;
      fs::remove_file(resolved_path)?;
      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Renames a file.
#[cfg(rename_file)]
pub async fn rename_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  old_path: PathBuf,
  new_path: PathBuf,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      let (old, new) = match options.and_then(|o| o.dir) {
        Some(dir) => (
          resolve_path(old_path, Some(dir.clone()))?,
          resolve_path(new_path, Some(dir))?,
        ),
        None => (old_path, new_path),
      };
      fs::rename(old, new).map_err(crate::Error::Io)
    },
    callback,
    error,
  )
  .await;
}

/// Writes a text file.
#[cfg(write_file)]
pub async fn write_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  contents: String,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      File::create(resolve_path(path, options.and_then(|o| o.dir))?)
        .map_err(crate::Error::Io)
        .and_then(|mut f| f.write_all(contents.as_bytes()).map_err(|err| err.into()))?;
      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Writes a binary file.
#[cfg(write_binary_file)]
pub async fn write_binary_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  contents: String,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      base64::decode(contents)
        .map_err(crate::Error::Base64Decode)
        .and_then(|c| {
          File::create(resolve_path(path, options.and_then(|o| o.dir))?)
            .map_err(|e| e.into())
            .and_then(|mut f| f.write_all(&c).map_err(|err| err.into()))
        })?;
      crate::Result::Ok(())
    },
    callback,
    error,
  )
  .await;
}

/// Reads a text file.
#[cfg(read_text_file)]
pub async fn read_text_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      file::read_string(resolve_path(path, options.and_then(|o| o.dir))?)
        .map_err(crate::Error::FailedToExecuteApi)
    },
    callback,
    error,
  )
  .await;
}

/// Reads a binary file.
#[cfg(read_binary_file)]
pub async fn read_binary_file<D: ApplicationDispatcherExt>(
  dispatcher: &mut D,
  path: PathBuf,
  options: Option<FileOperationOptions>,
  callback: String,
  error: String,
) {
  crate::execute_promise(
    dispatcher,
    async move {
      file::read_binary(resolve_path(path, options.and_then(|o| o.dir))?)
        .map_err(crate::Error::FailedToExecuteApi)
    },
    callback,
    error,
  )
  .await;
}

// test webview functionality.
#[cfg(test)]
mod test {
  // use super::*;
  // use web_view::*;

  // create a makeshift webview
  // fn create_test_webview() -> crate::Result<WebView<'static, ()>> {
  //   // basic html set into webview
  //   let content = r#"<html><head></head><body></body></html>"#;

  //   Ok(
  //     // use webview builder to create simple webview
  //     WebViewBuilder::new()
  //       .title("test")
  //       .size(800, 800)
  //       .resizable(true)
  //       .debug(true)
  //       .user_data(())
  //       .invoke_handler(|_wv, _arg| Ok(()))
  //       .content(Content::Html(content))
  //       .build()?,
  //   )
  // }

  /* #[test]
  #[cfg(not(any(target_os = "linux", target_os = "macos")))]
  // test the file_write functionality
  fn test_write_to_file() -> crate::Result<()> {
    // import read_to_string and write to be able to manipulate the file.
    use std::fs::{read_to_string, write};

    // create the webview
    let mut webview = create_test_webview()?;

    // setup the contents and the path.
    let contents = String::from(r#"Write to the Test file"#);
    let path = String::from("test/fixture/test.txt");

    // clear the file by writing nothing to it.
    write(&path, "")?;

    //call write file with the path and contents.
    write_file(
      &mut dispatcher,
      path.clone(),
      contents.clone(),
      String::from(""),
      String::from(""),
    );

    // sleep the main thread to wait for the promise to execute.
    std::thread::sleep(std::time::Duration::from_millis(200));

    // read from the file.
    let data = read_to_string(path)?;

    // check that the file contents is equal to the expected contents.
    assert_eq!(data, contents);

    Ok(())
  } */
}