how to detect if Firebase url is Photo or Video - Swift - swift

Question

I've been trying to figure how how to detect the url I retrieve from Firebase Storage is a Photo or a Video,
I followed this question but it doesn't seem to work with Firebase Storage URLs.
The solution implemented in that question work only when the url has the extension at the end.
This is the code I implemented:
extension String {
public func isImageType() -> Bool {
// image formats which you want to check
let imageFormats = ["jpg", "png", "gif"]
if URL(string: self) != nil {
let extensi = (self as NSString).pathExtension
return imageFormats.contains(extensi)
}
return false
}
}
To be clear, URLs retrieved from Firebase have this form.
Photo url example:
https://firebasestorage.googleapis.com/v0/b/myApp-1e48d.appspot.com/o/Images%2FD118DA58-C128-4E5E-BF24-AA820BEE5590.jpg?alt=media&token=49eie236-f9d1-45f8-bd0b-887382c61ccd
Video url example:
https://firebasestorage.googleapis.com/v0/b/myApp-1e48d.appspot.com/o/Videos%2F1E4B7CA0-4D0E-4AC5-9856-2F59D0811C47.mp4?alt=media&token=615teacf-0d20-48aa-bb8f-cew84a14d76d
As you can see the extension is not at the end (because there is the need of the token), hence the extension methods linked before doesn't work.
How can I still understand if the url retrieved from Firebase is of a Photo or a Video?

Answer 1

You can simply initialize an url with your link and get the path extension:
let link1 = "https://firebasestorage.googleapis.com/v0/b/myApp-1e48d.appspot.com/o/Images%2FD118DA58-C128-4E5E-BF24-AA820BEE5590.jpg?alt=media&token=49eie236-f9d1-45f8-bd0b-887382c61ccd"
if let url = URL(string: link1) {
let fileType = url.pathExtension // "jpg"
}
let link2 = "https://firebasestorage.googleapis.com/v0/b/myApp-1e48d.appspot.com/o/Videos%2F1E4B7CA0-4D0E-4AC5-9856-2F59D0811C47.mp4?alt=media&token=615teacf-0d20-48aa-bb8f-cew84a14d76d"
if let url = URL(string: link2) {
let fileType = url.pathExtension // "mp4"
}

Answer 2

I implemented a code that works:
First Remove all after "?":
extension String {
public func removeAfterExtension() -> String?{
if let index = (self.range(of: "?")?.lowerBound){
let beforeEqualsTo = String(self.prefix(upTo: index))
return beforeEqualsTo
}else{
return self
}
}
Then check if it is image Type:
public func isImageType() -> Bool {
// image formats which you want to check
let imageFormats = ["jpg", "png", "gif"]
if URL(string: self) != nil {
let extensi = (self as NSString).pathExtension
return imageFormats.contains(extensi)
}
return false
}