From c138e9ad3e365b227d6dbe8618fb93f195d95121 Mon Sep 17 00:00:00 2001
From: Gunwant Jain <mail@wantguns.dev>
Date: Tue, 29 Dec 2020 21:37:54 +0530
Subject: [PATCH] fix MIME types and add examples

cpp files were getting recognised as "text/csrc". checking subtrings is
a better way to compare MIME types.

Signed-off-by: Gunwant Jain <mail@wantguns.dev>
---
 src/main.rs                | 51 +++++++++++++++++++++++++++-----------
 templates/pretty.html.tera |  2 +-
 2 files changed, 37 insertions(+), 16 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 128d6e4..1d34221 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -31,9 +31,9 @@ fn pretty_retrieve(id: PasteId) -> Option<Template> {
     map.insert("code", contents);
     let rendered = Template::render("pretty", &map);
 
-    match tree_magic::from_filepath(filepath).as_str() {
-        "text/plain" => Some(rendered),
-            _   =>    Err("does not have the MIME type of a plaintext file").ok()
+    match tree_magic::from_filepath(filepath).contains("text") {
+        true    =>  Some(rendered),
+        false   =>  None
     }
 }
 
@@ -53,9 +53,9 @@ fn upload(paste: Data) -> Result<String, std::io::Error> {
 
     paste.stream_to_file(filepath)?;
 
-    let url = match tree_magic::from_filepath(filepath).as_str() {
-        "text/plain" => format!("{host}/p/{id}\n", host = "http://localhost:8000", id = id),
-            _ => format!("{host}/{id}\n", host = "http://localhost:8000", id = id),
+    let url = match tree_magic::from_filepath(filepath).as_str().contains("text") {
+        true    => format!("{host}/p/{id}\n", host = "http://localhost:8000", id = id),
+        false   => format!("{host}/{id}\n", host = "http://localhost:8000", id = id),
     };
 
     Ok(url)
@@ -64,21 +64,42 @@ fn upload(paste: Data) -> Result<String, std::io::Error> {
 #[get("/")]
 fn index() -> &'static str {
     "\
-USAGE
-=====
+    USAGE
+    -----
 
-    POST    / 
+        POST    / 
 
-        accepts raw data in the body of the request and responds with a URL
-        of a page containing the body's content
+            accepts raw data in the body of the request and responds with a URL
+            of a page containing the body's content
 
-    GET     /<id>
+        GET     /<id>
 
-        retrieves the content for the paste with id `<id>`
+            retrieves the content for the paste with id `<id>`
 
-    GET     /p/<id>
+        GET     /p/<id>
 
-        retrieves the content for the paste with id `<id>`, with syntax highlighting
+            retrieves the HTML page with syntax-highlighted content for the paste with id `<id>`
+ 
+    EXAMPLES
+    --------
+
+        Paste a file named 'file.txt' using cURL:
+
+            curl --data-binary @file.txt https://bin.example.com
+
+        Paste from stdin using cURL:
+
+            echo \"Hello, world.\" | curl --data-binary @- https://bin.example.com
+
+        Add this to your .zshrc to implement a quicker usage.
+
+            function paste() {
+              local file=${1:-/dev/stdin}
+              curl --data-binary @${file} https://paste.rs
+            }
+
+        If the uploaded data binary is parsed as \"text/*\", then the paste will be syntax
+        highlighted
     "
 }
 
diff --git a/templates/pretty.html.tera b/templates/pretty.html.tera
index 38865b6..1d030aa 100644
--- a/templates/pretty.html.tera
+++ b/templates/pretty.html.tera
@@ -15,7 +15,7 @@
             code {
                 background: #fff;
                 position: fixed;
-                width:100%;
+                width: 100%;
                 bottom: 0;
                 right: 0;
                 left: 0;