yt-stream/src/validate.js

Create easily readable streams from YouTube video url's

const validate = require('./request/url.js');

function validateID(ytstream, id) {
  if (typeof id !== 'string') return false;
  let reg = /^[A-Za-z0-9_-]*$/;
  if(!reg.test(id)) return false;
  if(id.length > 16 || id.length < 8) return false;
  return true;
}

function validatePlaylistID(ytstream, id){
  if(typeof id !== 'string') return false;
  let reg = /^[a-zA-Z0-9_-]*$/;
  if(!reg.test(id)) return false;
  if(id.length > 50 || id.length < 30) return false;
  return true;
}

function validateURL(ytstream, url) {
  if (typeof url !== 'string') return false;

  const _url = validate(url);
  if (!_url) return false;

  const hosts = ['music.youtube.com', 'youtube.com', 'youtu.be'];

  let ytid;

  const host = _url.hostname.toLowerCase().split('www.').join('');
  const index = hosts.indexOf(host);
  if (index < 0) return false;

  if (!_url.pathname.toLowerCase().startsWith('/watch')) {
    if (hosts[index] === `youtu.be` && _url.pathname.length > 0) ytid = _url.pathname.split('/').join('');
    else if (_url.pathname.startsWith('/embed/')) ytid = _url.pathname.split('/embed/').join('');
    else if (_url.pathname.startsWith('/v/')) ytid = _url.pathname.split('/v/').join('');
    else if(_url.pathname.startsWith('/shorts/')) ytid = _url.pathname.split('/shorts/').join('');
    else if(_url.pathname.startsWith('/playlist')){
      if(!_url.searchParams.get('list')) return false;
      ytid = _url.searchParams.get('list');
      if(!validatePlaylistID(ytstream, ytid)) return false;
      return true;
    }
  } else {
    if (!_url.searchParams.get('v')) return false;
    ytid = _url.searchParams.get('v');
  }

  if(!validateID(ytstream, ytid)) return false;
  else return true;
}

function validateVideoURL(ytstream, url) {
  if (typeof url !== 'string') return false;

  const _url = validate(url);
  if (!_url) return false;

  const hosts = ['music.youtube.com', 'youtube.com', 'youtu.be'];

  let ytid;

  const host = _url.hostname.toLowerCase().split('www.').join('');
  const index = hosts.indexOf(host);
  if (index < 0) return false;

  if (!_url.pathname.toLowerCase().startsWith('/watch')) {
    if (hosts[index] === `youtu.be` && _url.pathname.length > 0) ytid = _url.pathname.split('/').join('');
    else if (_url.pathname.startsWith('/embed/')) ytid = _url.pathname.split('/embed/').join('');
    else if (_url.pathname.startsWith('/v/')) ytid = _url.pathname.split('/v/').join('');
    else if(_url.pathname.startsWith('/shorts/')) ytid = _url.pathname.split('/shorts/').join('');
  } else {
    if (!_url.searchParams.get('v')) return false;
    ytid = _url.searchParams.get('v');
  }

  if(!validateID(ytstream, ytid)) return false;
  else return true;
}

function validatePlaylistURL(ytstream, url){
  if (typeof url !== 'string') return false;

  const _url = validate(url);
  if (!_url) return false;

  const hosts = ['music.youtube.com', 'youtube.com', 'youtu.be'];

  let ytid;

  const host = _url.hostname.toLowerCase().split('www.').join('');
  const index = hosts.indexOf(host);
  if (index < 0) return false;
  if(!_url.pathname.startsWith('/playlist')) return false;
  if(!_url.searchParams.get('list')) return false;

  ytid = _url.searchParams.get('list');
  if(!validatePlaylistID(ytstream, ytid)) return false;
  else return true;
}

module.exports = {
  validateURL,
  validateID,
  validatePlaylistID,
  validatePlaylistURL,
  validateVideoURL
};