1 files changed, 47 insertions, 86 deletions

diff --git a/src/main.rs b/src/main.rs
index d2caba3..b5665b9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,105 +15,66 @@
 // You should have received a copy of the GNU Affero General Public
 // License along with бирка-тян. If not, see <http://www.gnu.org/licenses/>.
 
+#![feature(proc_macro_hygiene, decl_macro)]
+
 #[macro_use]
 extern crate anyhow;
-extern crate dirs;
+extern crate image;
+#[macro_use]
+extern crate rocket;
+#[macro_use]
+extern crate rocket_contrib;
 
 mod database;
 
-use anyhow::Result;
-use blake2::{Blake2b, Digest};
 use database::TagDatabase;
+use image::{GenericImageView, ImageFormat};
+use rocket_contrib::json::{Json, JsonValue};
 use std::env;
-use std::io;
+use std::fs;
+use std::fs::File;
+use std::os::unix::fs::symlink;
 use std::path::Path;
 
-/// Return a base64-encoded BLAKE2b hash identifying the file at `path`.
-fn hash_file(path: &Path) -> Result<String> {
-    let mut file = std::fs::File::open(path)?;
-    let mut hasher = Blake2b::new();
-    io::copy(&mut file, &mut hasher)?;
-    Ok(base64::encode(&hasher.result()))
+#[get("/")]
+fn index() -> &'static str {
+    "Hello, world!"
 }
 
-fn main() {
-    let args: Vec<String> = env::args().collect();
-    if args.len() < 2 {
-        eprintln!("usage: {} ACTION [ARGS ...]", args[0]);
-        std::process::exit(1);
-    }
+#[get("/posts?<tags>")]
+fn posts(tags: String) -> Json<Vec<String>> {
+    let tags: Vec<String> = tags.split(",").map(|s| s.into()).collect();
+    Json(tags)
+}
 
-    let tb = if let Ok(val) = env::var("TB_PATH") {
-        TagDatabase::new(&val)
-    } else if let Some(dir) = dirs::config_dir() {
-        TagDatabase::new(dir.join("birka.db").to_str().unwrap())
-    } else {
-        TagDatabase::new(Path::new("/tmp").join("pometka.db").to_str().unwrap())
-    }
-    .unwrap();
+fn create_image_directory() {
+    let bin = env::current_exe().unwrap();
+    let image_dir = bin.parent().unwrap().join("image/");
+    fs::create_dir(&image_dir).ok();
 
-    let action = &args[1];
-    match action.to_lowercase().as_str() {
-        "add" => {
-            if args.len() < 3 {
-                eprintln!("usage: {} add PATH [TAGS ...]", args[0]);
-                std::process::exit(1);
-            }
-            let path = Path::new(&args[2]).canonicalize().unwrap();
-            let hash = hash_file(&path).unwrap();
-            let id = tb.import_image(&path, &hash).unwrap();
-            for arg in args[3..].iter() {
-                tb.add_tag(arg).unwrap();
-                tb.tag_image(id, arg).unwrap();
-            }
-        }
-        "id_for" => {
-            if args.len() < 3 {
-                eprintln!("usage: {} id_for PATH ", args[0]);
-                std::process::exit(1);
-            }
-            let path = Path::new(&args[2]).canonicalize().unwrap();
-            println!("{}", tb.image_id(&path).unwrap());
-        }
-        "add_tags" => {
-            if args.len() < 3 {
-                eprintln!("usage: {} add_tags ID [TAGS ...]", args[0]);
-                std::process::exit(1);
-            }
-            let id = args[2].parse::<i64>().unwrap();
-            for arg in args[3..].iter() {
-                tb.add_tag(arg).unwrap();
-                tb.tag_image(id, arg).unwrap();
-            }
-        }
-        "remove_tags" => {
-            if args.len() < 3 {
-                eprintln!("usage: {} remove_tags ID [TAGS ...]", args[0]);
-                std::process::exit(1);
-            }
-            let id = args[2].parse::<i64>().unwrap();
-            for arg in args[3..].iter() {
-                tb.untag_image(id, arg).unwrap();
-            }
-        }
-        "query" => {
-            let tags = args[2..].iter().map(|s| s.as_str()).collect::<Vec<&str>>();
-            for tag in tags.iter() {
-                if tb.tag_id(tag).is_err() {
-                    eprintln!("unknown tag: {}", tag);
-                    std::process::exit(1);
-                }
-            }
-            for image in tb.images_by_tags(&tags[..]).unwrap() {
-                println!(
-                    "{},{},{}{}",
-                    image.id, image.blake2, image.orig_dir, image.filename
-                );
-            }
-        }
-        _ => {
-            eprintln!("Unknown action '{}'", action);
-            std::process::exit(1);
+    let tb_path = bin.parent().unwrap().join("birka.db");
+    let tb = TagDatabase::new(tb_path.to_str().unwrap()).unwrap();
+
+    for image in tb.images_by_tags(&vec![][..]).unwrap() {
+        let src = format!("{}/{}", image.orig_dir, image.filename);
+        let dst = format!("{}/{}", image_dir.to_str().unwrap(), image.filename);
+        symlink(&src, &dst).ok();
+
+        let stem = Path::new(&dst).file_stem().unwrap().to_str().unwrap();
+        let thumb_path = format!("{}_thumb.png", stem);
+        let thumb_path = Path::new(&thumb_path);
+
+        if !thumb_path.exists() {
+            let im = image::open(&Path::new(&src)).unwrap();
+            let fout = &mut File::create(&thumb_path).unwrap();
+            im.thumbnail(100, 100)
+                .write_to(fout, ImageFormat::Png)
+                .unwrap();
         }
     }
 }
+
+fn main() {
+    create_image_directory();
+    rocket::ignite().mount("/", routes![index, posts]).launch();
+}