First commit

8 years ago · 4e7d63be3f
commit 4e7d63be3f
1 changed files with 100 additions and 0 deletions
--- a/main.go
+++ b/main.go
@ -0,0 +1,100 @@
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"crypto/sha256"
+	"log"
+	"path/filepath"
+	"os"
+	"time"
+	"io/ioutil"
+)
+
+
+func handle_get_paste(w http.ResponseWriter, r *http.Request){
+	
+	var paste_name, orig_name string
+	var err error
+
+	orig_name = filepath.Clean(r.URL.Path)
+	paste_name = "./pastes/" + orig_name
+
+	fmt.Printf("orig_name: '%s'\npaste_name: '%s'\n", orig_name, paste_name)
+
+	// The default is to serve index.html
+	if (orig_name == "/" ) || ( orig_name == "/index.html" ) {
+		http.ServeFile(w, r, "index.html")
+	} else {
+		// otherwise, if the requested paste exists, we serve it...
+		if _, err = os.Stat(paste_name); err == nil && orig_name != "./" {
+			http.ServeFile(w, r, paste_name)
+		}	else {
+			// otherwise, we give say we didn't find it 
+			fmt.Fprintf(w, "Paste '%s' not found\n", orig_name)
+			return
+		}
+	}
+}
+
+
+
+func handle_put_paste(w http.ResponseWriter, r *http.Request){
+
+
+	fmt.Printf("We are inside handle_put_paste\n");
+	
+	if err := r.ParseForm(); err != nil{
+		// Invalid POST -- let's serve the default file
+		http.ServeFile(w, r, "index.html")
+	} else {
+		h := sha256.New()
+		req_body := r.PostForm
+		// get title, body, and time
+		title := req_body.Get("title")
+		paste := req_body.Get("paste")
+		now := time.Now().String()
+		// format content 
+		content := fmt.Sprintf("# Title: %s\n# Pasted: %s\n------------\n%s", title, now, paste)
+		
+		// ccompute the sha256 hash using title, body, and time
+		h.Write([]byte (content))
+
+		paste_hash := fmt.Sprintf("%x", h.Sum(nil))
+		fmt.Printf("hash: %s fname: %s\n", paste_hash, paste_hash[:16])
+		paste_dir := "./pastes/"
+		
+		// Now we save the file
+		for i := 0; i < len(paste_hash) - 16; i ++ {
+
+			if _, err := os.Stat(paste_dir + paste_hash[i:i+16]); os.IsNotExist(err) {
+				// The file does not exist, so we can create it
+				if err := ioutil.WriteFile(paste_dir + paste_hash[i:i+16], []byte (content), 0644); err == nil{
+					// and then we return the URL:
+					fmt.Fprintf(w, "<html><body>Link: <a href='%s'>%s</a></body></html>",
+						paste_hash[i:i+16], paste_hash[i:i+16])
+					return
+				} else {
+					fmt.Fprintf(w, "Cannot create the paste!!!\n")
+				}
+			}
+		}
+	}
+}
+
+func req_handler(w http.ResponseWriter, r *http.Request){
+
+	switch r.Method {
+	case "GET":
+		handle_get_paste(w, r)
+	case "POST":
+		handle_put_paste(w, r)
+	default:
+		http.NotFound(w, r)
+	}
+}
+
+func main(){
+	http.HandleFunc("/", req_handler)
+	log.Fatal(http.ListenAndServe("localhost:8000", nil))
+}