1. const fs = require('fs');
  2. const path = require('path');
  3. const { program } = require('commander');
  5. /**
  6. * Backs up request headers for hypothesis validation.
  7. * @param {object} options - Configuration options.
  8. */
  9. function backupRequestHeaders(options) {
  10. const {
  11. sourceDirectory,
  12. backupDirectory,
  13. includeSpecificHeaders,
  14. excludeSpecificHeaders,
  15. format = 'json' // Default format
  16. } = options;
  18. if (!sourceDirectory || !backupDirectory) {
  19. console.error('Error: Source and backup directories are required.');
  20. process.exit(1);
  21. }
  23. // Ensure directories exist
  24. if (!fs.existsSync(sourceDirectory)) {
  25. console.error(`Error: Source directory "${sourceDirectory}" does not exist.`);
  26. process.exit(1);
  27. }
  29. if (!fs.existsSync(backupDirectory)) {
  30. fs.mkdirSync(backupDirectory, { recursive: true }); // create if doesn't exist
  31. }
  34. const files = fs.readdirSync(sourceDirectory);
  36. for (const file of files) {
  37. const filePath = path.join(sourceDirectory, file);
  39. if (fs.statSync(filePath).isFile()) {
  40. try {
  41. let headers = {};
  42. const headerData = fs.readFileSync(filePath, 'utf8');
  44. // Parse headers based on format
  45. if (format === 'json') {
  46. const parsedHeaders = JSON.parse(headerData);
  47. headers = parsedHeaders;
  48. } else if (format === 'text') {
  49. const headerLines = headerData.split('\n');
  50. for (const line of headerLines) {
  51. if (line.includes(':')) {
  52. const [key, value] = line.split(':').map(s => s.trim());
  53. headers[key] = value;
  54. }
  55. }
  56. } else {
  57. console.error(`Error: Unsupported format "${format}".`);
  58. continue;
  59. }
  61. // Apply filtering
  62. if (includeSpecificHeaders) {
  63. headers = Object.fromEntries(
  64. Object.entries(headers).filter(([key]) => includeSpecificHeaders.includes(key))
  65. );
  66. }
  68. if (excludeSpecificHeaders) {
  69. Object.keys(headers) = Object.keys(headers).filter(key => !excludeSpecificHeaders.includes(key));
  70. }
  73. const backupFilePath = path.join(backupDirectory, `${file}.backup`);
  74. fs.writeFileSync(backupFilePath, JSON.stringify(headers, null, 2), 'utf8'); // JSON format
  75. console.log(`Backed up ${file} to ${backupFilePath}`);
  77. } catch (err) {
  78. console.error(`Error processing file ${file}:`, err);
  79. }
  80. }
  81. }
  82. }
  85. // Command-line interface using Commander
  86. program
  87. .version('1.0.0')
  88. .description('Backs up request headers for hypothesis validation.')
  89. .requiredOption('-s, --source <directory>', 'The source directory containing header files.')
  90. .requiredOption('-b, --backup <directory>', 'The backup directory.')
  91. .option('-i, --include <headers...>', 'Specific headers to include (space-separated).')
  92. .option('-e, --exclude <headers...>', 'Specific headers to exclude (space-separated).')
  93. .option('-f, --format <format>', 'The format of the header files (json or text). Defaults to json.', 'json')
  94. .action((args) => {
  95. backupRequestHeaders(args);
  96. });
  98. program.parse(process.argv);

Add your comment