"use strict"; // Public modules from npm import axios, { AxiosResponse } from "axios"; import cheerio from "cheerio"; import axiosCookieJarSupport from "axios-cookiejar-support"; import tough from "tough-cookie"; // Modules from file import shared from "./shared.js"; import { urls as f95url } from "./constants/url.js"; import { selectors as f95selector } from "./constants/css-selector.js"; import LoginResult from "./classes/login-result.js"; import credentials from "./classes/credentials.js"; // Global variables const userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15) " + "AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0 Safari/605.1.15"; // @ts-ignore axiosCookieJarSupport.default(axios); const commonConfig = { headers: { "User-Agent": userAgent, "Connection": "keep-alive" }, withCredentials: true, jar: new tough.CookieJar() // Used to store the token in the PC }; /** * Gets the HTML code of a page. */ export async function fetchHTML(url: string): Promise { // Local variables let returnValue = null; // Fetch the response of the platform const response = await fetchGETResponse(url); // Manage response /* istambul ignore next */ if (!response) { shared.logger.warn(`Unable to fetch HTML for ${url}`); } /* istambul ignore next */ else if (!response.headers["content-type"].includes("text/html")) { // The response is not a HTML page shared.logger.warn(`The ${url} returned a ${response.headers["content-type"]} response`); } returnValue = response.data; return returnValue; } /** * It authenticates to the platform using the credentials * and token obtained previously. Save cookies on your * device after authentication. * @param {module:./classes/credentials.ts:Credentials} credentials Platform access credentials * @param {Boolean} force Specifies whether the request should be forced, ignoring any saved cookies * @returns {Promise} Result of the operation */ export async function authenticate(credentials: credentials, force: boolean = false): Promise { shared.logger.info(`Authenticating with user ${credentials.username}`); if (!credentials.token) throw new Error(`Invalid token for auth: ${credentials.token}`); // Secure the URL const secureURL = enforceHttpsUrl(f95url.F95_LOGIN_URL); // Prepare the parameters to send to the platform to authenticate const params = new URLSearchParams(); params.append("login", credentials.username); params.append("url", ""); params.append("password", credentials.password); params.append("password_confirm", ""); params.append("additional_security", ""); params.append("remember", "1"); params.append("_xfRedirect", "https://f95zone.to/"); params.append("website_code", ""); params.append("_xfToken", credentials.token); try { // Try to log-in let config = Object.assign({}, commonConfig); if (force) delete config.jar; const response = await axios.post(secureURL, params, config); // Parse the response HTML const $ = cheerio.load(response.data); // Get the error message (if any) and remove the new line chars const errorMessage = $("body").find(f95selector.LOGIN_MESSAGE_ERROR).text().replace(/\n/g, ""); // Return the result of the authentication const result = errorMessage.trim() === ""; const message = errorMessage.trim() === "" ? "Authentication successful" : errorMessage; return new LoginResult(result, message); } catch (e) { shared.logger.error(`Error ${e.message} occurred while authenticating to ${secureURL}`); return new LoginResult(false, `Error ${e.message} while authenticating`); } }; /** * Obtain the token used to authenticate the user to the platform. * @returns {Promise} Token or `null` if an error arise */ export async function getF95Token(): Promise { // Fetch the response of the platform const response = await fetchGETResponse(f95url.F95_LOGIN_URL); /* istambul ignore next */ if (!response) { shared.logger.warn("Unable to get the token for the session"); return null; } // The response is a HTML page, we need to find the with name "_xfToken" const $ = cheerio.load(response.data as string); return $("body").find(f95selector.GET_REQUEST_TOKEN).attr("value"); } //#region Utility methods /** * Performs a GET request to a specific URL and returns the response. * If the request generates an error (for example 400) `null` is returned. */ export async function fetchGETResponse(url: string): Promise> { // Secure the URL const secureURL = enforceHttpsUrl(url); try { // Fetch and return the response return await axios.get(secureURL, commonConfig); } catch (e) { shared.logger.error(`Error ${e.message} occurred while trying to fetch ${secureURL}`); return null; } } /** * Enforces the scheme of the URL is https and returns the new URL. * @param {String} url * @returns {String} Secure URL or `null` if the argument is not a string */ export function enforceHttpsUrl(url: string): string { return isStringAValidURL(url) ? url.replace(/^(https?:)?\/\//, "https://") : null; }; /** * Check if the url belongs to the domain of the F95 platform. * @param {String} url URL to check * @returns {Boolean} true if the url belongs to the domain, false otherwise */ export function isF95URL(url: string): boolean { if (url.toString().startsWith(f95url.F95_BASE_URL)) return true; else return false; }; /** * Checks if the string passed by parameter has a * properly formatted and valid path to a URL (HTTP/HTTPS). * @param {String} url String to check for correctness * @returns {Boolean} true if the string is a valid URL, false otherwise */ export function isStringAValidURL(url: string): boolean { // Many thanks to Daveo at StackOverflow (https://preview.tinyurl.com/y2f2e2pc) const expression = /https?:\/\/(www\.)?[-a-zA-Z0-9@:%._+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_+.~#?&//=]*)/; const regex = new RegExp(expression); if (url.match(regex)) return true; else return false; }; /** * Check if a particular URL is valid and reachable on the web. * @param {string} url URL to check * @param {boolean} [checkRedirect] * If true, the function will consider redirects a violation and return false. * Default: false * @returns {Promise} true if the URL exists, false otherwise */ export async function urlExists(url: string, checkRedirect: boolean = false): Promise { // Local variables let valid = false; if (isStringAValidURL(url)) { valid = await axiosUrlExists(url); if (valid && checkRedirect) { const redirectUrl = await getUrlRedirect(url); valid = redirectUrl === url; } } return valid; } /** * Check if the URL has a redirect to another page. * @param {String} url URL to check for redirect * @returns {Promise} Redirect URL or the passed URL */ export async function getUrlRedirect(url: string): Promise { const response = await axios.head(url); return response.config.url; } //#endregion Utility methods //#region Private methods /** * Check with Axios if a URL exists. */ async function axiosUrlExists(url: string): Promise { // Local variables const ERROR_CODES = ["ENOTFOUND", "ETIMEDOUT"]; let valid = false; try { const response = await axios.head(url, { timeout: 3000 }); valid = response && !/4\d\d/.test(response.status.toString()); } catch (error) { // Throw error only if the error is unknown if (!ERROR_CODES.includes(error.code)) throw error; } return valid; } //#endregion