F95API/app/index.js

604 lines
19 KiB
JavaScript
Raw Normal View History

"use strict";
// Core modules
const fs = require("fs");
2020-09-29 15:11:43 +00:00
// Modules from file
const shared = require("./scripts/shared.js");
2020-10-08 19:09:39 +00:00
const constURLs = require("./scripts/constants/urls.js");
const selectors = require("./scripts/constants/css-selectors.js");
const {
isStringAValidURL,
urlExists,
isF95URL,
} = require("./scripts/urls-helper.js");
2020-10-12 08:27:48 +00:00
const scraper = require("./scripts/game-scraper.js");
const {
prepareBrowser,
preparePage,
} = require("./scripts/puppeteer-helper.js");
2020-10-12 08:27:48 +00:00
const searcher = require("./scripts/game-searcher.js");
// Classes from file
const GameInfo = require("./scripts/classes/game-info.js");
const LoginResult = require("./scripts/classes/login-result.js");
const UserData = require("./scripts/classes/user-data.js");
2020-09-29 15:11:43 +00:00
2020-10-03 16:16:45 +00:00
//#region Export classes
module.exports.GameInfo = GameInfo;
module.exports.LoginResult = LoginResult;
module.exports.UserData = UserData;
2020-10-03 16:16:45 +00:00
//#endregion Export classes
2020-09-29 15:11:43 +00:00
2020-10-03 16:16:45 +00:00
//#region Export properties
2020-09-29 15:11:43 +00:00
/**
2020-10-02 15:43:14 +00:00
* Shows log messages and other useful functions for module debugging.
* @param {Boolean} value
2020-09-29 15:11:43 +00:00
*/
module.exports.debug = function (value) {
shared.debug = value;
};
2020-10-02 15:43:14 +00:00
/**
* @public
* Indicates whether a user is logged in to the F95Zone platform or not.
* @returns {String}
*/
module.exports.isLogged = function () {
return shared.isLogged;
2020-09-29 15:11:43 +00:00
};
2020-10-02 15:43:14 +00:00
/**
* @public
* If true, it opens a new browser for each request
2020-10-02 15:43:14 +00:00
* to the F95Zone platform, otherwise it reuses the same.
* @returns {String}
*/
module.exports.setIsolation = function (value) {
shared.isolation = value;
};
2020-10-02 15:43:14 +00:00
/**
* @public
* Path to the cache directory
* @returns {String}
*/
module.exports.getCacheDir = function () {
return shared.cacheDir;
};
2020-10-02 15:43:14 +00:00
/**
* @public
* Set path to the cache directory
* @returns {String}
*/
module.exports.setCacheDir = function (value) {
shared.cacheDir = value;
// Create directory if it doesn't exist
if (!fs.existsSync(shared.cacheDir)) fs.mkdirSync(shared.cacheDir);
};
2020-10-13 13:40:10 +00:00
/**
* @public
* Set local chromium path.
* @returns {String}
*/
module.exports.setChromiumPath = function (value) {
shared.chromiumLocalPath = value;
};
2020-10-03 16:16:45 +00:00
//#endregion Export properties
2020-09-29 15:11:43 +00:00
//#region Global variables
var _browser = null;
//#endregion
2020-09-29 15:11:43 +00:00
//#region Export methods
/**
* @public
* Log in to the F95Zone platform.
* This **must** be the first operation performed before accessing any other script functions.
* @param {String} username Username used for login
* @param {String} password Password used for login
* @returns {Promise<LoginResult>} Result of the operation
*/
module.exports.login = async function (username, password) {
if (shared.isLogged) {
if (shared.debug) console.log("Already logged in");
let result = new LoginResult();
result.success = true;
result.message = "Already logged in";
return result;
}
// If cookies are loaded, use them to authenticate
shared.cookies = loadCookies();
if (shared.cookies !== null) {
if (shared.debug) console.log("Valid session, no need to re-authenticate");
shared.isLogged = true;
let result = new LoginResult();
result.success = true;
result.message = "Logged with cookies";
return result;
}
2020-09-29 15:11:43 +00:00
// Else, log in throught browser
if (shared.debug)
console.log("No saved sessions or expired session, login on the platform");
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
let result = await loginF95(browser, username, password);
shared.isLogged = result.success;
2020-09-29 15:11:43 +00:00
if (result.success) {
// Reload cookies
shared.cookies = loadCookies();
if (shared.debug) console.log("User logged in through the platform");
} else {
console.warn("Error during authentication: " + result.message);
}
if (shared.isolation) await browser.close();
return result;
};
2020-09-29 15:11:43 +00:00
/**
* @public
* This method loads the main data from the F95 portal
* used to provide game information. You **must** be logged
2020-09-29 15:11:43 +00:00
* in to the portal before calling this method.
* @returns {Promise<Boolean>} Result of the operation
*/
module.exports.loadF95BaseData = async function () {
if (!shared.isLogged || !shared.cookies) {
console.warn("User not authenticated, unable to continue");
return false;
}
if (shared.debug) console.log("Loading base data...");
// Prepare a new web page
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
let page = await preparePage(browser); // Set new isolated page
await page.setCookie(...shared.cookies); // Set cookies to avoid login
// Go to latest update page and wait for it to load
await page.goto(constURLs.F95_LATEST_UPDATES, {
waitUntil: shared.WAIT_STATEMENT,
});
// Obtain engines (disc/online)
2020-10-03 16:16:45 +00:00
await page.waitForSelector(selectors.ENGINE_ID_SELECTOR);
shared.engines = await loadValuesFromLatestPage(
page,
shared.enginesCachePath,
2020-10-03 16:16:45 +00:00
selectors.ENGINE_ID_SELECTOR,
"engines"
);
// Obtain statuses (disc/online)
2020-10-03 16:16:45 +00:00
await page.waitForSelector(selectors.STATUS_ID_SELECTOR);
shared.statuses = await loadValuesFromLatestPage(
page,
shared.statusesCachePath,
2020-10-03 16:16:45 +00:00
selectors.STATUS_ID_SELECTOR,
"statuses"
);
if (shared.isolation) await browser.close();
if (shared.debug) console.log("Base data loaded");
return true;
};
2020-09-29 15:11:43 +00:00
/**
* @public
2020-10-10 09:33:54 +00:00
* Chek if exists a new version of the game.
2020-09-29 15:11:43 +00:00
* You **must** be logged in to the portal before calling this method.
* @param {GameInfo} info Information about the game to get the version for
2020-10-10 09:33:54 +00:00
* @returns {Promise<Boolean>} true if an update is available, false otherwise
2020-09-29 15:11:43 +00:00
*/
2020-10-10 09:33:54 +00:00
module.exports.chekIfGameHasUpdate = async function (info) {
if (!shared.isLogged || !shared.cookies) {
console.warn("user not authenticated, unable to continue");
return info.version;
}
2020-09-29 15:11:43 +00:00
// F95 change URL at every game update,
// so if the URL is different an update is available
let exists = await urlExists(info.f95url, true);
if (!exists) return true;
// Parse version from title
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
let onlineVersion = await scraper.getGameVersionFromTitle(browser, info);
if (shared.isolation) await browser.close();
return onlineVersion.toUpperCase() !== info.version.toUpperCase();
};
2020-09-29 15:11:43 +00:00
/**
* @public
* Starting from the name, it gets all the information about the game you are looking for.
* You **must** be logged in to the portal before calling this method.
* @param {String} name Name of the game searched
* @param {Boolean} includeMods Indicates whether to also take mods into account when searching
* @returns {Promise<GameInfo[]>} List of information obtained where each item corresponds to
* an identified game (in the case of homonymy). If no games were found, null is returned
2020-09-29 15:11:43 +00:00
*/
module.exports.getGameData = async function (name, includeMods) {
if (!shared.isLogged || !shared.cookies) {
console.warn("user not authenticated, unable to continue");
return null;
}
// Gets the search results of the game being searched for
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
2020-10-12 08:27:48 +00:00
let urlList = await searcher.getSearchGameResults(browser, name);
// Process previous partial results
let promiseList = [];
for (let url of urlList) {
// Start looking for information
2020-10-12 08:27:48 +00:00
promiseList.push(scraper.getGameInfo(browser, url));
}
// Filter for mods
let result = [];
for (let info of await Promise.all(promiseList)) {
// Skip mods if not required
2020-10-12 08:27:48 +00:00
if (!info) continue;
if (info.isMod && !includeMods) continue;
else result.push(info);
}
if (shared.isolation) await browser.close();
return result;
};
2020-10-10 09:45:43 +00:00
/**
* @public
* Starting from the url, it gets all the information about the game you are looking for.
* You **must** be logged in to the portal before calling this method.
* @param {String} url URL of the game to obtain information of
* @returns {Promise<GameInfo>} Information about the game. If no game was found, null is returned
*/
module.exports.getGameDataFromURL = async function (url) {
if (!shared.isLogged || !shared.cookies) {
2020-10-10 09:45:43 +00:00
console.warn("user not authenticated, unable to continue");
return null;
}
// Check URL
if (!urlExists(url)) return null;
2020-10-10 09:45:43 +00:00
if (!isF95URL(url)) throw url + " is not a valid F95Zone URL";
// Gets the search results of the game being searched for
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
2020-10-10 09:45:43 +00:00
// Get game data
2020-10-12 08:27:48 +00:00
let result = await scraper.getGameInfo(browser, url);
2020-10-10 09:45:43 +00:00
if (shared.isolation) await browser.close();
return result;
};
2020-09-29 15:11:43 +00:00
/**
* @public
* Gets the data of the currently logged in user.
2020-10-02 15:43:14 +00:00
* You **must** be logged in to the portal before calling this method.
* @returns {Promise<UserData>} Data of the user currently logged in or null if an error arise
2020-09-29 15:11:43 +00:00
*/
module.exports.getUserData = async function () {
if (!shared.isLogged || !shared.cookies) {
console.warn("user not authenticated, unable to continue");
return null;
}
// Prepare a new web page
if (_browser === null && !shared.isolation) _browser = await prepareBrowser();
let browser = shared.isolation ? await prepareBrowser() : _browser;
let page = await preparePage(browser); // Set new isolated page
await page.setCookie(...shared.cookies); // Set cookies to avoid login
await page.goto(constURLs.F95_BASE_URL); // Go to base page
// Explicitly wait for the required items to load
2020-10-03 16:16:45 +00:00
await page.waitForSelector(selectors.USERNAME_ELEMENT);
await page.waitForSelector(selectors.AVATAR_PIC);
let threads = getUserWatchedGameThreads(browser);
let username = await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector).innerText,
2020-10-03 16:16:45 +00:00
selectors.USERNAME_ELEMENT
);
let avatarSrc = await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector).getAttribute("src"),
2020-10-03 16:16:45 +00:00
selectors.AVATAR_PIC
);
let ud = new UserData();
ud.username = username;
ud.avatarSrc = isStringAValidURL(avatarSrc) ? avatarSrc : null;
ud.watchedThreads = await threads;
await page.close();
if (shared.isolation) await browser.close();
return ud;
};
2020-10-02 15:43:14 +00:00
/**
* @public
* Logout from the current user and gracefully close shared browser.
2020-10-02 15:43:14 +00:00
* You **must** be logged in to the portal before calling this method.
*/
module.exports.logout = async function () {
if (!shared.isLogged || !shared.cookies) {
console.warn("user not authenticated, unable to continue");
return;
}
shared.isLogged = false;
// Gracefully close shared browser
2020-10-08 20:39:03 +00:00
if (!shared.isolation && _browser !== null) {
await _browser.close();
_browser = null;
}
};
2020-09-29 15:11:43 +00:00
//#endregion
//#region Private methods
//#region Cookies functions
/**
* @private
* Loads and verifies the expiration of previously stored cookies from disk
* if they exist, otherwise it returns null.
* @return {object[]} List of dictionaries or null if cookies don't exist
*/
function loadCookies() {
// Check the existence of the cookie file
if (fs.existsSync(shared.cookiesCachePath)) {
// Read cookies
let cookiesJSON = fs.readFileSync(shared.cookiesCachePath);
let cookies = JSON.parse(cookiesJSON);
// Check if the cookies have expired
for (let cookie of cookies) {
if (isCookieExpired(cookie)) return null;
}
2020-09-29 15:11:43 +00:00
// Cookies loaded and verified
return cookies;
} else return null;
2020-09-29 15:11:43 +00:00
}
/**
* @private
* Check the validity of a cookie.
* @param {object} cookie Cookies to verify the validity. It's a dictionary
* @returns {Boolean} true if the cookie has expired, false otherwise
*/
function isCookieExpired(cookie) {
// Local variables
let expiredCookies = false;
// Ignore cookies that never expire
let expirationUnixTimestamp = cookie["expire"];
if (expirationUnixTimestamp !== "-1") {
// Convert UNIX epoch timestamp to normal Date
let expirationDate = new Date(expirationUnixTimestamp * 1000);
if (expirationDate < Date.now()) {
if (shared.debug)
console.log(
"Cookie " + cookie["name"] + " expired, you need to re-authenticate"
);
expiredCookies = true;
2020-09-29 15:11:43 +00:00
}
}
2020-09-29 15:11:43 +00:00
return expiredCookies;
2020-09-29 15:11:43 +00:00
}
//#endregion Cookies functions
//#region Latest Updates page parserer
/**
* @private
* If present, it reads the file containing the searched values (engines or states)
* from the disk, otherwise it connects to the F95 portal (at the page
2020-09-29 15:11:43 +00:00
* https://f95zone.to/latest) and downloads them.
* @param {puppeteer.Page} page Page used to locate the required elements
* @param {String} path Path to disk of the JSON file containing the data to read / write
* @param {String} selector CSS selector of the required elements
* @param {String} elementRequested Required element (engines or states) used to detail log messages
* @returns {Promise<String[]>} List of required values in uppercase
*/
async function loadValuesFromLatestPage(
page,
path,
selector,
elementRequested
) {
// If the values already exist they are loaded from disk without having to connect to F95
if (shared.debug) console.log("Load " + elementRequested + " from disk...");
if (fs.existsSync(path)) {
let valueJSON = fs.readFileSync(path);
return JSON.parse(valueJSON);
}
// Otherwise, connect and download the data from the portal
if (shared.debug)
console.log("No " + elementRequested + " cached, downloading...");
let values = await getValuesFromLatestPage(
page,
selector,
"Getting " + elementRequested + " from page"
);
fs.writeFileSync(path, JSON.stringify(values));
return values;
2020-09-29 15:11:43 +00:00
}
/**
* @private
* Gets all the textual values of the elements present
* in the F95 portal page and identified by the selector
2020-09-29 15:11:43 +00:00
* passed by parameter
* @param {puppeteer.Page} page Page used to locate items specified by the selector
* @param {String} selector CSS selector
* @param {String} logMessage Log message indicating which items the selector is requesting
* @return {Promise<String[]>} List of uppercase strings indicating the textual values of the elements identified by the selector
*/
async function getValuesFromLatestPage(page, selector, logMessage) {
if (shared.debug) console.log(logMessage);
2020-09-29 15:11:43 +00:00
let result = [];
let elements = await page.$$(selector);
2020-09-29 15:11:43 +00:00
for (let element of elements) {
let text = await element.evaluate(
/* istanbul ignore next */ (e) => e.innerText
);
2020-09-29 15:11:43 +00:00
// Save as upper text for better match if used in query
result.push(text.toUpperCase());
}
return result;
2020-09-29 15:11:43 +00:00
}
//#endregion
//#region User
/**
* @private
* Log in to the F95Zone portal and, if successful, save the cookies.
* @param {puppeteer.Browser} browser Browser object used for navigation
* @param {String} username Username to use during login
* @param {String} password Password to use during login
* @returns {Promise<LoginResult>} Result of the operation
*/
async function loginF95(browser, username, password) {
let page = await preparePage(browser); // Set new isolated page
await page.goto(constURLs.F95_LOGIN_URL); // Go to login page
// Explicitly wait for the required items to load
2020-10-03 16:16:45 +00:00
await page.waitForSelector(selectors.USERNAME_INPUT);
await page.waitForSelector(selectors.PASSWORD_INPUT);
await page.waitForSelector(selectors.LOGIN_BUTTON);
2020-10-03 16:16:45 +00:00
await page.type(selectors.USERNAME_INPUT, username); // Insert username
await page.type(selectors.PASSWORD_INPUT, password); // Insert password
2020-10-14 13:12:35 +00:00
await Promise.all([
page.click(selectors.LOGIN_BUTTON), // Click on the login button
page.waitForNavigation({
waitUntil: shared.WAIT_STATEMENT,
}), // Wait for page to load
2020-10-14 13:12:35 +00:00
]);
// Prepare result
let result = new LoginResult();
// Check if the user is logged in
result.success = await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector) !== null,
2020-10-03 16:16:45 +00:00
selectors.AVATAR_INFO
);
// Save cookies to avoid re-auth
if (result.success) {
let c = await page.cookies();
fs.writeFileSync(shared.cookiesCachePath, JSON.stringify(c));
result.message = "Authentication successful";
}
// Obtain the error message
else if (
await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector) !== null,
2020-10-03 16:16:45 +00:00
selectors.LOGIN_MESSAGE_ERROR
)
) {
let errorMessage = await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector).innerText,
2020-10-03 16:16:45 +00:00
selectors.LOGIN_MESSAGE_ERROR
);
if (errorMessage === "Incorrect password. Please try again.") {
result.message = "Incorrect password";
} else if (
errorMessage ===
"The requested user '" + username + "' could not be found."
) {
result.message = "Incorrect username";
} else result.message = errorMessage;
} else result.message = "Unknown error";
await page.close(); // Close the page
return result;
2020-09-29 15:11:43 +00:00
}
/**
* @private
* Gets the list of URLs of threads the user follows.
* @param {puppeteer.Browser} browser Browser object used for navigation
* @returns {Promise<String[]>} URL list
2020-09-29 15:11:43 +00:00
*/
async function getUserWatchedGameThreads(browser) {
let page = await preparePage(browser); // Set new isolated page
await page.goto(constURLs.F95_WATCHED_THREADS); // Go to the thread page
// Explicitly wait for the required items to load
2020-10-03 16:16:45 +00:00
await page.waitForSelector(selectors.WATCHED_THREAD_FILTER_POPUP_BUTTON);
// Show the popup
2020-10-03 16:16:45 +00:00
await page.click(selectors.WATCHED_THREAD_FILTER_POPUP_BUTTON);
await page.waitForSelector(selectors.UNREAD_THREAD_CHECKBOX);
await page.waitForSelector(selectors.ONLY_GAMES_THREAD_OPTION);
await page.waitForSelector(selectors.FILTER_THREADS_BUTTON);
// Set the filters
await page.evaluate(
/* istanbul ignore next */ (selector) =>
document.querySelector(selector).removeAttribute("checked"),
2020-10-03 16:16:45 +00:00
selectors.UNREAD_THREAD_CHECKBOX
); // Also read the threads already read
2020-10-03 16:16:45 +00:00
await page.click(selectors.ONLY_GAMES_THREAD_OPTION);
// Filter the threads
2020-10-03 16:16:45 +00:00
await page.click(selectors.FILTER_THREADS_BUTTON);
await page.waitForSelector(selectors.WATCHED_THREAD_URLS);
// Get the threads urls
let urls = [];
let nextPageExists = false;
do {
// Get all the URLs
2020-10-03 16:16:45 +00:00
for (let handle of await page.$$(selectors.WATCHED_THREAD_URLS)) {
let src = await page.evaluate(
/* istanbul ignore next */ (element) => element.href,
handle
);
// If 'unread' is left, it will redirect to the last unread post
let url = src.replace("/unread", "");
urls.push(url);
}
nextPageExists = await page.evaluate(
/* istanbul ignore next */ (selector) => document.querySelector(selector),
2020-10-03 16:16:45 +00:00
selectors.WATCHED_THREAD_NEXT_PAGE
);
// Click to next page
if (nextPageExists) {
2020-10-03 16:16:45 +00:00
await page.click(selectors.WATCHED_THREAD_NEXT_PAGE);
await page.waitForSelector(selectors.WATCHED_THREAD_URLS);
}
} while (nextPageExists);
await page.close();
return urls;
2020-09-29 15:11:43 +00:00
}
//#endregion User
//#endregion Private methods