WhatToWatch/server/services/plex/auth.js

import parser from 'xml2json';
import uuid from 'uuid';
import btoa from 'btoa';
import request from 'request-promise';
import models from '../../db/models';

const rxAuthToken = /authenticationToken="([^"]+)"/;

const tokenUrlParams = (username, password) => ({
  url: 'https://plex.tv/users/sign_in.xml',
  headers: {
    'X-Plex-Client-Identifier': uuid(),
    Authorization: `Basic ${encryptUserCreds(username, password)}`,
  },
});

const encryptUserCreds = (username, password) => {
  const creds = `${username}:${password}`;
  return btoa(creds);
};

const fetchToken = async (username, password) => {
  try {
    const res = await request.post(tokenUrlParams(username, password));
    const token = res.match(rxAuthToken)[1];
    return token;
  } catch (error) {
    return error.message;
  }
};

const plexUrlParams = (plexToken, user) => ({
  url: 'https://plex.tv/pms/servers.xml',
  headers: {
    'X-Plex-Client-Identifier': user.googleId,
    'X-Plex-Token': plexToken,
  },
});

const getPlexPin = async user => {
  try {
    const params = {
      url: 'https://plex.tv/pins.xml',
      headers: {
        'X-Plex-Client-Identifier': user.googleId,
      },
    };
    const res = await request.post(params);
    const formattedResponse = JSON.parse(parser.toJson(res));

    return formattedResponse;
  } catch (error) {
    console.log(error);
    return error.message;
  }
};

const checkPlexPin = async (pinId, user) => {
  try {
    const params = {
      url: `https://plex.tv/pins/${pinId}.xml`,
      headers: {
        'X-Plex-Client-Identifier': user.googleId,
      },
    };
    const res = await request.get(params);
    const formattedResponse = JSON.parse(parser.toJson(res));
    console.log(
      'TCL: checkPlexPin -> formattedResponse',
      formattedResponse.pin.auth_token,
    );
    return formattedResponse.pin.auth_token;
  } catch (error) {
    console.log(error);
    return error.message;
  }
};

const getPlexUrl = async (plexToken, user) => {
  try {
    console.log('Madeit', user);
    const res = await request.get(plexUrlParams(plexToken, user));
    console.log('Madeit', res);
    let formattedResponse = JSON.parse(parser.toJson(res)).MediaContainer
      .Server;
    if (!Array.isArray(formattedResponse)) {
      formattedResponse = [formattedResponse];
    }
    console.log('formatted response', formattedResponse);
    const server = formattedResponse.slice(-1)[0];
    console.log('server', server);
    await models.User.update(
      {
        plexToken: plexToken.trim(),
        plexUrl: `http://${server.address}:${server.port}`.trim(),
      },
      {where: {googleId: user.googleId}},
    );
    console.log('server--', server);
    return `http://${server[0].address}:${server[0].port}`;
  } catch (error) {
    console.log(error.message);
    return error.message;
  }
};

export default {fetchToken, getPlexPin, checkPlexPin, getPlexUrl};