From 8e15d802865ed57db0018c15ea5559c8bd44c01f Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Wed, 7 Sep 2022 02:48:46 +0800
Subject: First working version. Just a POC.

---
 common/urlparse.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 common/urlparse.go

(limited to 'common/urlparse.go')

diff --git a/common/urlparse.go b/common/urlparse.go
new file mode 100644
index 0000000..dc72cee
--- /dev/null
+++ b/common/urlparse.go
@@ -0,0 +1,21 @@
+package common
+
+import (
+	"errors"
+	"net/url"
+	"strings"
+)
+
+// GetFileExtensionFromUrl
+// copied from https://elisegev.medium.com/get-a-file-extension-from-a-url-in-golang-5061d4a298a
+func GetFileExtensionFromUrl(rawUrl string) (string, error) {
+	u, err := url.Parse(rawUrl)
+	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