ssh-transfer 0.4.0

License to transfer file via SFTP protocol
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use crate::error::{Error::SftpError, Result};
use ssh2::{File, OpenFlags, OpenType, Session};
use std::{io::Write, path::Path};

pub struct SftpWriter {
  path: String,
  file: File,
}

impl SftpWriter {
  pub fn new(session: &Session, path: &str) -> Result<Self> {
    let file = session.sftp()?.open_mode(
      Path::new(path),
      OpenFlags::WRITE | OpenFlags::TRUNCATE,
      0o644,
      OpenType::File,
    )?;
    Ok(SftpWriter {
      path: path.to_string(),
      file,
    })
  }

  pub fn get_size(&mut self) -> Result<u64> {
    self
      .file
      .stat()?
      .size
      .ok_or_else(|| SftpError(format!("Cannot retrieve size for path: {}", self.path)))
  }
}

impl Write for SftpWriter {
  fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
    self.file.write(buf)
  }

  fn flush(&mut self) -> std::io::Result<()> {
    self.file.flush()
  }
}