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
42
43

#![crate_name = "smart_open"]
extern crate flate2;

use std::fs::File;
use std::io::{BufReader, Read};
use std::path::Path;
use std::string::String;
use flate2::read::GzDecoder;

/// Returns the contents of the file that has been passed as filepath
///
/// # Examples
///
/// ```
/// let filepath = "tests/foo.txt";
/// assert_eq!(smart_open::smart_open(&filepath).unwrap(), "Hello, world!");
/// ```
pub fn smart_open(filepath: &str) -> std::io::Result<String> {
    // TODO: make the function more modular.
    let path = Path::new(filepath);
    let content_type = match path.extension() {
        None => panic!("Paths without extension is not allowed!!"),
        Some(os_str) => {
            match os_str.to_str() {
                None => panic!("None has seeped in somehow. Please contact developers."),
                Some("gz") => "gz",
                _ => "text",
            }
        }
    };
    let file_handler = File::open(&path)?;
    let mut buf_reader = BufReader::new(file_handler);
    let mut contents = String::new();
    if content_type == "gz" {
        let mut reader = GzDecoder::new(buf_reader);
        reader.read_to_string(&mut contents).unwrap();
    } else if content_type == "text" {
        buf_reader.read_to_string(&mut contents)?;
    } else {
        panic!("Content type {} is not allowed.", content_type);
    }
    Ok(contents.trim().to_string())
}