138 lines
2.9 KiB
Go
138 lines
2.9 KiB
Go
package filemanager
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/asdine/storm"
|
|
"github.com/asdine/storm/q"
|
|
)
|
|
|
|
type shareLink struct {
|
|
Hash string `json:"hash" storm:"id,index"`
|
|
Path string `json:"path" storm:"index"`
|
|
Expires bool `json:"expires"`
|
|
ExpireDate time.Time `json:"expireDate"`
|
|
}
|
|
|
|
func shareHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
r.URL.Path = sanitizeURL(r.URL.Path)
|
|
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
return shareGetHandler(c, w, r)
|
|
case http.MethodDelete:
|
|
return shareDeleteHandler(c, w, r)
|
|
case http.MethodPost:
|
|
return sharePostHandler(c, w, r)
|
|
}
|
|
|
|
return http.StatusNotImplemented, nil
|
|
}
|
|
|
|
func shareGetHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
var (
|
|
s []*shareLink
|
|
path = filepath.Join(string(c.User.FileSystem), r.URL.Path)
|
|
)
|
|
|
|
err := c.db.Find("Path", path, &s)
|
|
if err == storm.ErrNotFound {
|
|
return http.StatusNotFound, nil
|
|
}
|
|
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
for i, link := range s {
|
|
if link.Expires && link.ExpireDate.Before(time.Now()) {
|
|
c.db.DeleteStruct(&shareLink{Hash: link.Hash})
|
|
s = append(s[:i], s[i+1:]...)
|
|
}
|
|
}
|
|
|
|
return renderJSON(w, s)
|
|
}
|
|
|
|
func sharePostHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
path := filepath.Join(string(c.User.FileSystem), r.URL.Path)
|
|
|
|
var s shareLink
|
|
expire := r.URL.Query().Get("expires")
|
|
unit := r.URL.Query().Get("unit")
|
|
|
|
if expire == "" {
|
|
err := c.db.Select(q.Eq("Path", path), q.Eq("Expires", false)).First(&s)
|
|
if err == nil {
|
|
w.Write([]byte(c.RootURL() + "/share/" + s.Hash))
|
|
return 0, nil
|
|
}
|
|
}
|
|
|
|
bytes, err := generateRandomBytes(32)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
str := hex.EncodeToString(bytes)
|
|
|
|
s = shareLink{
|
|
Path: path,
|
|
Hash: str,
|
|
Expires: expire != "",
|
|
}
|
|
|
|
if expire != "" {
|
|
num, err := strconv.Atoi(expire)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
var add time.Duration
|
|
switch unit {
|
|
case "seconds":
|
|
add = time.Second * time.Duration(num)
|
|
case "minutes":
|
|
add = time.Minute * time.Duration(num)
|
|
case "days":
|
|
add = time.Hour * 24 * time.Duration(num)
|
|
default:
|
|
add = time.Hour * time.Duration(num)
|
|
}
|
|
|
|
s.ExpireDate = time.Now().Add(add)
|
|
}
|
|
|
|
err = c.db.Save(&s)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
return renderJSON(w, s)
|
|
}
|
|
|
|
func shareDeleteHandler(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, error) {
|
|
var s shareLink
|
|
|
|
err := c.db.One("Hash", strings.TrimPrefix(r.URL.Path, "/"), &s)
|
|
if err == storm.ErrNotFound {
|
|
return http.StatusNotFound, nil
|
|
}
|
|
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
err = c.db.DeleteStruct(&s)
|
|
if err != nil {
|
|
return http.StatusInternalServerError, err
|
|
}
|
|
|
|
return http.StatusOK, nil
|
|
}
|