first commit
This commit is contained in:
104
lib/util.go
Normal file
104
lib/util.go
Normal file
@ -0,0 +1,104 @@
|
||||
package lib
|
||||
|
||||
import (
|
||||
"archive/tar"
|
||||
"bytes"
|
||||
"compress/gzip"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
)
|
||||
|
||||
func RenderError(c *fiber.Ctx, code int) error{
|
||||
var msg string = "Server Error"
|
||||
c.Status(code)
|
||||
|
||||
switch code {
|
||||
case 404:
|
||||
msg = "Not Found"
|
||||
}
|
||||
|
||||
return c.Render("error", fiber.Map{
|
||||
"msg": msg,
|
||||
})
|
||||
}
|
||||
|
||||
func SizeFromBytes(size int) string {
|
||||
if(size > 1024*1024*1024) {
|
||||
return fmt.Sprintf("%dGB", (size/1024/1024/1024))
|
||||
}else if(size > 1024*1024) {
|
||||
return fmt.Sprintf("%dMB", (size/1024/1024))
|
||||
}else if(size > 1024) {
|
||||
return fmt.Sprintf("%dKB", (size/1024))
|
||||
}
|
||||
return fmt.Sprintf("%dB", size)
|
||||
}
|
||||
|
||||
func ExtractFile(raw []byte, file string) (string, error){
|
||||
stream := bytes.NewReader(raw)
|
||||
ustream, err := gzip.NewReader(stream)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
reader := tar.NewReader(ustream)
|
||||
var header *tar.Header
|
||||
|
||||
for header, err = reader.Next(); err == nil; header, err = reader.Next() {
|
||||
if header.Typeflag != tar.TypeReg {
|
||||
return "", errors.New("Found invalid entry in archive")
|
||||
}
|
||||
|
||||
if header.Name != file {
|
||||
continue
|
||||
}
|
||||
|
||||
buf := new(strings.Builder)
|
||||
_, err := io.Copy(buf, reader)
|
||||
if err != nil {
|
||||
return "", errors.New("Failed to extract file to memory")
|
||||
}
|
||||
|
||||
return buf.String(), nil
|
||||
}
|
||||
|
||||
return "", errors.New("File not found")
|
||||
}
|
||||
|
||||
func GetTimePassed(t time.Time) string {
|
||||
diff := time.Since(t)
|
||||
res := fmt.Sprintf(
|
||||
"%ds ago",
|
||||
int(diff.Seconds()),
|
||||
)
|
||||
|
||||
if diff.Minutes() > 1 {
|
||||
res = fmt.Sprintf(
|
||||
"%dm and %ds ago",
|
||||
int(diff.Minutes()), int(diff.Seconds())-(int(diff.Minutes())*60),
|
||||
)
|
||||
}
|
||||
|
||||
if diff.Hours() > 1 {
|
||||
res = fmt.Sprintf("%dh and %dm ago",
|
||||
int(diff.Hours()),
|
||||
int(diff.Minutes())-(int(diff.Hours())*60),
|
||||
)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func CleanString(s string) string{
|
||||
var possible_bad_char_list_that_i_made_up_which_may_contain_bad_xss_chars_look_im_not_gonna_search_for_a_xss_charlist_just_to_prevent_self_xss []string = []string{"'", "\"", "/", "<", ">", "?", "=", "#", "(", ")", "{", "}", "*", "!", "`", "[", "]"}
|
||||
|
||||
for _, c := range possible_bad_char_list_that_i_made_up_which_may_contain_bad_xss_chars_look_im_not_gonna_search_for_a_xss_charlist_just_to_prevent_self_xss {
|
||||
s = strings.ReplaceAll(s, c, "")
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
Reference in New Issue
Block a user