1 files changed, 86 insertions, 0 deletions
diff --git a/src/imgview.rs b/src/imgview.rs
new file mode 100644
index 0000000..c52abee
--- /dev/null
+++ b/src/imgview.rs
@@ -0,0 +1,86 @@
+use crate::widget::{Widget, WidgetCore};
+use crate::fail::HResult;
+
+use std::path::{Path, PathBuf};
+
+impl std::cmp::PartialEq for ImgView {
+    fn eq(&self, other: &Self) -> bool {
+        self.core == other.core &&
+            self.buffer == other.buffer
+    }
+}
+
+pub struct ImgView {
+    pub core: WidgetCore,
+    pub buffer: Vec<String>,
+    pub file: PathBuf
+}
+
+impl ImgView {
+    pub fn new_from_file(core: WidgetCore, file: &Path) -> HResult<ImgView> {
+        let (xsize, ysize) = core.coordinates.size_u();
+
+        let output = std::process::Command::new("preview-gen")
+            .arg(format!("{}", (xsize)))
+            .arg(format!("{}", (ysize+1)))
+            .arg("image")
+            .arg(format!("true"))
+            .arg(format!("true"))
+            .arg(file.to_string_lossy().to_string())
+            .output()?
+            .stdout;
+
+        let output = std::str::from_utf8(&output)?;
+        let output = output.lines()
+            .map(|l| l.to_string())
+            .collect();
+
+        Ok(ImgView {
+            core: core,
+            buffer: output,
+            file: file.to_path_buf()
+        })
+    }
+
+    pub fn set_image_data(&mut self, img_data: Vec<String>) {
+        self.buffer = img_data;
+    }
+
+    pub fn lines(&self) -> usize {
+        self.buffer.len()
+    }
+}
+
+
+impl Widget for ImgView {
+    fn get_core(&self) -> HResult<&WidgetCore> {
+        Ok(&self.core)
+    }
+
+    fn get_core_mut(&mut self) -> HResult<&mut WidgetCore> {
+        Ok(&mut self.core)
+    }
+
+    fn refresh(&mut self) -> HResult<()> {
+
+        Ok(())
+    }
+
+    fn get_drawlist(&self) -> HResult<String> {
+        let (xpos, ypos) = self.core.coordinates.position_u();
+
+        let mut draw = self.buffer
+            .iter()
+            .enumerate()
+            .fold(String::new(), |mut draw, (pos, line)| {
+                draw += &format!("{}", crate::term::goto_xy_u(xpos+1,
+                                                              ypos + pos));
+                draw += line;
+                draw
+            });
+
+        draw += &format!("{}", termion::style::Reset);
+
+        Ok(draw)
+    }
+}