mirror of https://github.com/bobwen-dev/htpdate
130 lines
3.0 KiB
CoffeeScript
130 lines
3.0 KiB
CoffeeScript
util = require 'util'
|
|
got = require 'got'
|
|
dayjs = require 'dayjs'
|
|
info = require './package.json'
|
|
median = require './median'
|
|
adjust_time = require './adjust_time'
|
|
|
|
|
|
argv = require('./argv') {
|
|
threshold: {
|
|
alias: 't'
|
|
describe: 'At least how many milliseconds are considered to adjust system time'
|
|
default: 1500
|
|
type: 'number'
|
|
}
|
|
set: {
|
|
describe: 'Adjust system time if necessary'
|
|
alias: 's'
|
|
default: false
|
|
type: 'boolean'
|
|
}
|
|
protocol: {
|
|
describe: 'Use this protocol when no protocol is specified in the url'
|
|
alias: 'p'
|
|
default: 'https'
|
|
type: 'string'
|
|
}
|
|
method: {
|
|
describe: 'HTTP method'
|
|
alias: 'm'
|
|
default: 'HEAD'
|
|
type: 'string'
|
|
}
|
|
count: {
|
|
describe: 'The number of requests for each url'
|
|
alias: 'c'
|
|
default: 4
|
|
type: 'number'
|
|
}
|
|
redirect: {
|
|
describe: 'If redirect responses should be followed'
|
|
alias: 'r'
|
|
default: false
|
|
type: 'boolean'
|
|
}
|
|
timeout: {
|
|
alias: 'T'
|
|
default: 6000
|
|
type: 'number'
|
|
}
|
|
command: {
|
|
describe: 'Command to adjust system time, in https://day.js.org/ display format'
|
|
alias: 'C'
|
|
default: adjust_time.command
|
|
type: 'string'
|
|
}
|
|
interval: {
|
|
describe: 'The minimum milliseconds between requests'
|
|
alias: 'i'
|
|
default: 500
|
|
type: 'number'
|
|
}
|
|
'user-agent': {
|
|
alias: 'u'
|
|
default: "#{info.name}/#{info.version}"
|
|
type: 'string'
|
|
}
|
|
}
|
|
adjust_time.command = argv.command
|
|
|
|
|
|
req_opt = {
|
|
method: argv.method.trim().toUpperCase()
|
|
followRedirect: argv.redirect
|
|
timeout: argv.timeout
|
|
retry: 0
|
|
dnsCache: true
|
|
cache: false
|
|
headers: {
|
|
'user-agent': argv['user-agent']
|
|
}
|
|
}
|
|
|
|
|
|
get_time_delta = (url) ->
|
|
if not /^https?:\/\//.test url
|
|
url = "#{argv.protocol}://#{url}"
|
|
console.log "#{argv.method} #{url}"
|
|
for i in [1 .. argv.count]
|
|
step = "\##{i}: ".padStart 8
|
|
start_at = Date.now()
|
|
try
|
|
r = await got url, req_opt
|
|
catch e
|
|
if not e.response?
|
|
console.log "#{step}#{e}"
|
|
wait_ms = argv.interval + start_at - Date.now()
|
|
if wait_ms > 0
|
|
await delay wait_ms
|
|
continue
|
|
r = e.response
|
|
wait_ms = argv.interval + start_at - Date.now()
|
|
if wait_ms > 0
|
|
await delay wait_ms
|
|
duration = r.timings.end - r.timings.upload
|
|
server_moment = dayjs r.headers.date
|
|
delta = Math.round(server_moment - r.timings.end - duration / 2 + 500)
|
|
console.log "#{step}" + "#{delta} ms".padStart 10
|
|
delta
|
|
|
|
|
|
delay = util.promisify (ms, cb) ->
|
|
setTimeout cb, ms
|
|
|
|
|
|
do ->
|
|
values = []
|
|
for url in argv._
|
|
values.push (await get_time_delta url)...
|
|
if values.length is 0
|
|
console.log "Network failure"
|
|
process.exit 2
|
|
return
|
|
delta = median values
|
|
console.log "Median: " + "#{delta} ms".padStart 10
|
|
return if not argv.set
|
|
if Math.abs(delta) < argv.threshold
|
|
console.log "There is no need to adjust the time"
|
|
return
|
|
await adjust_time delta |