app.js 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  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 HOVER_DELAY = 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. HOVER_DELAY = cookieValueSplit[4]
  32. })
  33. }
  34. function sha384(data) {
  35. const hash = crypto.createHash('sha384')
  36. hash.update(data)
  37. return hash.digest('base64')
  38. }
  39. async function requestListener(req, res) {
  40. let headers = {
  41. 'Content-Type': 'text/html',
  42. }
  43. let pathString = req.url.substr(1)
  44. let page = parseInt(pathString)
  45. if (pathString == '') {
  46. page = 1
  47. }
  48. let content = ''
  49. const jsContent = await fsPromises.readFile(path.resolve(__dirname, '../instantpage.js'))
  50. const jsHash = sha384(jsContent)
  51. if (pathString == 'instantpage.js') {
  52. headers['Content-Type'] = 'text/javascript'
  53. content += jsContent
  54. }
  55. else if (!isNaN(page)) {
  56. handleCookies(req)
  57. await sleep(SLEEP_TIME)
  58. if (CACHE_MAX_AGE) {
  59. headers['Cache-Control'] = `max-age=${CACHE_MAX_AGE}`
  60. }
  61. content += await fsPromises.readFile(path.resolve(__dirname, 'header.html'))
  62. if (!DATA_INSTANT) {
  63. content = content.replace('<body>', '<body data-instant-allow-query-string data-instant-allow-external-links>')
  64. }
  65. if (USE_WHITELIST) {
  66. content = content.replace('<body', '<body data-instant-whitelist')
  67. }
  68. if (HOVER_DELAY != 65) {
  69. content = content.replace('<body', `<body data-instant-intensity="${HOVER_DELAY}"`)
  70. }
  71. dataInstantAttribute = DATA_INSTANT || USE_WHITELIST ? `data-instant` : ``
  72. content = content.replace(':checked_aqsael', DATA_INSTANT ? 'checked' : '')
  73. content = content.replace(':checked_whitelist', USE_WHITELIST ? 'checked' : '')
  74. content = content.replace(':value_sleep', `value="${SLEEP_TIME}"`)
  75. content = content.replace(':value_cacheAge', `value="${CACHE_MAX_AGE}"`)
  76. content = content.replace(':value_hoverDelay', `value="${HOVER_DELAY}"`)
  77. content += `<h1>Page ${page}</h1>`
  78. for (let i = 1; i <= 3; i++) {
  79. if (page != i) {
  80. content += `<a href="/${i}?${Math.random()}" ${dataInstantAttribute}><span>Page ${i}</span></a>`
  81. }
  82. }
  83. content += `<a href="/${page}?${Math.random()}" target="_blank" ${dataInstantAttribute}><span>Opens in a new tab</span></a>`
  84. content += `<a href="/${page}?${Math.random()}#anchor" ${dataInstantAttribute}><span>Other page anchor</span></a>`
  85. content += `<a href="${req.url}#anchor" id="anchor"><span>Same-page anchor</span></a>`
  86. content += `<a href="/${page}?${Math.random()}" data-no-instant><span>Manually blacklisted link</span></a>`
  87. content += `<a href="/${page}?${Math.random()}"><span>Non-whitelisted link</span></a>`
  88. content += `<a href="https://www.google.com/" ${dataInstantAttribute}><span>External link</span></a>`
  89. content += `<a><span>&lt;a&gt; without <code>href</code></span></a>`
  90. content += `<a href="file:///C:/"><span>file: link</span></a>`
  91. let footer = await fsPromises.readFile(path.resolve(__dirname, 'footer.html'))
  92. footer = footer.toString().replace('__HASH__', jsHash)
  93. content += footer
  94. }
  95. res.writeHead(200, headers)
  96. res.write(content)
  97. res.end()
  98. }
  99. http.createServer(requestListener).listen(PORT)
  100. console.log(`-> Running on http://127.0.0.1:${PORT}/`)