From a7589f140887e35c573a1caf51f6610ba6047a4d Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Mon, 14 Mar 2016 14:31:28 -0400
Subject: [PATCH] Added noprompting for encryption

---
 routes.go            | 46 +++++++++++++++-----------------------------
 templates/index.tmpl |  2 +-
 templates/view.tmpl  |  6 ++++--
 3 files changed, 21 insertions(+), 33 deletions(-)

diff --git a/routes.go b/routes.go
index f1194ee..816979d 100644
--- a/routes.go
+++ b/routes.go
@@ -29,11 +29,9 @@ func encryptionRoute(c *gin.Context) {
 	fmt.Println(option, title)
 	var jsonLoad EncryptionPost
 	if option == "/decrypt" {
-		fmt.Println("Decrypting...")
 		if c.BindJSON(&jsonLoad) == nil {
 			var err error
 			currentText, _, _, _, encrypted := getCurrentText(title, -1)
-			fmt.Println(currentText, encrypted, jsonLoad.Password)
 			if encrypted == true {
 				currentText, err = decryptString(currentText, jsonLoad.Password)
 				if err != nil {
@@ -65,7 +63,6 @@ func encryptionRoute(c *gin.Context) {
 	}
 	if option == "/encrypt" {
 		if c.BindJSON(&jsonLoad) == nil {
-			fmt.Println(jsonLoad)
 			p := WikiData{strings.ToLower(title), "", []string{}, []string{}, true}
 			p.save(encryptString(jsonLoad.Text, jsonLoad.Password))
 			c.JSON(200, gin.H{
@@ -114,30 +111,17 @@ func editNote(c *gin.Context) {
 				c.Redirect(302, "/"+title+"/view")
 			}
 			numRows := len(strings.Split(currentText, "\n")) + 10
-			if currentVersion {
-				c.HTML(http.StatusOK, "index.tmpl", gin.H{
-					"Title":       title,
-					"WikiName":    RuntimeArgs.WikiName,
-					"ExternalIP":  RuntimeArgs.ExternalIP,
-					"CurrentText": currentText,
-					"NumRows":     numRows,
-					"Versions":    versions,
-					"TotalTime":   totalTime,
-					"SocketType":  RuntimeArgs.Socket,
-				})
-			} else {
-				c.HTML(http.StatusOK, "index.tmpl", gin.H{
-					"Title":       title,
-					"WikiName":    RuntimeArgs.WikiName,
-					"ExternalIP":  RuntimeArgs.ExternalIP,
-					"CurrentText": currentText,
-					"NumRows":     numRows,
-					"Versions":    versions,
-					"TotalTime":   totalTime,
-					"SocketType":  RuntimeArgs.Socket,
-					"NoEdit":      true,
-				})
-			}
+			c.HTML(http.StatusOK, "index.tmpl", gin.H{
+				"Title":       title,
+				"WikiName":    RuntimeArgs.WikiName,
+				"ExternalIP":  RuntimeArgs.ExternalIP,
+				"CurrentText": currentText,
+				"NumRows":     numRows,
+				"Versions":    versions,
+				"TotalTime":   totalTime,
+				"SocketType":  RuntimeArgs.Socket,
+				"NoEdit":      !currentVersion,
+			})
 
 		}
 	}
@@ -148,6 +132,7 @@ func everythingElse(c *gin.Context) {
 	title := c.Param("title")
 	if option == "/view" {
 		version := c.DefaultQuery("version", "-1")
+		noprompt := c.DefaultQuery("noprompt", "-1")
 		versionNum, _ := strconv.Atoi(version)
 		if strings.ToLower(title) == "about" {
 			versionNum = -1
@@ -159,9 +144,9 @@ func everythingElse(c *gin.Context) {
 			p := WikiData{strings.ToLower(title), "", []string{}, []string{}, false}
 			p.save("")
 		}
-		renderMarkdown(c, currentText, title, versions, "", totalTime, encrypted)
+		renderMarkdown(c, currentText, title, versions, "", totalTime, encrypted, noprompt == "-1")
 	} else if title == "ls" && option == "/"+RuntimeArgs.AdminKey && len(RuntimeArgs.AdminKey) > 1 {
-		renderMarkdown(c, listEverything(), "ls", nil, RuntimeArgs.AdminKey, time.Now().Sub(time.Now()), false)
+		renderMarkdown(c, listEverything(), "ls", nil, RuntimeArgs.AdminKey, time.Now().Sub(time.Now()), false, false)
 	} else if option == "/list" {
 		renderList(c, title)
 	} else if title == "static" {
@@ -180,7 +165,7 @@ func serveStaticFile(c *gin.Context, option string) {
 	}
 }
 
-func renderMarkdown(c *gin.Context, currentText string, title string, versions []versionsInfo, AdminKey string, totalTime time.Duration, encrypted bool) {
+func renderMarkdown(c *gin.Context, currentText string, title string, versions []versionsInfo, AdminKey string, totalTime time.Duration, encrypted bool, noprompt bool) {
 	r, _ := regexp.Compile("\\[\\[(.*?)\\]\\]")
 	for _, s := range r.FindAllString(currentText, -1) {
 		currentText = strings.Replace(currentText, s, "["+s[2:len(s)-2]+"](/"+s[2:len(s)-2]+"/view)", 1)
@@ -218,6 +203,7 @@ func renderMarkdown(c *gin.Context, currentText string, title string, versions [
 		"TotalTime": totalTime.String(),
 		"AdminKey":  AdminKey,
 		"Encrypted": encrypted,
+		"Prompt":    noprompt,
 	})
 
 }
diff --git a/templates/index.tmpl b/templates/index.tmpl
index d9d3384..b3359e0 100644
--- a/templates/index.tmpl
+++ b/templates/index.tmpl
@@ -174,7 +174,7 @@
                   success: function (data) {
                     if (data['success'] == true) {
                       swal("Encryption", "Encrypted!", "success");
-                      window.location.href = '/{{ .Title }}/view';
+                      window.location.href = '/{{ .Title }}/view?noprompt=1';
                     } else {
                       swal("Encryption", "Something went wrong.", "error");
                     }
diff --git a/templates/view.tmpl b/templates/view.tmpl
index 9ec3e7b..2420479 100644
--- a/templates/view.tmpl
+++ b/templates/view.tmpl
@@ -74,9 +74,10 @@ a.deleteable {
 $( document ).ready(function() {
 
   {{ if .Encrypted }}
+  {{ if .Prompt }}
     var pass1 = "";
     swal({
-      title: "Encryption",
+      title: "Decryption",
       text: "Enter your passphrase:",
       type: "input",
       showCancelButton: true,
@@ -109,11 +110,12 @@ $( document ).ready(function() {
           }
       });
     });
+    {{ end }}
     $('.postdecrypt').click(function(event) {
     var pass1 = "";
     event.preventDefault();
     swal({
-      title: "Encryption",
+      title: "Decryption",
       text: "Enter your passphrase:",
       type: "input",
       showCancelButton: true,