Provide error page for age gated profiles

src/lib/collectors.js

@@ -101,7 +101,9 @@ function fetchUserFromHTML(username) {
 				// require down here or have to deal with require loop. require cache will take care of it anyway.
 				// User -> Timeline -> TimelineEntry -> collectors -/> User
 				const User = require("./structures/User")
-				const sharedData = extractSharedData(text)
+				const result = extractSharedData(text)
+				if (result.status === constants.symbols.extractor_results.SUCCESS) {
+					const sharedData = result.value
 					const user = new User(sharedData.entry_data.ProfilePage[0].graphql.user)
 					history.report("user", true)
 					if (constants.caching.db_user_id) {
@@ -127,6 +129,13 @@ function fetchUserFromHTML(username) {
 						})
 					}
 					return user
+				} else if (result.status === constants.symbols.extractor_results.AGE_RESTRICTED) {
+					// I don't like this code.
+					history.report("user", true)
+					throw constants.symbols.extractor_results.AGE_RESTRICTED
+				} else {
+					throw result.status
+				}
 			}
 		}).catch(error => {
 			if (error === constants.symbols.INSTAGRAM_DEMANDS_LOGIN || error === constants.symbols.RATE_LIMITED) {

src/lib/constants.js

@@ -100,11 +100,15 @@ let constants = {
 		TYPE_GALLERY_IMAGE: Symbol("TYPE_GALLERY_IMAGE"),
 		TYPE_GALLERY_VIDEO: Symbol("TYPE_GALLERY_VIDEO"),
 		NOT_FOUND: Symbol("NOT_FOUND"),
-		NO_SHARED_DATA: Symbol("NO_SHARED_DATA"),
 		INSTAGRAM_DEMANDS_LOGIN: Symbol("INSTAGRAM_DEMANDS_LOGIN"),
 		RATE_LIMITED: Symbol("RATE_LIMITED"),
 		ENDPOINT_OVERRIDDEN: Symbol("ENDPOINT_OVERRIDDEN"),
 		NO_ASSISTANTS_AVAILABLE: Symbol("NO_ASSISTANTS_AVAILABLE"),
+		extractor_results: {
+			SUCCESS: Symbol("SUCCESS"),
+			AGE_RESTRICTED: Symbol("AGE_RESTRICTED"),
+			NO_SHARED_DATA: Symbol("NO_SHARED_DATA")
+		},
 		assistant_statuses: {
 			OFFLINE: Symbol("OFFLINE"),
 			BLOCKED: Symbol("BLOCKED"),

src/lib/utils/body.js

@@ -3,17 +3,40 @@ const {Parser} = require("./parser/parser")
 
 /**
  * @param {string} text
+ * @returns {{status: symbol, value: any}}
  */
 function extractSharedData(text) {
 	const parser = new Parser(text)
 	const index = parser.seek("window._sharedData = ", {moveToMatch: true, useEnd: true})
-	if (index === -1) throw constants.symbols.NO_SHARED_DATA
+	if (index === -1) {
+		// Maybe the profile is age restricted?
+		const age = getRestrictedAge(text)
+		if (age !== null) { // Correct.
+			return {status: constants.symbols.extractor_results.AGE_RESTRICTED, value: age}
+		}
+		return {status: constants.symbols.extractor_results.NO_SHARED_DATA, value: null}
+	}
 	parser.store()
 	const end = parser.seek(";</script>")
 	parser.restore()
 	const sharedDataString = parser.slice(end - parser.cursor)
 	const sharedData = JSON.parse(sharedDataString)
-	return sharedData
+	return {status: constants.symbols.extractor_results.SUCCESS, value: sharedData}
+}
+
+/**
+ * @param {string} text
+ */
+function getRestrictedAge(text) {
+	const parser = new Parser(text)
+	let index = parser.seek("<h2>Restricted profile</h2>", {moveToMatch: true, useEnd: true})
+	if (index === -1) return null
+	index = parser.seek("<p>", {moveToMatch: true, useEnd: true})
+	if (index === -1) return null
+	const explanation = parser.get({split: "</p>"}).trim()
+	const match = explanation.match(/You must be (\d+?) years? old or over to see this profile/)
+	if (!match) return null
+	return +match[1] // the age
 }
 
 module.exports.extractSharedData = extractSharedData

src/lib/utils/parser/parser.js

@@ -37,6 +37,7 @@ class Parser {
 	}
 
 	/**
+	 * Get the next element from the buffer, either up to a token or between two tokens, and update the cursor.
 	 * @param {GetOptions} [options]
 	 * @returns {String}
 	 */
@@ -123,7 +124,7 @@ class Parser {
 	}
 
 	/**
-	 * Seek past the next occurance of the string.
+	 * Seek to or past the next occurance of the string.
 	 * @param {string} toFind
 	 * @param {{moveToMatch?: boolean, useEnd?: boolean}} options both default to false
 	 */

src/site/api/feed.js

@@ -47,6 +47,8 @@ module.exports = [
 							expiresMinutes: userRequestCache.getTtl("user/"+fill[0], 1000*60)
 						})
 					}
+				} else if (error === constants.symbols.extractor_results.AGE_RESTRICTED) {
+					return render(403, "pug/age_gated.pug")
 				} else {
 					throw error
 				}

src/site/api/routes.js

@@ -89,6 +89,8 @@ module.exports = [
 							expiresMinutes: userRequestCache.getTtl("user/"+fill[0], 1000*60)
 						})
 					}
+				} else if (error === constants.symbols.extractor_results.AGE_RESTRICTED) {
+					return render(403, "pug/age_gated.pug")
 				} else {
 					throw error
 				}

src/site/pug/age_gated.pug

@@ -0,0 +1,11 @@
+include includes/error.pug
+
+doctype html
+html
+	head
+		title= `Restricted profile | Bibliogram`
+		include includes/head
+	body.error-page
+		+error(403, "Restricted profile", false)
+			| This profile is age restricted.
+			| You must log in to Instagram to view this profile.

src/site/pug/blocked.pug

@@ -2,8 +2,6 @@
 
 include includes/error.pug
 
-- const numberFormat = new Intl.NumberFormat().format
-
 doctype html
 html
 	head

test/body.js

@@ -4,16 +4,27 @@ const {extractSharedData} = require("../src/lib/utils/body")
 const fs = require("fs").promises
 
 tap.test("extract shared data", async childTest => {
-	childTest.throws(() => extractSharedData(""), constants.symbols.NO_SHARED_DATA, "not found in blank")
+	{
+		const result = extractSharedData("")
+		childTest.equal(result.status, constants.symbols.extractor_results.NO_SHARED_DATA, "not found in blank")
+	}
 	{
 		const page = await fs.readFile("test/files/page-user-instagram.html", "utf8")
-		const sharedData = extractSharedData(page)
+		const result = extractSharedData(page)
-		childTest.equal(sharedData.entry_data.ProfilePage[0].graphql.user.username, "instagram", "can extract user page")
+		childTest.equal(result.status, constants.symbols.extractor_results.SUCCESS, "extractor status success")
+		childTest.equal(result.value.entry_data.ProfilePage[0].graphql.user.username, "instagram", "can extract user page")
 	}
 	{
 		const page = await fs.readFile("test/files/page-login.html", "utf8")
-		const sharedData = extractSharedData(page)
+		const result = extractSharedData(page)
-		childTest.true(sharedData.entry_data.LoginAndSignupPage[0], "can extract login page")
+		childTest.equal(result.status, constants.symbols.extractor_results.SUCCESS, "extractor status success")
+		childTest.true(result.value.entry_data.LoginAndSignupPage[0], "can extract login page")
+	}
+	{
+		const page = await fs.readFile("test/files/page-age-gated.html", "utf8")
+		const result = extractSharedData(page)
+		childTest.equal(result.status, constants.symbols.extractor_results.AGE_RESTRICTED, "extractor detects age restricted")
+		childTest.equal(result.value, 21, "correct age is extracted")
 	}
 	childTest.end()
 })