135 lines
2.8 KiB
Go
135 lines
2.8 KiB
Go
package plak
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.gnous.eu/gnouseu/plakken/internal/database"
|
|
"git.gnous.eu/gnouseu/plakken/internal/utils"
|
|
"github.com/redis/go-redis/v9"
|
|
|
|
"html/template"
|
|
)
|
|
|
|
var ctx = context.Background()
|
|
|
|
type WebConfig struct {
|
|
DB *redis.Client
|
|
UrlLength uint8
|
|
}
|
|
|
|
// Plak "Object" for plak
|
|
type Plak struct {
|
|
Key string
|
|
Content string
|
|
Expiration time.Duration
|
|
DB *redis.Client
|
|
}
|
|
|
|
// Create manage POST request for create Plak
|
|
func (config WebConfig) Create(w http.ResponseWriter, r *http.Request) {
|
|
content := r.FormValue("content")
|
|
if content == "" {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
}
|
|
|
|
dbConf := database.DBConfig{
|
|
DB: config.DB,
|
|
}
|
|
|
|
secret := utils.GenerateSecret()
|
|
key := utils.GenerateUrl(config.UrlLength)
|
|
rawExpiration := r.FormValue("exp")
|
|
expiration, err := utils.ParseExpiration(rawExpiration)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
} else if expiration == 0 {
|
|
dbConf.InsertPaste(key, content, secret, -1)
|
|
} else {
|
|
dbConf.InsertPaste(key, content, secret, time.Duration(expiration*int(time.Second)))
|
|
}
|
|
|
|
http.Redirect(w, r, key, http.StatusSeeOther)
|
|
}
|
|
|
|
// View for plak
|
|
func (config WebConfig) View(w http.ResponseWriter, r *http.Request) {
|
|
dbConf := database.DBConfig{
|
|
DB: config.DB,
|
|
}
|
|
var plak Plak
|
|
key := r.PathValue("key")
|
|
|
|
if dbConf.UrlExist(key) {
|
|
plak = Plak{
|
|
Key: key,
|
|
DB: config.DB,
|
|
}
|
|
plak = plak.GetContent()
|
|
if r.PathValue("settings") == "raw" {
|
|
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
|
_, err := io.WriteString(w, plak.Content)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
} else {
|
|
t, err := template.ParseFiles("templates/paste.html")
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
err = t.Execute(w, plak)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
}
|
|
|
|
// Delete manage plak deletion endpoint
|
|
func (config WebConfig) Delete(w http.ResponseWriter, r *http.Request) {
|
|
dbConf := database.DBConfig{
|
|
DB: config.DB,
|
|
}
|
|
key := r.PathValue("key")
|
|
|
|
if dbConf.UrlExist(key) {
|
|
secret := r.URL.Query().Get("secret")
|
|
if dbConf.VerifySecret(key, secret) {
|
|
plak := Plak{
|
|
Key: key,
|
|
DB: config.DB,
|
|
}
|
|
err := plak.deletePlak()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
w.WriteHeader(http.StatusNoContent)
|
|
} else {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
}
|
|
}
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
|
|
// deletePlak Delete plak from database
|
|
func (plak Plak) deletePlak() error {
|
|
err := plak.DB.Del(ctx, plak.Key).Err()
|
|
if err != nil {
|
|
log.Println(err)
|
|
return &DeletePlakError{Name: plak.Key, Err: err}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetContent get plak content
|
|
func (plak Plak) GetContent() Plak {
|
|
plak.Content = plak.DB.HGet(ctx, plak.Key, "content").Val()
|
|
return plak
|
|
}
|