Refactor: encapsulate custom url manipulation function into object method.

1 files changed, 27 insertions, 0 deletions

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
+}