package main import ( "bytes" "fmt" "html/template" "io" "io/ioutil" "log" "net/http" "os" "path" "regexp" "strconv" "strings" "time" "github.com/boltdb/bolt" "github.com/gin-gonic/gin" "github.com/microcosm-cc/bluemonday" "github.com/russross/blackfriday" ) func putFile(c *gin.Context) { filename := c.Param("title") contentLength := c.Request.ContentLength var reader io.Reader reader = c.Request.Body if contentLength == -1 { // queue file to disk, because s3 needs content length var err error var f io.Reader f = reader var b bytes.Buffer n, err := io.CopyN(&b, f, _24K+1) if err != nil && err != io.EOF { log.Printf("%s", err.Error()) } if n > _24K { file, err := ioutil.TempFile("./", "transfer-") if err != nil { log.Printf("%s", err.Error()) } defer file.Close() n, err = io.Copy(file, io.MultiReader(&b, f)) if err != nil { os.Remove(file.Name()) log.Printf("%s", err.Error()) } reader, err = os.Open(file.Name()) } else { reader = bytes.NewReader(b.Bytes()) } contentLength = n } buf := new(bytes.Buffer) buf.ReadFrom(reader) // p := WikiData{filename, "", []string{}, []string{}, false, ""} // p.save(buf.String()) var p WikiData p.load(strings.ToLower(filename)) p.save(buf.String()) fmt.Println(c.ClientIP()) c.Data(200, "text/plain", []byte("File uploaded to http://"+RuntimeArgs.ExternalIP+"/"+filename)) } type EncryptionPost struct { Text string `form:"text" json:"text" binding:"required"` Password string `form:"password" json:"password" binding:"required"` } func encryptionRoute(c *gin.Context) { title := c.Param("title") option := c.Param("option") fmt.Println(option, title) var jsonLoad EncryptionPost if option == "/decrypt" { if c.BindJSON(&jsonLoad) == nil { var err error currentText, _, _, _, encrypted, _ := getCurrentText(title, -1) if encrypted == true { currentText, err = decryptString(currentText, jsonLoad.Password) if err != nil { c.JSON(200, gin.H{ "status": "Inorrect passphrase.", "title": title, "option": option, "success": false, }) } else { p := WikiData{strings.ToLower(title), "", []string{}, []string{}, false, ""} p.save(currentText) c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": true, }) } } } else { c.JSON(200, gin.H{ "status": "Could not bind", "title": title, "option": option, "success": false, }) } } if option == "/encrypt" { if c.BindJSON(&jsonLoad) == nil { p := WikiData{strings.ToLower(title), "", []string{}, []string{}, true, ""} p.save(encryptString(jsonLoad.Text, jsonLoad.Password)) c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": true, }) } else { c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": false, }) } } if option == "/lock" { if c.BindJSON(&jsonLoad) == nil { var p WikiData err := p.load(strings.ToLower(title)) if err != nil { panic(err) } p.Locked = jsonLoad.Password p.save(p.CurrentText) c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": true, }) } else { c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": false, }) } } if option == "/unlock" { if c.BindJSON(&jsonLoad) == nil { var p WikiData err := p.load(strings.ToLower(title)) if err != nil { panic(err) } if len(p.Locked) > 0 && p.Locked == jsonLoad.Password { p.Locked = "" p.save(p.CurrentText) c.JSON(200, gin.H{ "status": "Unlocked!", "title": title, "option": option, "success": true, }) } else { c.JSON(200, gin.H{ "status": "Incorrect password!", "title": title, "option": option, "success": false, }) } } else { c.JSON(200, gin.H{ "status": "posted", "title": title, "option": option, "success": false, }) } } } func newNote(c *gin.Context) { title := randomAlliterateCombo() c.Redirect(302, "/"+title) } func editNote(c *gin.Context) { title := c.Param("title") if title == "ws" { wshandler(c.Writer, c.Request) } else if title == "robots.txt" { robotsTxtFile, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "static/text/robots.txt")) c.Data(200, "text/plain", robotsTxtFile) } else if title == "sitemap.xml" { robotsTxtFile, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "static/text/sitemap.xml")) c.Data(200, "text/plain", robotsTxtFile) } else if strings.ToLower(title) == "help" { //}&& strings.Contains(AllowedIPs, c.ClientIP()) != true { c.Redirect(302, "/Help/view") } else { version := c.DefaultQuery("version", "-1") versionNum, _ := strconv.Atoi(version) currentText, versions, currentVersion, totalTime, encrypted, locked := getCurrentText(title, versionNum) if strings.Contains(c.Request.Header.Get("User-Agent"), "curl/") { c.Data(200, "text/plain", []byte(currentText)) return } if encrypted || len(locked) > 0 { c.Redirect(302, "/"+title+"/view") return } if strings.Contains(currentText, "self-destruct\n") || strings.Contains(currentText, "\nself-destruct") { c.Redirect(302, "/"+title+"/view") return } numRows := len(strings.Split(currentText, "\n")) + 10 totalTimeString := totalTime.String() if totalTime.Seconds() < 1 { totalTimeString = "< 1 s" } c.HTML(http.StatusOK, "index.tmpl", gin.H{ "Title": title, "WikiName": RuntimeArgs.WikiName, "ExternalIP": RuntimeArgs.ExternalIP, "CurrentText": currentText, "NumRows": numRows, "Versions": versions, "TotalTime": totalTimeString, "SocketType": RuntimeArgs.Socket, "NoEdit": !currentVersion, }) } } func everythingElse(c *gin.Context) { option := c.Param("option") 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) == "help" { versionNum = -1 } currentText, versions, _, totalTime, encrypted, locked := getCurrentText(title, versionNum) if (strings.Contains(currentText, "self-destruct\n") || strings.Contains(currentText, "\nself-destruct")) && strings.ToLower(title) != "help" { currentText = strings.Replace(currentText, "self-destruct\n", `> *This page has been deleted, you cannot return after closing.*`+"\n", 1) currentText = strings.Replace(currentText, "\nself-destruct", "\n"+`> *This page has been deleted, you cannot return after closing.*`, 1) p := WikiData{strings.ToLower(title), "", []string{}, []string{}, false, ""} p.save("") } renderMarkdown(c, currentText, title, versions, "", totalTime, encrypted, noprompt == "-1", len(locked) > 0) } else if title == "ls" && option == "/"+RuntimeArgs.AdminKey && len(RuntimeArgs.AdminKey) > 1 { renderMarkdown(c, listEverything(), "ls", nil, RuntimeArgs.AdminKey, time.Now().Sub(time.Now()), false, false, false) } else if option == "/list" { renderList(c, title) } else if title == "static" { serveStaticFile(c, option) } else { c.Redirect(302, "/"+title) } } func serveStaticFile(c *gin.Context, option string) { staticFile, err := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "static") + option) if err != nil { c.AbortWithStatus(404) } else { c.Data(200, contentType(option), []byte(staticFile)) } } func renderMarkdown(c *gin.Context, currentText string, title string, versions []versionsInfo, AdminKey string, totalTime time.Duration, encrypted bool, noprompt bool, locked 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) } unsafe := blackfriday.MarkdownCommon([]byte(currentText)) pClean := bluemonday.UGCPolicy() pClean.AllowElements("img") pClean.AllowAttrs("alt").OnElements("img") pClean.AllowAttrs("src").OnElements("img") pClean.AllowAttrs("class").OnElements("a") pClean.AllowAttrs("href").OnElements("a") pClean.AllowAttrs("id").OnElements("a") pClean.AllowDataURIImages() html := pClean.SanitizeBytes(unsafe) html2 := string(html) r, _ = regexp.Compile("\\$\\$(.*?)\\$\\$") for _, s := range r.FindAllString(html2, -1) { html2 = strings.Replace(html2, s, "", 1) } r, _ = regexp.Compile("\\$(.*?)\\$") for _, s := range r.FindAllString(html2, -1) { html2 = strings.Replace(html2, s, "", 1) } html2 = strings.Replace(html2, "&#36;", "$", -1) html2 = strings.Replace(html2, "&#91;", "[", -1) html2 = strings.Replace(html2, "&#93;", "]", -1) html2 = strings.Replace(html2, "&35;", "#", -1) totalTimeString := totalTime.String() if totalTime.Seconds() < 1 { totalTimeString = "< 1 s" } c.HTML(http.StatusOK, "view.tmpl", gin.H{ "Title": title, "WikiName": RuntimeArgs.WikiName, "Body": template.HTML([]byte(html2)), "Versions": versions, "TotalTime": totalTimeString, "AdminKey": AdminKey, "Encrypted": encrypted, "Locked": locked, "Prompt": noprompt, "LockedOrEncrypted": locked || encrypted, }) } func reorderList(text string) ([]template.HTML, []string) { listItemsString := "" for _, lineString := range strings.Split(text, "\n") { if len(lineString) > 1 { if string(lineString[0]) != "-" { listItemsString += "- " + lineString + "\n" } else { listItemsString += lineString + "\n" } } } // get ordering of template.HTML for rendering renderedListString := string(blackfriday.MarkdownCommon([]byte(listItemsString))) listItems := []template.HTML{} endItems := []template.HTML{} for _, lineString := range strings.Split(renderedListString, "\n") { if len(lineString) > 1 { if strings.Contains(lineString, "") || strings.Contains(lineString, "") { endItems = append(endItems, template.HTML(lineString)) } else { listItems = append(listItems, template.HTML(lineString)) } } } // get ordering of strings for deleting listItemsStringArray := []string{} endItemsStringArray := []string{} for _, lineString := range strings.Split(listItemsString, "\n") { if len(lineString) > 1 { if strings.Contains(lineString, "~~") { endItemsStringArray = append(endItemsStringArray, lineString) } else { listItemsStringArray = append(listItemsStringArray, lineString) } } } return append(listItems, endItems...), append(listItemsStringArray, endItemsStringArray...) } func renderList(c *gin.Context, title string) { if strings.ToLower(title) == "help" { //}&& strings.Contains(AllowedIPs, c.ClientIP()) != true { c.Redirect(302, "/Help/view") } var p WikiData err := p.load(strings.ToLower(title)) if err != nil { panic(err) } currentText := p.CurrentText if strings.Contains(currentText, "self-destruct\n") || strings.Contains(currentText, "\nself-destruct") { c.Redirect(302, "/"+title+"/view") } if p.Encrypted || len(p.Locked) > 0 { c.Redirect(302, "/"+title+"/view") } pClean := bluemonday.UGCPolicy() pClean.AllowElements("img") pClean.AllowAttrs("alt").OnElements("img") pClean.AllowAttrs("src").OnElements("img") pClean.AllowAttrs("class").OnElements("a") pClean.AllowAttrs("href").OnElements("a") pClean.AllowAttrs("id").OnElements("a") pClean.AllowDataURIImages() text := pClean.SanitizeBytes([]byte(p.CurrentText)) listItems, _ := reorderList(string(text)) fmt.Println(string(text)) c.HTML(http.StatusOK, "list.tmpl", gin.H{ "Title": title, "WikiName": RuntimeArgs.WikiName, "ListItems": listItems, }) } func deleteListItem(c *gin.Context) { lineNum, err := strconv.Atoi(c.DefaultQuery("lineNum", "None")) title := c.Query("title") // shortcut for c.Request.URL.Query().Get("lastname") if err == nil { var p WikiData err := p.load(strings.ToLower(title)) if err != nil { panic(err) } _, listItems := reorderList(p.CurrentText) newText := p.CurrentText for i, lineString := range listItems { // fmt.Println(i, lineString, lineNum) if i+1 == lineNum { // fmt.Println("MATCHED") if strings.Contains(lineString, "~~") == false { // fmt.Println(p.Text, "("+lineString[2:]+"\n"+")", "~~"+lineString[2:]+"~~"+"\n") newText = strings.Replace(newText+"\n", lineString[2:]+"\n", "~~"+strings.TrimSpace(lineString[2:])+"~~"+"\n", 1) } else { newText = strings.Replace(newText+"\n", lineString[2:]+"\n", lineString[4:len(lineString)-2]+"\n", 1) } p.save(newText) break } } c.JSON(200, gin.H{ "message": "Done.", }) } else { c.JSON(404, gin.H{ "message": "?", }) } } func deletePage(c *gin.Context) { deleteName := c.DefaultQuery("DeleteName", "None") adminKey := c.DefaultQuery("AdminKey", "None") fmt.Println(adminKey) fmt.Println(deleteName) // if adminKey == RuntimeArgs.AdminKey || true == true { if strings.ToLower(deleteName) != "help" { p := WikiData{strings.ToLower(deleteName), "", []string{}, []string{}, false, ""} p.save("") } // // remove from program data // var deleteKey []byte // foundKey := false // err := db.View(func(tx *bolt.Tx) error { // b := tx.Bucket([]byte("programdata")) // c := b.Cursor() // for k, v := c.First(); k != nil; k, v = c.Next() { // if strings.ToLower(string(v)) == strings.ToLower(deleteName) { // fmt.Println("FOUND " + string(v)) // deleteKey = k // foundKey = true // break // } // } // return nil // }) // if err != nil { // panic(err) // } // if foundKey == true { // fmt.Println(len([]string{})) // fmt.Println(deleteKey) // db.View(func(tx *bolt.Tx) error { // b := tx.Bucket([]byte("programdata")) // err := b.Delete(deleteKey) // return err // }) // } // return OKAY c.JSON(200, gin.H{ "message": "Done.", }) // } else { // c.JSON(404, gin.H{ // "message": "?", // }) // } } func listEverything() string { Open(RuntimeArgs.DatabaseLocation) defer Close() everything := `| Title | Current size | Changes | Total Size | | | --------- |-------------| -----| ------------- | ------------- | ` db.View(func(tx *bolt.Tx) error { // Assume bucket exists and has keys b := tx.Bucket([]byte("datas")) c := b.Cursor() for k, v := c.First(); k != nil; k, v = c.Next() { var p WikiData p.load(string(k)) if len(p.CurrentText) > 1 { contentSize := strconv.Itoa(len(p.CurrentText)) numChanges := strconv.Itoa(len(p.Diffs)) totalSize := strconv.Itoa(len(v)) everything += "| [" + p.Title + "](/" + p.Title + "/view) | " + contentSize + " | " + numChanges + " | " + totalSize + ` | Delete | ` + "\n" } } return nil }) return everything } func dumpEverything() { Open(RuntimeArgs.DatabaseLocation) defer Close() err := os.MkdirAll("dump", 0777) if err != nil { fmt.Println("Already exists") } db.View(func(tx *bolt.Tx) error { // Assume bucket exists and has keys b := tx.Bucket([]byte("datas")) c := b.Cursor() for k, _ := c.First(); k != nil; k, _ = c.Next() { var p WikiData p.load(string(k)) fmt.Println(string(k), len(p.CurrentText)) ioutil.WriteFile(path.Join("dump", string(k)+".md"), []byte(p.CurrentText), 0644) } return nil }) }