[Solved] iOS Swift, how to do if statement for different file extension? [closed]

if (fileURL.pathExtension == "png") {
    // image
} else if (fileURL.pathExtension == "mp4") {
    // video
}