YouTube.js/lib/Actions.js

'use strict';

const Uuid = require('uuid');
const Axios = require('axios');
const Utils = require('./Utils');
const Constants = require('./Constants');

async function engage(session, engagement_type, args = {}) {
  if (!session.logged_in) throw new Error('You are not signed-in');
 
  let data;
  switch (engagement_type) {
    case 'like/like':
    case 'like/dislike':
    case 'like/removelike':
      data = {
        context: session.context,
        target: {
          videoId: args.video_id
        }
      };
      break;
    case 'subscription/subscribe':
    case 'subscription/unsubscribe':
      data = {
        context: session.context,
        channelIds: [args.channel_id]
      };
      break;
    case 'comment/create_comment':
      data = {
        context: session.context,
        commentText: args.text,
        createCommentParams: Utils.generateCommentParams(args.video_id)
      };
      break;
    default:
  }

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/${engagement_type}${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
  JSON.stringify(data), Constants.INNERTUBE_REQOPTS({ session, id: args.video_id, data })).catch((error) => error);
  
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
 
  return {
    success: true,
    status_code: response.status
  };
}

async function browse(session, action_type) {
  if (!session.logged_in) throw new Error('You are not signed-in');

  let data;
  switch (action_type) { // TODO: Handle more actions
    case 'subscriptions_feed':
      data = {
        context: session.context,
        browseId: 'FEsubscriptions'
      };
      break;
    default:
  }

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/browse${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
  JSON.stringify(data), Constants.INNERTUBE_REQOPTS({ session })).catch((error) => error);
  
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
 
  return {
    success: true,
    status_code: response.status,
    data: response.data
  };
}

async function search(session, args = {}) {
  if (!args.query) throw new Error('No query was provided');

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/search${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`,
  JSON.stringify({ 
    context: session.context, 
    params: Utils.encodeFilter(args.options.period, args.options.duration, args.options.order), 
    query: args.query 
  }), Constants.INNERTUBE_REQOPTS({ session })).catch((error) => error);
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
  
  return {
    success: true,
    status_code: response.status,
    data: response.data
  };
}

async function notifications(session, action_type, args = {}) {
  if (!session.logged_in) throw new Error('You are not signed-in');
 
  let data;

  switch (action_type) {
    case 'modify_channel_preference':
      let pref_types = { PERSONALIZED: 1, ALL: 2, NONE: 3 };
      data = {
        context: session.context,
        params: Utils.encodeNotificationPref(args.channel_id, pref_types[args.pref.toUpperCase()])
      };
      break;
    case 'get_notification_menu':
      data = {
        context: session.context,
        notificationsMenuRequestType: 'NOTIFICATIONS_MENU_REQUEST_TYPE_INBOX'
      };
      break;
    case 'get_unseen_count':
      data = {
        context: session.context
      };
      break;
    default:
  }

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/notification/${action_type}${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
  JSON.stringify(data), Constants.INNERTUBE_REQOPTS({ session })).catch((error) => error);
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
  if (action_type === 'modify_channel_preference') return { success: true, status_code: response.status };
 
  return {
    success: true,
    status_code: response.status,
    data: response.data
  };
}

async function livechat(session, action_type, args = {}) {
  let data;
  switch (action_type) {
    case 'live_chat/send_message':
      data = {
        context: session.context,
        params: Utils.generateMessageParams(args.channel_id, args.video_id),
        clientMessageId: `ytjs-${Uuid.v4()}`,
        richMessage: {
          textSegments: [{ text: args.text }]
        }
      };
      break;
    case 'live_chat/get_item_context_menu':
      data = {
        context: session.context
      };
      break;
    case 'live_chat/moderate':
      data = {
        context: session.context,
        params: args.cmd_params
      };
      break;
    default:
  }

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/${action_type}${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
  JSON.stringify(data), Constants.INNERTUBE_REQOPTS({ session, params: args.params })).catch((error) => error);
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
  
  return {
    success: true,
    status_code: response.status,
    data: response.data
  };
}

async function getVideoInfo(session, args = {}) {
  let response;
 
  !args.is_desktop && (response = await Axios.get(`${Constants.URLS.YT_WATCH_PAGE}?v=${args.id}&t=8s&pbj=1`, Constants.INNERTUBE_REQOPTS({ session, id: args.id, desktop: false })).catch((error) => error)) ||
    (response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/player${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
     JSON.stringify(Constants.VIDEO_INFO_REQBODY(args.id, session.sts, session.context)), Constants.INNERTUBE_REQOPTS({ session, id: args.id, desktop: true })).catch((error) => error));
  if (response instanceof Error) throw new Error(`Could not get video info: ${response.message}`);

  return response.data;
}

async function getContinuation(session, info = {}) {
  let data = { context: session.context };
  info.continuation_token && (data.continuation = info.continuation_token);

  if (info.video_id) {
    data.videoId = info.video_id;
    data.racyCheckOk = true;
    data.contentCheckOk = false;
    data.autonavState = 'STATE_NONE';
    data.playbackContext = {
      vis: 0,
      lactMilliseconds: '-1'
    };
    data.captionsRequested = false;
  }

  const response = await Axios.post(`${Constants.URLS.YT_BASE_URL}/youtubei/v1/next${session.logged_in && session.cookie.length < 1 ? '' : `?key=${session.key}`}`, 
  JSON.stringify(data), Constants.INNERTUBE_REQOPTS({ session })).catch((error) => error);
  if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };
 
  return {
    success: true,
    status_code: response.status,
    data: response.data
  };
}

module.exports = { engage, browse, search, notifications, livechat, getVideoInfo, getContinuation };