app.js 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. const http = require('http')
  2. const fsPromises = require('fs').promises
  3. const path = require('path')
  4. const crypto = require('crypto')
  5. const sleep = require('util').promisify(setTimeout)
  6. const argvIndexOfFile = process.argv.indexOf(__filename)
  7. let PORT = parseInt(process.argv[argvIndexOfFile + 1])
  8. if (isNaN(PORT)) {
  9. PORT = 8000
  10. }
  11. let DATA_INSTANT = 0
  12. let SLEEP_TIME = 200
  13. let CACHE_MAX_AGE = 0
  14. let USE_WHITELIST = 0
  15. let INTENSITY = 65
  16. function handleCookies(req) {
  17. const cookies = req.headers.cookie
  18. if (!cookies) {
  19. return
  20. }
  21. cookies.split('; ').map((cookie) => {
  22. const [key, value] = cookie.split('=')
  23. if (key != 'instantpage_test') {
  24. return
  25. }
  26. const cookieValueSplit = value.split(',').map((param) => parseInt(param))
  27. DATA_INSTANT = cookieValueSplit[0]
  28. SLEEP_TIME = cookieValueSplit[1]
  29. CACHE_MAX_AGE = cookieValueSplit[2]
  30. USE_WHITELIST = cookieValueSplit[3]
  31. INTENSITY = cookieValueSplit[4]
  32. if (isNaN(INTENSITY)) {
  33. INTENSITY = value.split(',')[4]
  34. }
  35. })
  36. }
  37. function sha384(data) {
  38. const hash = crypto.createHash('sha384')
  39. hash.update(data)
  40. return hash.digest('base64')
  41. }
  42. async function requestListener(req, res) {
  43. let headers = {
  44. 'Content-Type': 'text/html',
  45. }
  46. let pathString = req.url.substr(1)
  47. let page = parseInt(pathString)
  48. if (pathString == '') {
  49. page = 1
  50. }
  51. let content = ''
  52. const jsContent = await fsPromises.readFile(path.resolve(__dirname, '../instantpage.js'))
  53. const jsHash = sha384(jsContent)
  54. if (pathString == 'instantpage.js') {
  55. headers['Content-Type'] = 'text/javascript'
  56. content += jsContent
  57. }
  58. else if (!isNaN(page)) {
  59. handleCookies(req)
  60. await sleep(SLEEP_TIME)
  61. if (CACHE_MAX_AGE) {
  62. headers['Cache-Control'] = `max-age=${CACHE_MAX_AGE}`
  63. }
  64. content += await fsPromises.readFile(path.resolve(__dirname, 'header.html'))
  65. if (!DATA_INSTANT) {
  66. content = content.replace('<body>', '<body data-instant-allow-query-string data-instant-allow-external-links>')
  67. }
  68. if (USE_WHITELIST) {
  69. content = content.replace('<body', '<body data-instant-whitelist')
  70. }
  71. else if (INTENSITY != 65) {
  72. content = content.replace('<body', `<body data-instant-intensity="${INTENSITY}"`)
  73. }
  74. dataInstantAttribute = DATA_INSTANT || USE_WHITELIST ? `data-instant` : ``
  75. content = content.replace(':checked_aqsael', DATA_INSTANT ? 'checked' : '')
  76. content = content.replace(':checked_whitelist', USE_WHITELIST ? 'checked' : '')
  77. content = content.replace(':value_sleep', `value="${SLEEP_TIME}"`)
  78. content = content.replace(':value_cacheAge', `value="${CACHE_MAX_AGE}"`)
  79. content = content.replace(':value_intensity', `value="${INTENSITY}"`)
  80. content += `<h1>Page ${page}</h1>`
  81. for (let i = 1; i <= 3; i++) {
  82. if (page != i) {
  83. content += `<a href="/${i}?${Math.random()}" ${dataInstantAttribute}><span>Page ${i}</span></a>`
  84. }
  85. }
  86. content += `<a href="/${page}?${Math.random()}" target="_blank" ${dataInstantAttribute}><span>Opens in a new tab</span></a>`
  87. content += `<a href="/${page}?${Math.random()}#anchor" ${dataInstantAttribute}><span>Other page anchor</span></a>`
  88. content += `<a href="${req.url}#anchor" id="anchor"><span>Same-page anchor</span></a>`
  89. content += `<a href="/${page}?${Math.random()}" data-no-instant><span>Manually blacklisted link</span></a>`
  90. content += `<a href="/${page}?${Math.random()}"><span>Non-whitelisted link</span></a>`
  91. content += `<a href="https://www.google.com/" ${dataInstantAttribute}><span>External link</span></a>`
  92. content += `<a><span>&lt;a&gt; without <code>href</code></span></a>`
  93. content += `<a href="file:///C:/"><span>file: link</span></a>`
  94. let footer = await fsPromises.readFile(path.resolve(__dirname, 'footer.html'))
  95. footer = footer.toString().replace('__HASH__', jsHash)
  96. content += footer
  97. }
  98. res.writeHead(200, headers)
  99. res.write(content)
  100. res.end()
  101. }
  102. http.createServer(requestListener).listen(PORT)
  103. console.log(`-> Running on http://127.0.0.1:${PORT}/`)