mirror of
https://github.com/diamondburned/cchat-discord.git
synced 2024-12-22 20:36:45 +00:00
58 lines
895 B
Go
58 lines
895 B
Go
package urlutils
|
|
|
|
import (
|
|
"net/url"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// AvatarURL wraps the URL with URL queries for the avatar.
|
|
func AvatarURL(URL string) string {
|
|
return Sized(URL, 64)
|
|
}
|
|
|
|
// Sized wraps the URL with the size query.
|
|
func Sized(URL string, size int) string {
|
|
u, err := url.Parse(URL)
|
|
if err != nil {
|
|
return URL
|
|
}
|
|
|
|
q := u.Query()
|
|
q.Set("size", strconv.Itoa(size))
|
|
u.RawQuery = q.Encode()
|
|
|
|
return u.String()
|
|
}
|
|
|
|
// Ext returns the lowercased file extension of the URL.
|
|
func Ext(URL string) string {
|
|
u, err := url.Parse(URL)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return strings.ToLower(path.Ext(u.Path))
|
|
}
|
|
|
|
func Name(URL string) string {
|
|
u, err := url.Parse(URL)
|
|
if err != nil {
|
|
return URL
|
|
}
|
|
return path.Base(u.Path)
|
|
}
|
|
|
|
func ExtIs(URL string, exts []string) bool {
|
|
var ext = Ext(URL)
|
|
|
|
for _, e := range exts {
|
|
if e == ext {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|