Merge pull request #23 from krzd/patch-2

Fix #22
gilesvangruisen · Jul 16, 2015 · 4a84bfe · 4a84bfe
2 parents 8782c7f + 13791e6
commit 4a84bfe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/YouTubePlayer/YouTubePlayer/YouTubePlayer.swift b/YouTubePlayer/YouTubePlayer/YouTubePlayer.swift
@@ -62,6 +62,9 @@ private extension NSURL {
 }
 
 public func videoIDFromYouTubeURL(videoURL: NSURL) -> String? {
+    if let host = videoURL.host, pathComponents = videoURL.pathComponents where pathComponents.count > 1 && host.hasSuffix("youtu.be") {
+        return pathComponents[1] as? String
+    }
     return videoURL.queryStringComponents()["v"] as? String
 }