From 8f07f6f4d7e91a9e4b7164a39759907fac5fb8a1 Mon Sep 17 00:00:00 2001 From: Keuin Date: Sat, 1 Jul 2023 22:06:01 +0800 Subject: Refactor: encapsulate custom url manipulation function into object method. --- common/myurl/urlparse.go | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 common/myurl/urlparse.go (limited to 'common/myurl/urlparse.go') diff --git a/common/myurl/urlparse.go b/common/myurl/urlparse.go new file mode 100644 index 0000000..69695d1 --- /dev/null +++ b/common/myurl/urlparse.go @@ -0,0 +1,27 @@ +package myurl + +import ( + "errors" + "net/url" + "strings" +) + +type Url string + +func (o Url) Url() string { + return string(o) +} + +// FileExtension returns file extension of file name from this url. +// copied from https://elisegev.medium.com/get-a-file-extension-from-a-url-in-golang-5061d4a298a +func (o Url) FileExtension() (string, error) { + u, err := url.Parse(o.Url()) + if err != nil { + return "", err + } + pos := strings.LastIndex(u.Path, ".") + if pos == -1 { + return "", errors.New("couldn't find a period to indicate a file extension") + } + return u.Path[pos+1 : len(u.Path)], nil +} -- cgit v1.2.3