Compare commits

..

3 commits

4 changed files with 65 additions and 29 deletions

2
db.go
View file

@ -18,7 +18,7 @@ func connectDB() *redis.Client {
return db return db
} }
func insert_paste(key string, content string, secret string, ttl time.Duration) { func insertPaste(key string, content string, secret string, ttl time.Duration) {
type dbSchema struct { type dbSchema struct {
content string content string
secret string secret string

15
main.go
View file

@ -2,6 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"io"
"net/http" "net/http"
"strings" "strings"
) )
@ -18,11 +19,23 @@ func handleRequest(w http.ResponseWriter, r *http.Request) {
} else if strings.HasPrefix(path, "/static/") { } else if strings.HasPrefix(path, "/static/") {
fs := http.FileServer(http.Dir("./static")) fs := http.FileServer(http.Dir("./static"))
http.Handle("/static/", http.StripPrefix("/static/", fs)) http.Handle("/static/", http.StripPrefix("/static/", fs))
} else {
if urlExist(path) {
io.WriteString(w, "exist")
} else { } else {
w.WriteHeader(http.StatusNotFound) w.WriteHeader(http.StatusNotFound)
} }
}
case "POST": case "POST":
fmt.Println("Post!") if path == "/" {
secret := generateSecret()
url := "/" + generateUrl()
content := r.FormValue("content")
insertPaste(url, content, secret, -1)
http.Redirect(w, r, url, http.StatusSeeOther)
} else {
w.WriteHeader(http.StatusMethodNotAllowed)
}
} }
} }

View file

@ -9,6 +9,7 @@
<link href="/static/style.css" rel="stylesheet"> <link href="/static/style.css" rel="stylesheet">
</head> </head>
<body> <body>
<form>
<label for="content"></label> <label for="content"></label>
<textarea id="content" name="content" placeholder="Your paste here"></textarea> <textarea id="content" name="content" placeholder="Your paste here"></textarea>
<nav> <nav>
@ -34,5 +35,6 @@
</svg> </svg>
</button> </button>
</nav> </nav>
</form>
</body> </body>
</html> </html>

View file

@ -1,6 +1,8 @@
package main package main
import "math/rand" import (
"math/rand"
)
func generateUrl() string { func generateUrl() string {
length := currentConfig.urlLength length := currentConfig.urlLength
@ -12,3 +14,22 @@ func generateUrl() string {
return string(b) return string(b)
} }
func generateSecret() string {
length := 32
listChars := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
b := make([]rune, length)
for i := range b {
b[i] = listChars[rand.Intn(len(listChars))]
}
return string(b)
}
func urlExist(url string) bool {
exist := connectDB().Exists(ctx, url).Val()
if exist == 1 {
return true
}
return false
}