fix(url): return only urls with 200 statusCode

Author: lgaticaq

src/index.js

@@ -1,6 +1,8 @@
 'use strict'
 
 const http = require('http')
+const https = require('https')
+const { URL } = require('url')
 const cheerio = require('cheerio')
 const querystring = require('querystring')
 const Fuse = require('fuse.js')
@@ -18,6 +20,34 @@ const requestOptions = uri => {
   }
 }
 
+const testLink = uri => {
+  return new Promise((resolve, reject) => {
+    const myURL = new URL(uri)
+    const options = {
+      headers: { 'User-Agent': userAgent },
+      hostname: myURL.hostname,
+      port: myURL.port,
+      path: myURL.pathname,
+      method: 'GET'
+    }
+    let request
+    if (myURL.protocol === 'https:') {
+      request = https.request
+    } else {
+      request = http.request
+    }
+    const req = request(options, res => {
+      if (res.statusCode !== 200) {
+        resolve(null)
+      } else {
+        resolve(uri)
+      }
+    })
+    req.on('error', () => resolve(null))
+    req.end()
+  })
+}
+
 const getOriginalLinks = uri => {
   return new Promise((resolve, reject) => {
     const options = requestOptions(uri)
@@ -57,15 +87,17 @@ const getLinksByUrl = uri => {
               .get()
               .filter(x => regex.test(x))
               .map(x =>
-                getOriginalLinks(`/stream/jkmedia/${regex.exec(x)[1]}/`)
+                getOriginalLinks(`/stream/jkmedia/${regex.exec(x)[1]}/`).then(
+                  testLink
+                )
               )
             Promise.all(promises)
               .then(urls => {
                 resolve({
                   title: title,
                   codeName: codeName,
                   chapter: chapter,
-                  urls: urls
+                  urls: urls.filter(uri => uri !== null)
                 })
               })
               .catch(reject)

test/test.js

@@ -82,6 +82,11 @@ describe('anime-dl', function () {
             }
           ]
         })
+      nock('https://www2.mp4upload.com:282')
+        .get(
+          '/d/rgx3762vz3b4quuoykue6jcwkwyu7xsrj57jctgg65tu5hmtf53xlbb4/video.mp4'
+        )
+        .reply(200)
     })
 
     it('should return valid data of a chapter of a anime', done => {
@@ -128,6 +133,11 @@ describe('anime-dl', function () {
             }
           ]
         })
+      nock('https://www2.mp4upload.com:282')
+        .get(
+          '/d/rgx3762vz3b4quuoykue6jcwkwyu7xsrj57jctgg65tu5hmtf53xlbb4/video.mp4'
+        )
+        .reply(200)
     })
 
     it('should return valid data of a chapter of a anime', done => {