99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
package route
|
|
|
|
import (
|
|
"leech/config"
|
|
"leech/html"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func RecursivelyGetSize(completePath string) (int64, error) {
|
|
files, err := os.ReadDir(completePath)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
var sum int64
|
|
for _, f := range files {
|
|
dirEntryInfo, err := f.Info()
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
dirEntrySize := dirEntryInfo.Size()
|
|
if f.IsDir() {
|
|
dirEntrySize, err = RecursivelyGetSize(path.Join(completePath, f.Name()))
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
}
|
|
sum += dirEntrySize
|
|
}
|
|
return sum, nil
|
|
}
|
|
|
|
func HandleList(c *fiber.Ctx) error {
|
|
encodedReq := c.Params("req")
|
|
req, err := url.QueryUnescape(encodedReq)
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).SendString(err.Error())
|
|
}
|
|
c.Set(fiber.HeaderContentType, fiber.MIMETextHTML)
|
|
if req == "" {
|
|
keys := make([]string, 0, len(config.Config.ServeDirs))
|
|
for k := range config.Config.ServeDirs {
|
|
keys = append(keys, k)
|
|
}
|
|
sort.Slice(keys, func(i, j int) bool {
|
|
return i < j
|
|
})
|
|
|
|
entries := make([]html.Entry, 0, len(config.Config.ServeDirs))
|
|
for _, key := range keys {
|
|
entries = append(entries, html.Entry{
|
|
Name: key,
|
|
Size: -1,
|
|
IsDir: true,
|
|
})
|
|
}
|
|
|
|
return c.SendString(html.FileListPage(req, entries))
|
|
} else {
|
|
pathSlice := strings.Split(req, ".")
|
|
pathBase, ok := config.Config.ServeDirs[pathSlice[0]]
|
|
if ok {
|
|
pathSlice[0] = pathBase
|
|
completePath := path.Join(pathSlice...)
|
|
files, err := os.ReadDir(completePath)
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).SendString(err.Error())
|
|
}
|
|
entries := []html.Entry{}
|
|
for _, f := range files {
|
|
dirEntryInfo, err := f.Info()
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).SendString("Internal server error!")
|
|
}
|
|
dirEntrySize := dirEntryInfo.Size()
|
|
if f.IsDir() {
|
|
dirEntrySize, err = RecursivelyGetSize(path.Join(completePath, f.Name()))
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).SendString(err.Error())
|
|
}
|
|
}
|
|
entries = append(entries, html.Entry{
|
|
Name: f.Name(),
|
|
Size: dirEntrySize,
|
|
IsDir: f.IsDir(),
|
|
})
|
|
}
|
|
return c.SendString(html.FileListPage(req, entries))
|
|
} else {
|
|
return c.Status(fiber.StatusNotFound).SendString("Sorry can't find that!")
|
|
}
|
|
}
|
|
}
|