diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/format/format.go | 7 | ||||
| -rw-r--r-- | utils/validators/links.go | 12 |
2 files changed, 18 insertions, 1 deletions
diff --git a/utils/format/format.go b/utils/format/format.go index 53c813e..e57bc1f 100644 --- a/utils/format/format.go +++ b/utils/format/format.go @@ -13,7 +13,12 @@ func FileSize(size int64) string { exp++ } - return fmt.Sprintf("%.2f %sB", float64(size)/float64(div), "KMGTPE"[exp:exp+1]) + val := float64(size) / float64(div) + unitStr := "KMGTPE"[exp : exp+1] + if val == float64(int64(val)) { + return fmt.Sprintf("%d %sB", int64(val), unitStr) + } + return fmt.Sprintf("%.2f %sB", val, unitStr) } func Count(count int64) string { diff --git a/utils/validators/links.go b/utils/validators/links.go new file mode 100644 index 0000000..cc9dd9b --- /dev/null +++ b/utils/validators/links.go @@ -0,0 +1,12 @@ +package validators + +import "strings" + +func GetRefererForURL(url string) string { + switch { + case strings.Contains(url, "i.pximg.net") || strings.Contains(url, "pixiv.net"): + return "https://www.pixiv.net/" + default: + return "" + } +} |