hlsdump 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. #!/usr/bin/env node
  2. var hlsdump = require('commander');
  3. hlsdump.version('0.0.0')
  4. .usage('[options] <url>')
  5. .option('-o, --output <path>', 'target file')
  6. .option('-u, --udp [host:port]', 'relay TS over UDP', function(val) {
  7. var r = { host:'localhost', port:1234 };
  8. if (val) {
  9. var s = val.split(':');
  10. if (s.length === 1) {
  11. r.port = parseInt(s[0], 10);
  12. } else {
  13. r.host = s[0];
  14. r.port = parseInt(s[1], 10);
  15. }
  16. }
  17. return r;
  18. })
  19. .option('-b, --buffer-size <bytes>|full', 'try to buffer <bytes> of input data (implies -s)', function(val) {
  20. if (val === 'full') return 0x80000000-1;
  21. return parseInt(val, 0);
  22. })
  23. .option('-s, --sync', 'clock sync using stream PCR')
  24. .option('-k, --keep-connection', 'don\'t give up once connected')
  25. .option('-f, --full-stream', 'fetch all stream data')
  26. .option('-a, --user-agent <string>', 'HTTP User-Agent')
  27. .parse(process.argv);
  28. var util = require('util'),
  29. url = require('url'),
  30. fs = require('fs');
  31. var reader = require('../lib/reader'),
  32. tssmooth = require('../lib/tssmooth'),
  33. tsblast = require('../lib/tsblast');
  34. var src = process.argv[2];
  35. if (!src) return hlsdump.help();
  36. if (hlsdump.bufferSize) hlsdump.sync = true;
  37. var r = reader(src, {highWaterMark:hlsdump.bufferSize, keepConnection:hlsdump.keepConnection, fullStream:hlsdump.fullStream});
  38. var time = 0;
  39. r.on('segment', function(seqNo, duration, meta) {
  40. // console.error(new Date().toJSON() + sep + meta.size + sep + duration.toFixed(3) + sep + (meta.size / (duration * 1024/8)).toFixed(3));
  41. console.error('new segment at '+time.toFixed(0)+' seconds, avg bitrate (kbps):', (meta.size / (duration * 1024/8)).toFixed(1));
  42. time += duration;
  43. });
  44. r.on('error', function(err) {
  45. console.error('error', err.stack || err);
  46. });
  47. r.on('end', function() {
  48. console.error('done');
  49. });
  50. var stream = r;
  51. if (hlsdump.sync)
  52. stream = stream.pipe(tssmooth());
  53. if (hlsdump.udp)
  54. stream.pipe(tsblast(hlsdump.udp));
  55. if (hlsdump.output) {
  56. var dst;
  57. if (hlsdump.output === '-')
  58. dst = process.stdout;
  59. else
  60. dst = fs.createWriteStream(hlsdump.output);
  61. if (dst)
  62. stream.pipe(dst);
  63. }