diff --git a/plugins/af_youtube_thumbnail/init.php b/plugins/af_youtube_thumbnail/init.php
index 650f1ad..ac516b0 100644
--- a/plugins/af_youtube_thumbnail/init.php
+++ b/plugins/af_youtube_thumbnail/init.php
@@ -1,23 +1,33 @@
add_hook($host::HOOK_ARTICLE_FILTER, $this);
- }
-
- function hook_article_filter($article) {
- if(strpos($article["link"], "youtube.com") !== FALSE) {
- $video_id = str_replace('https://www.youtube.com/watch?v=', '', $article["link"]);
- $article["content"] = '
';
+class Af_Youtube_Thumbnail extends Plugin
+{
+ function about()
+ {
+ return array(0.1,
+ "Add the thumbnail for Youtube videos",
+ "gdott9");
}
- return $article;
- }
+ function init($host)
+ {
+ $host->add_hook($host::HOOK_ARTICLE_FILTER, $this);
+ }
- function api_version() { return 2; }
+ function hook_article_filter($article)
+ {
+ if (strpos($article["link"], "youtube.com") !== false) {
+ $video_url_prefix = 'https://www.youtube.com/watch?v=';
+ $short_url_prefix = 'https://www.youtube.com/shorts/';
+ $is_short = substr($article["link"], 0, strlen($short_url_prefix)) === $short_url_prefix;
+ $video_id = str_replace($is_short ? $short_url_prefix : $video_url_prefix, '', $article["link"]);
+ $article["content"] = '
';
+ }
+
+ return $article;
+ }
+
+ function api_version()
+ {
+ return 2;
+ }
}