asimov_runner/
importer.rs

1// This is free and unencumbered software released into the public domain.
2
3use crate::{Runner, RunnerError};
4use async_trait::async_trait;
5use std::{ffi::OsStr, io::Cursor, process::Stdio};
6
7pub use asimov_patterns::ImporterOptions;
8
9pub type ImporterResult = std::result::Result<Cursor<Vec<u8>>, RunnerError>; // TODO
10
11/// RDF dataset importer. Consumes a URL input, produces RDF output.
12#[derive(Debug)]
13#[allow(unused)]
14pub struct Importer {
15    runner: Runner,
16    options: ImporterOptions,
17}
18
19impl Importer {
20    pub fn new(program: impl AsRef<OsStr>, options: ImporterOptions) -> Self {
21        let mut runner = Runner::new(program);
22
23        runner
24            .command()
25            .arg(&options.input_url)
26            .stdin(Stdio::null())
27            .stdout(Stdio::piped())
28            .stderr(Stdio::piped());
29
30        Self { runner, options }
31    }
32}
33
34impl asimov_patterns::Importer<Cursor<Vec<u8>>, RunnerError> for Importer {}
35
36#[async_trait]
37impl asimov_patterns::Execute<Cursor<Vec<u8>>, RunnerError> for Importer {
38    async fn execute(&mut self) -> ImporterResult {
39        let stdout = self.runner.execute().await?;
40        Ok(stdout)
41    }
42}
43
44#[cfg(test)]
45mod tests {
46    use super::*;
47    use asimov_patterns::Execute;
48
49    #[tokio::test]
50    async fn test_execute() {
51        let mut runner = Importer::new(
52            "curl",
53            ImporterOptions {
54                input_url: "https://www.google.com/robots.txt".to_string(),
55            },
56        );
57        let result = runner.execute().await;
58        assert!(result.is_ok());
59    }
60}
asimov_runner/importer.rs

asimov_runner/
importer.rs
