Add option to show either magnet or torrent links in feed, closes #13

PeterTheOne · Aug 19, 2015 · bbd047f · bbd047f
1 parent 317fadc
commit bbd047f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 9 deletions.
diff --git a/config-sample.php b/config-sample.php
@@ -29,6 +29,8 @@
 define('MINIMUM_FILESIZE', 3); /* in GB */
 define('MAXIMUM_FILESIZE', 19); /* in GB */
 
+define('FEED_DOWNLOAD_LINK', 'both'); /* torrent, magnet, both */
+
 $sites = array( /* prioritize sites to search */
     'extratorrent',
     'kickasstorrents'

diff --git a/index.php b/index.php
@@ -58,19 +58,29 @@
         $item->appendChild( $torrentContentLength );
     }
 
-    if ($film->torrentFile) {
-        $enclosure = $xml->createElement( "enclosure" );
-        $enclosure->setAttribute( "url", $film->torrentFile );
-        if ($film->torrentSize) $enclosure->setAttribute( "length", $film->torrentSize );
-        $enclosure->setAttribute( "type", "application/x-bittorrent" );
-        $item->appendChild( $enclosure );
+    if ((!$film->torrentFile && !$film->torrentMagnet) ||
+        (FEED_DOWNLOAD_LINK === 'torrent' && !$film->torrentFile) ||
+        (FEED_DOWNLOAD_LINK === 'magnet' && !$film->torrentMagnet) ) {
+        continue;
+    }
+
+    $enclosure = $xml->createElement( "enclosure" );
+    if ($film->torrentSize) $enclosure->setAttribute( "length", $film->torrentSize );
+    $enclosure->setAttribute( "type", "application/x-bittorrent" );
+    $link = $xml->createElement( "link" );
 
-        $link = $xml->createElement( "link" );
+    if ((FEED_DOWNLOAD_LINK === 'both' || FEED_DOWNLOAD_LINK === 'torrent') && $film->torrentFile) {
+        $enclosure->setAttribute( "url", $film->torrentFile );
         $link->appendChild( new DOMText( $film->torrentFile ) );
-        $item->appendChild( $link );
+    } else if (FEED_DOWNLOAD_LINK === 'magnet' && $film->torrentMagnet) {
+        $enclosure->setAttribute( "url", $film->torrentMagnet );
+        $link->appendChild( new DOMText( $film->torrentMagnet ) );
     }
 
-    if ($film->torrentMagnet) {
+    $item->appendChild( $enclosure );
+    $item->appendChild( $link );
+
+    if ((FEED_DOWNLOAD_LINK === 'both' || FEED_DOWNLOAD_LINK === 'magnet') && $film->torrentMagnet) {
         $torrentMagnetURI = $xml->createElement( "torrent:magnetURI" );
         $torrentMagnetURI->appendChild( new DOMText( $film->torrentMagnet ) );
         $item->appendChild( $torrentMagnetURI );